prismic-ts-codegen 0.0.7 → 0.1.2

package/dist/index.js

@@ -1,6 +1,6 @@
 import { Project, ModuleDeclarationKind } from 'ts-morph';
 import { stripIndent } from 'common-tags';
-import { CustomTypeModelFieldType, CustomTypeModelLinkSelectType, CustomTypeModelSliceType } from '@prismicio/types';
+import { CustomTypeModelSliceType, CustomTypeModelFieldType, CustomTypeModelLinkSelectType } from '@prismicio/types';
 import { pascalCase as pascalCase$1, pascalCaseTransformMerge } from 'pascal-case';
 
 const BLANK_LINE_IDENTIFIER = "// ___BLANK_LINE_TO_BE_REPLACED___";
@@ -26,19 +26,24 @@ const FIELD_DOCUMENTATION_URLS = {
   GeoPoint: "https://prismic.io/docs/core-concepts/geopoint",
   Group: "https://prismic.io/docs/core-concepts/group",
   IntegrationFields: "https://prismic.io/docs/core-concepts/integration-fields",
-  Slices: "https://prismic.io/docs/core-concepts/slices"
+  Slices: "https://prismic.io/docs/core-concepts/slices",
+  LegacySlices: "https://prismic.io/docs/core-concepts/slices"
 };
 
 const isCustomTypeModelField = (model) => {
-  return typeof model === "object" && model !== null && "type" in model && "config" in model;
+  return typeof model === "object" && model !== null && "type" in model;
+};
+
+const hasOwnProperty = (obj, prop) => {
+  return obj.hasOwnProperty(prop);
 };
 
 const isCustomTypeModelSlice = (model) => {
-  return typeof model === "object" && model !== null && "non-repeat" in model && "repeat" in model;
+  return typeof model === "object" && model !== null && hasOwnProperty(model, "type") && model.type === CustomTypeModelSliceType.Slice;
 };
 
 const isSharedSliceModel = (model) => {
-  return typeof model === "object" && model !== null && "variations" in model;
+  return typeof model === "object" && model !== null && hasOwnProperty(model, "type") && model.type === CustomTypeModelSliceType.SharedSlice;
 };
 
 const getAPIIDPath = (config) => {
@@ -67,16 +72,19 @@ const isCustomTypeModel = (model) => {
 
 const getHumanReadableModelName = (config) => {
   if (isCustomTypeModel(config.model)) {
-    return config.model.label;
+    return config.model.label || config.model.id;
   } else if (isSharedSliceModel(config.model)) {
     return config.model.name;
-  } else if (isCustomTypeModelField(config.model) && "label" in config.model.config) {
-    return config.model.config.label;
+  } else if (isCustomTypeModelField(config.model)) {
+    if (config.model.config && "label" in config.model.config) {
+      return config.model.config.label || config.id;
+    } else if (config.model.config && "fieldset" in config.model) {
+      return config.model.fieldset || config.id;
+    }
   } else if (isCustomTypeModelSlice(config.model)) {
-    return config.model.fieldset;
-  } else {
-    return `Slice Zone (\`${config.id}\`)`;
+    return config.model.fieldset || config.id;
   }
+  return `\`${config.id}\``;
 };
 
 const getHumanReadableFieldPath = (config) => {
@@ -97,14 +105,15 @@ const getHumanReadableFieldPath = (config) => {
 const getHumanReadableFieldType = (config) => {
   switch (config.model.type) {
     case "StructuredText": {
-      const isTitleField = "single" in config.model.config && config.model.config.single.split(",").every((blockType) => /heading/.test(blockType));
+      const isTitleField = config.model.config && "single" in config.model.config && config.model.config.single && config.model.config.single.split(",").every((blockType) => /heading/.test(blockType));
       return isTitleField ? "Title" : "Rich Text";
     }
     case "IntegrationFields": {
-      return `Integration Fields (Catalog: \`${config.model.config.catalog}\`)`;
+      const catalog = config.model.config?.catalog;
+      return `Integration Fields (Catalog: ${catalog ? `\`${catalog}\`` : "*unknown*"})`;
     }
     case "Link": {
-      switch (config.model.config.select) {
+      switch (config.model.config?.select) {
         case CustomTypeModelLinkSelectType.Document: {
           return "Content Relationship";
         }
@@ -165,8 +174,8 @@ const buildFieldDocs = (config) => {
         const fieldType = getHumanReadableFieldType({
           model: config.model
         });
-        const placeholder = "placeholder" in config.model.config && config.model.config.placeholder ? config.model.config.placeholder : `*None*`;
-        const defaultValue = "default_value" in config.model.config ? config.model.config.default_value : void 0;
+        const placeholder = config.model.config && "placeholder" in config.model.config && config.model.config.placeholder ? config.model.config.placeholder : `*None*`;
+        const defaultValue = config.model.config && "default_value" in config.model.config ? config.model.config.default_value : void 0;
         const apiIDPath = getAPIIDPath({
           path: [...config.path, { id: config.id, model: config.model }]
         });
@@ -271,7 +280,7 @@ const addInterfacePropertyForField = (config) => {
       break;
     }
     case "Image": {
-      if (config.model.config.thumbnails.length > 0) {
+      if (config.model.config?.thumbnails && config.model.config.thumbnails.length > 0) {
         const thumbnailNames = config.model.config.thumbnails.map((thumbnail) => `"${thumbnail.name}"`).join(" | ");
         config.interface.addProperty({
           name: config.id,
@@ -286,7 +295,7 @@ const addInterfacePropertyForField = (config) => {
       } else {
         config.interface.addProperty({
           name: config.id,
-          type: "prismicT.ImageField<null>",
+          type: "prismicT.ImageField<never>",
           docs: buildFieldDocs({
             id: config.id,
             model: config.model,
@@ -298,7 +307,7 @@ const addInterfacePropertyForField = (config) => {
       break;
     }
     case "IntegrationFields": {
-      const catalogType = config.fieldConfigs.integrationFields?.catalogTypes?.[config.model.config.catalog];
+      const catalogType = config.model.config?.catalog ? config.fieldConfigs.integrationFields?.catalogTypes?.[config.model.config.catalog] : void 0;
       config.interface.addProperty({
         name: config.id,
         type: catalogType ? `prismicT.IntegrationFields<${catalogType}>` : "prismicT.IntegrationFields",
@@ -312,11 +321,12 @@ const addInterfacePropertyForField = (config) => {
       break;
     }
     case "Link": {
-      switch (config.model.config.select) {
+      switch (config.model.config?.select) {
         case CustomTypeModelLinkSelectType.Document: {
+          config.model.config;
           config.interface.addProperty({
             name: config.id,
-            type: config.model.config.customtypes && config.model.config.customtypes.length > 0 ? `prismicT.RelationField<${config.model.config.customtypes.map((type) => `"${type}"`).join(" | ")}>` : "prismicT.RelationField",
+            type: "customtypes" in config.model.config && config.model.config.customtypes && config.model.config.customtypes.length > 0 ? `prismicT.RelationField<${config.model.config.customtypes.map((type) => `"${type}"`).join(" | ")}>` : "prismicT.RelationField",
             docs: buildFieldDocs({
               id: config.id,
               model: config.model,
@@ -368,7 +378,7 @@ const addInterfacePropertyForField = (config) => {
       break;
     }
     case "StructuredText": {
-      const isTitleField = "single" in config.model.config && config.model.config.single.split(",").every((blockType) => /heading/.test(blockType));
+      const isTitleField = config.model.config && "single" in config.model.config && config.model.config.single && config.model.config.single.split(",").every((blockType) => /heading/.test(blockType));
       if (isTitleField) {
         config.interface.addProperty({
           name: config.id,
@@ -395,12 +405,12 @@ const addInterfacePropertyForField = (config) => {
       break;
     }
     case "Select": {
-      const options = config.model.config.options.map((option) => `"${option}"`).join(" | ");
-      const hasDefault = Boolean(config.model.config.default_value);
+      const options = config.model.config?.options?.map((option) => `"${option}"`).join(" | ");
+      const hasDefault = Boolean(config.model.config?.default_value);
       if (hasDefault) {
         config.interface.addProperty({
           name: config.id,
-          type: `prismicT.SelectField<${options}, "filled">`,
+          type: `prismicT.SelectField<${options || "string"}, "filled">`,
           docs: buildFieldDocs({
             id: config.id,
             model: config.model,
@@ -469,19 +479,21 @@ const addInterfacePropertyForField = (config) => {
         ],
         isExported: true
       });
-      addInterfacePropertiesForFields({
-        interface: itemInterface,
-        sourceFile: config.sourceFile,
-        fields: config.model.config.fields,
-        path: [
-          ...config.path,
-          {
-            id: config.id,
-            model: config.model
-          }
-        ],
-        fieldConfigs: config.fieldConfigs
-      });
+      if (config.model.config?.fields) {
+        addInterfacePropertiesForFields({
+          interface: itemInterface,
+          sourceFile: config.sourceFile,
+          fields: config.model.config.fields,
+          path: [
+            ...config.path,
+            {
+              id: config.id,
+              model: config.model
+            }
+          ],
+          fieldConfigs: config.fieldConfigs
+        });
+      }
       config.interface.addProperty({
         name: config.id,
         type: `prismicT.GroupField<Simplify<${itemInterface.getName()}>>`,
@@ -496,120 +508,122 @@ const addInterfacePropertyForField = (config) => {
     }
     case "Slices": {
       const choiceInterfaceNames = [];
-      for (const choiceId in config.model.config.choices) {
-        const choice = config.model.config.choices[choiceId];
-        if (choice.type === CustomTypeModelSliceType.SharedSlice) {
-          choiceInterfaceNames.push(buildSharedSliceInterfaceName({ id: choiceId }));
-        } else if (choice.type === CustomTypeModelSliceType.Slice) {
-          let primaryInterface;
-          if (Object.keys(choice["non-repeat"]).length > 0) {
-            primaryInterface = config.sourceFile.addInterface({
-              name: pascalCase(`${config.path[0].id} Document Data ${config.id} ${choiceId} Slice Primary`),
-              docs: [
-                {
-                  description: (writer) => {
-                    const humanReadablePath = getHumanReadableFieldPath({
-                      path: [
-                        ...config.path,
-                        {
-                          id: config.id,
-                          model: config.model
-                        },
-                        {
-                          id: choiceId,
-                          model: choice
-                        },
-                        {
-                          id: "primary",
-                          label: "Primary"
-                        }
-                      ]
-                    });
-                    writer.writeLine(`Primary content in ${humanReadablePath}`);
+      if (config.model.config?.choices) {
+        for (const choiceId in config.model.config.choices) {
+          const choice = config.model.config.choices[choiceId];
+          if (choice.type === CustomTypeModelSliceType.SharedSlice) {
+            choiceInterfaceNames.push(buildSharedSliceInterfaceName({ id: choiceId }));
+          } else if (choice.type === CustomTypeModelSliceType.Slice) {
+            let primaryInterface;
+            if (choice["non-repeat"] && Object.keys(choice["non-repeat"]).length > 0) {
+              primaryInterface = config.sourceFile.addInterface({
+                name: pascalCase(`${config.path[0].id} Document Data ${config.id} ${choiceId} Slice Primary`),
+                docs: [
+                  {
+                    description: (writer) => {
+                      const humanReadablePath = getHumanReadableFieldPath({
+                        path: [
+                          ...config.path,
+                          {
+                            id: config.id,
+                            model: config.model
+                          },
+                          {
+                            id: choiceId,
+                            model: choice
+                          },
+                          {
+                            id: "primary",
+                            label: "Primary"
+                          }
+                        ]
+                      });
+                      writer.writeLine(`Primary content in ${humanReadablePath}`);
+                    }
                   }
-                }
-              ]
-            });
-            addInterfacePropertiesForFields({
-              interface: primaryInterface,
-              sourceFile: config.sourceFile,
-              fields: choice["non-repeat"],
-              path: [
-                ...config.path,
-                {
-                  id: config.id,
-                  model: config.model
-                },
-                {
-                  id: choiceId,
-                  model: choice
-                },
-                {
-                  id: "primary",
-                  label: "Primary"
-                }
-              ],
-              fieldConfigs: config.fieldConfigs
-            });
-          }
-          let itemInterface;
-          if (Object.keys(choice.repeat).length > 0) {
-            itemInterface = config.sourceFile.addInterface({
-              name: pascalCase(`${config.path[0].id} Document Data ${config.id} ${choiceId} Slice Item`),
-              docs: [
-                {
-                  description: (writer) => {
-                    const humanReadablePath = getHumanReadableFieldPath({
-                      path: [
-                        ...config.path,
-                        {
-                          id: config.id,
-                          model: config.model
-                        },
-                        {
-                          id: choiceId,
-                          model: choice
-                        },
-                        {
-                          id: "items",
-                          label: "Items"
-                        }
-                      ]
-                    });
-                    writer.writeLine(`Item in ${humanReadablePath}`);
+                ]
+              });
+              addInterfacePropertiesForFields({
+                interface: primaryInterface,
+                sourceFile: config.sourceFile,
+                fields: choice["non-repeat"],
+                path: [
+                  ...config.path,
+                  {
+                    id: config.id,
+                    model: config.model
+                  },
+                  {
+                    id: choiceId,
+                    model: choice
+                  },
+                  {
+                    id: "primary",
+                    label: "Primary"
                   }
-                }
-              ],
+                ],
+                fieldConfigs: config.fieldConfigs
+              });
+            }
+            let itemInterface;
+            if (choice.repeat && Object.keys(choice.repeat).length > 0) {
+              itemInterface = config.sourceFile.addInterface({
+                name: pascalCase(`${config.path[0].id} Document Data ${config.id} ${choiceId} Slice Item`),
+                docs: [
+                  {
+                    description: (writer) => {
+                      const humanReadablePath = getHumanReadableFieldPath({
+                        path: [
+                          ...config.path,
+                          {
+                            id: config.id,
+                            model: config.model
+                          },
+                          {
+                            id: choiceId,
+                            model: choice
+                          },
+                          {
+                            id: "items",
+                            label: "Items"
+                          }
+                        ]
+                      });
+                      writer.writeLine(`Item in ${humanReadablePath}`);
+                    }
+                  }
+                ],
+                isExported: true
+              });
+              addInterfacePropertiesForFields({
+                interface: itemInterface,
+                sourceFile: config.sourceFile,
+                fields: choice.repeat,
+                path: [
+                  ...config.path,
+                  {
+                    id: config.id,
+                    model: config.model
+                  },
+                  {
+                    id: choiceId,
+                    model: choice
+                  },
+                  {
+                    id: "items",
+                    label: "Items"
+                  }
+                ],
+                fieldConfigs: config.fieldConfigs
+              });
+            }
+            const sliceType = config.sourceFile.addTypeAlias({
+              name: pascalCase(`${config.path[0].id} Document Data ${config.id} ${choiceId} Slice`),
+              type: `prismicT.Slice<"${choiceId}", ${primaryInterface ? `Simplify<${primaryInterface.getName()}>` : "Record<string, never>"}, ${itemInterface ? `Simplify<${itemInterface.getName()}>` : "never"}>`,
               isExported: true
             });
-            addInterfacePropertiesForFields({
-              interface: itemInterface,
-              sourceFile: config.sourceFile,
-              fields: choice.repeat,
-              path: [
-                ...config.path,
-                {
-                  id: config.id,
-                  model: config.model
-                },
-                {
-                  id: choiceId,
-                  model: choice
-                },
-                {
-                  id: "items",
-                  label: "Items"
-                }
-              ],
-              fieldConfigs: config.fieldConfigs
-            });
+            choiceInterfaceNames.push(sliceType.getName());
           }
-          const sliceType = config.sourceFile.addTypeAlias({
-            name: pascalCase(`${config.path[0].id} Document Data ${config.id} ${choiceId} Slice`),
-            type: `prismicT.Slice<"${choiceId}", ${primaryInterface ? `Simplify<${primaryInterface.getName()}>` : "Record<string, never>"}, ${itemInterface ? `Simplify<${itemInterface.getName()}>` : "never"}>`,
-            isExported: true
-          });
-          choiceInterfaceNames.push(sliceType.getName());
         }
       }
       const slicesType = config.sourceFile.addTypeAlias({
@@ -662,7 +676,7 @@ const addInterfacePropertiesForFields = (config) => {
   for (const name in config.fields) {
     addInterfacePropertyForField({
       ...config,
-      id: name.includes("-") ? `"${name}"` : name,
+      id: name.includes("-") || /^[0-9]/.test(name) ? `"${name}"` : name,
       model: config.fields[name],
       tabName: config.tabName
     });
@@ -753,7 +767,7 @@ const addTypeAliasForSharedSlice = (config) => {
   const variationTypeNames = [];
   for (const variation of config.model.variations) {
     let primaryInterface;
-    if (Object.keys(variation.primary).length > 0) {
+    if (variation.primary && Object.keys(variation.primary).length > 0) {
       primaryInterface = config.sourceFile.addInterface({
         name: pascalCase(`${buildSharedSliceInterfaceName({ id: config.model.id })} ${variation.id} Primary`),
         docs: [
@@ -794,7 +808,7 @@ const addTypeAliasForSharedSlice = (config) => {
       });
     }
     let itemInterface;
-    if (Object.keys(variation.items).length > 0) {
+    if (variation.items && Object.keys(variation.items).length > 0) {
       itemInterface = config.sourceFile.addInterface({
         name: pascalCase(`${buildSharedSliceInterfaceName({ id: config.model.id })} ${variation.id} Item`),
         docs: [