import { WHIRLPOOL_PROGRAM_ID } from '../programId';
import * as anchor from './anchor';
import * as custom from './custom';

export function fromCode(code: number, logs?: string[]): custom.CustomError | anchor.AnchorError | null {
  return code >= 6000 ? custom.fromCode(code, logs) : anchor.fromCode(code, logs);
}

function hasOwnProperty<X extends object, Y extends PropertyKey>(obj: X, prop: Y): obj is X & Record<Y, unknown> {
  return Object.hasOwnProperty.call(obj, prop);
}

const errorRe = /Program (\w+) failed: custom program error: (\w+)/;

export function fromTxError(err: unknown): custom.CustomError | anchor.AnchorError | null {
  if (typeof err !== 'object' || err === null || !hasOwnProperty(err, 'logs') || !Array.isArray(err.logs)) {
    return null;
  }

  let firstMatch: RegExpExecArray | null = null;
  for (const logLine of err.logs) {
    firstMatch = errorRe.exec(logLine);
    if (firstMatch !== null) {
      break;
    }
  }

  if (firstMatch === null) {
    return null;
  }

  const [programIdRaw, codeRaw] = firstMatch.slice(1);
  if (programIdRaw !== WHIRLPOOL_PROGRAM_ID.toString()) {
    return null;
  }

  let errorCode: number;
  try {
    errorCode = parseInt(codeRaw, 16);
  } catch (parseErr) {
    return null;
  }

  return fromCode(errorCode, err.logs);
}