import { z } from 'zod/v4';

const DepObject = z.object({
  currentValue: z.string().optional(),
  datasource: z.string().optional(),
  depName: z.string().optional(),
  packageName: z.string().optional(),
  currentDigest: z.string().optional(),
  versioning: z.string().optional(),
  depType: z.string().optional(),
  registryUrl: z.string().optional(),
  extractVersion: z.string().optional(),
  indentation: z.string().optional(),
});

export const QueryResultZod = z
  .union([z.array(DepObject), DepObject])
  .transform((input) => {
    return Array.isArray(input) ? input : [input];
  });