import Web3 from "web3";
import "../../../utils/polyfills";

interface ParityBlockReward {
  author: string;
  rewardType: 'block';
  value: string;
}
interface ParityCall {
  callType: 'call';
  from: string;
  gas: string;
  input: string;
  to: string;
  value: string;
}
export interface ParityTraceResponse {
  action: ParityBlockReward | ParityCall;
  blockHash: string;
  blockNumber: number;
  result?: { gasUsed: string; output: string };
  subtraces: number;
  traceAddress: [];
  transactionHash?: string;
  transactionPosition?: number;
  type: 'reward' | 'call';
}

export class ParityRPC {
  web3: Web3;

  constructor(web3: Web3) {
    this.web3 = web3;
  }

  public async *getTransactionsForAddress(bestHeight: number, address: string) {
    const txs = await this.scan(0, bestHeight, address);
    for (const tx of txs) {
      yield {
        id: null,
        txid: tx.transactionHash,
        fee: tx.result ? tx.result.gasUsed : null,
        category: 'receive',
        satoshis: tx.action.value,
        height: tx.blockNumber,
        address,
        outputIndex: tx.result ? tx.result.output : null
      };
    }
  }

  scan(fromHeight: number, toHeight: number, address: string) {
    return new Promise<Array<ParityTraceResponse>>(resolve =>
      this.web3.eth.currentProvider.send(
        {
          method: 'trace_filter',
          params: [
            {
              fromBlock: this.web3.utils.toHex(fromHeight),
              toBlock: this.web3.utils.toHex(toHeight),
              toAddress: [address.toLowerCase()]
            }
          ],
          jsonrpc: '2.0',
          id: 0
        },
        (_, data) => resolve(data.result as Array<ParityTraceResponse>)
      )
    );
  }
}