import { StatusCodes } from "@ledgerhq/hw-transport/errors";
import { DeviceAppVerifyNotSupported, UserRefusedAddress } from "../errors";
import { log } from "@ledgerhq/logs";
import type { GetAddressResult, GetAddressOptions } from "../derivation";

export type GetAddressFn = (
  deviceId: string,
  addressOpt: GetAddressOptions,
) => Promise<GetAddressResult>;

const getAddressWrapper =
  (getAddressFn: GetAddressFn) =>
  async (deviceId: string, opts: GetAddressOptions): Promise<GetAddressResult> => {
    const { currency, path, verify } = opts;
    return getAddressFn(deviceId, opts)
      .then(result => {
        log("hw", `getAddress ${currency.id} on ${path}`, result);
        return result;
      })
      .catch(e => {
        log("hw", `getAddress ${currency.id} on ${path} FAILED ${String(e)}`);

        if (e && e.name === "TransportStatusError") {
          if (e.statusCode === StatusCodes.INCORRECT_P1_P2 && verify) {
            throw new DeviceAppVerifyNotSupported();
          }

          if (
            e.statusCode === StatusCodes.CONDITIONS_OF_USE_NOT_SATISFIED ||
            e.statusCode === StatusCodes.USER_REFUSED_ON_DEVICE
          ) {
            throw new UserRefusedAddress();
          }
        }

        throw e;
      });
  };

export default getAddressWrapper;