Projekt

Obecné

Profil

Stáhnout (3.34 KB) Statistiky
| Větev: | Revize:
1
"use strict";
2

    
3
exports.__esModule = true;
4

    
5
var _symbol = require("babel-runtime/core-js/symbol");
6

    
7
var _symbol2 = _interopRequireDefault(_symbol);
8

    
9
exports.default = function (code, opts) {
10
  var stack = void 0;
11
  try {
12
    throw new Error();
13
  } catch (error) {
14
    if (error.stack) {
15
      stack = error.stack.split("\n").slice(1).join("\n");
16
    }
17
  }
18

    
19
  opts = (0, _assign2.default)({
20
    allowReturnOutsideFunction: true,
21
    allowSuperOutsideMethod: true,
22
    preserveComments: false
23
  }, opts);
24

    
25
  var _getAst = function getAst() {
26
    var ast = void 0;
27

    
28
    try {
29
      ast = babylon.parse(code, opts);
30

    
31
      ast = _babelTraverse2.default.removeProperties(ast, { preserveComments: opts.preserveComments });
32

    
33
      _babelTraverse2.default.cheap(ast, function (node) {
34
        node[FROM_TEMPLATE] = true;
35
      });
36
    } catch (err) {
37
      err.stack = err.stack + "from\n" + stack;
38
      throw err;
39
    }
40

    
41
    _getAst = function getAst() {
42
      return ast;
43
    };
44

    
45
    return ast;
46
  };
47

    
48
  return function () {
49
    for (var _len = arguments.length, args = Array(_len), _key = 0; _key < _len; _key++) {
50
      args[_key] = arguments[_key];
51
    }
52

    
53
    return useTemplate(_getAst(), args);
54
  };
55
};
56

    
57
var _cloneDeep = require("lodash/cloneDeep");
58

    
59
var _cloneDeep2 = _interopRequireDefault(_cloneDeep);
60

    
61
var _assign = require("lodash/assign");
62

    
63
var _assign2 = _interopRequireDefault(_assign);
64

    
65
var _has = require("lodash/has");
66

    
67
var _has2 = _interopRequireDefault(_has);
68

    
69
var _babelTraverse = require("babel-traverse");
70

    
71
var _babelTraverse2 = _interopRequireDefault(_babelTraverse);
72

    
73
var _babylon = require("babylon");
74

    
75
var babylon = _interopRequireWildcard(_babylon);
76

    
77
var _babelTypes = require("babel-types");
78

    
79
var t = _interopRequireWildcard(_babelTypes);
80

    
81
function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
82

    
83
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
84

    
85
var FROM_TEMPLATE = "_fromTemplate";
86
var TEMPLATE_SKIP = (0, _symbol2.default)();
87

    
88
function useTemplate(ast, nodes) {
89
  ast = (0, _cloneDeep2.default)(ast);
90
  var _ast = ast,
91
      program = _ast.program;
92

    
93

    
94
  if (nodes.length) {
95
    (0, _babelTraverse2.default)(ast, templateVisitor, null, nodes);
96
  }
97

    
98
  if (program.body.length > 1) {
99
    return program.body;
100
  } else {
101
    return program.body[0];
102
  }
103
}
104

    
105
var templateVisitor = {
106
  noScope: true,
107

    
108
  enter: function enter(path, args) {
109
    var node = path.node;
110

    
111
    if (node[TEMPLATE_SKIP]) return path.skip();
112

    
113
    if (t.isExpressionStatement(node)) {
114
      node = node.expression;
115
    }
116

    
117
    var replacement = void 0;
118

    
119
    if (t.isIdentifier(node) && node[FROM_TEMPLATE]) {
120
      if ((0, _has2.default)(args[0], node.name)) {
121
        replacement = args[0][node.name];
122
      } else if (node.name[0] === "$") {
123
        var i = +node.name.slice(1);
124
        if (args[i]) replacement = args[i];
125
      }
126
    }
127

    
128
    if (replacement === null) {
129
      path.remove();
130
    }
131

    
132
    if (replacement) {
133
      replacement[TEMPLATE_SKIP] = true;
134
      path.replaceInline(replacement);
135
    }
136
  },
137
  exit: function exit(_ref) {
138
    var node = _ref.node;
139

    
140
    if (!node.loc) _babelTraverse2.default.clearNode(node);
141
  }
142
};
143
module.exports = exports["default"];
    (1-1/1)