Initial commit 🎉

fbshipit-source-id: b6fc29740c6875d2e78953b8a7123890a67930f2
Co-authored-by: Sebastian McKenzie <sebmck@fb.com>
Co-authored-by: John Knox <jknox@fb.com>
Co-authored-by: Emil Sjölander <emilsj@fb.com>
Co-authored-by: Pritesh Nandgaonkar <prit91@fb.com>

static/transforms/dynamic-requires.js

@@ -0,0 +1,31 @@
+/**
+ * Copyright 2018-present Facebook.
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ * @format
+ */
+function isDynamicRequire(node) {
+  return (
+    node.type === 'CallExpression' &&
+    node.callee.type === 'Identifier' &&
+    node.callee.name === 'require' &&
+    (node.arguments.length !== 1 || node.arguments[0].type !== 'StringLiteral')
+  );
+}
+
+module.exports = function(babel) {
+  const t = babel.types;
+
+  return {
+    name: 'replace-dynamic-requires',
+    visitor: {
+      CallExpression(path) {
+        if (!isDynamicRequire(path.node)) {
+          return;
+        }
+
+        path.replaceWith(t.identifier('triggerDynamicRequireError'));
+      },
+    },
+  };
+};

static/transforms/electron-requires.js

@@ -0,0 +1,87 @@
+/**
+ * Copyright 2018-present Facebook.
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ * @format
+ */
+
+const BUILTINS = [
+  'electron',
+  'buffer',
+  'child_process',
+  'crypto',
+  'dgram',
+  'dns',
+  'fs',
+  'http',
+  'https',
+  'net',
+  'os',
+  'readline',
+  'stream',
+  'string_decoder',
+  'tls',
+  'tty',
+  'zlib',
+  'constants',
+  'events',
+  'url',
+  'assert',
+  'util',
+  'path',
+  'punycode',
+  'querystring',
+  'cluster',
+  'console',
+  'module',
+  'process',
+  'vm',
+  'domain',
+  'v8',
+  'repl',
+  'timers',
+];
+
+const IGNORED_MODULES = [
+  'bufferutil',
+  'utf-8-validate',
+  'spawn-sync',
+  './src/logcat',
+  './src/monkey',
+  './src/adb',
+];
+
+function isRequire(node) {
+  return (
+    node.type === 'CallExpression' &&
+    node.callee.type === 'Identifier' &&
+    node.callee.name === 'require' &&
+    node.arguments.length === 1 &&
+    node.arguments[0].type === 'StringLiteral'
+  );
+}
+
+module.exports = function(babel) {
+  const t = babel.types;
+
+  return {
+    name: 'infinity-import-react',
+    visitor: {
+      CallExpression(path) {
+        if (!isRequire(path.node)) {
+          return;
+        }
+
+        const source = path.node.arguments[0].value;
+
+        if (BUILTINS.includes(source)) {
+          path.node.callee.name = 'electronRequire';
+        }
+
+        if (IGNORED_MODULES.includes(source)) {
+          path.replaceWith(t.identifier('triggerReferenceError'));
+        }
+      },
+    },
+  };
+};

static/transforms/fb-stubs.js

@@ -0,0 +1,45 @@
+/**
+ * Copyright 2018-present Facebook.
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ * @format
+ */
+
+const fs = require('fs');
+const path = require('path');
+const replaceFBStubs = fs.existsSync(
+  path.join(__dirname, '..', '..', 'src', 'fb'),
+);
+
+const requireFromFolder = (folder, path) =>
+  new RegExp(folder + '/[A-Za-z0-9.-_]+(.js)?$', 'g').test(path);
+
+module.exports = function(babel) {
+  return {
+    name: 'replace-dynamic-requires',
+    visitor: {
+      CallExpression(path) {
+        if (
+          replaceFBStubs &&
+          path.node.type === 'CallExpression' &&
+          path.node.callee.type === 'Identifier' &&
+          path.node.callee.name === 'require' &&
+          path.node.arguments.length > 0
+        ) {
+          if (requireFromFolder('fb', path.node.arguments[0].value)) {
+            throw new Error(
+              'Do not requrie directly from fb/, but rather from fb-stubs/ to not break flow-typing and make sure stubs are uptodate.',
+            );
+          } else if (
+            requireFromFolder('fb-stubs', path.node.arguments[0].value)
+          ) {
+            path.node.arguments[0].value = path.node.arguments[0].value.replace(
+              '/fb-stubs/',
+              '/fb/',
+            );
+          }
+        }
+      },
+    },
+  };
+};

static/transforms/import-react.js

