Merge pull request #18 from LF-Engineering/add-common-util

Add common util

Author: enkhalifapro

affiliation/dto.go

@@ -48,16 +48,6 @@ type EnrollmentsResponse struct {
 	UUID        string       `json:"uuid"`
 }
 
-// Profile ...
-type Profile struct {
-	IsBot     int    `json:"is_bot"`
-	Name      string `json:"name"`
-	UUID      string `json:"uuid"`
-	Email     string `json:"email"`
-	Gender    string `json:"gender"`
-	GenderAcc int    `json:"gender_acc"`
-}
-
 // ProfileResponse ...
 type ProfileResponse struct {
 	Enrollments  []Enrollment `json:"enrollments"`
@@ -66,3 +56,57 @@ type ProfileResponse struct {
 	Profile      Profile      `json:"profile"`
 	UUID         string       `json:"uuid"`
 }
+
+// IdentityData ...
+type IdentityData struct {
+	Email        *string    `json:"email,omitempty"`
+	ID           string     `json:"id,omitempty"`
+	LastModified *time.Time `json:"last_modified,omitempty"`
+	Name         *string    `json:"name,omitempty"`
+	Source       string     `json:"source,omitempty"`
+	Username     *string    `json:"username,omitempty"`
+	UUID         *string    `json:"uuid,omitempty"`
+}
+
+// UniqueIdentityFullProfile ...
+type UniqueIdentityFullProfile struct {
+	Enrollments []*Enrollments  `json:"enrollments"`
+	Identities  []*IdentityData `json:"identities"`
+	Profile     *Profile        `json:"profile,omitempty"`
+	UUID        string          `json:"uuid,omitempty"`
+}
+
+// Enrollments ...
+type Enrollments struct {
+	Organization *Organization `json:"organization,omitempty"`
+}
+
+// Organization ...
+type Organization struct {
+	Name string `json:"name,omitempty"`
+}
+
+// Profile ...
+type Profile struct {
+	Email     *string `json:"email,omitempty"`
+	Gender    *string `json:"gender,omitempty"`
+	GenderAcc *int64  `json:"gender_acc,omitempty"`
+	IsBot     *int64  `json:"is_bot,omitempty"`
+	Name      *string `json:"name,omitempty"`
+	UUID      string  `json:"uuid,omitempty"`
+}
+
+// AffIdentity contains affiliation user Identity
+type AffIdentity struct {
+	ID            *string `json:"id"`
+	UUID          *string
+	Name          string
+	Username      string
+	Email         string
+	Domain        string
+	Gender        *string  `json:"gender"`
+	GenderACC     *int64   `json:"gender_acc"`
+	OrgName       *string  `json:"org_name"`
+	IsBot         *int64   `json:"is_bot"`
+	MultiOrgNames []string `json:"multi_org_names"`
+}

affiliation/identity.go

@@ -190,6 +190,71 @@ func (a *Affiliation) GetProfile(uuid, projectSlug string) *ProfileResponse {
 	return &response
 }
 
+// GetIdentityByUser ...
+func (a *Affiliation) GetIdentityByUser(key string, value string) (*AffIdentity, error) {
+	if key == "" || value == "" {
+		nilKeyOrValueErr := "repository: GetIdentityByUser: key or value is null"
+		log.Println(nilKeyOrValueErr)
+		return nil, fmt.Errorf(nilKeyOrValueErr)
+	}
+	token, err := a.auth0Client.ValidateToken(a.Environment)
+	if err != nil {
+		log.Println(err)
+		return nil, err
+	}
+
+	headers := make(map[string]string, 0)
+	headers["Authorization"] = fmt.Sprintf("%s %s", "Bearer", token)
+	endpoint := a.AffBaseURL + "/affiliation/" + "identity/" + key + "/" + value
+	_, res, err := a.httpClient.Request(strings.TrimSpace(endpoint), "GET", headers, nil, nil)
+	if err != nil {
+		log.Println("Repository: GetIdentityByUser: Could not get the identity: ", err)
+		return nil, err
+	}
+
+	var ident IdentityData
+	err = json.Unmarshal(res, &ident)
+	if err != nil {
+		return nil, err
+	}
+
+	profileEndpoint := a.AffBaseURL + "/affiliation/" + url.PathEscape(a.ProjectSlug) + "/get_profile/" + *ident.UUID
+	_, profileRes, err := a.httpClient.Request(strings.TrimSpace(profileEndpoint), "GET", headers, nil, nil)
+	if err != nil {
+		log.Println("Repository: GetIdentityByUser: Could not get the identity: ", err)
+		return nil, err
+	}
+
+	var profile UniqueIdentityFullProfile
+	err = json.Unmarshal(profileRes, &profile)
+	if err != nil {
+		return nil, err
+	}
+	var identity AffIdentity
+	identity.UUID = ident.UUID
+	identity.Name = *ident.Name
+	identity.Username = *ident.Username
+	identity.Email = *ident.Email
+	identity.ID = &ident.ID
+
+	identity.IsBot = profile.Profile.IsBot
+	identity.Gender = profile.Profile.Gender
+	identity.GenderACC = profile.Profile.GenderAcc
+
+	if len(profile.Enrollments) > 1 {
+		identity.OrgName = &profile.Enrollments[0].Organization.Name
+		for _, org := range profile.Enrollments {
+			identity.MultiOrgNames = append(identity.MultiOrgNames, org.Organization.Name)
+		}
+	} else if len(profile.Enrollments) == 1 {
+		identity.OrgName = &profile.Enrollments[0].Organization.Name
+		identity.MultiOrgNames = append(identity.MultiOrgNames, profile.Enrollments[0].Organization.Name)
+	}
+
+	return &identity, nil
+
+}
+
 func buildServices(a *Affiliation) (httpClientProvider *http.ClientProvider, esClientProvider *elastic.ClientProvider, auth0ClientProvider *auth0.ClientProvider, err error) {
 	esClientProvider, err = elastic.NewClientProvider(&elastic.Params{
 		URL:      a.ESCacheURL,

go.mod

@@ -5,9 +5,16 @@ go 1.15
 require (
 	github.com/avast/retry-go v3.0.0+incompatible
 	github.com/aws/aws-sdk-go v1.36.15
+	github.com/dgrijalva/jwt-go v3.2.0+incompatible
 	github.com/elastic/go-elasticsearch/v7 v7.10.0
+	github.com/go-git/go-git/v5 v5.2.0
+	github.com/google/go-github v17.0.0+incompatible
+	github.com/google/go-querystring v1.0.0 // indirect
+	github.com/google/uuid v1.1.4
 	github.com/json-iterator/go v1.1.10
 	github.com/stretchr/testify v1.6.1
 	golang.org/x/crypto v0.0.0-20201221181555-eec23a3978ad
+	golang.org/x/oauth2 v0.0.0-20201208152858-08078c50e5b5
 	golang.org/x/text v0.3.4
+	gopkg.in/yaml.v2 v2.2.8
 )