import { existsSync } from "node:fs";
import { fileURLToPath, pathToFileURL } from "node:url";
import { join } from "node:path";

import { moduleAliases } from "../loader/module-alias.mjs";
import { applyAlias } from "./utils.mjs";
import * as sys from "../sys.mjs";

const alias = moduleAliases();

const cwd = sys.cwd();
let options, outDir;
const resolveCache = new Map();
export async function initialize(data) {
  options = data?.options || {};
  outDir = options.outDir || ".react-server";
}

export async function resolve(specifier, context, nextResolve) {
  const cacheKey = specifier + "\0" + (context.parentURL ?? "");
  const cached = resolveCache.get(cacheKey);
  if (cached) return { ...cached, shortCircuit: true };

  const result = await resolveUncached(specifier, context, nextResolve);
  resolveCache.set(cacheKey, result);
  return result;
}

async function resolveUncached(specifier, context, nextResolve) {
  switch (specifier) {
    case "@lazarv/react-server/dist/server/preload-manifest":
      return nextResolve(
        pathToFileURL(join(cwd, outDir, "server/preload-manifest.mjs")).href
      );
    case "@lazarv/react-server/dist/manifest-registry":
    case "@lazarv/react-server/dist/client/manifest-registry":
      return nextResolve(
        pathToFileURL(join(cwd, outDir, "server/client/manifest-registry.mjs"))
          .href
      );
    case "@lazarv/react-server/dist/server/client-reference-map":
      return nextResolve(
        pathToFileURL(join(cwd, outDir, "server/client-reference-map.mjs")).href
      );
    case "@lazarv/react-server/dist/server/server-reference-map":
      return nextResolve(
        pathToFileURL(join(cwd, outDir, "server/server-reference-map.mjs")).href
      );
    case "@lazarv/react-server/dist/server/build-manifest":
      return nextResolve(
        pathToFileURL(join(cwd, outDir, "server/build-manifest.mjs")).href
      );
    case "@lazarv/react-server/dist/server/action-secret":
      try {
        return await nextResolve(
          pathToFileURL(join(cwd, outDir, "server/action-secret.mjs")).href
        );
      } catch {
        return nextResolve("@lazarv/react-server/dist/server/action-secret");
      }
    case "@lazarv/react-server/dist/server/server-manifest":
    case "@lazarv/react-server/dist/server/client-manifest":
    case "@lazarv/react-server/dist/client/browser-manifest":
      return nextResolve("@lazarv/react-server/lib/loader/manifest-loader.mjs");
  }

  try {
    return await nextResolve(applyAlias(alias, specifier), context);
  } catch (e) {
    if (e.code === "ERR_MODULE_NOT_FOUND" && !specifier.endsWith(".js")) {
      const jsSpecifier = `${specifier}.js`;

      if (jsSpecifier.startsWith("file:")) {
        const candidatePath = fileURLToPath(jsSpecifier);
        if (existsSync(candidatePath)) {
          return await nextResolve(pathToFileURL(candidatePath).href, context);
        }
      } else {
        try {
          return await nextResolve(jsSpecifier, context);
        } catch {
          throw e;
        }
      }
    }
    throw e;
  }
}