import { z } from 'zod';

import { isoDateTimeSchema, timelineMessageBase } from './base';
import { turnOriginSchema } from './turn-origin';

export const turnUsageSchema = z.object({
  input_tokens: z.number().int().nonnegative().optional(),
  output_tokens: z.number().int().nonnegative().optional(),
  cached_tokens: z.number().int().nonnegative().optional(),
  cost: z.number().nonnegative().optional(),
});

export type TurnUsage = z.infer<typeof turnUsageSchema>;

export const turnMessageSchema = z.object({
  type: z.literal('turn'),
  ...timelineMessageBase,
  turn_id: z.string().min(1),
  ordinal: z.number().int().nonnegative(),
  status: z.enum(['running', 'completed']),
  origin: turnOriginSchema,
  user_message_id: z.string().min(1).optional(),
  attachment_ids: z.array(z.string().min(1)).optional(),
  started_at: isoDateTimeSchema.optional(),
  ended_at: isoDateTimeSchema.optional(),
  usage: turnUsageSchema.optional(),
  duration_ms: z.number().nonnegative().optional(),
});

export type TurnMessage = z.infer<typeof turnMessageSchema>;