documind/extractor/src/utils/convertToZodSchema.js

Intelligent document processing and extraction.

import { z } from 'zod';

/**
 * Converts an array of field definitions into a Zod schema.
 * @param {Array} object - Array of field definitions.
 * @returns {ZodObject} - A Zod object schema.
 */
export const convertToZodSchema = (object) => {
  const createZodSchema = (fields) => {
    const schema = {};

    fields.forEach((item) => {
      let zodType;

      switch (item.type) {
        case 'string':
          zodType = z.string();
          break;

        case 'number':
          zodType = z.number();
          break;

        case 'boolean':
          zodType = z.boolean();
          break;

        case 'enum':
          if (item.values && Array.isArray(item.values)) {
            zodType = z.enum(item.values);
          } else {
            throw new Error(`Invalid "enum" type definition for ${item.name}`);
          }
          break;

        case 'object':
          if (item.children) {
            zodType = z.object(createZodSchema(item.children));
          } else {
            throw new Error(`Invalid "object" type definition for ${item.name}`);
          }
          break;

        case 'array':
          if (item.children && item.children.length > 0) {
            if (item.children.length === 1) {
              const singleChild = item.children[0];
              const childSchema = createZodSchema([singleChild]);
              const childType = childSchema[singleChild.name];
              
              zodType = z.array(childType);
            } else {
              const arraySchema = createZodSchema(item.children);
              zodType = z.array(z.object(arraySchema));
            }
          } else {
            throw new Error(
              `Invalid or unsupported "array" type definition for ${item.name}`
            );
          }
          break;

        default:
          throw new Error(`Unsupported type "${item.type}" for field ${item.name}`);
      }

      if (item.description) {
        zodType = zodType.describe(item.description);
      }

      schema[item.name] = zodType;
    });

    return schema;
  };

  return z.object(createZodSchema(object));
};