URLPlugin.js 3.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Ivan Kopeykin @vankop
  4. */
  5. "use strict";
  6. const { approve } = require("../javascript/JavascriptParserHelpers");
  7. const InnerGraph = require("../optimize/InnerGraph");
  8. const URLDependency = require("./URLDependency");
  9. /** @typedef {import("estree").NewExpression} NewExpressionNode */
  10. /** @typedef {import("../Compiler")} Compiler */
  11. /** @typedef {import("../javascript/JavascriptParser")} JavascriptParser */
  12. class URLPlugin {
  13. /**
  14. * @param {Compiler} compiler compiler
  15. */
  16. apply(compiler) {
  17. compiler.hooks.compilation.tap(
  18. "URLPlugin",
  19. (compilation, { normalModuleFactory }) => {
  20. compilation.dependencyFactories.set(URLDependency, normalModuleFactory);
  21. compilation.dependencyTemplates.set(
  22. URLDependency,
  23. new URLDependency.Template()
  24. );
  25. /**
  26. * @param {JavascriptParser} parser parser
  27. * @param {object} parserOptions options
  28. */
  29. const parserCallback = (parser, parserOptions) => {
  30. if (parserOptions.url === false) return;
  31. const relative = parserOptions.url === "relative";
  32. /**
  33. * @param {NewExpressionNode} expr expression
  34. * @returns {undefined | string} request
  35. */
  36. const getUrlRequest = expr => {
  37. if (expr.arguments.length !== 2) return;
  38. const [arg1, arg2] = expr.arguments;
  39. if (
  40. arg2.type !== "MemberExpression" ||
  41. arg1.type === "SpreadElement"
  42. )
  43. return;
  44. const chain = parser.extractMemberExpressionChain(arg2);
  45. if (
  46. chain.members.length !== 1 ||
  47. chain.object.type !== "MetaProperty" ||
  48. chain.object.meta.name !== "import" ||
  49. chain.object.property.name !== "meta" ||
  50. chain.members[0] !== "url"
  51. )
  52. return;
  53. const request = parser.evaluateExpression(arg1).asString();
  54. return request;
  55. };
  56. parser.hooks.canRename.for("URL").tap("URLPlugin", approve);
  57. parser.hooks.new.for("URL").tap("URLPlugin", _expr => {
  58. const expr = /** @type {NewExpressionNode} */ (_expr);
  59. const request = getUrlRequest(expr);
  60. if (!request) return;
  61. const [arg1, arg2] = expr.arguments;
  62. const dep = new URLDependency(
  63. request,
  64. [arg1.range[0], arg2.range[1]],
  65. expr.range,
  66. relative
  67. );
  68. dep.loc = expr.loc;
  69. parser.state.current.addDependency(dep);
  70. InnerGraph.onUsage(parser.state, e => (dep.usedByExports = e));
  71. return true;
  72. });
  73. parser.hooks.isPure.for("NewExpression").tap("URLPlugin", _expr => {
  74. const expr = /** @type {NewExpressionNode} */ (_expr);
  75. const { callee } = expr;
  76. if (callee.type !== "Identifier") return;
  77. const calleeInfo = parser.getFreeInfoFromVariable(callee.name);
  78. if (!calleeInfo || calleeInfo.name !== "URL") return;
  79. const request = getUrlRequest(expr);
  80. if (request) return true;
  81. });
  82. };
  83. normalModuleFactory.hooks.parser
  84. .for("javascript/auto")
  85. .tap("URLPlugin", parserCallback);
  86. normalModuleFactory.hooks.parser
  87. .for("javascript/esm")
  88. .tap("URLPlugin", parserCallback);
  89. }
  90. );
  91. }
  92. }
  93. module.exports = URLPlugin;