index.js 6.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178
  1. var unparse = require('escodegen').generate;
  2. module.exports = function (ast, vars) {
  3. if (!vars) vars = {};
  4. var FAIL = {};
  5. var result = (function walk (node, scopeVars) {
  6. if (node.type === 'Literal') {
  7. return node.value;
  8. }
  9. else if (node.type === 'UnaryExpression'){
  10. var val = walk(node.argument)
  11. if (node.operator === '+') return +val
  12. if (node.operator === '-') return -val
  13. if (node.operator === '~') return ~val
  14. if (node.operator === '!') return !val
  15. return FAIL
  16. }
  17. else if (node.type === 'ArrayExpression') {
  18. var xs = [];
  19. for (var i = 0, l = node.elements.length; i < l; i++) {
  20. var x = walk(node.elements[i]);
  21. if (x === FAIL) return FAIL;
  22. xs.push(x);
  23. }
  24. return xs;
  25. }
  26. else if (node.type === 'ObjectExpression') {
  27. var obj = {};
  28. for (var i = 0; i < node.properties.length; i++) {
  29. var prop = node.properties[i];
  30. var value = prop.value === null
  31. ? prop.value
  32. : walk(prop.value)
  33. ;
  34. if (value === FAIL) return FAIL;
  35. obj[prop.key.value || prop.key.name] = value;
  36. }
  37. return obj;
  38. }
  39. else if (node.type === 'BinaryExpression' ||
  40. node.type === 'LogicalExpression') {
  41. var l = walk(node.left);
  42. if (l === FAIL) return FAIL;
  43. var r = walk(node.right);
  44. if (r === FAIL) return FAIL;
  45. var op = node.operator;
  46. if (op === '==') return l == r;
  47. if (op === '===') return l === r;
  48. if (op === '!=') return l != r;
  49. if (op === '!==') return l !== r;
  50. if (op === '+') return l + r;
  51. if (op === '-') return l - r;
  52. if (op === '*') return l * r;
  53. if (op === '/') return l / r;
  54. if (op === '%') return l % r;
  55. if (op === '<') return l < r;
  56. if (op === '<=') return l <= r;
  57. if (op === '>') return l > r;
  58. if (op === '>=') return l >= r;
  59. if (op === '|') return l | r;
  60. if (op === '&') return l & r;
  61. if (op === '^') return l ^ r;
  62. if (op === '&&') return l && r;
  63. if (op === '||') return l || r;
  64. return FAIL;
  65. }
  66. else if (node.type === 'Identifier') {
  67. if ({}.hasOwnProperty.call(vars, node.name)) {
  68. return vars[node.name];
  69. }
  70. else return FAIL;
  71. }
  72. else if (node.type === 'ThisExpression') {
  73. if ({}.hasOwnProperty.call(vars, 'this')) {
  74. return vars['this'];
  75. }
  76. else return FAIL;
  77. }
  78. else if (node.type === 'CallExpression') {
  79. var callee = walk(node.callee);
  80. if (callee === FAIL) return FAIL;
  81. if (typeof callee !== 'function') return FAIL;
  82. var ctx = node.callee.object ? walk(node.callee.object) : FAIL;
  83. if (ctx === FAIL) ctx = null;
  84. var args = [];
  85. for (var i = 0, l = node.arguments.length; i < l; i++) {
  86. var x = walk(node.arguments[i]);
  87. if (x === FAIL) return FAIL;
  88. args.push(x);
  89. }
  90. return callee.apply(ctx, args);
  91. }
  92. else if (node.type === 'MemberExpression') {
  93. var obj = walk(node.object);
  94. // do not allow access to methods on Function
  95. if((obj === FAIL) || (typeof obj == 'function')){
  96. return FAIL;
  97. }
  98. if (node.property.type === 'Identifier') {
  99. return obj[node.property.name];
  100. }
  101. var prop = walk(node.property);
  102. if (prop === FAIL) return FAIL;
  103. return obj[prop];
  104. }
  105. else if (node.type === 'ConditionalExpression') {
  106. var val = walk(node.test)
  107. if (val === FAIL) return FAIL;
  108. return val ? walk(node.consequent) : walk(node.alternate)
  109. }
  110. else if (node.type === 'ExpressionStatement') {
  111. var val = walk(node.expression)
  112. if (val === FAIL) return FAIL;
  113. return val;
  114. }
  115. else if (node.type === 'ReturnStatement') {
  116. return walk(node.argument)
  117. }
  118. else if (node.type === 'FunctionExpression') {
  119. var bodies = node.body.body;
  120. // Create a "scope" for our arguments
  121. var oldVars = {};
  122. Object.keys(vars).forEach(function(element){
  123. oldVars[element] = vars[element];
  124. })
  125. for(var i=0; i<node.params.length; i++){
  126. var key = node.params[i];
  127. if(key.type == 'Identifier'){
  128. vars[key.name] = null;
  129. }
  130. else return FAIL;
  131. }
  132. for(var i in bodies){
  133. if(walk(bodies[i]) === FAIL){
  134. return FAIL;
  135. }
  136. }
  137. // restore the vars and scope after we walk
  138. vars = oldVars;
  139. var keys = Object.keys(vars);
  140. var vals = keys.map(function(key) {
  141. return vars[key];
  142. });
  143. return Function(keys.join(', '), 'return ' + unparse(node)).apply(null, vals);
  144. }
  145. else if (node.type === 'TemplateLiteral') {
  146. var str = '';
  147. for (var i = 0; i < node.expressions.length; i++) {
  148. str += walk(node.quasis[i]);
  149. str += walk(node.expressions[i]);
  150. }
  151. str += walk(node.quasis[i]);
  152. return str;
  153. }
  154. else if (node.type === 'TaggedTemplateExpression') {
  155. var tag = walk(node.tag);
  156. var quasi = node.quasi;
  157. var strings = quasi.quasis.map(walk);
  158. var values = quasi.expressions.map(walk);
  159. return tag.apply(null, [strings].concat(values));
  160. }
  161. else if (node.type === 'TemplateElement') {
  162. return node.value.cooked;
  163. }
  164. else return FAIL;
  165. })(ast);
  166. return result === FAIL ? undefined : result;
  167. };