import { z } from 'zod/v4';
import { Toml } from '../../../util/schema-utils/index.ts';

const BuildpackByName = z.object({
  id: z.string(),
  version: z.string().optional(),
});

const BuildpackByURI = z.object({
  uri: z.string(),
});

const BuildpackGroup = BuildpackByName.or(BuildpackByURI);

type BuildpackByName = z.infer<typeof BuildpackByName>;
type BuildpackByURI = z.infer<typeof BuildpackByURI>;
type BuildpackGroup = z.infer<typeof BuildpackGroup>;

export function isBuildpackByName(
  group: BuildpackGroup,
): group is BuildpackByName {
  return 'id' in group;
}

export function isBuildpackByURI(
  group: BuildpackGroup,
): group is BuildpackByURI {
  return 'uri' in group;
}

const IoBuildpacks = z.object({
  builder: z.string().optional(),
  group: z.array(BuildpackGroup).optional(),
});

export const ProjectDescriptor = z.object({
  _: z.object({
    'schema-version': z.string(),
  }),
  io: z
    .object({
      buildpacks: IoBuildpacks.optional(),
    })
    .optional(),
});

export type ProjectDescriptor = z.infer<typeof ProjectDescriptor>;
export const ProjectDescriptorToml = Toml.pipe(ProjectDescriptor);