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>
This commit is contained in:
51
static/transforms/import-react.js
Normal file
51
static/transforms/import-react.js
Normal file
@@ -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'),
|
||||
),
|
||||
);
|
||||
}
|
||||
},
|
||||
},
|
||||
};
|
||||
};
|
||||
Reference in New Issue
Block a user