import type { Client } from '@notionhq/client';
import { z } from 'astro/zod';
import * as rawPropertyType from './schemas/raw-properties.js';
import type { DatabasePropertyConfigResponse } from './types.js';

export async function propertiesSchemaForDatabase(client: Client, databaseId: string) {
  const database = await client.databases.retrieve({ database_id: databaseId });

  const schemaForDatabaseProperty: (propertyConfig: DatabasePropertyConfigResponse) => z.ZodTypeAny = (
    propertyConfig
  ) => rawPropertyType[propertyConfig.type];

  const schema = Object.fromEntries(
    Object.entries(database.properties).map(([key, value]: [string, DatabasePropertyConfigResponse]) => {
      let propertySchema = schemaForDatabaseProperty(value);
      if (value.description) {
        propertySchema = propertySchema.describe(value.description);
      }
      if (key !== 'Name') {
        // propertySchema = propertySchema.optional();
      }

      return [key, propertySchema];
    })
  );

  return z.object(schema);
}