let fs = require(‘fs‘);
let path = require(‘path‘);
let babylon = require(‘babylon‘); // Babylon 把源码转换为AST
let t = require(‘@babel/types‘); // @babel-types 替换节点
let traverse = require(‘@babel/traverse‘).default; // @babel-traverse 遍历节点
let generator = require(‘@babel/generator‘).default; // @babel/generator 生成
let ejs = require(‘ejs‘); // js模版
class Compiler {
// 构造函数
constructor(config) {
// entry out
this.config = config;
// 1、保存入口文件的路径
this.entryId;
// 2、博阿村所有的模块依赖
this.modules = {};
// 入口路径
this.entry = config.entry;
// 工作路径
this.root = process.cwd();
}
// 获取文件源码
getSource(modulePath){
let content = fs.readFileSync(modulePath,‘utf8‘);
return content;
}
// 解析源码
parse(source,parentPath){
// AST解析语法树
let ast = babylon.parse(source);
let dependencies = [];
traverse(ast,{
CallExpression(p){
let node = p.node; // 对应的节点
if(node.callee.name === ‘require‘){
node.callee.name = ‘__webpack_require__‘;
let moduleName = node.arguments[0].value; // 取到的就是模块的引用名字
moduleName = moduleName + (path.extname(moduleName)?‘‘:‘.js‘);
moduleName = ‘./‘+path.join(parentPath,moduleName);
dependencies.push(moduleName);
node.arguments = [t.stringLiteral(moduleName)];
}
}
});
let sourceCode = generator(ast).code;
return { sourceCode, dependencies }
}
// 建立模块
buildModule(modulePath,isEntry){
// 拿到模块的内容
let source = this.getSource(modulePath);
// 模块id modulePath modulePath-this.root = src/index.js
let moduleName = ‘./‘+path.relative(this.root,modulePath);
if(isEntry){
// 保存入口的名字
this.entryId = moduleName;
}
// 解析需要把source源码进行改造,返回一个依赖列表
let {sourceCode,dependencies} = this.parse(source,path.dirname(moduleName));
this.modules[moduleName] = sourceCode;
dependencies.forEach(
// 父模块的加载,递归加载
(dep)=>{
this.buildModule(path.join(this.root,dep),false)
}
);
}
emitFile(){
// 发射文件
// 用数据渲染
// 输出路径
let main = path.join(this.config.output.path,this.config.output.filename);
// 模板的路径
let tempateStr = this.getSource(path.join(__dirname,‘main.ejs‘));
let code = ejs.render(tempateStr,{
entryId:this.entryId,
modules: this.modules
});
// this.assets = {};
// // 路径对应的代码
// this.assets[main] = code;
// fs.writeFileSync(main,this.assets[main]);
fs.writeFileSync(main,code);
}
run() {
// 执行并创建模块依赖关系
this.buildModule(path.resolve(this.root, this.entry),true);
// 发射一个打包后的文件
this.emitFile();
}
}
module.exports = Compiler;