getFunctionExpression.js 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  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. * Returns function expression with additional information.
  13. * @param {Expression | SpreadElement} expr expressions
  14. * @returns {FunctionExpressionResult | undefined} function expression with additional information
  15. */
  16. module.exports = (expr) => {
  17. // <FunctionExpression>
  18. if (
  19. expr.type === "FunctionExpression" ||
  20. expr.type === "ArrowFunctionExpression"
  21. ) {
  22. return {
  23. fn: expr,
  24. expressions: [],
  25. needThis: false
  26. };
  27. }
  28. // <FunctionExpression>.bind(<Expression>)
  29. if (
  30. expr.type === "CallExpression" &&
  31. expr.callee.type === "MemberExpression" &&
  32. expr.callee.object.type === "FunctionExpression" &&
  33. expr.callee.property.type === "Identifier" &&
  34. expr.callee.property.name === "bind" &&
  35. expr.arguments.length === 1
  36. ) {
  37. return {
  38. fn: expr.callee.object,
  39. expressions: [expr.arguments[0]],
  40. needThis: undefined
  41. };
  42. }
  43. // (function(_this) {return <FunctionExpression>})(this) (Coffeescript)
  44. if (
  45. expr.type === "CallExpression" &&
  46. expr.callee.type === "FunctionExpression" &&
  47. expr.callee.body.type === "BlockStatement" &&
  48. expr.arguments.length === 1 &&
  49. expr.arguments[0].type === "ThisExpression" &&
  50. expr.callee.body.body &&
  51. expr.callee.body.body.length === 1 &&
  52. expr.callee.body.body[0].type === "ReturnStatement" &&
  53. expr.callee.body.body[0].argument &&
  54. expr.callee.body.body[0].argument.type === "FunctionExpression"
  55. ) {
  56. return {
  57. fn: expr.callee.body.body[0].argument,
  58. expressions: [],
  59. needThis: true
  60. };
  61. }
  62. };