Add more methods to affiliation (#15)

* Add methods to affiliation
- GetIdentity
- GetOrganizations
- GetProfile

Author: linonymous

affiliation/dto.go

@@ -25,3 +25,44 @@ type AffiliationsResponse struct {
 	} `json:"profile"`
 	UUID string `json:"uuid"`
 }
+
+// Enrollment ...
+type Enrollment struct {
+	End          time.Time `json:"end"`
+	ID           int       `json:"id"`
+	Organization struct {
+		ID   int    `json:"id"`
+		Name string `json:"name"`
+	} `json:"organization"`
+	OrganizationID int       `json:"organization_id"`
+	Role           string    `json:"role"`
+	Start          time.Time `json:"start"`
+	UUID           string    `json:"uuid"`
+}
+
+// EnrollmentsResponse ...
+type EnrollmentsResponse struct {
+	Enrollments []Enrollment `json:"enrollments"`
+	Scope       string       `json:"scope"`
+	User        string       `json:"user"`
+	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"`
+	Identities   []Identity   `json:"identities"`
+	LastModified time.Time    `json:"last_modified"`
+	Profile      Profile      `json:"profile"`
+	UUID         string       `json:"uuid"`
+}

affiliation/identity.go

@@ -84,7 +84,7 @@ func (a *Affiliation) AddIdentity(identity *Identity) bool {
 	queryParams["email"] = identity.Email
 	queryParams["uuid"] = identity.UUID
 
-	endpoint := a.AffBaseURL + "/affiliation/" + url.PathEscape(a.ProjectSlug)+ "/add_identity/" + url.PathEscape(identity.Source)
+	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)
@@ -99,6 +99,96 @@ func (a *Affiliation) AddIdentity(identity *Identity) bool {
 	return true
 }
 
+// GetIdentity ...
+func (a *Affiliation) GetIdentity(uuid string) *Identity {
+	if uuid == "" {
+		log.Println("GetIdentity: uuid is empty")
+		return nil
+	}
+	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)
+
+	endpoint := a.AffBaseURL + "/affiliation/get_identity/" + uuid
+
+	_, res, err := a.httpClient.Request(strings.TrimSpace(endpoint), "GET", headers, nil, nil)
+	if err != nil {
+		log.Println("GetIdentity: Could not get the identity: ", err)
+		return nil
+	}
+	var identity Identity
+	err = json.Unmarshal(res, &identity)
+	if err != nil {
+		log.Println("GetIdentity: failed to unmarshal identity: ", err)
+		return nil
+	}
+	return &identity
+}
+
+// GetOrganizations ...
+func (a *Affiliation) GetOrganizations(uuid, projectSlug string) *[]Enrollment {
+	if uuid == "" || projectSlug == "" {
+		return nil
+	}
+	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)
+
+	endpoint := a.AffBaseURL + "/affiliation/" + url.PathEscape(projectSlug) + "/enrollments/" + uuid
+
+	_, res, err := a.httpClient.Request(strings.TrimSpace(endpoint), "GET", headers, nil, nil)
+	if err != nil {
+		log.Println("GetOrganizations: Could not get the organizations: ", err)
+		return nil
+	}
+
+	var response EnrollmentsResponse
+	err = json.Unmarshal(res, &response)
+	if err != nil {
+		log.Println("GetOrganizations: failed to unmarshal enrollments response: ", err)
+		return nil
+	}
+	return &response.Enrollments
+}
+
+// GetProfile ...
+func (a *Affiliation) GetProfile(uuid, projectSlug string) *ProfileResponse {
+	if uuid == "" || projectSlug == "" {
+		return nil
+	}
+	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)
+
+	endpoint := a.AffBaseURL + "/affiliation/" + url.PathEscape(projectSlug) + "/get_profile/" + uuid
+
+	_, res, err := a.httpClient.Request(strings.TrimSpace(endpoint), "GET", headers, nil, nil)
+	if err != nil {
+		log.Println("GetProfile: Could not get the profile: ", err)
+		return nil
+	}
+
+	var response ProfileResponse
+	err = json.Unmarshal(res, &response)
+	if err != nil {
+		log.Println("GetProfile: failed to unmarshal profile response: ", err)
+		return nil
+	}
+	return &response
+}
+
 func buildServices(a *Affiliation) (httpClientProvider *http.ClientProvider, esClientProvider *elastic.ClientProvider, auth0ClientProvider *auth0.ClientProvider, err error) {
 	esClientProvider, err = elastic.NewClientProvider(&elastic.Params{
 		URL:      a.ESCacheURL,