import { FaceSolanaNearTransactionPayload, JsonRpcMethod, Network } from '@haechi-labs/face-types';
import { BigNumber } from 'ethers';
import * as nearAPI from 'near-api-js';

import { Internal } from '../Internal';

export class NearProvider {
  internal: Internal;

  constructor(internal: Internal) {
    this.internal = internal;
  }

  async getPublicKeys(): Promise<nearAPI.utils.PublicKey[]> {
    const strPublicKeys = await this.internal.getAddresses();
    return strPublicKeys.map((strPublicKey) =>
      nearAPI.utils.PublicKey.fromString(
        nearAPI.utils.serialize.base_encode(Buffer.from(strPublicKey, 'hex'))
      )
    );
  }

  async getBalance(address: string, contractAddress?: string): Promise<BigNumber> {
    const res = await this.internal.sendRpc({
      method: JsonRpcMethod.near_get_balance,
      params: [address, contractAddress],
    });

    return BigNumber.from(res);
  }

  async signAndSendTransaction(transaction: nearAPI.transactions.Transaction): Promise<string> {
    const txPayload: FaceSolanaNearTransactionPayload = {
      serializedHex: Buffer.from(
        nearAPI.utils.serialize.serialize(nearAPI.transactions.SCHEMA, transaction)
      ).toString('hex'),
    };
    const result = await this.internal.sendRpc<any>({
      method: JsonRpcMethod.near_broadcast_tx_async,
      params: [txPayload],
    });

    return result!;
  }
}