import { z } from 'zod';
import { USER_REACTION, JOB_STATUS } from '../../constants';
import { AttachedFileSchema } from './attached-file.schema';

export const SolvingEduTaskJobParamsSchema = z.object({
    attachedFiles: z.array(AttachedFileSchema),
});

export type SolvingEduTaskJobParams = z.infer<typeof SolvingEduTaskJobParamsSchema>;

export const SolvingEduTaskJobSchema = z.object({
    uuid: z.string().uuid(),
    prompt: z.string(),
    aiOutput: z.string().nullable().optional(),
    reaction: z.nativeEnum(USER_REACTION).nullable(),
    title: z.string(),
    status: z.nativeEnum(JOB_STATUS),
    modelId: z.string().uuid(),
    price: z.number(),
    params: SolvingEduTaskJobParamsSchema,
    error: z.string().nullable(),
    userId: z.string().uuid().nullable().optional(),
    createdAt: z.date(),
    updatedAt: z.date(),
});

export type SolvingEduTaskJob = z.infer<typeof SolvingEduTaskJobSchema>;