Support React JSX functions via Babel

package.json

@@ -4,16 +4,30 @@
   "description": "Builds static call graph for JavaScript with a field-based algorithm.",
   "main": "index.js",
   "dependencies": {
-    "@babel/core": "^7.14.6",
-    "@babel/preset-flow": "^7.14.5",
+    "@babel/core": "^7.18.2",
+    "@babel/plugin-proposal-class-properties": "^7.1.0",
+    "@babel/plugin-proposal-optional-chaining": "^7.17.12",
+    "@babel/plugin-transform-react-jsx": "^7.17.12",
+    "@babel/preset-env": "^7.18.2",
+    "@babel/preset-flow": "^7.0.0",
+    "@babel/preset-typescript": "^7.1.0",
     "@typescript-eslint/typescript-estree": "^5.37.0",
-    "argparse": "^2.0.1",
-    "babel-core": "^6.26.3",
+    "amdefine": "^1.0.1",
+    "argparse": "^1.0.10",
+    "body-parser": "^1.18.2",
+    "esprima": "^4.0.0",
+    "express": "^4.16.3",
+    "JSONStream": "1.3.3",
+    "npm": "^5.10.0",
+    "sloc": "^0.2.0",
+    "source-map": "^0.7.3",
     "typescript": "^4.8.3",
-    "underscore": "^1.13.1"
+    "underscore": "^1.13.1",
+    "vue-parser": "^1.1.6"
   },
   "devDependencies": {
-    "jest": "^27.0.5"
+    "@babel/cli": "^7.17.10",
+    "jest": "^23.1.0"
   },
   "scripts": {
     "test": "jest && python3 ./tests/test.py"

src/astutil.js

@@ -284,7 +284,7 @@ function parse(src,jsxCheck) {
     return tsEstree.parse(src, {
         loc: true,
         range: true,
-        jsx: jsxCheck
+        jsx: false
     });
 }
 
@@ -302,6 +302,8 @@ function buildProgram(fname, src) {
 
     // trim hashbang
     src = prep.trimHashbangPrep(src);
+    // Transform JSX
+    src = prep.jsxPrep(src);
     // extract script from .vue file
     try {
         if (fname.endsWith('.vue')) {

src/srcPreprocessor.js

@@ -18,6 +18,7 @@
 const babel = require('@babel/core');
 
 const nameToFunc = {
+    'jsx': jsxPrep,
     'flow': stripFlowPrep,
     'hashbang': trimHashbangPrep
 };
@@ -51,6 +52,22 @@ function stripFlowPrep(src) {
     }).code;
 }
 
+function jsxPrep(src) {
+    src = babel.transform(src, {
+        presets: ["@babel/preset-env"],
+        plugins: [
+            ['@babel/plugin-transform-react-jsx',
+            {
+            pragma: 'callGraphCreateElement',
+            pragmaFrag: 'Fragment',
+            }],
+        ],
+        retainLines: true,
+        parserOpts: {strictMode: false}
+    }).code;
+    return src.replace(/callGraphCreateElement\(([^,]+), /g, '$1(')
+}
+
 /* Trim hashbang to avoid the parser blowing up
 Example:
     #!/usr/bin/env node
@@ -71,4 +88,5 @@ function trimHashbangPrep(src) {
 
 module.exports.applyPreps = applyPreps;
 module.exports.stripFlowPrep = stripFlowPrep;
+module.exports.jsxPrep = jsxPrep;
 module.exports.trimHashbangPrep = trimHashbangPrep;

tests/jsx/nested.js

@@ -0,0 +1,7 @@
+const SubComponent = () => null;
+const InternalComponent = () => null;
+const Component = () => <SubComponent />;
+
+const Root = () => <Component> <InternalComponent /> </Component>;
+
+Root();

tests/jsx/nested.truth

@@ -0,0 +1,4 @@
+'Component' (nested.js@3:108-126) -> 'SubComponent' (nested.js@1:21-31)
+'Root' (nested.js@5:195-218) -> 'InternalComponent' (nested.js@2:59-69)
+'Root' (nested.js@5:161-224) -> 'Component' (nested.js@3:89-126)
+'global' (nested.js@7:227-233) -> 'Root' (nested.js@5:142-224)

tests/jsx/simple.js

@@ -0,0 +1,5 @@
+const Component = () => null;
+
+const Root = () => <Component />;
+
+Root();

tests/jsx/simple.truth

@@ -0,0 +1,2 @@
+'Root' (simple.js@3:63-78) -> 'Component' (simple.js@1:18-28)
+'global' (simple.js@5:81-87) -> 'Root' (simple.js@3:44-78)