from typing import Dict

from gql import gql

ADD_USER_IDENTITY_MUTATION = gql("""
mutation createIdentity(
    $identity: IdentityInput!,
    $masterContact: Boolean,
    $sendInvite: Boolean,
    $tenantProfileRoles: TenantProfileRolesCreateInput!
) {
    createIdentity(
        identity: $identity,
        masterContact: $masterContact,
        sendInvite: $sendInvite,
        tenantProfileRoles: $tenantProfileRoles
    ) {
        id
        email
        active
        firstName
        lastName
    }
}
""")


def get_variables(email: str) -> Dict:
    return {
        'identity': {'email': email, 'firstName': 'GQL', 'lastName': 'User'},
        'masterContact': True,
        'sendInvite': True,
        'tenantProfileRoles': {  # Changed from tenantRoles
            'tenantUuid': '2e79b9b8-29ed-44e1-832f-2aa3b47f13c8',  # Platform Test Vendor
            'tenantType': 'ACCOUNT',
            'rolesToAttach': [
                'COLLABORATORS_BASE_ROLE',
                'INSIGHTS_BASE_ROLE',
                'WORKSTATION_ANALYTICS_BASE_ROLE',
                'WORKSTATION_CATALOG_ROLE',
                'WORKSTATION_MARKETING_ROLE',
                'WORKSTATION_ADVERTISING_ROLE',
            ],
        },
    }