写一个umi插件自动生成代码解放cv的双手

引言

最近在写一个中台项目，使用的react的umi框架。
各种增删改查。基本是列表页新建页详情页这种页面
为了避免不必要的简单重复（主要是想偷懒）于是想去实现自己的一个代码生成器

探索

首先，在官网上看到了官方写的一个生成器

再去源码里扒一扒找到关键所在

简而言之，就是利用插件的api注册了一个生成model的指令，生成器指向目录里的model.js

代码如下

import { join } from 'path';
import assert from 'assert';

export default api => {
  const { paths, config } = api;
  const absTemplatePath = join(__dirname, '../template/generators');

  return class Generator extends api.Generator {
    writing() {
       ...
       // 判断目录名是models还是model
      const models = config.singular ? 'model' : 'models';
      const name = this.args[0].toString();
      ...
     // 将模板目录下里的model代码 拷贝到项目的model目录下 并命名为指令输入的文件名
      this.fs.copyTpl(
        join(absTemplatePath, 'model.js'),
        join(paths.absSrcPath, models, `${name}.js`),
        {
          name,
        },
      );
    }
  };
};

../template/generators/model.js

export default {
  state: '<%= name %>',
  subscriptions: {
    setup({ dispatch, history }) {
    },
  },
  reducers: {
    update(state) {
      return `${state}_<%= name %>`;
    },
  },
  effects: {
    *fetch({ type, payload }, { put, call, select }) {
    },
  },
}

model是一个常规的dva的model
里面的<%= name %>是ejs语法,对应着copyTpl方法的第三个参数中的name
模板js里的这个占位会被参数name替换

因为我们项目中习惯将model写到模块文件夹下，而且model里的代码有些我们的自己的书写
所以需要自定义一个生成方法了。

上手

以下是我写的一个生成规则

import { join } from 'path';
const fs=require('fs');
export default api => {
  const  {paths} = api;
  const configPath=join(paths.absSrcPath,'generatorConfig.js');
  const absTemplatePath = join(__dirname, '../template/generators');
  return class Generator extends api.Generator {
    writing() {
      const name = this.args[0].toString();
      // assert(!name.includes('/'), `model name should not contains /, bug got ${name}`);
      const type =this.args[1]&& this.args[1].toString();
     // type即为命令后跟的参数
      switch (type) {
        case 'list':
          if(!fs.existsSync(configPath)) {
            api.log.error('新建列表模板缺少generatorConfig.js')
            return
          }
          const genConfig=require(configPath);
          this.fs.copyTpl(join(absTemplatePath, 'list.js'),join(paths.absSrcPath, `pages/${name}/${type}`, `index.js`), {
            name,
            queryFormItems:genConfig[name]['queryFormItems'],
            columns:genConfig[name]['columns']
          });
      }
      this.fs.copyTpl(join(absTemplatePath, 'model.js'), join(paths.absSrcPath, `pages/${name}`, `model.js`), {
        name
      });
      this.fs.copyTpl(join(absTemplatePath, 'index.less'), join(paths.absSrcPath, `pages/${name}`, `index.less`), {
        name
      });
      this.fs.copyTpl(join(absTemplatePath, 'service.js'), join(paths.absSrcPath, `pages/${name}`, `service.js`), {
        name
      });
    }
  };
};

添加了如下功能

结合项目中的目录结构约定进行目录生成（比如我们约定用service来进行接口方法管理）
增加在命令后面加不同参数生成不同的特征模块（比如列表详情）
增加了配置项可以在node环境下去读取配置再生成到代码里去（比如 antd的列表的columns）

再仿照umi-dva-plugin的流程进行命令注册和插件导出

import { join } from 'path';
export default(api, opts = {})=> {
  api.registerGenerator('dva:newPage', {
    Generator: require('./model').default(api),
    resolved: join(__dirname, './model'),
  });
}

遇到问题

在探索和上手遇到挺多问题，总结如下
1.阅读源码加以甄别，因为umi-dva-plugin的代码贼多，模板功能只是其中的非核心功能，所以也是看了好几遍发现这个功能其实和其他代码并不存在耦合可以单独提出来
2.探索模板语法一开始不知道是ejs 找了下copyTpl方法

然后就恍然大悟，怪不得看起来那么熟悉,顺便学了一下ejs模板<%= %>和<%- %>的区别
3.兼容性问题遇到的一个贼奇怪的问题 node环境兼容的问题
一开始不知道用babel转成es5了一直报错class constructor Generator cannot be invoked without ‘new
看上去就是个兼容问题然后用web版的babel转换器关闭preset es2015 调整node版本到6.4主要是把对象的解构赋值要转换掉 不然依赖的三方Generator可能不认

总结

现在看来其实写这个插件其实并不难，但是在当时很多知识都不了解的情况下去看，确实还是有些许棘手，了解用法和原理比较有挑战，毕竟不是自己写的代码，所以还是要加强代码方便的阅读。

项目链接

戳我查看

原文地址：https://www.cnblogs.com/amigod/p/umi-plugin-a.html

时间： 2024-10-01 04:29:34

写一个umi插件自动生成代码解放cv的双手

引言

探索

上手

遇到问题

总结

项目链接