背景

最近搭建毕设前端框架的时候，每当创建一个页面，我都需要创建这个页面组件，创建它的Route，最后将该Route加入到总的Route。当然，这里的流程还不算复杂，基本也是复制粘贴改改变量，但是后面还要用到Redux，可能还会使用saga…再将它们加入这一条流程线，我需要改的东西又多了。

在公司实习的脚手架里，发现有大佬造的轮子，之前也只是照着命令敲拿来用，这次顺带研究了一下核心功能，结合我的毕设框架需要，加入了最简单的自动化“脚本”。

所需环境和工具

Node环境下执行。
命令映射，使用commander。

让文件可以通过命令行的形式执行
文件读写，这里我使用的是fs-extra，也可以使用Node自带的File System，但是前者支持Promise和Async, Await。

文件读写只是读取模板文件内容，然后写入到新的文件为我们所用
模板字符串，使用lodash/string的模板字符串方法template。

模板字符串：我们可以使用xxx.tmpl格式的文件存储我们的模板，需要替换的内容使用<%= xxx %>表示即可，下面会给出文件原型
TS文件修改，使用ts-simple-ast。

ts-simple-ast封装了TS Compiler API，能够更便捷的修改原来的TS文件内容，加入所需的东西。例如修改变量值，这也是这篇文章中提到的较为简单的一个用途，其他更复杂的也可以参考文档学习

文件原型和需求

0. 需求

每当新增一个页面，我们需要创建一个基本框架组件，一个Route，最后把这个Route自动插入到总的Router里。

1. xxxComponent

这里创建了一个非常简单的组件，带有Props和State，interface使用Ixxx命名。

import React from 'react';

interface I<%= featureUpperName %>Props {}

interface I<%= featureUpperName %>State {}

export default class <%= featureUpperName %> extends React.Component<I<%= featureUpperName %>Props, I<%= featureUpperName %>State> {
    constructor(props: I<%= featureUpperName %>Props) {
        super(props);
        this.state = {};
    }

    render() {
        return (
            <h2>My Home</h2>
        )
    }
}

2. xxx.index

这个文件里加入所有需要导出的Component，并作为统一导出出口。

export { default as <%= featureUpperName %> } from './<%= featureUpperName %>'

3. Route

自定义的Route，属性也基本遵循原生Route，加入loadable component，支持按需加载。

import App from '../common/component/App';
import { IRoute } from '@src/common/routeConfig';

const loader = (name: string) => async () => {
    const entrance = await import('./');
    return entrance[name];
};

const childRoutes: IRoute[] = [{
    path: '/<%= featureName %>',
    name: '<%= featureUpperName %>',
    loader: loader('<%= featureUpperName %>'),
}]

export default {
    path: '/<%= featureName %>',
    name: '',
    component: App,
    childRoutes
};

上面三个便作为基本模板文件，下面这个则是总的Route。

4. routeConfig

完成一个页面的创建并生成它的route后，需要在该文件引入这个route，然后修改变量childRoutes，插入该route，这样我们的工作就算完成啦。

import HomeRoute from "../features/home/route";

export interface IRoute {
    path: string
    name: string
    component?: React.ComponentClass | React.SFC;
    childRoutes?: IRoute[]
    loader?: AsyncComponentLoader
    exact?: boolean
    redirect?: string
}

const childRoutes: IRoute[] = [HomeRoute]

const routes = [{
    path: '/',
    name: 'app',
    exact: true,
    redirect: '/home'
}, ...childRoutes]

export default routes;

步骤

零、源代码

1. kit.js

用于读取模板文件，写入新的文件

首先第一行，告诉shell此文件默认执行环境为Node。

接下来我们来看addFeatureItem(忽略我的命名╮(╯▽╰)╭)，这个函数有三个参数：

srcPath，template文件位置
targetPath，写入的文件位置
option，渲染模板时使用，简而言之可以替换掉模板中的变量为里面我们设定的值

我们先确认文件是否存在，然后读取模板文件，写入新的文件即可，中间加了个已有文件判断。

是不是很简单！

最后加入使用commander创建自己的命令即可，更详细的用法可以查看commander的文档，这里添加一个简单的add命令，后跟一个featureName，键入命令后执行action函数，里面的参数即我们刚刚键入的featureName，读取后便可以从模板创建新的feature。

当然，我们还需要修改routeConfig.ts这个文件，我将这个操作放到了下面的ts-ast.ts文件。

#! /usr/bin/env node

const fse = require('fs-extra');
const path = require('path');
const _ = require('lodash/string')
const commander = require('commander')
const ast = require('./ts-ast')

