// Copyright Abridged, Inc. 2022. All Rights Reserved.
// Node module: @collabland/staking-contracts
// This file is licensed under the MIT License.
// License text available at https://opensource.org/licenses/MIT

import {pMap} from '@collabland/common';
import {BindingScope, extensionFor, injectable} from '@loopback/core';
import {BigNumber} from 'ethers';
import {STAKING_ADAPTERS_EXTENSION_POINT} from '../keys.js';
import {BaseStakingContractAdapter, StakingAsset} from '../staking.js';
import {NftwEscrow} from '../types/NftwEscrow.js';
import {MtgStaking__factory} from '../types/factories/MtgStaking__factory.js';
import {NftwEscrow__factory} from '../types/factories/NftwEscrow__factory.js';

@injectable(
  {
    scope: BindingScope.SINGLETON,
  },
  extensionFor(STAKING_ADAPTERS_EXTENSION_POINT),
)
export class NFTWEscrowStakingContractAdapter extends BaseStakingContractAdapter {
  contractAddress = '0x69f0b8c5e94f6b64d832b7d9b15f3a88cb2f6f4b';

  supportedAssets: StakingAsset[] = [
    {
      name: 'NFTWorlds',
      asset: 'ERC721:0xBD4455dA5929D5639EE098ABFaa3241e9ae111Af',
    },
  ];

  async getStakedTokenIds(owner: string): Promise<BigNumber[]> {
    const contract = NftwEscrow__factory.connect(
      this.contractAddress,
      this.provider,
    );
    const tokenIdsByOwner: Record<string, BigNumber[]> =
      await this.getTokenIdsByOwner(contract);
    return tokenIdsByOwner[owner];
  }

  private async getTokenIdsByOwner(contract: NftwEscrow) {
    const tokenIdsByOwner: Record<string, BigNumber[]> = {};
    const tokenIds = new Array<number>(10001);
    for (let i = 1; i <= 10000; i++) {
      tokenIds[i - 1] = i;
    }
    await pMap(tokenIds, async i => {
      const tokenId = BigNumber.from(i);
      const info = await contract.getWorldInfo(tokenId);
      tokenIdsByOwner[info.owner] = tokenIdsByOwner[info.owner] ?? [];
      tokenIdsByOwner[info.owner].push(tokenId);
    });
    return tokenIdsByOwner;
  }

  async getStakedTokenBalance(owner: string): Promise<BigNumber> {
    const contract = MtgStaking__factory.connect(
      this.contractAddress,
      this.provider,
    );
    return contract.numOfTokenStaked(owner);
  }
}