41 lines
1.2 KiB
JavaScript
41 lines
1.2 KiB
JavaScript
//@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";
|
||
|
||
|
||
const code = fs.readFileSync("code/code3.js", "utf-8");
|
||
let ast = parse(code)
|
||
|
||
traverse(ast, {
|
||
IfStatement(path) {
|
||
let {consequent, alternate} = path.node;
|
||
let testpath = path.get("test");
|
||
const evaluateTest = testpath.evaluateTruthy();
|
||
if (evaluateTest === true) {
|
||
if (types.isBlockStatement(consequent)) {
|
||
consequent = consequent.body;
|
||
}
|
||
path.replaceWithMultiple(consequent);
|
||
} else if (evaluateTest === false) {
|
||
if (alternate != null) {
|
||
if (types.isBlockStatement(alternate)) {
|
||
alternate = alternate.body;
|
||
}
|
||
path.replaceWithMultiple(alternate);
|
||
} else {
|
||
path.remove();
|
||
}
|
||
}
|
||
},
|
||
});
|
||
|
||
const {code: output} = CodeGenerator.default(ast);
|
||
console.log(output) |