@@ -0,0 +1,51 @@
+/**
+ * Copyright 2018-present Facebook.
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ * @format
+ */
+
+module.exports = function(babel) {
+  const t = babel.types;
+
+  return {
+    name: 'infinity-import-react',
+    visitor: {
+      Program: {
+        exit(path, state) {
+          if (state.get('NEEDS_REACT')) {
+            path.unshiftContainer('body', [
+              t.variableDeclaration('var', [
+                t.variableDeclarator(
+                  t.identifier('React'),
+                  t.callExpression(t.identifier('require'), [
+                    t.stringLiteral('react'),
+                  ]),
+                ),
+              ]),
+            ]);
+          }
+        },
+      },
+
+      ReferencedIdentifier(path, state) {
+        // mark react as needing to be imported
+        if (path.node.name === 'React' && !path.scope.getBinding('React')) {
+          state.set('NEEDS_REACT', true);
+        }
+
+        // replace Buffer with require('buffer')
+        if (path.node.name === 'Buffer' && !path.scope.getBinding('Buffer')) {
+          path.replaceWith(
+            t.memberExpression(
+              t.callExpression(t.identifier('require'), [
+                t.stringLiteral('buffer'),
+              ]),
+              t.identifier('Buffer'),
+            ),
+          );
+        }
+      },
+    },
+  };
+};

static/transforms/index.js

@@ -0,0 +1,64 @@
+/**
+ * Copyright 2018-present Facebook.
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ * @format
+ */
+
+const generate = require('@babel/generator').default;
+const babylon = require('babylon');
+const babel = require('@babel/core');
+const metro = require('metro');
+
+exports.transform = function({filename, options, src}) {
+  const presets = [require('../node_modules/babel-preset-react')];
+  const isSonarPlugin = !__dirname.startsWith(options.projectRoot);
+
+  let ast = babylon.parse(src, {
+    filename,
+    plugins: ['jsx', 'flow', 'classProperties', 'objectRestSpread'],
+    sourceType: 'module',
+  });
+
+  // run babel
+  const plugins = [
+    require('../node_modules/babel-plugin-transform-object-rest-spread'),
+    require('../node_modules/babel-plugin-transform-class-properties'),
+    require('../node_modules/babel-plugin-transform-flow-strip-types'),
+    require('./electron-requires.js'),
+    require('./fb-stubs.js'),
+    require('./dynamic-requires.js'),
+  ];
+  if (isSonarPlugin) {
+    plugins.push(require('./sonar-requires.js'));
+  } else {
+    plugins.push(require('./import-react.js'));
+  }
+  plugins.unshift(require('babel-plugin-transform-es2015-modules-commonjs'));
+
+  ast = babel.transformFromAst(ast, src, {
+    babelrc: !filename.includes('node_modules'),
+    code: false,
+    comments: false,
+    compact: false,
+    filename,
+    plugins,
+    presets,
+    sourceMaps: true,
+  }).ast;
+  const result = generate(
+    ast,
+    {
+      filename,
+      sourceFileName: filename,
+      sourceMaps: true,
+    },
+    src,
+  );
+  return {
+    ast,
+    code: result.code,
+    filename,
+    map: result.rawMappings.map(metro.sourceMaps.compactMapping),
+  };
+};

static/transforms/sonar-requires.js

@@ -0,0 +1,64 @@
+/**
+ * Copyright 2018-present Facebook.
+ * This source code is licensed under the MIT license found in the
+ * LICENSE file in the root directory of this source tree.
+ * @format
+ */
+
+// do not apply this transform for these paths
+const EXCLUDE_PATHS = [
+  '/node_modules/react-devtools-core/',
+  'relay-devtools/DevtoolsUI',
+];
+
+function isExcludedPath(path) {
+  for (const epath of EXCLUDE_PATHS) {
+    if (path.indexOf(epath) > -1) {
+      return true;
+    }
+  }
+  return false;
+} // $FlowFixMe
+module.exports = ({types: t}) => ({
+  visitor: {
+    // $FlowFixMe
+    CallExpression(path, state) {
+      if (isExcludedPath(state.file.opts.filename)) {
+        return;
+      }
+      const node = path.node;
+      const args = node.arguments || [];
+      if (
+        node.callee.name === 'require' &&
+        args.length === 1 &&
+        t.isStringLiteral(args[0]) &&
+        args[0].value === 'sonar'
+      ) {
+        path.replaceWith(t.identifier('window.Sonar'));
+      } else if (
+        node.callee.name === 'require' &&
+        args.length > 0 &&
+        t.isStringLiteral(args[0]) &&
+        args[0].value === 'react'
+      ) {
+        path.replaceWith(t.identifier('window.React'));
+      } else if (
+        node.callee.name === 'require' &&
+        args.length > 0 &&
+        t.isStringLiteral(args[0]) &&
+        args[0].value === 'react-dom'
+      ) {
+        path.replaceWith(t.identifier('window.ReactDOM'));
+      }
+    },
+    Identifier(path, state) {
+      if (
+        path.node.name === 'React' &&
+        path.parentPath.node.id !== path.node &&
+        !isExcludedPath(state.file.opts.filename)
+      ) {
+        path.replaceWith(t.identifier('window.React'));
+      }
+    },
+  },
+});