fbbf8cf16b
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>
52 lines
1.4 KiB
JavaScript
52 lines
1.4 KiB
JavaScript
/**
|
|
* 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'),
|
|
),
|
|
);
|
|
}
|
|
},
|
|
},
|
|
};
|
|
};
|