Merge branch 'main' of github.com:LF-Engineering/dev-analytics-libraries into add-common-util

Author: enkhalifapro

Makefile

@@ -4,7 +4,7 @@ lint: fmt
 	golint -set_exit_status $(PKG_LIST)
 
 fmt:
-	./scripts/for_each_go_file.sh gofmt -s -w
+	./scripts/for_each_go_file.sh 'gofmt -s -w'
 
 test:
 	go test -v $(PKG_LIST)

affiliation/dto.go

@@ -0,0 +1,27 @@
+package affiliation
+
+import "time"
+
+// Identity ...
+type Identity struct {
+	ID           string    `json:"id"`
+	LastModified time.Time `json:"last_modified"`
+	Name         string    `json:"name"`
+	Source       string    `json:"source"`
+	Username     string    `json:"username"`
+	Email        string    `json:"email"`
+	UUID         string    `json:"uuid"`
+}
+
+// AffiliationsResponse ...
+type AffiliationsResponse struct {
+	Code         string      `json:"Code"`
+	Message      string      `json:"Message"`
+	Enrollments  interface{} `json:"enrollments"`
+	Identities   interface{} `json:"identities"`
+	LastModified time.Time   `json:"last_modified"`
+	Profile      struct {
+		UUID string `json:"uuid"`
+	} `json:"profile"`
+	UUID string `json:"uuid"`
+}

affiliation/identity.go

@@ -0,0 +1,120 @@
+package affiliation
+
+import (
+	"encoding/json"
+	"fmt"
+	"log"
+	"net/url"
+	"strings"
+
+	"github.com/LF-Engineering/dev-analytics-libraries/auth0"
+	"github.com/LF-Engineering/dev-analytics-libraries/elastic"
+	"github.com/LF-Engineering/dev-analytics-libraries/http"
+)
+
+// Affiliations interface
+type Affiliations interface {
+	AddIdentity(identity *Identity) bool
+}
+
+// Affiliation struct
+type Affiliation struct {
+	AffBaseURL       string
+	ProjectSlug      string
+	ESCacheURL       string
+	ESCacheUsername  string
+	ESCachePassword  string
+	AuthGrantType    string
+	AuthClientID     string
+	AuthClientSecret string
+	AuthAudience     string
+	AuthURL          string
+	Environment      string
+	httpClient       *http.ClientProvider
+	esClient         *elastic.ClientProvider
+	auth0Client      *auth0.ClientProvider
+}
+
+// NewAffiliationsClient consumes
+//  affBaseURL, projectSlug, esCacheUrl, esCacheUsername, esCachePassword, esCacheIndex, env, authGrantType, authClientID, authClientSecret, authAudience, authURL
+func NewAffiliationsClient(affBaseURL, projectSlug, esCacheURL, esCacheUsername, esCachePassword, env, authGrantType, authClientID, authClientSecret, authAudience, authURL string) (*Affiliation, error) {
+	aff := &Affiliation{
+		AffBaseURL:       affBaseURL,
+		ProjectSlug:      projectSlug,
+		ESCacheURL:       esCacheURL,
+		ESCacheUsername:  esCacheUsername,
+		ESCachePassword:  esCachePassword,
+		AuthGrantType:    authGrantType,
+		AuthClientID:     authClientID,
+		AuthClientSecret: authClientSecret,
+		AuthAudience:     authAudience,
+		AuthURL:          authURL,
+		Environment:      env,
+	}
+
+	httpClientProvider, esClientProvider, auth0ClientProvider, err := buildServices(aff)
+	if err != nil {
+		return nil, err
+	}
+
+	aff.esClient = esClientProvider
+	aff.httpClient = httpClientProvider
+	aff.auth0Client = auth0ClientProvider
+
+	return aff, nil
+}
+
+// AddIdentity ...
+func (a *Affiliation) AddIdentity(identity *Identity) bool {
+	if identity == nil {
+		log.Println("Repository: AddIdentity: Identity is nil")
+		return false
+	}
+	token, err := a.auth0Client.ValidateToken(a.Environment)
+	if err != nil {
+		log.Println(err)
+	}
+	headers := make(map[string]string, 0)
+	headers["Content-type"] = "application/json"
+	headers["Authorization"] = fmt.Sprintf("%s %s", "Bearer", token)
+
+	queryParams := make(map[string]string, 0)
+	queryParams["name"] = identity.Name
+	queryParams["username"] = identity.Username
+	queryParams["email"] = identity.Email
+	queryParams["uuid"] = identity.UUID
+
+	endpoint := a.AffBaseURL + "/affiliation/" + url.PathEscape(a.ProjectSlug)+ "/add_identity/" + url.PathEscape(identity.Source)
+	_, res, err := a.httpClient.Request(strings.TrimSpace(endpoint), "POST", headers, nil, queryParams)
+	if err != nil {
+		log.Println("Repository: AddIdentity: Could not insert the identity: ", err)
+		return false
+	}
+	var errMsg AffiliationsResponse
+	err = json.Unmarshal(res, &errMsg)
+	if err != nil || errMsg.Message != "" {
+		log.Println("Repository: AddIdentity: failed to add identity: ", err)
+		return false
+	}
+	return true
+}
+
+func buildServices(a *Affiliation) (httpClientProvider *http.ClientProvider, esClientProvider *elastic.ClientProvider, auth0ClientProvider *auth0.ClientProvider, err error) {
+	esClientProvider, err = elastic.NewClientProvider(&elastic.Params{
+		URL:      a.ESCacheURL,
+		Username: a.ESCacheUsername,
+		Password: a.ESCachePassword,
+	})
+	if err != nil {
+		return
+	}
+
+	httpClientProvider = http.NewClientProvider(60)
+
+	auth0ClientProvider, err = auth0.NewAuth0Client(a.ESCacheURL, a.ESCacheUsername, a.ESCachePassword, a.Environment, a.AuthGrantType, a.AuthClientID, a.AuthClientSecret, a.AuthAudience, a.AuthURL)
+	if err != nil {
+		return
+	}
+
+	return
+}

auth0/dto.go

@@ -0,0 +1,44 @@
+package auth0
+
+// AuthToken Struct
+type AuthToken struct {
+	Name  string `json:"name"`
+	Token string `json:"token"`
+}
+
+// Resp struct
+type Resp struct {
+	AccessToken string `json:"access_token"`
+	Scope       string `json:"scope"`
+	ExpiresIn   int    `json:"expires_in"`
+	TokenType   string `json:"token_type"`
+}
+
+// ESTokenSchema for AuthToken
+type ESTokenSchema struct {
+	Took     int  `json:"took"`
+	TimedOut bool `json:"timed_out"`
+	Shards   struct {
+		Total      int `json:"total"`
+		Successful int `json:"successful"`
+		Skipped    int `json:"skipped"`
+		Failed     int `json:"failed"`
+	} `json:"_shards"`
+	Hits struct {
+		Total struct {
+			Value    int    `json:"value"`
+			Relation string `json:"relation"`
+		} `json:"total"`
+		MaxScore float64 `json:"max_score"`
+		Hits     []struct {
+			Index  string  `json:"_index"`
+			Type   string  `json:"_type"`
+			ID     string  `json:"_id"`
+			Score  float64 `json:"_score"`
+			Source struct {
+				Name  string `json:"name"`
+				Token string `json:"token"`
+			} `json:"_source"`
+		} `json:"hits"`
+	} `json:"hits"`
+}