Added flag to disable type map generation

Author: max-leuthaeuser

bin/astgen.js

@@ -28,6 +28,11 @@ async function main(argvs) {
       type: "boolean",
       description: "Recurse mode suitable for mono-repos",
     })
+    .option("tsTypes", {
+      default: true,
+      type: "boolean",
+      description: "Generate type mappings using the Typescript Compiler API",
+    })
     .version()
     .help("h").argv;
 

index.js

@@ -223,7 +223,10 @@ const createJSAst = async (options) => {
   try {
     const promiseMap = await getAllSrcJSAndTSFiles(options.src);
     const srcFiles = promiseMap.flatMap((d) => d);
-    const ts = createTsc(srcFiles);
+    let ts;
+    if (options.tsTypes) {
+      ts = createTsc(srcFiles);
+    }
 
     for (const file of srcFiles) {
       try {
@@ -305,17 +308,18 @@ const writeAstFile = (file, ast, options) => {
     outAstFile,
     JSON.stringify(data, getCircularReplacer(), "  ")
   );
-  console.log("Converted", relativePath, "to", outAstFile);
+  console.log("Converted AST for", relativePath, "to", outAstFile);
 };
 
 const writeTypesFile = (file, seenTypes, options) => {
   const relativePath = file.replace(new RegExp("^" + options.src + "/"), "");
-  const outAstFile = path.join(options.output, relativePath + ".typemap");
-  fs.mkdirSync(path.dirname(outAstFile), { recursive: true });
+  const outTypeFile = path.join(options.output, relativePath + ".typemap");
+  fs.mkdirSync(path.dirname(outTypeFile), { recursive: true });
   fs.writeFileSync(
-    outAstFile,
+    outTypeFile,
     JSON.stringify(mapToObj(seenTypes))
   );
+  console.log("Converted types for", relativePath, "to", outTypeFile);
 };
 
 

jest.config.js

@@ -1,7 +1,7 @@
 {
   "name": "@joernio/astgen",
-  "version": "2.0.4",
-  "description": "Generate AST in json format for a number of languages by invoking appropriate tool",
+  "version": "0.1.0",
+  "description": "Generate JS/TS AST in json format with Babel",
   "exports": "./index.js",
   "keywords": [
     "AST",
@@ -10,17 +10,15 @@
   "bin": {
     "astgen": "./bin/astgen.js"
   },
-  "repository": "https://github.com/joernio/astgen",
-  "author": "Team joernio",
+  "repository": "https://github.com/max-leuthaeuser/astgen",
+  "author": "Max Leuthaeuser",
   "license": "Apache-2.0",
   "dependencies": {
     "@babel/parser": "^7.17.9",
     "typescript": "^4.9.4",
     "yargs": "^17.4.1"
   },
   "scripts": {
-    "test": "cross-env NODE_OPTIONS=--experimental-vm-modules jest",
-    "watch": "cross-env NODE_OPTIONS=--experimental-vm-modules jest --watch",
     "postinstall": "pkg . --no-bytecode --no-native-build --public --compress GZip"
   },
   "engines": {
@@ -32,7 +30,6 @@
   ],
   "devDependencies": {
     "cross-env": "^7.0.3",
-    "jest": "^26.6.3",
     "pkg": "^5.8.0"
   }
 }