Custom helper support and getDistinctForeignModels helper (#9)

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@gatewayapps/cradle-template-emitter",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "description": "Cradle emitter for reading Handlebars based template files",
   "main": "./dist/index.js",
   "scripts": {

src/TemplateEmitter/ITemplateEmitterOptions.ts

@@ -1,11 +1,13 @@
 import { CradleModel } from '@gatewayapps/cradle'
+import { HelperDelegate } from 'handlebars'
 
 export interface ITemplateEmitterOptions {
     readonly sourcePath: string
     readonly outputPath: string
     readonly overwriteExisting: boolean
     readonly mode: string
     readonly languageType: string
+    readonly registerCustomHelpers?: (register: (name: string, fn: HelperDelegate) => void) => void
     readonly shouldEmit: (meta: CradleModel) => boolean
     readonly onFileEmitted: (path: string) => any
 }

src/TemplateEmitter/TemplateEmitter.ts

@@ -195,6 +195,26 @@ export class TemplateEmitter implements ICradleEmitter {
             return got
         })
 
+        handlebars.registerHelper('getDistinctForeignModels', (context, options) => {
+            const got: any[] = []
+
+            if (!context || Object.keys(context).length === 0) {
+                return []
+            }
+
+            for (const c in context) {
+                const ref = context[c]
+                if (!ref.ForeignModel) {
+                    continue
+                }
+                if (got.indexOf(ref.ForeignModel) === -1) {
+                    got.push(ref.ForeignModel)
+                }
+            }
+
+            return got
+        })
+
         handlebars.registerHelper('getObjectProps', (context, options) => {
             const got: any = []
 
@@ -217,6 +237,11 @@ export class TemplateEmitter implements ICradleEmitter {
 
             return got
         })
+
+        if (this.config && this.config.options.registerCustomHelpers) {
+            const register = handlebars.registerHelper.bind(handlebars)
+            this.config.options.registerCustomHelpers(register)
+        }
     }
     public writeFileContents(filePath: string, content: string) {
         const fileExists = fs.existsSync(filePath)