import { asyncFn, basePartial, fn, globPartial, injectPartial, requiredGlobDirectoryPartial, requiredSwDestPartial } from "@serwist/build/schema";
import { z } from "zod";

export const hooks = z.object({
  beforeBuildServiceWorker: z
    .union([
      fn({
        input: [z.any()],
        output: z.void(),
      }),
      asyncFn({
        input: [z.any()],
        output: z.void(),
      }),
    ])
    .optional(),
  closeBundleOrder: z.union([z.literal("pre"), z.literal("post"), z.null()]).optional(),
  configureOptions: z
    .union([
      fn({
        input: [z.any(), z.any()],
        output: z.void(),
      }),
      asyncFn({
        input: [z.any(), z.any()],
        output: z.void(),
      }),
    ])
    .optional(),
});

export const devOptions = z.object({
  bundle: z.boolean().default(true),
  minify: z.union([z.boolean(), z.literal("terser"), z.literal("esbuild")]).default(false),
});

export const injectManifestPartial = z.object({
  mode: z.union([z.literal("development"), z.literal("production")]),
  type: z.union([z.literal("classic"), z.literal("module")]).default("classic"),
  scope: z.string(),
  base: z.string(),
  disable: z.boolean().default(false),
  integration: hooks.default({}),
  swUrl: z.string().default("/sw.js"),
  plugins: z.array(z.any()).default([]),
  rollupFormat: z.union([z.literal("es"), z.literal("iife")]).default("es"),
  rollupOptions: z.record(z.string(), z.any()).default({}),
  devOptions: devOptions.prefault({}),
});

export const injectManifestOptions = z.strictObject({
  ...basePartial.shape,
  ...globPartial.shape,
  ...injectPartial.shape,
  ...requiredSwDestPartial.shape,
  ...requiredGlobDirectoryPartial.shape,
  ...injectManifestPartial.shape,
});