feat: support for component schemas

Author: cyp3rius

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sensinum/astro-strapi-loader",
-  "version": "1.0.7",
+  "version": "1.0.8",
   "description": "Astro loader for Strapi CMS",
   "keywords": [
     "astro",

src/types/strapi.ts

@@ -34,6 +34,19 @@ export interface StrapiContentType {
   };
 }
 
+export interface StrapiComponent {
+  uid: string;
+  category: string;
+  apiId: string;
+  schema: {
+    displayName: string;
+    description: string;
+    icon: string;
+    collectionName: string;
+    attributes: Array<StrapiAttribute>;
+  }
+}
+
 export interface StrapiAttribute {
   type: string;
   required?: boolean;

src/utils/__tests__/schema.test.ts

@@ -35,7 +35,7 @@ describe("StrapiSchemaGenerator", () => {
   let generator: StrapiSchemaGenerator;
 
   beforeEach(() => {
-    generator = new StrapiSchemaGenerator([mockContentType]);
+    generator = new StrapiSchemaGenerator([mockContentType], []);
   });
 
   describe("generateSchema", () => {

src/utils/schema.ts

@@ -1,12 +1,14 @@
 import { z } from "zod";
-import type { StrapiContentType, StrapiAttribute } from "../types/strapi";
+import type { StrapiComponent, StrapiContentType, StrapiAttribute } from "../types/strapi";
 
 export class StrapiSchemaGenerator {
   private contentTypes: Array<StrapiContentType> = [];
+  private components: Array<StrapiComponent> = [];
   private strict: boolean = false;
 
-  constructor(contentTypes: Array<StrapiContentType>, strict: boolean = false) {
+  constructor(contentTypes: Array<StrapiContentType>, components: Array<StrapiComponent>, strict: boolean = false) {
     this.contentTypes = contentTypes;
+    this.components = components;
     this.strict = strict;
   }
 
@@ -123,8 +125,16 @@ export class StrapiSchemaGenerator {
       case "component":
         if (!attribute.component)
           throw new Error("Component type requires component name");
-        // TODO: Implement component schema generation
-        return z.any();
+        const component = this.components.find(
+          (component) => component.uid === attribute.component,
+        );
+        if (!component)
+          throw new Error(`Component ${attribute.component} not found`);
+
+        if (attribute.repeatable) {
+          return z.array(this.generateComponentSchema(component.schema));
+        }
+        return this.generateComponentSchema(component.schema);
 
       case "dynamiczone":
         // TODO: Implement dynamiczone schema generation
@@ -172,6 +182,16 @@ export class StrapiSchemaGenerator {
     return z.object(shape);
   }
 
+  private generateComponentSchema(componentSchema: StrapiComponent["schema"]): z.ZodObject<any> {
+    const ref = this;
+    const shape: Record<string, z.ZodTypeAny> = Object.entries(componentSchema.attributes).reduce((acc, [key, attribute]) => {
+      const schema = ref.generateAttributeSchema(attribute);
+      return { ...acc, [key]: schema };
+    }, {});
+
+    return z.object(shape);
+  }
+
   public generateSchema(contentTypeName: string): z.ZodObject<any> {
     const contentType = this.contentTypes.find(
       (contentType) => contentType.apiID === contentTypeName,

src/utils/strapi.ts

@@ -1,6 +1,6 @@
 import { z, defineCollection, CollectionConfig } from "astro:content";
 
-import type { StrapiContentType, StrapiResponse } from "../types/strapi";
+import type { StrapiComponent, StrapiContentType, StrapiResponse } from "../types/strapi";
 import { StrapiSchemaGenerator } from "./schema";
 import { strapiLoader } from "./loader";
 
@@ -55,6 +55,18 @@ export async function fetchContentTypes(
   return contentTypes.data.filter((contentType) => !contentType.plugin);
 }
 
+export async function fetchComponents(
+  options: Omit<StrapiRequestOptions, "path">,
+): Promise<Array<StrapiComponent>> {
+  const components = await strapiRequest<
+    StrapiResponse<Array<StrapiComponent>>
+  >({
+    ...options,
+    path: "content-type-builder/components",
+  });
+  return components.data;
+}
+
 export async function fetchContent(
   options: Omit<StrapiRequestOptions, "path"> & { contentType: string },
 ): Promise<any> {
@@ -73,7 +85,8 @@ export async function generateStrapiSchema(
   const { url, token, strict } = options;
 
   const contentTypes = await fetchContentTypes({ url, token });
-  const schemaGenerator = new StrapiSchemaGenerator(contentTypes, strict);
+  const components = await fetchComponents({ url, token });
+  const schemaGenerator = new StrapiSchemaGenerator(contentTypes, components, strict);
   return schemaGenerator.generateAllSchemas();
 }