import Transport from '@ledgerhq/hw-transport-u2f';
import Eth from '@ledgerhq/hw-app-eth';
import BN from 'bn.js';

export const LEDGER_LEGACY_PATH = "m/44'/60'/0'/x";
export const LEDGER_LIVE_PATH = "m/44'/60'/x'/0/0";

export const chopHdPrefix = str =>
  str.slice(0, 2) === 'm/' ? str.slice(2) : str;
export const addHdPrefix = str => (str.slice(0, 2) === 'm/' ? str : 'm/' + str);

export const ledgerSignTransaction = async (txn, hdPath) => {
  const transport = await Transport.create();
  const eth = new Eth(transport);

  const path = chopHdPrefix(hdPath);
  const serializedTx = txn.serialize().toString('hex');
  const sig = await eth.signTransaction(path, serializedTx);

  txn.v = new BN(sig.v, 'hex');
  txn.r = new BN(sig.r, 'hex');
  txn.s = new BN(sig.s, 'hex');

  return txn;
};

export const ledgerSignMessage = async (message, hdPath) => {
  const transport = await Transport.create();
  const eth = new Eth(transport);

  const path = chopHdPrefix(hdPath);
  const { r, s, v } = await eth.signPersonalMessage(
    path,
    Buffer.from(message).toString('hex')
  );
  const ethSignature = new Uint8Array(65);
  ethSignature.set(Buffer.from(r, 'hex'));
  ethSignature.set(Buffer.from(s, 'hex'), 32);
  ethSignature[64] = v;

  return `0x${Buffer.from(ethSignature).toString('hex')}`;
};