import { createResolver, defineIntegration } from "astro-integration-kit";
import dotenv from "dotenv";
import { readFileSync, writeFileSync, existsSync } from "node:fs";
import type { CollectionModel } from "pocketbase";
import { Config, Credentials, defaultConfig, fetchCollections } from "zod-pocketbase-continue";
import { generate } from "zod-pocketbase-continue/server";

dotenv.config();

export const integration = defineIntegration({
  name: "astro-pocketbase-continue",
  optionsSchema: Config.omit({ adminEmail: true, adminPassword: true, output: true, url: true }).default(defaultConfig),
  setup({ options }) {
    const { resolve } = createResolver(import.meta.url);

    let collections: CollectionModel[] = [];

    return {
      hooks: {
        "astro:config:setup": async (params) => {
          const { addDevToolbarApp, addMiddleware, config, logger } = params;
          const { srcDir } = config;

          const {
            ASTRO_POCKETBASE_ADMIN_EMAIL: adminEmail,
            ASTRO_POCKETBASE_ADMIN_PASSWORD: adminPassword,
            PUBLIC_ASTRO_POCKETBASE_URL: url,
          } = process.env;

          try {
            const output = `${srcDir.pathname}lib/pocketbase/schemas.ts`;
            const config = Config.parse({ ...options, adminEmail, adminPassword, url, output });
            const credentials = Credentials.parse(config);
            const allCollections = await fetchCollections(credentials);
            collections = allCollections.filter(({ name }) => !config.ignore.includes(name));
            await generate(collections, config);
          } catch (error) {
            logger.error(error instanceof Error ? error.message : "unknown error");
          }

          if (!existsSync(new URL("lib/pocketbase/loader.ts", srcDir))) {
            const loaderContent = readFileSync(resolve("../assets/loader.ts"), "utf-8");
            writeFileSync(new URL("lib/pocketbase/loader.ts", srcDir), loaderContent);
          }

          if (!existsSync(new URL("lib/pocketbase/middleware.ts", srcDir))) {
            const middlewareContent = readFileSync(resolve("../assets/middleware.ts"), "utf-8");
            writeFileSync(new URL("lib/pocketbase/middleware.ts", srcDir), middlewareContent);
          }
          addMiddleware({ entrypoint: new URL("lib/pocketbase/middleware.ts", srcDir), order: "pre" });

          addDevToolbarApp({
            id: "astro-pocketbase-continue",
            name: "Astro PocketBase Continued",
            icon: `<svg role="img" viewBox="0 0 24 24" xmlns="http://www.w3.org/2000/svg"><title>PocketBase</title><path fill="currentColor" d="M5.684 12a.632.632 0 0 1-.631-.632V4.421c0-.349.282-.632.631-.632h2.37c.46 0 .889.047 1.287.139.407.084.758.23 1.053.44.303.202.541.475.715.82.173.335.26.75.26 1.246 0 .479-.092.894-.273 1.247a2.373 2.373 0 0 1-.715.869 3.11 3.11 0 0 1-1.053.503c-.398.11-.823.164-1.273.164h-.46a.632.632 0 0 0-.632.632v1.52a.632.632 0 0 1-.632.631Zm1.279-4.888c0 .349.283.632.632.632h.343c1.04 0 1.56-.437 1.56-1.31 0-.428-.135-.73-.404-.907-.26-.176-.645-.264-1.156-.264h-.343a.632.632 0 0 0-.632.631Zm6.3 13.098a.632.632 0 0 1-.631-.631v-6.947a.63.63 0 0 1 .631-.632h2.203c.44 0 .845.034 1.216.1.38.06.708.169.984.328.276.16.492.37.647.63.164.26.246.587.246.982 0 .185-.03.37-.09.554a1.537 1.537 0 0 1-.26.516 1.857 1.857 0 0 1-1.076.7.031.031 0 0 0-.023.03c0 .015.01.028.025.03.591.111 1.04.32 1.346.626.311.31.466.743.466 1.297 0 .42-.082.78-.246 1.083a2.153 2.153 0 0 1-.685.755 3.4 3.4 0 0 1-1.036.441 5.477 5.477 0 0 1-1.268.139zm1.271-5.542c0 .349.283.631.632.631h.21c.465 0 .802-.088 1.009-.264.207-.176.31-.424.31-.743 0-.302-.107-.516-.323-.642-.207-.135-.535-.202-.984-.202h-.222a.632.632 0 0 0-.632.632Zm0 3.463c0 .349.283.631.632.631h.39c1.019 0 1.528-.369 1.528-1.108 0-.36-.125-.621-.376-.78-.241-.16-.625-.24-1.152-.24h-.39a.632.632 0 0 0-.632.632zM1.389 0C.629 0 0 .629 0 1.389V15.03a1.4 1.4 0 0 0 1.389 1.39H8.21a.632.632 0 0 0 .63-.632.632.632 0 0 0-.63-.63H1.389c-.078 0-.125-.05-.125-.128V1.39c0-.078.047-.125.125-.125H15.03c.078 0 .127.047.127.125v6.82a.632.632 0 0 0 .631.63.632.632 0 0 0 .633-.63V1.389A1.4 1.4 0 0 0 15.032 0ZM15.79 7.578a.632.632 0 0 0-.632.633.632.632 0 0 0 .631.63h6.822c.078 0 .125.05.125.128V22.61c0 .078-.047.125-.125.125H8.97c-.077 0-.127-.047-.127-.125v-6.82a.632.632 0 0 0-.631-.63.632.632 0 0 0-.633.63v6.822A1.4 1.4 0 0 0 8.968 24h13.643c.76 0 1.389-.629 1.389-1.389V8.97a1.4 1.4 0 0 0-1.389-1.39Z"/></svg>`,
            entrypoint: resolve("../assets/toolbar.ts"),
          });
        },

        "astro:config:done": ({ injectTypes }) => {
          const content = readFileSync(resolve("../assets/env.d.ts"), "utf-8");
          injectTypes({ filename: "env.d.ts", content });
        },

        "astro:server:setup": ({ refreshContent, toolbar }) => {
          toolbar.on("astro-pocketbase-continue:refresh", async () => {
            if (!refreshContent) console.warn("Content can only be refreshed in Astro v5.0.0 or higher");
            await refreshContent?.({ loaders: ["pocketbase-loader"] });
          });
        },
      },
    };
  },
});