import path from "path";
import * as z from 'zod';

import { BuilderError } from "../internal/core/errors";
import { ERRORS } from "../internal/core/errors-list";
import { ASSETS_DIR } from "../internal/core/project-structure";
import { parseZodError } from "../internal/core/validation-errors";
import { Accounts, ASADef, ASADefs } from "../types";
import { ASADefsSchema } from "../types-input";
import { loadFromYamlFileSilentWithMessage } from "./files";

function validateSingle (accounts: Accounts, filename: string, asaDef: ASADef): void {
  if (!asaDef.optInAccNames || asaDef.optInAccNames.length === 0) {
    return;
  }
  for (const accName of asaDef.optInAccNames) {
    if (!accounts.get(accName)) {
      throw new BuilderError(
        ERRORS.SCRIPT.ASA_PARAM_ERROR_NO_NAMED_OPT_IN_ACCOUNT, {
          filename: filename,
          optInAccName: accName
        });
    }
  }
}

function validateParsedASADefs (accounts: Accounts, asaDefs: ASADefs, filename: string): void {
  for (const def of Object.values(asaDefs)) {
    validateSingle(accounts, filename, def);
  }
}

export function validateASADefs (obj: Object, accounts: Accounts, filename: string): ASADefs {
  try {
    const parsed = ASADefsSchema.parse(obj);
    Object.keys(parsed).forEach(k => {
      if (parsed[k].defaultFrozen === undefined) {
        parsed[k].defaultFrozen = false;
      }
    });
    validateParsedASADefs(accounts, parsed, filename);
    return parsed;
  } catch (e) {
    if (e instanceof z.ZodError) {
      throw new BuilderError(
        ERRORS.SCRIPT.ASA_PARAM_PARSE_ERROR, {
          reason: parseZodError(e),
          filename: filename
        }, e);
    }
    throw e;
  }
}

export function loadASAFile (accounts: Accounts): ASADefs {
  const filename = path.join(ASSETS_DIR, "asa.yaml");
  return validateASADefs(
    loadFromYamlFileSilentWithMessage(filename, path.join(ASSETS_DIR, "asa.yaml") + " doesn't exist."),
    accounts,
    filename);
}