List possible values when inviting a team within an environment

backend/dataall/api/Objects/Group/queries.py

@@ -1,5 +1,5 @@
 from ... import gql
-from .resolvers import get_group, list_datasets_owned_by_env_group, list_data_items_shared_with_env_group
+from .resolvers import get_group, list_datasets_owned_by_env_group, list_data_items_shared_with_env_group, list_cognito_groups
 
 getGroup = gql.QueryField(
     name='getGroup',
@@ -33,3 +33,9 @@
     type=gql.Ref('EnvironmentPublishedItemSearchResults'),
     test_scope='Dataset',
 )
+
+listCognitoGroups = gql.QueryField(
+    name='listCognitoGroups',
+    type=gql.ArrayType(gql.Ref('CognitoGroup')),
+    resolver=list_cognito_groups
+)

backend/dataall/api/Objects/Group/resolvers.py

@@ -1,7 +1,11 @@
+import os
 from .... import db
 from ....db import exceptions
 from ....db.models import Group
 from ...constants import *
+from ....aws.handlers.parameter_store import ParameterStoreManager
+from ....aws.handlers.sts import SessionHelper
+from ....aws.handlers.cognito import Cognito
 
 
 def resolve_group_environment_permissions(context, source, environmentUri):
@@ -70,3 +74,19 @@ def list_data_items_shared_with_env_group(
             data=filter,
             check_perm=True,
         )
+
+
+def list_cognito_groups(context, source):
+    current_account = SessionHelper.get_account()
+    current_region = os.getenv('AWS_REGION', 'eu-west-1')
+    envname = os.getenv('envname', 'local')
+    if envname in ['local', 'dkrcompose']:
+        return [{"groupName": 'DAAdministrators'}, {"groupName": 'Engineers'}, {"groupName": 'Scientists'}]
+    parameter_path = f'/dataall/{envname}/cognito/userpool'
+    user_pool_id = ParameterStoreManager.get_parameter_value(current_account, current_region, parameter_path)
+    groups = Cognito.list_cognito_groups(current_account, current_region, user_pool_id)
+    res = []
+    for group in groups:
+        res.append({"groupName": group['GroupName']})
+
+    return res

backend/dataall/api/Objects/Group/schema.py

@@ -46,3 +46,10 @@
         gql.Field(name='nodes', type=gql.ArrayType(Group)),
     ],
 )
+
+CognitoGroup = gql.ObjectType(
+    name='CognitoGroup',
+    fields=[
+        gql.Field(name='groupName', type=gql.String),
+    ],
+)

backend/dataall/aws/handlers/cognito.py

@@ -0,0 +1,25 @@
+import logging
+
+from .sts import SessionHelper
+
+
+log = logging.getLogger(__name__)
+
+
+class Cognito:
+    @staticmethod
+    def client(account_id: str, region_name: str, client_type: str):
+        session = SessionHelper.remote_session(account_id)
+        return session.client(client_type, region_name=region_name)
+
+    @staticmethod
+    def list_cognito_groups(account_id: str, region: str, user_pool_id: str):
+        try:
+            cognitoCli = Cognito.client(account_id, region, "cognito-idp")
+            response = cognitoCli.list_groups(UsePoolId=user_pool_id)
+        except Exception as e:
+            log.error(
+                f'Failed to list groups of user pool {user_pool_id} due to {e}'
+            )
+        else:
+            return response['Groups']

frontend/src/api/Groups/listCognitoGroups.js

@@ -0,0 +1,13 @@
+import { gql } from 'apollo-boost';
+
+const listCognitoGroups = () => ({
+  query: gql`
+    query listCognitoGroups {
+      listCognitoGroups{
+        groupName
+      }
+    }
+  `
+});
+
+export default listCognitoGroups;

frontend/src/views/Environments/EnvironmentTeamInviteForm.js

@@ -28,6 +28,7 @@ import useClient from '../../hooks/useClient';
 import listEnvironmentGroupInvitationPermissions from '../../api/Environment/listEnvironmentPermissions';
 import inviteGroupOnEnvironment from '../../api/Environment/inviteGroup';
 import listEnvironmentNotInvitedGroups from '../../api/Environment/listNotInvitedGroups';
+import listCognitoGroups from '../../api/Groups/listCognitoGroups';
 
 const EnvironmentTeamInviteForm = (props) => {
   const { environment, onClose, open, reloadTeams, ...other } = props;
@@ -44,17 +45,15 @@ const EnvironmentTeamInviteForm = (props) => {
   const fetchGroups = useCallback(async () => {
     try {
       setLoadingGroups(true);
-      const response = await client.query(
-        listEnvironmentNotInvitedGroups({
-          environmentUri: environment.environmentUri
-        })
-      );
+      console.log("fetchgroups")
+      const response = await client.query(listCognitoGroups());
+      console.log(response)
       if (!response.errors) {
         setGroupOptions(
-          response.data.listEnvironmentNotInvitedGroups.nodes.map((g) => ({
+          response.data.listCognitoGroups.map((g) => ({
             ...g,
-            value: g.groupUri,
-            label: g.groupUri
+            value: g.groupName,
+            label: g.groupName
           }))
         );
       } else {