From 30e2bee5e674d2e5d9bee5ee36a9af908ab50d7c Mon Sep 17 00:00:00 2001
From: David Enke <david@enke.dev>
Date: Tue, 12 Nov 2024 21:42:58 +0100
Subject: [PATCH] feat: handle file collections

---
 src/utils/transform.utils.ts | 29 +++++++++++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)

diff --git a/src/utils/transform.utils.ts b/src/utils/transform.utils.ts
index c22c4f7..08b15a2 100644
--- a/src/utils/transform.utils.ts
+++ b/src/utils/transform.utils.ts
@@ -91,8 +91,33 @@ export function transformCollection(
   const { format = 'frontmatter' } = collection;
   if (format === 'frontmatter') {
     collection.fields = collection.fields?.filter(({ name }) => name !== 'body');
+    collection.files = collection.files?.reduce((acc, file) => {
+      return [...acc, { ...file, fields: file.fields.filter(({ name }) => name !== 'body') }];
+    }, [] as Decap.CmsCollectionFile[]);
   }
 
-  // we can process the collection with its fields like every other object
-  return transformObjectField(collection as Decap.CmsFieldBase & Decap.CmsFieldObject, zod);
+  // we can process folder collections with its fields like every other object
+  if ('folder' in collection) {
+    const { name, fields = [] } = collection;
+    const field = { name, fields, widget: 'object' as const };
+    return transformObjectField(field, zod);
+  }
+
+  // file collections are for now union types of all defined types; that might
+  // change in the future, as it should be possible to define different collection
+  // data types in astro... (?)
+  if (Array.isArray(collection.files)) {
+    const results = collection.files.map(file => {
+      const { name, fields = [] } = file;
+      const field = { name, fields, widget: 'object' as const };
+      return transformObjectField(field, zod);
+    });
+    return {
+      cptime: `z.union([${results.map(({ cptime }) => cptime).join(', ')}])`,
+      runtime: zod.union(results.map(({ runtime }) => runtime) as any),
+    };
+  }
+
+  // a collection without a folder OR a file list is invalid, thus we define `never`
+  return { runtime: zod.never(), cptime: 'z.never()' };
 }