import type { PackageScripts } from "@pnpm/types";
import { omit, pick } from "remeda";
import type { IsolateConfigResolved } from "../config";
import { usePackageManager } from "../package-manager";
import type { PackageManifest, PackagesRegistry } from "../types";
import {
  adaptManifestInternalDeps,
  adoptPnpmFieldsFromRoot,
  resolveCatalogDependencies,
} from "./helpers";

/**
 * Adapt the output package manifest, so that:
 *
 * - Its internal dependencies point to the isolated ./packages/* directory.
 * - The devDependencies are possibly removed
 * - Scripts are picked or omitted and otherwise removed
 */
export async function adaptTargetPackageManifest({
  manifest,
  packagesRegistry,
  workspaceRootDir,
  config,
}: {
  manifest: PackageManifest;
  packagesRegistry: PackagesRegistry;
  workspaceRootDir: string;
  config: IsolateConfigResolved;
}): Promise<PackageManifest> {
  const packageManager = usePackageManager();
  const {
    includeDevDependencies,
    pickFromScripts,
    omitFromScripts,
    omitPackageManager,
    forceNpm,
  } = config;

  /** Dev dependencies are omitted by default */
  const inputManifest = includeDevDependencies
    ? manifest
    : omit(manifest, ["devDependencies"]);

  /** Resolve catalog dependencies before adapting internal deps */
  const manifestWithResolvedCatalogs = {
    ...inputManifest,
    dependencies: await resolveCatalogDependencies(
      inputManifest.dependencies,
      workspaceRootDir,
    ),
  };

  const adaptedManifest =
    packageManager.name === "pnpm" && !forceNpm
      ? /**
         * For PNPM the output itself is a workspace so we can preserve the specifiers
         * with "workspace:*" in the output manifest, but we do want to adopt the
         * pnpm.overrides field from the root package.json.
         */
        await adoptPnpmFieldsFromRoot(
          manifestWithResolvedCatalogs,
          workspaceRootDir,
        )
      : /** For other package managers we replace the links to internal dependencies */
        adaptManifestInternalDeps({
          manifest: manifestWithResolvedCatalogs,
          packagesRegistry,
        });

  return {
    ...adaptedManifest,
    /**
     * Adopt the package manager definition from the root manifest if available.
     * The option to omit is there because some platforms might not handle it
     * properly (Cloud Run, April 24th 2024, does not handle pnpm v9)
     */
    packageManager: omitPackageManager
      ? undefined
      : packageManager.packageManagerString,
    /**
     * Scripts are removed by default if not explicitly picked or omitted via
     * config.
     */
    scripts: pickFromScripts
      ? (pick(manifest.scripts ?? {}, pickFromScripts) as PackageScripts)
      : omitFromScripts
        ? (omit(manifest.scripts ?? {}, omitFromScripts) as PackageScripts)
        : {},
  };
}