import { z } from "zod";

/** Kinds a library folder indexes; everything else in the folder is ignored. */
export const MediaLibraryKindSchema = z.enum(["photo", "video"]);
export type MediaLibraryKind = z.infer<typeof MediaLibraryKindSchema>;

export const MediaLibrarySourceStatusSchema = z.enum(["idle", "syncing", "failed"]);
export type MediaLibrarySourceStatus = z.infer<typeof MediaLibrarySourceStatusSchema>;

export const MediaLibrarySourceSchema = z.object({
  id: z.string(),
  path: z.string(),
  status: MediaLibrarySourceStatusSchema,
  error: z.string().nullable(),
  assetCount: z.number().int().nonnegative(),
  lastSyncedAt: z.string().nullable(),
});
export type MediaLibrarySource = z.infer<typeof MediaLibrarySourceSchema>;

export const MediaLibrarySourceAddInputSchema = z.object({
  path: z.string().trim().min(1).max(4096),
});
export type MediaLibrarySourceAddInput = z.infer<typeof MediaLibrarySourceAddInputSchema>;

/** Arguments of the `search_media` agent tool. */
export const MediaLibrarySearchInputSchema = z.object({
  query: z.string().trim().max(200).optional(),
  kind: MediaLibraryKindSchema.optional(),
  taken_after: z.string().datetime({ offset: true }).optional(),
  taken_before: z.string().datetime({ offset: true }).optional(),
  limit: z.number().int().min(1).max(50).optional(),
});
export type MediaLibrarySearchInput = z.infer<typeof MediaLibrarySearchInputSchema>;

/** Arguments of the `import_media` agent tool. */
export const MediaLibraryImportInputSchema = z.object({
  asset_id: z.string().min(1).max(64),
});
export type MediaLibraryImportInput = z.infer<typeof MediaLibraryImportInputSchema>;