Initial cloud-services repo - gateway service + pkg modules

pkg/auth/get_users_list.go

@@ -0,0 +1,92 @@
+package auth
+
+import (
+	"strings"
+	"sync"
+
+	"fiskerinc.com/modules/common"
+	"fiskerinc.com/modules/logger"
+	"fiskerinc.com/modules/utils/envtool"
+	"github.com/aws/aws-sdk-go/aws"
+	"github.com/aws/aws-sdk-go/aws/session"
+	"github.com/aws/aws-sdk-go/service/cognitoidentityprovider"
+)
+
+var (
+	ConsumerPoolId  string
+	cognitoOnce     sync.Once
+	cognitoInstance *cognitoidentityprovider.CognitoIdentityProvider
+)
+
+func GetUsersList(users []string) ([]common.JSONUserProfile, error) {
+	var userList []common.JSONUserProfile
+
+	for _, userid := range users {
+		cognitoClient := getAWS()
+
+		filter := strings.Replace("username = \"userId\"", "userId", userid, -1)
+
+		request := &cognitoidentityprovider.ListUsersInput{
+			Filter:     &filter,
+			UserPoolId: &ConsumerPoolId,
+		}
+		resp, err := cognitoClient.ListUsers(request)
+		if err != nil {
+			return nil, err
+		}
+
+		userList = append(userList, convertAWSUsers(resp.Users)...)
+	}
+
+	return userList, nil
+}
+
+func convertAWSUsers(users []*cognitoidentityprovider.UserType) []common.JSONUserProfile {
+	var userList []common.JSONUserProfile
+
+	for _, user := range users {
+		userList = append(userList, findUserAttributes(user))
+	}
+
+	return userList
+}
+
+func findUserAttributes(awsUser *cognitoidentityprovider.UserType) common.JSONUserProfile {
+	attributes := awsUser.Attributes
+	user := common.JSONUserProfile{}
+
+	user.UserName = *awsUser.Username
+	for _, attribute := range attributes {
+		switch *attribute.Name {
+		case "email":
+			user.Email = *attribute.Value
+		case "phone_number":
+			user.Phone = *attribute.Value
+		case "given_name":
+			user.FirstName = *attribute.Value
+		case "family_name":
+			user.LastName = *attribute.Value
+		}
+	}
+
+	return user
+}
+
+func getAWS() *cognitoidentityprovider.CognitoIdentityProvider {
+	cognitoOnce.Do(func() {
+		if cognitoInstance != nil {
+			return
+		}
+		logger.Info().Msg("Init cognito provider instance")
+		setPoolId()
+		mySession := session.Must(session.NewSession())
+		cognitoInstance = cognitoidentityprovider.New(mySession, aws.NewConfig().WithRegion("us-west-2"))
+	})
+
+	return cognitoInstance
+}
+
+func setPoolId() {
+	//default to dev pool
+	ConsumerPoolId = envtool.GetEnv("CONSUMER_COGNITO_CLIENT_ID", "us-west-2_c7Qu91m3J")
+}