Merge pull request #7 from gatewayapps/should-emit-model

depracated isModelTopLevel, added shouldEmitModel
gatewayapps · Jan 14, 2019 · fe11af8 · fe11af8
2 parents 68e0c15 + 56b6e07
commit fe11af8
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@gatewayapps/cradle-apollo-emitter",
-  "version": "0.3.0",
+  "version": "0.3.1",
   "description": "",
   "main": "./dist/index.js",
   "scripts": {

diff --git a/src/ApolloEmitter.ts b/src/ApolloEmitter.ts
@@ -35,7 +35,7 @@ export default class ApolloEmitter implements ICradleEmitter {
 
   public async emitSchema(schema: CradleSchema) {
     schema.Models.forEach((model) => {
-      if (!this.options.options.isModelToplevel || this.options.options.isModelToplevel(model)) {
+      if (this.shouldEmitModel(model)) {
         this.writeTypeDefsForModel(model)
 
         if (this.shouldGenerateResolvers(model) && this.options.options.shouldOutputResolverFiles !== false) {
@@ -204,6 +204,15 @@ ${localFields.join('\n')}
     }
   }
 
+  private shouldEmitModel(model: CradleModel): boolean {
+    if (this.options.options.shouldEmitModel) {
+      return this.options.options.shouldEmitModel(model)
+    } else if (this.options.options.isModelToplevel) {
+      return this.options.options.isModelToplevel(model)
+    }
+    return true
+  }
+
   private getDirectiveForResolver(model: CradleModel, resolverType: string, resolverName: string): string {
     if (!this.options.options.getDirectiveForResolver) {
       return ''

diff --git a/src/IApolloEmitterOptions.ts b/src/IApolloEmitterOptions.ts
@@ -8,6 +8,10 @@ export interface IApolloEmitterOptions {
   readonly verbose: boolean
   readonly outputType: string
   readonly useMongoObjectIds: boolean
+  readonly shouldEmitModel?: (model: CradleModel) => boolean
+  /**
+   * @deprecated since version 0.3.1
+   */
   readonly isModelToplevel?: (model: CradleModel) => boolean
   readonly shouldOutputResolverFiles: boolean
   readonly shouldGenerateResolvers?: (model: CradleModel) => boolean