add priority of models

hennessyevan · Sep 13, 2019 · ff4ef07 · ff4ef07
1 parent de232b6
commit ff4ef07
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 5 deletions.
diff --git a/package.json b/package.json
@@ -59,14 +59,14 @@
     "/oclif.manifest.json"
   ],
   "homepage": "https://github.com/hennessyevan/gooseberry",
-  "keywords": [
-    "oclif"
-  ],
   "gooseberry": {
     "modelDir": "./demo/models",
     "mongoURI": "mongodb://localhost:27017/test",
     "dataDir": "./demo/data",
-    "dropDatabase": true
+    "dropDatabase": true,
+    "top": [
+      "User"
+    ]
   },
   "license": "MIT",
   "main": "lib/index.js",

diff --git a/src/tasks/createDocsFromData.ts b/src/tasks/createDocsFromData.ts
@@ -5,7 +5,23 @@ import { ListrContext } from "../commands/seed"
 
 export async function createDocsFromData(ctx: ListrContext, task: ListrTaskWrapper) {
   if (ctx.cache) {
-    const creationPromises = Object.entries(ctx.cache).map(async ([modelName, { data }]) => {
+    let order = Object.keys(ctx.cache)
+    if (ctx.config.top) {
+      order.sort((a, b) => {
+        if (ctx.config.top!.includes(a)) {
+          if (ctx.config.top!.includes(b)) {
+            if (ctx.config.top!.indexOf(a) < ctx.config.top!.indexOf(b)) {
+              return -1
+            }
+            return 1
+          }
+          return -1
+        }
+        return 1
+      })
+    }
+    const creationPromises = order.map(async modelName => {
+      const { data } = ctx.cache![modelName]
       const model = mongoose.model(modelName)
       task.output = modelName
 

diff --git a/src/utils/parseOptions.ts b/src/utils/parseOptions.ts
@@ -5,6 +5,7 @@ export type Options = {
   modelDir: string
   mongoURI: string
   dataDir: string
+  top?: string[]
   dropDatabase?: boolean
 }