import { z } from "zod";
import zodToCamelCase from "zod-to-camel-case";

import { programmeFieldsSchema } from "@/schema/public/components/programmeFields/programmeFields.schema";
import { subjectSlugs } from "@/schema/public/components/base/base.schema";

export const queriesSchema = z.enum([
  "pupilSubjectListingQuery",
  "pupilProgrammeListingQuery",
  "pupilUnitListingQuery",
  "pupilLessonListingQuery",
  "pupilLessonQuery",
  "unitListingQuery",
  "lessonListingQuery",
  "programmeListingQuery",
  "subjectListingQuery",
  "lessonOverviewQuery",
  "lessonDownloadsQuery",
  "lessonBetaDownloadsQuery",
  "lessonShareQuery",
  "lessonMediaClipsQuery",
  "teacherPreviewLessonQuery",
  "teachersPreviewLessonDownloadQuery",
  "teachersPreviewUnitListingQuery",
]);

export const journeysSchema = z.enum(["pupil", "teacher", "curriculum"]);

export const journeysAndQueriesSchema = z.union([
  queriesSchema,
  journeysSchema,
]);

export const actionsSchema = z
  .object({
    exclusions: z.array(journeysAndQueriesSchema),
    opt_out: z.array(journeysAndQueriesSchema),
    programme_field_overrides: programmeFieldsSchema.partial(),
    display_expiring_banner: z.boolean(),
    disable_pupil_share: z.boolean(),
    display_PE_title: z.boolean(),
    display_vocab_button: z.boolean(),
    override_pathway: z.string(), // we have to override pathway in this way because we don't have the functionality to override different selection of programme fields across different queries
    is_pe_practical: z.boolean(),
    related_subject_slugs: z.array(subjectSlugs),
    group_units_as: z.string(),
    subject_category_actions: z.object({
      all_disabled: z.boolean(),
      default_category_id: z.number().int().positive(),
      group_by_subjectcategory: z.boolean(),
    }),
  })
  .partial();
export type Actions = z.infer<typeof actionsSchema>;

export const actionsSchemaCamel = zodToCamelCase(actionsSchema);
export type ActionsCamel = z.infer<typeof actionsSchemaCamel>;