import { EnvironmentUtils } from '@affinidi-tdk/common'
import { Configuration, IotaApi } from '@affinidi-tdk/iota-client'
import {
  CognitoIdentityClient,
  GetCredentialsForIdentityCommand,
} from '@aws-sdk/client-cognito-identity'

export interface IAuthProviderParams {
  region: string
  apiGW: string
}

export interface IotaCredentials {
  readonly credentials: Credentials
  readonly connectionClientId: string
}

export interface Credentials {
  readonly accessKeyId?: string
  readonly secretKey?: string
  readonly sessionToken?: string
  readonly expiration?: Date
}

export interface IdentityCredentials {
  readonly identityId: string
  readonly token: string
}

export class IotaAuthProvider {
  region: string
  apiGW: string

  constructor(param?: { [key: string]: any }) {
    this.region = param?.region ?? EnvironmentUtils.fetchRegion()
    this.apiGW = param?.apiGW ?? EnvironmentUtils.fetchApiGwUrl()
  }

  async limitedTokenToIotaCredentials(
    limitedToken: string,
  ): Promise<IotaCredentials> {
    const iotaAPIClient = new IotaApi(
      new Configuration({ basePath: `${this.apiGW}/ais` }),
    )
    const response = await iotaAPIClient.awsExchangeCredentials({
      assertion: limitedToken,
    })
    const { connectionClientId, credentials: identityCredentials } =
      response.data

    const credentials =
      await this.exchangeIdentityCredentials(identityCredentials)

    return {
      credentials,
      connectionClientId,
    }
  }

  async exchangeIdentityCredentials(
    identityCredentials: IdentityCredentials,
  ): Promise<Credentials> {
    const cognitoIdentityClient = new CognitoIdentityClient({
      region: this.region,
    })
    const responseCmd = await cognitoIdentityClient.send(
      new GetCredentialsForIdentityCommand({
        IdentityId: identityCredentials.identityId,
        Logins: {
          'cognito-identity.amazonaws.com': identityCredentials.token,
        },
      }),
    )
    const credentials = responseCmd.Credentials
    if (!credentials) {
      throw new Error('Error fetching credentials')
    }
    return {
      accessKeyId: credentials.AccessKeyId,
      secretKey: credentials.SecretKey,
      sessionToken: credentials.SessionToken,
      expiration: credentials.Expiration,
    }
  }
}