import { readFile } from "node:fs/promises";
import { join, relative } from "node:path";

import { select, Separator } from "@inquirer/prompts";

import { theme } from "../lib/theme.mjs";

export default async (context) => {
  const adapter =
    context.env.options.deploy ??
    (context.env.hasOptions
      ? "none"
      : !context.props.custom
        ? "vercel"
        : await select(
            {
              message: "Deployment adapter",
              choices: [
                {
                  name: "None",
                  value: "none",
                  description: "No deployment adapter",
                },
                new Separator(),
                {
                  name: "Docker",
                  value: "docker",
                  description:
                    "Build a Docker image and deploy to a container registry",
                },
                {
                  name: "Vercel",
                  value: "vercel",
                  description: "Deploy to Vercel",
                },
                {
                  name: "Netlify",
                  value: "netlify",
                  description: "Deploy to Netlify",
                },
                {
                  name: "Cloudflare Workers/Pages",
                  value: "cloudflare",
                  description: "Deploy to Cloudflare Workers/Pages",
                },
                {
                  name: "Bun",
                  value: "bun",
                  description: "Deploy to Bun runtime",
                },
                {
                  name: "Deno",
                  value: "deno",
                  description: "Deploy to Deno runtime",
                },
                {
                  name: "Azure Functions",
                  value: "azure",
                  description:
                    "Deploy to Azure Functions with streaming support",
                },
                {
                  name: "Azure Static Web Apps",
                  value: "azure-swa",
                  description: "Deploy to Azure Static Web Apps",
                },
                {
                  name: "AWS",
                  value: "aws",
                  description: "Deploy to AWS Lambda",
                },
              ],
              theme,
            },
            context
          ));

  const partials = {
    ...context.partials,
  };
  const files = context.files;
  const adapterName = {
    vercel: "Vercel",
    netlify: "Netlify",
    cloudflare: "Cloudflare Workers/Pages",
    azure: "Azure Functions",
    "azure-swa": "Azure Static Web Apps",
    bun: "Bun",
    deno: "Deno",
    aws: "AWS Lambda",
  };
  const adapterIgnore = {
    vercel: [".vercel", "vercel.json"],
    netlify: ["netlify.toml", "netlify", ".netlify"],
    cloudflare: [".cloudflare", ".wrangler", "wrangler.toml"],
    azure: [".azure"],
    "azure-swa": [".azure-swa", "staticwebapp.config.json"],
    bun: [".bun"],
    deno: [".deno"],
    aws: [".aws", "template.json", "samconfig.toml"],
  };

  if (adapter in adapterIgnore) {
    partials["react-server.config.json"] = {
      ...partials?.["react-server.config.json"],
      type: "json",
      merge: [
        ...(partials?.["react-server.config.json"]?.merge ?? []),
        {
          adapter,
        },
      ],
    };
    if (context.features.includes("git")) {
      partials[".gitignore"] = {
        ...partials?.[".gitignore"],
        merge: [
          ...(partials?.[".gitignore"]?.merge ?? []),
          `\n# ${adapterName[adapter] ?? adapter}\n${adapterIgnore[adapter].join("\n")}\n`,
        ],
      };
    }
  }

  if (adapter === "docker") {
    const merge = [
      await readFile(join(context.env.templateDir, ".dockerignore"), "utf8"),
    ];
    merge.push(
      ...Object.keys(context.partials).map((filename) => `!${filename}`)
    );
    partials[".dockerignore"] = {
      ...partials[".dockerignore"],
      type: "text",
      merge: [...(partials[".dockerignore"]?.merge ?? []), ...merge],
    };
    partials["README.md"] = {
      ...context.partials["README.md"],
      template: `${context.partials["README.md"].template}\n\n${await readFile(join(context.env.templateDir, "README.docker.md"), "utf8")}`,
    };
  }

  return {
    ...context,
    files,
    props: {
      ...context.props,
      adapter,
    },
    partials,
  };
};

export async function prepare(context) {
  const { adapter, packageManager } = context.props;
  if (adapter === "docker") {
    const dockerfile = join(
      context.env.templateDir,
      `Dockerfile.${packageManager.name}`
    );
    return {
      ...context,
      partials: {
        ...context.partials,
        Dockerfile: {
          type: "text",
          content: await readFile(dockerfile, "utf8"),
        },
        ".dockerignore": {
          ...context.partials[".dockerignore"],
          merge: [
            ...(context.partials[".dockerignore"]?.merge ?? []),
            `!${context.props.packageManager.lock}`,
            ...context.files
              .map((filename) =>
                Array.isArray(filename)
                  ? `!${relative(context.env.projectDir, filename[1])}`
                  : `!${relative(context.env.templateDir, filename)}`
              )
              .filter((filename) => filename.split("/").length === 1),
          ],
        },
      },
    };
  }
  return context;
}