porter/internal/analytics/identifiers.go

package analytics

import (
	"fmt"

	"github.com/porter-dev/porter/internal/models"
	segment "gopkg.in/segmentio/analytics-go.v3"
)

type segmentIdentifier interface {
	getUserId() string
	getTraits() segment.Traits
}

type segmentIdentifyNewUser struct {
	userId    string
	userEmail string
	isGithub  bool
}

// Creates a segment Identifier struct for new users. As we handle registration with github, it also accepts a param
// to check if the new user has registered with github or not.
func CreateSegmentIdentifyNewUser(user *models.User, registeredViaGithub bool) *segmentIdentifyNewUser {
	userId := fmt.Sprintf("%v", user.ID)
	return &segmentIdentifyNewUser{
		userId:    userId,
		userEmail: user.Email,
		isGithub:  registeredViaGithub,
	}
}

func (i segmentIdentifyNewUser) getUserId() string {
	return i.userId
}

func (i segmentIdentifyNewUser) getTraits() segment.Traits {
	var githubTrait string

	if i.isGithub {
		githubTrait = "true"
	} else {
		githubTrait = "false"
	}

	return segment.NewTraits().
		SetEmail(i.userEmail).
		Set("github", githubTrait)
}