//@Author : dingjiawen
//@Date : 2024/03/25 15:26
//@Usage :
//@Desc :使用AST技术，还原code1.js


import {traverse, types} from "@babel/core";
import {parse} from "@babel/parser";
import CodeGenerator from "@babel/generator";
// import * as types from "@babel/types";
import fs from "fs";

//从系统盘读取文件  code4.js

const code = fs.readFileSync("code/code4.js", "utf-8");
let ast = parse(code)

traverse(ast, {
    WhileStatement(path) {
        const {node, scope} = path;
        const {test, body} = node;
        let switchNode = body.body[0];
        let {discriminant, cases} = switchNode;
        let {object, property} = discriminant;

        let arrName = object.name;
        let binding = scope.getBinding(arrName);
        let {init} = binding.path.node;
        object = init.callee.object;
        property = init.callee.property;
        let argument = init.arguments[0].value;
        let arrayFlow = object.value[property.name](argument);

        let resultBody = [];
        arrayFlow.forEach((index) => {
                let switchCase = cases.filter((c) => c.test.value == index)[0];
                let caseBody = switchCase.consequent;
                if (types.isContinueStatement(caseBody[caseBody.length - 1])) {
                    caseBody.pop();
                }
                resultBody = resultBody.concat(caseBody);
            }
        );
        path.replaceWithMultiple(resultBody);
    },
});

const {code: output} = CodeGenerator.default(ast);
console.log(output)