Add relative directory argument

Author: vs-cristian

src/bin.ts

@@ -1,12 +1,42 @@
 #!/usr/bin/env node
 
-import { argv } from 'yargs';
+import yargs from 'yargs';
 import { Logger } from './core/Logger';
-import { runCLI } from './index';
-
-runCLI(argv._[0] === 'init')
-  .then(() => process.exit(0))
-  .catch(err => {
-    Logger.error(err);
-    process.exit(1);
-  });
+import { IArgs, runGenerator } from './index';
+import { initConfigFile } from './init';
+
+yargs
+  .command(
+    'init',
+    'Run config initialization wizard',
+    () => {},
+    () => {
+      Logger.log('Starting config initialization wizard');
+
+      initConfigFile()
+        .then(() => process.exit(0))
+        .catch(err => {
+          Logger.error(err);
+          process.exit(1);
+        });
+    }
+  )
+  .command({
+    command: '*',
+    handler: (args: IArgs) => {
+      const { directory } = args;
+
+      runGenerator({ directory })
+        .then(() => process.exit(0))
+        .catch(err => {
+          Logger.error(err);
+          process.exit(1);
+        });
+    },
+  })
+  .option('directory', {
+    alias: 'd',
+    type: 'string',
+    description: 'Generate files in custom directory relative to root path',
+  })
+  .parse();

src/constants/index.ts

@@ -1,7 +1,7 @@
 import path from 'path';
 
 import { getAppRoot, getConfig, getStylesExtension } from '../utils';
-import { StyleFormats } from '../types/types';
+import { StyleFormats } from '../types';
 
 export const STYLE_FORMATS: StyleFormats[] = ['CSS', 'SCSS', 'SASS', 'Less', 'Stylus'];
 export const APP_ROOT = getAppRoot();

src/index.ts

@@ -1,21 +1,26 @@
 import inquirer from 'inquirer';
 
-import { init } from './init';
+import path from 'path';
 import { Logger } from './core/Logger';
 import * as questionTypes from './questions/questionTypes';
 import { type as typeQuestion } from './questions/questions';
 import { FileGenerateManager } from './core/FileGenerateManager';
+import { CONFIG } from './constants';
 
-export async function runCLI(shouldInit: boolean) {
-  if (shouldInit) {
-    await init();
-    return;
-  }
+export interface IArgs {
+  directory?: string;
+}
 
+export async function runGenerator(args: IArgs) {
+  const { directory } = args;
   const { type } = await inquirer.prompt(typeQuestion as any);
 
   process.stdout.write('\n');
 
+  if (directory) {
+    CONFIG.path = path.resolve(process.cwd(), directory);
+  }
+
   switch (type) {
     case 'component': {
       const answers = await inquirer.prompt(questionTypes.getComponentQuestions());

src/init/index.ts

@@ -8,7 +8,7 @@ import { question } from './questions';
 import { APP_ROOT } from '../constants';
 import { updatePackage } from './helpers';
 
-export async function init() {
+export async function initConfigFile() {
   clear(true);
   Logger.log(
     chalk => `

src/utils/getConfig.ts

@@ -4,7 +4,7 @@ import { cosmiconfigSync } from 'cosmiconfig';
 import defaultConfig from '../config/default.json';
 import { APP_ROOT, STYLE_FORMATS } from '../constants';
 import { Logger } from '../core/Logger';
-import { IConfig } from '../types/types';
+import { IConfig } from '../types';
 
 export function getConfig() {
   let config = defaultConfig as IConfig;