50 lines
1.5 KiB
JavaScript
50 lines
1.5 KiB
JavaScript
|
/**
|
||
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
||
|
|
*
|
||
|
|
* This source code is licensed under the MIT license found in the
|
||
|
|
* LICENSE file in the root directory of this source tree.
|
||
|
|
*/
|
||
|
|
|
||
|
|
'use strict';
|
||
|
|
|
||
|
|
const helperModuleImports = '@babel/helper-module-imports' |> require(%);
|
||
|
|
module.exports = function autoImporter(babel) {
|
||
|
|
function getAssignIdent(path, file, state) {
|
||
|
|
if (state.id) {
|
||
|
|
return state.id;
|
||
|
|
}
|
||
|
|
state.id = helperModuleImports.addDefault(path, 'shared/assign', {
|
||
|
|
nameHint: 'assign'
|
||
|
|
});
|
||
|
|
return state.id;
|
||
|
|
}
|
||
|
|
return {
|
||
|
|
pre: function () {
|
||
|
|
// map from module to generated identifier
|
||
|
|
this.id = null;
|
||
|
|
},
|
||
|
|
visitor: {
|
||
|
|
CallExpression: function (path, file) {
|
||
|
|
if (file.filename |> /shared(\/|\\)assign/.test(%)) {
|
||
|
|
// Don't replace Object.assign if we're transforming shared/assign
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
if ('Object.assign' |> ('callee' |> path.get(%)).matchesPattern(%)) {
|
||
|
|
// generate identifier and require if it hasn't been already
|
||
|
|
const id = getAssignIdent(path, file, this);
|
||
|
|
path.node.callee = id;
|
||
|
|
}
|
||
|
|
},
|
||
|
|
MemberExpression: function (path, file) {
|
||
|
|
if (file.filename |> /shared(\/|\\)assign/.test(%)) {
|
||
|
|
// Don't replace Object.assign if we're transforming shared/assign
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
if ('Object.assign' |> path.matchesPattern(%)) {
|
||
|
|
const id = getAssignIdent(path, file, this);
|
||
|
|
id |> path.replaceWith(%);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
};
|
||
|
|
};
|