@nathapp/na-cli 1.0.0-rc.12 → 1.0.0-rc.13

package/src/utils/template.ts

@@ -1,4 +1,25 @@
 import * as ejs from 'ejs';
+import * as fs from 'fs';
+import * as inquirer from 'inquirer';
+import * as shell from 'shelljs';
+import chalk from 'chalk';
+import * as yargs from 'yargs';
+import * as path from 'path';
+
+const { name, version } = require("../../package.json");
+
+export interface TemplateConfig {
+    files?: string[]
+    postMessage?: string
+}
+
+export interface CliOptions {
+    projectName: string
+    templateName: string
+    templatePath: string
+    tartgetPath: string
+    config: TemplateConfig
+}
 
 export interface TemplateData {
     projectName: string
@@ -7,4 +28,186 @@ export interface TemplateData {
 
 export function render(content: string, data: TemplateData) {
     return ejs.render(content, data);
+}
+
+
+export function generateTemplate(templateDir: string, targetDirectory: string) {
+    const templateChoices = fs.readdirSync(templateDir);
+
+    const QUESTIONS = [
+        {
+            name: 'template',
+            type: 'list',
+            message: 'What project template would you like to generate?',
+            choices: templateChoices,
+            when: () => !yargs.argv['template']
+        },
+        {
+            name: 'name',
+            type: 'input',
+            message: 'Project name:',
+            when: () => !yargs.argv['name'],
+            validate: (input: string) => {
+                if (/^([A-Za-z\-\_\d])+$/.test(input)) return true;
+                else return 'Project name may only include letters, numbers, underscores and hashes.';
+            }
+        },
+        {
+            name: 'author',
+            type: 'input',
+            message: 'Author:',
+            when: () => !yargs.argv['author']
+        },
+        {
+            name: "run_install",
+            type: "confirm",
+            message: "Do you want to run npm install after template created?",
+        },
+    ];
+
+
+    console.log(`Welcome to ${name} version ${version}`);
+    inquirer.prompt(QUESTIONS)
+        .then(answers => {
+
+            answers = Object.assign({}, answers, yargs.argv);
+            // @ts-ignore
+            const projectChoice = answers['template'];
+            // @ts-ignore
+            const projectName = answers['name'];
+            // @ts-ignore
+            const author = answers['author'];
+            // @ts-ignore
+            const runInstall = answers['run_install'];
+            const templatePath = path.join(templateDir, projectChoice);
+            const tartgetPath = path.join(targetDirectory, projectName);
+            const templateConfig = getTemplateConfig(templatePath);
+
+            const options: CliOptions = {
+                projectName,
+                templateName: projectChoice,
+                templatePath,
+                tartgetPath,
+                config: templateConfig
+            }
+
+            if (!createProject(tartgetPath)) {
+                return;
+            }
+
+            createDirectoryContents(templatePath, projectName, templateConfig, author);
+
+            if (runInstall && !postProcess(options)) {
+                return;
+            }
+
+            showMessage(options);
+        });
+
+    function showMessage(options: CliOptions) {
+        console.log('');
+        console.log(chalk.green('Done.'));
+        console.log(chalk.green(`Go into the project: cd ${options.projectName}`));
+
+        const message = options.config.postMessage;
+
+        if (message) {
+            console.log('');
+            console.log(chalk.yellow(message));
+            console.log('');
+        }
+
+    }
+
+    function getTemplateConfig(templatePath: string): TemplateConfig {
+        const configPath = path.join(templatePath, '.template.json');
+
+        if (!fs.existsSync(configPath)) return {};
+
+        const templateConfigContent = fs.readFileSync(configPath);
+
+        if (templateConfigContent) {
+            return JSON.parse(templateConfigContent.toString());
+        }
+
+        return {};
+    }
+
+    function createProject(projectPath: string) {
+        if (fs.existsSync(projectPath)) {
+            console.log(chalk.red(`Folder ${projectPath} exists. Delete or use another name.`));
+            return false;
+        }
+
+        fs.mkdirSync(projectPath);
+        return true;
+    }
+
+    function postProcess(options: CliOptions) {
+        console.log('');
+        console.log("running post installation..please wait...")
+        if (isNode(options)) {
+            return postProcessNode(options);
+        }
+        return true;
+    }
+
+    function isNode(options: CliOptions) {
+        return fs.existsSync(path.join(options.templatePath, 'package.json'));
+    }
+
+    function postProcessNode(options: CliOptions) {
+        shell.cd(options.tartgetPath);
+
+        let cmd = '';
+
+        if (shell.which('yarn')) {
+            cmd = 'yarn';
+        } else if (shell.which('npm')) {
+            cmd = 'npm install';
+        }
+
+        if (cmd) {
+            const result = shell.exec(cmd);
+
+            if (result.code !== 0) {
+                return false;
+            }
+        } else {
+            console.log(chalk.red('No yarn or npm found. Cannot run installation.'));
+        }
+
+        return true;
+    }
+
+    const SKIP_FILES = ['node_modules', '.template.json'];
+
+    function createDirectoryContents(templatePath: string, projectName: string, config: TemplateConfig, author?: string) {
+        const filesToCreate = fs.readdirSync(templatePath);
+
+        filesToCreate.forEach(file => {
+            const origFilePath = path.join(templatePath, file);
+
+            // get stats about the current file
+            const stats = fs.statSync(origFilePath);
+
+            if (SKIP_FILES.indexOf(file) > -1) return;
+
+            if (stats.isFile()) {
+                let contents = fs.readFileSync(origFilePath, 'utf8');
+
+                if (file === '.npmignore-rename') file = '.npmignore';
+
+                contents = render(contents, { projectName, author });
+
+                const writePath = path.join(targetDirectory, projectName, file);
+                fs.writeFileSync(writePath, contents, 'utf8');
+            } else if (stats.isDirectory()) {
+                fs.mkdirSync(path.join(targetDirectory, projectName, file));
+
+                // recursive call
+                createDirectoryContents(path.join(templatePath, file), path.join(projectName, file), config, author);
+            }
+        });
+    }
 }