import { UninitializedEntityManagerError } from '../../errors'
import type { EntityManagerService } from '../../services'
import {
  Action,
  AdvancedOptions,
  EntityType
} from '../../services/EntityManager/types'
import { parseParams } from '../../utils/parseParams'
import {
  Configuration,
  DashboardWalletUsersApi as GeneratedDashboardWalletUsersApi
} from '../generated/default'

import {
  CreateDashboardWalletUser,
  CreateDashboardWalletUserRequest,
  DeleteDashboardWalletUserRequest,
  DeleteDashboardWalletUserSchema,
  type DashboardWalletUsersApiServicesConfig
} from './types'

export class DashboardWalletUsersApi extends GeneratedDashboardWalletUsersApi {
  private readonly entityManager?: EntityManagerService

  constructor(
    config: Configuration,
    services: DashboardWalletUsersApiServicesConfig
  ) {
    super(config)
    this.entityManager = services.entityManager
  }

  /**
   * Connect an Audius user to a wallet on the protocol dashboard
   */
  async connectUserToDashboardWallet(
    params: CreateDashboardWalletUserRequest,
    advancedOptions?: AdvancedOptions
  ) {
    const { wallet, userId, walletSignature, userSignature } =
      await parseParams(
        'createDashboardWalletUser',
        CreateDashboardWalletUser
      )(params)

    const signatureMetadata = walletSignature
      ? {
          wallet_signature: {
            message: walletSignature.message,
            signature: walletSignature.signature
          }
        }
      : {
          user_signature: {
            message: userSignature!.message,
            signature: userSignature!.signature
          }
        }

    if (!this.entityManager) {
      throw new UninitializedEntityManagerError()
    }
    const response = await this.entityManager.manageEntity({
      userId,
      entityType: EntityType.DASHBOARD_WALLET_USER,
      entityId: 0, // Unused
      action: Action.CREATE,
      metadata: JSON.stringify({
        wallet,
        ...signatureMetadata
      }),
      ...advancedOptions
    })

    return {
      ...response
    }
  }

  /**
   * Disconnect an Audius user from a wallet on the protocol dashboard
   */
  async disconnectUserFromDashboardWallet(
    params: DeleteDashboardWalletUserRequest
  ) {
    const { userId, wallet } = await parseParams(
      'deleteDashboardWalletUser',
      DeleteDashboardWalletUserSchema
    )(params)

    if (!this.entityManager) {
      throw new UninitializedEntityManagerError()
    }
    return await this.entityManager.manageEntity({
      userId,
      entityType: EntityType.DASHBOARD_WALLET_USER,
      entityId: 0, // Unused
      action: Action.DELETE,
      metadata: JSON.stringify({
        wallet
      })
    })
  }
}