feat: add asset reference in model creation if thumb_field command op…

…tion is specified

README.md

@@ -38,4 +38,4 @@ You can provide the following optional parameters to the CLI:
 
 - Add tests
 - Add content media upload
-- Add possibility to exclude certain fields from the extracted model
+- Add possibility to exclude specific fields from the extracted model

index.js

@@ -35,8 +35,7 @@ yargs(argv).command(
         type: "string",
         description: "Your graphCMS token",
       })
-      .option("post-thumbnail", {
-        alias: "pt",
+      .option("thumb_field", {
         type: "string",
         description:
           'The name of the yaml field to recognize as a post thumbnail (it will be uploaded automatically to the "Asset" model)',
@@ -59,6 +58,12 @@ yargs(argv).command(
       return logger.error("You must specify your graphcms token");
     }
 
+    if (!argv.thumb_field) {
+      logger.warn(
+        "You didn't specify any yaml field to be used as post thumbnail"
+      );
+    }
+
     const response = await prompts(
       {
         type: "text",
@@ -76,7 +81,7 @@ yargs(argv).command(
       restartSpinner("Fetching and parsing mds file");
       const mds = await fetchMds(argv.path);
       restartSpinner("Extracting model from first md file");
-      const model = extractModel(mds?.[0]);
+      const model = extractModel(mds?.[0], argv.thumb_field);
       restartSpinner("Creating the model inside your graphCMS instance");
       await createModel(
         argv.url,
@@ -94,7 +99,9 @@ yargs(argv).command(
       );
       spinner.succeed();
     } catch (e) {
+      spinner.fail();
       logger.error("Something went wrong", e);
+      process.exit();
     }
   }
 ).argv;

src/createModel.js

@@ -1,5 +1,5 @@
 const { newMigration } = require("@graphcms/management");
-const { logger } = require("./utils");
+const { logger, spinner } = require("./utils");
 
 const createModel = async (url, token, model, modelName) => {
   try {
@@ -10,14 +10,24 @@ const createModel = async (url, token, model, modelName) => {
 
     model.reduce(
       (acc, modelField) => {
-        acc.addSimpleField({
-          apiId: modelField.name,
-          displayName: modelField.name,
-          type: modelField.type,
-          ...(modelField.formRenderer && {
-            formRenderer: modelField.formRenderer,
-          }),
-        });
+        if (modelField.model) {
+          acc.addRelationalField({
+            apiId: modelField.name,
+            displayName: modelField.name,
+            relationType: modelField.relationType,
+            model: modelField.model,
+          });
+        } else {
+          acc.addSimpleField({
+            apiId: modelField.name,
+            displayName: modelField.name,
+            type: modelField.type,
+            ...(modelField.formRenderer && {
+              formRenderer: modelField.formRenderer,
+            }),
+          });
+        }
+
         return acc;
       },
       migration.createModel({
@@ -38,7 +48,9 @@ const createModel = async (url, token, model, modelName) => {
       logger.info(name);
     }
   } catch (e) {
+    spinner.fail();
     logger.error("errors creating model", e);
+    process.exit();
   }
 };
 

src/extractModel.js

@@ -1,8 +1,8 @@
-const { FieldType, Renderer } = require("@graphcms/management");
+const { FieldType, Renderer, RelationType } = require("@graphcms/management");
 const string = require("string-sanitizer");
 const matter = require("gray-matter");
 
-const extractModel = (md) => {
+const extractModel = (md, thumbField) => {
   const { data } = matter(md);
 
   if (!data) {
@@ -12,6 +12,13 @@ const extractModel = (md) => {
   return Object.keys({ ...data, content: "" }).map((frontKey) => {
     const value = data[frontKey];
 
+    if (frontKey === thumbField) {
+      return {
+        name: frontKey,
+        model: "Asset",
+        relationType: RelationType.OneToOne,
+      };
+    }
     if (frontKey === "content") {
       return {
         name: "content",