fix: CreateFile GQL field

Author: adrians5j

packages/webiny-api-files/package.json

@@ -15,9 +15,6 @@
   "license": "MIT",
   "dependencies": {
     "@babel/runtime": "^7.0.0",
-      "@commodo/fields": "^0.1.1",
-      "@commodo/hooks": "^0.0.8",
-      "@commodo/name": "^0.0.11",
     "@svgr/webpack": "^4.1.0",
     "bcryptjs": "^2.4.3",
     "graphql-shield": "^5.1.0",

packages/webiny-api-files/src/entities/File.entity.js

@@ -0,0 +1,87 @@
+// @flow
+import { Entity } from "webiny-entity";
+
+export interface IFile extends Entity {
+    createdBy: ?Entity;
+    src: string;
+    description: string;
+    name: string;
+    tags: Array<string>;
+}
+
+export function fileFactory(context: Object): Class<IFile> {
+    return class File extends Entity {
+        static classId = "File";
+
+        createdBy: ?Entity;
+        src: string;
+        description: string;
+        name: string;
+        type: string;
+        tags: Array<string>;
+        constructor() {
+            super();
+
+            const { user = {}, security, files } = context;
+            const { User } = security.entities;
+
+            this.attr("createdBy")
+                .entity(User)
+                .setSkipOnPopulate();
+
+            this.attr("size")
+                .integer()
+                .setValidators("required");
+            this.attr("type")
+                .char()
+                .setValidators("required,maxLength:50");
+            this.attr("src")
+                .char()
+                .setValidators("required,maxLength:200");
+            this.attr("name")
+                .char()
+                .setValidators("required,maxLength:100");
+            this.attr("tags")
+                .array()
+                .onSet(value => {
+                    if (Array.isArray(value)) {
+                        return value.map(item => item.toLowerCase());
+                    }
+
+                    return value;
+                })
+                .setValidators(tags => {
+                    if (Array.isArray(tags)) {
+                        if (tags.length > 15) {
+                            throw Error("You cannot set more than 15 tags.");
+                        }
+
+                        for (let i = 0; i < tags.length; i++) {
+                            let tag = tags[i];
+                            if (typeof tag !== "string") {
+                                throw Error("Tag must be typeof string.");
+                            }
+
+                            if (tag.length > 50) {
+                                throw Error(`Tag ${tag} is more than 50 characters long.`);
+                            }
+                        }
+                    }
+                });
+
+            this.on("beforeCreate", async () => {
+                if (!this.src.startsWith("/") || this.src.startsWith("http")) {
+                    throw Error(
+                        `File "src" must be a relative path, starting with forward slash ("/").`
+                    );
+                }
+
+                if (await files.entities.File.findOne({ query: { src: this.src } })) {
+                    throw Error(`File "src" must be unique. `);
+                }
+
+                this.createdBy = user.id;
+            });
+        }
+    };
+}

packages/webiny-api-files/src/entities/File.model.js

@@ -1,2 +1,2 @@
 // @flow
-export { File } from "webiny-api-files/entities/File.model";
+export { fileFactory } from "webiny-api-files/entities/File.entity";

packages/webiny-api-files/src/entities/index.js

@@ -0,0 +1,15 @@
+// @flow
+import { type EntityPluginType } from "webiny-api/types";
+import * as entities from "webiny-api-files/entities";
+
+const file: EntityPluginType = {
+    name: "entity-files-file",
+    type: "entity",
+    namespace: "files",
+    entity: {
+        name: "File",
+        factory: entities.fileFactory
+    }
+};
+
+export default [file];

packages/webiny-api-files/src/plugins/entities.js

@@ -5,7 +5,7 @@ import listFiles from "./resolvers/listFiles";
 import listTags from "./resolvers/listTags";
 import updateFileBySrc from "./resolvers/updateFileBySrc";
 
-const fileFetcher = ctx => ctx.models.File;
+const fileFetcher = ctx => ctx.files.entities.File;
 
 import {
     FileType,

packages/webiny-api-files/src/plugins/graphql/file.js

@@ -1,8 +1,8 @@
 // @flow
 import { ListResponse } from "webiny-api/graphql/responses";
 
-export default modelFetcher => async (root: any, args: Object, context: Object) => {
-    const model = modelFetcher(context);
+export default (entityFetcher: Function) => async (root: any, args: Object, context: Object) => {
+    const entityClass = entityFetcher(context);
     const { page = 1, perPage = 10, sort = null, search = "", types = [], tags = [] } = args;
     const findArgs = { page, perPage, sort };
 
@@ -30,6 +30,6 @@ export default modelFetcher => async (root: any, args: Object, context: Object)
         findArgs.query = { $and };
     }
 
-    const data = await model.find(findArgs);
+    const data = await entityClass.find(findArgs);
     return new ListResponse(data, data.getMeta());
 };

packages/webiny-api-files/src/plugins/graphql/resolvers/listFiles.js

@@ -7,7 +7,7 @@ export default (modelFetcher: ModelFetcher) => async (root: any, args: Object, c
     const model = modelFetcher(context);
 
     const results = await model
-        .getStorageDriver()
+        .getDriver()
         .getDatabase()
         .collection("File")
         .aggregate([

packages/webiny-api-files/src/plugins/graphql/resolvers/listTags.js

@@ -1,5 +1,5 @@
 // @flow
-import models from "./models";
+import entities from "./entities";
 import graphql from "./graphql";
 
-export default [models, graphql];
+export default [entities, graphql];