Allow custom template directory fix #30

src/generators.js

@@ -2,7 +2,7 @@ import ReactCrudGenerator from './generators/ReactCrudGenerator';
 import TypescriptInterfaceGenerator from './generators/TypescriptInterfaceGenerator';
 
 function wrap (cl) {
-  return (prefix) => new cl(prefix)
+  return ({hydraPrefix, templateDirectory}) => new cl({hydraPrefix, templateDirectory})
 }
 
 function generators (generator = 'react') {

src/generators/ReactCrudGenerator.js

@@ -7,8 +7,8 @@ import chalk from 'chalk';
 export default class ReactCrudGenerator {
   templates = {};
 
-  constructor(hydraPrefix) {
-    const templatePath = `${__dirname}/../../templates/react/`;
+  constructor({hydraPrefix, templateDirectory}) {
+    const templatePath = `${templateDirectory}/react/`;
 
     this.hydraPrefix = hydraPrefix;
 

src/generators/ReactCrudGenerator.test.js

@@ -7,7 +7,7 @@ import ReactCrudGenerator from './ReactCrudGenerator';
 
 
 test('Generate a React app', () => {
-  const generator = new ReactCrudGenerator('hydra:');
+  const generator = new ReactCrudGenerator({hydraPrefix: 'hydra:', templateDirectory: `${__dirname}/../../templates`});
   const tmpobj = tmp.dirSync({unsafeCleanup: true});
 
   const fields = [new Field('bar', {

src/generators/TypescriptInterfaceGenerator.js

@@ -5,8 +5,8 @@ import fs from 'fs';
 export default class TypescriptInterfaceGenerator {
   templates = {};
 
-  constructor() {
-    const templatePath = `${__dirname}/../../templates/typescript`;
+  constructor({templateDirectory}) {
+    const templatePath = `${templateDirectory}/typescript/`;
     this.template = handlebars.compile(fs.readFileSync(`${templatePath}/interface.ts`).toString())
   }
 

src/generators/TypescriptInterfaceGenerator.test.js

@@ -7,7 +7,7 @@ import TypescriptInterfaceGenerator from './TypescriptInterfaceGenerator';
 
 
 test('Generate a typescript interface', () => {
-  const generator = new TypescriptInterfaceGenerator('hydra:');
+  const generator = new TypescriptInterfaceGenerator({templateDirectory: `${__dirname}/../../templates`});
   const tmpobj = tmp.dirSync({unsafeCleanup: true});
 
   const resource = new Resource('abc', 'http://example.com/foos', {

src/index.js

@@ -13,6 +13,7 @@ program
   .option('-r, --resource [resourceName]', 'Generate CRUD for the given resource')
   .option('-p, --hydra-prefix [hydraPrefix]', 'The hydra prefix used by the API', 'hydra:')
   .option('-g, --generator [generator]', 'The generator to use, one of "react", "angular" etc.', 'react')
+  .option('-t, --template-directory [templateDirectory]', 'The templates directory base to use. Final directory will be ${templateDirectory}/${generator}', `${__dirname}/../templates/`)
   .parse(process.argv);
 
 if (2 !== program.args.length && (!process.env.API_PLATFORM_CLIENT_GENERATOR_ENTRYPOINT || !process.env.API_PLATFORM_CLIENT_GENERATOR_OUTPUT)) {
@@ -22,7 +23,10 @@ if (2 !== program.args.length && (!process.env.API_PLATFORM_CLIENT_GENERATOR_ENT
 const entrypoint = program.args[0] || process.env.API_PLATFORM_CLIENT_GENERATOR_ENTRYPOINT;
 const outputDirectory = program.args[1] || process.env.API_PLATFORM_CLIENT_GENERATOR_OUTPUT;
 
-const generator = generators(program.generator)(program.hydraPrefix);
+const generator = generators(program.generator)({
+  hydraPrefix: program.hydraPrefix,
+  templateDirectory: program.templateDirectory
+});
 const resourceToGenerate = program.resource ? program.resource.toLowerCase() : null;
 
 parseHydraDocumentation(entrypoint).then(api => {