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;