getFunctionExpression.js 1.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. /** @typedef {import("estree").ArrowFunctionExpression} ArrowFunctionExpression */
  7. /** @typedef {import("estree").Expression} Expression */
  8. /** @typedef {import("estree").FunctionExpression} FunctionExpression */
  9. /** @typedef {import("estree").SpreadElement} SpreadElement */
  10. /** @typedef {{ fn: FunctionExpression | ArrowFunctionExpression, expressions: (Expression | SpreadElement)[], needThis: boolean | undefined }} FunctionExpressionResult */
  11. /**
  12. * @param {Expression | SpreadElement} expr expressions
  13. * @returns {FunctionExpressionResult | undefined} function expression with additional information
  14. */
  15. module.exports = (expr) => {
  16. // <FunctionExpression>
  17. if (
  18. expr.type === "FunctionExpression" ||
  19. expr.type === "ArrowFunctionExpression"
  20. ) {
  21. return {
  22. fn: expr,
  23. expressions: [],
  24. needThis: false
  25. };
  26. }
  27. // <FunctionExpression>.bind(<Expression>)
  28. if (
  29. expr.type === "CallExpression" &&
  30. expr.callee.type === "MemberExpression" &&
  31. expr.callee.object.type === "FunctionExpression" &&
  32. expr.callee.property.type === "Identifier" &&
  33. expr.callee.property.name === "bind" &&
  34. expr.arguments.length === 1
  35. ) {
  36. return {
  37. fn: expr.callee.object,
  38. expressions: [expr.arguments[0]],
  39. needThis: undefined
  40. };
  41. }
  42. // (function(_this) {return <FunctionExpression>})(this) (Coffeescript)
  43. if (
  44. expr.type === "CallExpression" &&
  45. expr.callee.type === "FunctionExpression" &&
  46. expr.callee.body.type === "BlockStatement" &&
  47. expr.arguments.length === 1 &&
  48. expr.arguments[0].type === "ThisExpression" &&
  49. expr.callee.body.body &&
  50. expr.callee.body.body.length === 1 &&
  51. expr.callee.body.body[0].type === "ReturnStatement" &&
  52. expr.callee.body.body[0].argument &&
  53. expr.callee.body.body[0].argument.type === "FunctionExpression"
  54. ) {
  55. return {
  56. fn: expr.callee.body.body[0].argument,
  57. expressions: [],
  58. needThis: true
  59. };
  60. }
  61. };