explicit-globals/replaceFunctionDeclaration.js at master · cvializ/explicit-globals · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
function functionExpressionNode(functionDeclarationNode, globalName) {
  return {
    "type": "ExpressionStatement",
    "expression": {
      "type": "AssignmentExpression",
      "operator": "=",
      "left": {
        "type": "MemberExpression",
        "object": {
          "type": "Identifier",
          "name": globalName
        },
        "property": functionDeclarationNode.id,
        "computed": false
      },
      "right": {
        "type": "FunctionExpression",
        "id": null,
        "params": functionDeclarationNode.params,
        "body": functionDeclarationNode.body,
        "expression": false
      }
    }
  };
}

function replace(replaceFn, ast, test, globalName) {
  globalName = globalName || 'window';
  var replacement = {
    'FunctionDeclaration': {
      replace: function (node) {
        return functionExpressionNode(node, globalName);
      },
      test: test
    }
  };
  replaceFn(ast, replacement);
}

module.exports = replace;