import { getPublic } from "@toruslabs/eccrypto";
import BN from "bn.js";
import type { curve, ec } from "elliptic";

import { BNString } from "../baseTypes/commonTypes";
import { ecCurve } from "../utils";
import Point from "./Point";

// These functions are here because BN can't be extended
export const toPrivKeyEC = (bn: BN): ec.KeyPair => ecCurve.keyFromPrivate(bn.toString("hex", 64));

export const toPrivKeyECC = (bn: BNString): Buffer => {
  const tmp = new BN(bn, "hex");
  return Buffer.from(tmp.toString("hex", 64), "hex");
};

export const getPubKeyEC = (bn: BN): curve.base.BasePoint => ecCurve.keyFromPrivate(bn.toString("hex", 64)).getPublic();

export const getPubKeyECC = (bn: BN): Buffer => getPublic(toPrivKeyECC(bn));

export const getPubKeyPoint = (bn: BN): Point => {
  const pubKeyEc = getPubKeyEC(bn);
  return new Point(pubKeyEc.getX().toString("hex"), pubKeyEc.getY().toString("hex"));
};