import { z } from 'zod';

export const TaskTypeSchema = z.enum([
  'debug',
  'analyze',
  'review',
  'design',
  'second_opinion',
  'reality_check',
  'synthesis',
]);

export const SpecializedTaskRequestSchema = z.object({
  task_type: TaskTypeSchema,
  context: z.string(),
  files: z.array(z.string()).optional(),
  model: z.string().optional(),
});

export const ReadFileRequestSchema = z.object({
  file_path: z.string(),
});

export const CreateDiffRequestSchema = z.object({
  file_path: z.string(),
  new_content: z.string(),
  reason: z.string(),
});

export const ApplyDiffRequestSchema = z.object({
  operation_id: z.string(),
  approved: z.boolean(),
});

export type SpecializedTaskRequest = z.infer<typeof SpecializedTaskRequestSchema>;
export type ReadFileRequest = z.infer<typeof ReadFileRequestSchema>;
export type CreateDiffRequest = z.infer<typeof CreateDiffRequestSchema>;
export type ApplyDiffRequest = z.infer<typeof ApplyDiffRequestSchema>;