import { z } from 'zod';
import { TOOL_TYPE } from '../../common';
import { JOB_STATUS } from '../../constants';
import { WRITER_DOCUMENT_STAGE } from '../../constants';

export const UnifiedToolJobSchema = z.object({
    uuid: z.string().uuid(),
    title: z.string(),
    type: z.nativeEnum(TOOL_TYPE),
    status: z.nativeEnum(JOB_STATUS).optional().nullable(),
    stage: z.nativeEnum(WRITER_DOCUMENT_STAGE).optional().nullable(),
    documentTypeId: z.string().uuid().optional().nullable(),
    createdAt: z.date().or(z.string()),
    updatedAt: z.date().or(z.string()),
});

export type UnifiedToolJob = z.infer<typeof UnifiedToolJobSchema>;