Add not found errors

packages/strapi-plugin-content-type-builder/controllers/Components.js

@@ -1,5 +1,7 @@
 'use strict';
 
+const _ = require('lodash');
+
 const {
   validateComponentInput,
   validateUpdateComponentInput,
@@ -81,6 +83,10 @@ module.exports = {
     const { uid } = ctx.params;
     const { body } = ctx.request;
 
+    if (!_.has(strapi.components, uid)) {
+      return ctx.send({ error: 'component.notFound' }, 404);
+    }
+
     try {
       await validateUpdateComponentInput(body);
     } catch (error) {
@@ -112,6 +118,10 @@ module.exports = {
   async deleteComponent(ctx) {
     const { uid } = ctx.params;
 
+    if (!_.has(strapi.components, uid)) {
+      return ctx.send({ error: 'component.notFound' }, 404);
+    }
+
     try {
       strapi.reload.isWatching = false;
 

packages/strapi-plugin-content-type-builder/controllers/ContentTypes.js

@@ -76,6 +76,10 @@ module.exports = {
     const { uid } = ctx.params;
     const { body } = ctx.request;
 
+    if (!_.has(strapi.contentTypes, uid)) {
+      return ctx.send({ error: 'contentType.notFound' }, 404);
+    }
+
     try {
       await validateUpdateContentTypeInput(body);
     } catch (error) {
@@ -102,6 +106,10 @@ module.exports = {
   async deleteContentType(ctx) {
     const { uid } = ctx.params;
 
+    if (!_.has(strapi.contentTypes, uid)) {
+      return ctx.send({ error: 'contentType.notFound' }, 404);
+    }
+
     try {
       strapi.reload.isWatching = false;
 

packages/strapi-plugin-content-type-builder/controllers/validation/component.js

@@ -34,18 +34,21 @@ const VALID_TYPES = [
 
 const componentSchema = createSchema(VALID_TYPES, VALID_RELATIONS, {
   modelType: modelTypes.COMPONENT,
-}).shape({
+})
-  icon: yup
+  .shape({
-    .string()
+    icon: yup
-    .nullable()
+      .string()
-    .test(isValidName)
+      .nullable()
-    .required('icon.required'),
+      .test(isValidName)
-  category: yup
+      .required('icon.required'),
-    .string()
+    category: yup
-    .nullable()
+      .string()
-    .test(isValidName)
+      .nullable()
-    .required('category.required'),
+      .test(isValidName)
-});
+      .required('category.required'),
+  })
+  .required()
+  .noUnknown();
 
 const nestedComponentSchema = yup.array().of(
   componentSchema
@@ -63,19 +66,16 @@ const nestedComponentSchema = yup.array().of(
       },
     })
     .required()
+    .noUnknown()
 );
 
-const createComponentSchema = () => {
+const validateComponentInput = data => {
   return yup
     .object({
-      component: componentSchema.required().noUnknown(),
+      component: componentSchema,
       components: nestedComponentSchema,
     })
-    .noUnknown();
+    .noUnknown()
-};
-
-const validateComponentInput = data => {
-  return createComponentSchema()
     .validate(data, {
       strict: true,
       abortEarly: false,
@@ -105,7 +105,12 @@ const validateUpdateComponentInput = data => {
     });
   }
 
-  return createComponentSchema()
+  return yup
+    .object({
+      component: componentSchema,
+      components: nestedComponentSchema,
+    })
+    .noUnknown()
     .validate(data, {
       strict: true,
       abortEarly: false,
@@ -116,6 +121,6 @@ const validateUpdateComponentInput = data => {
 module.exports = {
   validateComponentInput,
   validateUpdateComponentInput,
-  componentSchema,
+
   nestedComponentSchema,
 };

packages/strapi-plugin-content-type-builder/controllers/validation/content-type.js

@@ -44,17 +44,15 @@ const contentTypeSchema = createSchema(VALID_TYPES, VALID_RELATIONS, {
   modelType: modelTypes.CONTENT_TYPE,
 });
 
-const createContentTypeSchema = () => {
+const createContentTypeSchema = yup
-  return yup
+  .object({
-    .object({
+    contentType: contentTypeSchema.required().noUnknown(),
-      contentType: contentTypeSchema.required().noUnknown(),
+    components: nestedComponentSchema,
-      components: nestedComponentSchema,
+  })
-    })
+  .noUnknown();
-    .noUnknown();
-};
 
 const validateContentTypeInput = data => {
-  return createContentTypeSchema()
+  return createContentTypeSchema
     .validate(data, {
       strict: true,
       abortEarly: false,
@@ -72,7 +70,19 @@ const validateUpdateContentTypeInput = data => {
     });
   }
 
-  return createContentTypeSchema()
+  if (_.has(data, 'components') && Array.isArray(data.components)) {
+    data.components.forEach(data => {
+      if (_.has(data, 'attributes') && _.has(data, 'uid')) {
+        Object.keys(data.attributes).forEach(attribute => {
+          if (data.attributes[attribute].default === '') {
+            data.attributes[attribute].default = undefined;
+          }
+        });
+      }
+    });
+  }
+
+  return createContentTypeSchema
     .validate(data, {
       strict: true,
       abortEarly: false,