const templatesDir = path.join(__dirname, 'templates');
const targetDir = path.join(__dirname, '..', 'src', 'features');

async function addFeatureItem(srcPath, targetPath, option) {
    let res;
    try {
        await fse.ensureFile(srcPath)
        res = await fse.readFile(srcPath, 'utf-8')
        
        // avoid override
        const exists = await fse.pathExists(targetPath)
        if(exists) {
            console.log(`${targetPath} is already added!`);
            return
        }

        await fse.outputFile(targetPath, _.template(res)(option), {
            encoding: "utf-8"
        })
        console.log(`Add ${srcPath} success!`);
    } catch (err) {
        console.error(err)
    }
}

async function addFeature(name) {
    const renderOpt = {
        featureName: name,
        featureUpperName: _.upperFirst(name)
    }

    const componentTmpl = `${templatesDir}/Component.tsx.tmpl`;
    const component = `${targetDir}/${name}/${_.upperFirst(name)}.tsx`;
    addFeatureItem(componentTmpl, component, renderOpt);

    const indexTmpl = `${templatesDir}/index.ts.tmpl`;
    const index = `${targetDir}/${name}/index.ts`;
    addFeatureItem(indexTmpl, index, renderOpt);

    const routeTmpl = `${templatesDir}/route.ts.tmpl`;
    const route = `${targetDir}/${name}/route.ts`;
    addFeatureItem(routeTmpl, route, renderOpt);
}

commander
    .version(require('../package.json').version)
    .command('add <feature>')
    .action((featureName) => {
        // add features
        addFeature(featureName);
        // manipulate some ts file like route
        ast(featureName);
    })

commander.parse(process.argv);

2. ts-ast.js

用于修改rootConfig.ts文件

先给出ts-simple-ast的地址，自己还是觉得这个操作是比较复杂的，我也是参考了文档再加上项目脚手架代码才看明白，至于原理性的东西，可能还需要查看Typescript Compiler API，因为这个包也只是Wrapper，文档也还不是很完善，更复杂的需求还有待学习。

这里关键就两个操作，一个是添加一个import，其次则是修改childRoutes变量的值。但是一些函数的英文字面意思理解起来可能比上面的文件读写要困难。

我们首先需要新建一个Project
然后需要获取待操作的文件，拿到待操作文件(srcFile)之后，直接使用addImportDeclaration这个方法便可以添加default import，如果需要named import，也可以使用addNamedImport。定义好default name(defaultImport prop)以及path(moduleSpecifier)即可。
最后是对childRoutes变量的值进行修改，这一过程比较复杂：
- 首先需要通过变量名拿到一个VariableStatement
- 然后再拿到变量声明（VariableDeclaration），因为一个文件中可能有多处声明（函数中，全局中）
- 由于该例中只有一处声明，所以我们直接forEach遍历声明即可，遍历时拿到initializer（这里是一个类似array的东东）
- 再使用它的forEachChild遍历拿到node，node.getText终于拿到了里面的一个值（比如HomeRoute）
- 我们将这些值添加到一个新的数组
- 直到遍历完毕，再将它们拼接起来，加入新的route，以字符串形式setInitializer即可
最后保存所有操作，Project.save()即可

const { Project } = require('ts-simple-ast')
const _ = require('lodash/string')
const path = require('path')

const project = new Project();
const srcDir = path.join(__dirname, '..', 'src', 'common');

async function addRoute(featureName) {
  const FeatureName = _.upperFirst(featureName);
  try {
    const srcFile = project.addExistingSourceFile(`${srcDir}/routeConfig.ts`);
    srcFile.addImportDeclaration({
      defaultImport: `${FeatureName}Route`,
      moduleSpecifier: `../features/${featureName}/route`
    })
    const routeVar = srcFile.getVariableStatementOrThrow('childRoutes');

    let newRoutes = [];

    routeVar.getDeclarations().forEach((decl, i) => {
      decl.getInitializer().forEachChild(node => {
        newRoutes.push(node.getText());
      })
      decl.setInitializer(`[${newRoutes.join(', ')}, ${FeatureName}Route]`);
    })

    await project.save();
    console.log("Add route successful");
  } catch(err) {
    console.log(err);
  }
  
}

module.exports = addRoute;

一、修改文件权限，使其可执行

只要头部加入了#! /usr/bin/env node，简单的一行命令即可搞定chmod +x /filePath/yourExeFile

然后，我们便可以使用/filePath/yourExeFile add featureName的方式添加一个新的页面！

参考

what-does-chmod-x-filename-do-and-how-do-i-use-it