import {Wallet, ethers} from 'ethers';
import {
  State,
  simpleEthAllocation,
  signState,
  getSignerAddress,
  toNitroState,
  makeAddress,
} from '@statechannels/wallet-core';
import _ from 'lodash';

import {participant} from '../../engine/__test__/fixtures/participants';
import {recoverAddress, signState as wasmSignState} from '../signatures';
import {addHash} from '../../state-utils';
import {createLogger} from '../../logger';
import {defaultTestWalletConfig} from '../../config';

const logger = createLogger(defaultTestWalletConfig());
it('sign vs wasmSign', async () => {
  const promises = _.range(5).map(async channelNonce => {
    const {address: ethAddress, privateKey} = Wallet.createRandom();
    const address = makeAddress(ethAddress);
    const state: State = {
      chainId: '0x1',
      channelNonce,
      participants: [participant({signingAddress: address})],
      outcome: simpleEthAllocation([]),
      turnNum: 1,
      isFinal: false,
      appData: '0x00',
      appDefinition: makeAddress(ethers.constants.AddressZero),
      challengeDuration: 0x5,
    };

    const signedState = signState(state, privateKey);
    const wasmSignedState = wasmSignState(addHash(state), privateKey);
    try {
      expect(signedState).toEqual((await wasmSignedState).signature);
    } catch (error) {
      logger.info({error, state, privateKey});
      throw error;
    }
  });
  await Promise.all(promises);
});

it('getSignerAddress vs fastRecover', async () => {
  const promises = _.range(5).map(async channelNonce => {
    const {address: ethAddress, privateKey} = Wallet.createRandom();
    const address = makeAddress(ethAddress);

    const state: State = {
      chainId: '0x1',
      channelNonce,
      participants: [participant({signingAddress: address})],
      outcome: simpleEthAllocation([]),
      turnNum: 1,
      isFinal: false,
      appData: '0x00',
      appDefinition: makeAddress(ethers.constants.AddressZero),
      challengeDuration: 0x5,
    };

    const signedState = await wasmSignState(addHash(state), privateKey);
    try {
      const recovered = makeAddress(getSignerAddress(signedState.state, signedState.signature));
      const wasmRecovered = recoverAddress(signedState.signature, toNitroState(signedState.state));
      expect(recovered).toEqual(wasmRecovered);
    } catch (error) {
      logger.info({error, state, privateKey});
      throw error;
    }
  });
  await Promise.all(promises);
});