import {HexString} from "aptos";
import {CreationHelper} from "../momentum-safe/creation";
import {MY_ACCOUNT} from "../web3/global";
import {
  printSeparator,
  promptUntilString,
  registerState,
  setState,
  State,
  printMyMessage,
  CmdOption,
  executeCmdOptions,
  promptForYN
} from "./common";
import {Transaction} from "../web3/transaction";
import * as Aptos from '../web3/global';
import * as Gen from "aptos/src/generated";
import {HexBuffer} from "../utils/buffer";

interface creationDetailsArg {
  address: HexString,
}

export function registerCreationDetails() {
  registerState(State.PendingCreate, (a) => creationDetails(a));
}

async function creationDetails(rawArg: any) {
  const address = (rawArg as creationDetailsArg).address;

  console.clear();
  await printMyMessage();

  let creation: CreationHelper;
  try {
    creation = await CreationHelper.fromPendingCreation(address);
  } catch (e) {
    if (e instanceof Error && e.message.includes('Table Item not found by Table handle')) {
      console.log("Cannot get creation data. Is multi-sig already registered?");
      // TODO: Check with momentum safe and go to msafe detail page.
      printSeparator();
      console.log("\tb)\tback");
      await promptUntilString("", "", s => s === 'b');
      setState(State.List);
      return;
    } else {
      throw e;
    }
  }

  console.log(`Momentum Safe Address: ${creation.address.hex()}`);
  console.log();

  const collectedSigs = await creation.collectedSignatures();

  console.log(`Collected signatures from public keys: ${collectedSigs.length} / ${creation.threshold}`);
  collectedSigs.forEach( (pk, i) => {
    console.log(`pk ${i}:\t${pk}`);
  });
  const isMeSigned = collectedSigs.find( pk => pk.hex() === MY_ACCOUNT.publicKey().hex()) !== undefined;

  printSeparator();

  // Do the check first. Corner case when the transaction was not executed last
  // time even enough signature was collected.
  const userBreak = await checkCreationEnoughSigsAndAssemble(creation);
  if (userBreak) {
    await executeCmdOptions(
      "User breaks the signature submission",
      [{shortage: 'b', showText: 'Back', handleFunc: () => setState(State.List)}],
    );
    return;
  }

  let optionPromptStr: string;
  if (isMeSigned) {
    optionPromptStr = 'Already signed. Waiting for other confirmations.';
  } else {
    optionPromptStr = 'Waiting for my signature. Sign?';
  }

  let isReturn = true;
  const opts: CmdOption[] = [];
  if (!isMeSigned) {
    opts.push({shortage: 's', showText: 'Sign', handleFunc: () => { isReturn = false }});
  }
  opts.push(
    {shortage: 'r', showText: 'Refresh', handleFunc: () =>
        setState(State.PendingCreate, {address: creation.address})},
    {shortage: 'b', showText: 'Back', handleFunc: () =>
        setState(State.List)}
    );

  await executeCmdOptions(optionPromptStr, opts);
  if (isReturn) {return}

  console.log();

  const isReadyExecute = await creation.isReadyToSubmit(MY_ACCOUNT.publicKey());
  let tx: Gen.PendingTransaction;
  if (isReadyExecute) {
    tx = await creation.assembleAndSubmitTx(MY_ACCOUNT);
  } else {
    tx = await creation.submitSignature(MY_ACCOUNT, {
      estimateGasPrice: true,
      estimateMaxGas: true,
    });
  }
  console.log(`\tTransaction ${tx.hash} submitted. Waiting for confirmation`);
  await Aptos.waitForTransaction(tx.hash);
  console.log(`\tTransaction confirmed on chain.`);

  printSeparator();

  await executeCmdOptions(
    'Choose your next step',
    [
      {shortage: 'r', showText: 'Refresh', handleFunc: () =>
        setState(State.PendingCreate, {address: creation.address})},
      {shortage: 'b', showText: 'Back', handleFunc: () =>
        setState(State.List)},
    ]
  );
}

// check whether enough signatures are collected, and then assemble, submit.
export async function checkCreationEnoughSigsAndAssemble(creation: CreationHelper): Promise<boolean> {
  const isReadyToExecute = await creation.isReadyToSubmit();
  if (!isReadyToExecute) {
    return false;
  }
  const opt = await promptForYN("Already collected enough signature. Submit?", true);
  if (!opt) {
    return true;
  }
  const tx = await creation.assembleAndSubmitTx(MY_ACCOUNT);
  console.log(`\tTransaction ${tx.hash} submitted. Waiting for confirmation`);
  await Aptos.waitForTransaction(tx.hash);
  console.log(`\tTransaction confirmed on chain.`);

  printSeparator();
  return false;
}