SymlinkPlugin.js 2.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. const forEachBail = require("./forEachBail");
  7. const getPaths = require("./getPaths");
  8. const { getType, PathType } = require("./util/path");
  9. /** @typedef {import("./Resolver")} Resolver */
  10. /** @typedef {import("./Resolver").ResolveRequest} ResolveRequest */
  11. /** @typedef {import("./Resolver").ResolveStepHook} ResolveStepHook */
  12. module.exports = class SymlinkPlugin {
  13. /**
  14. * @param {string | ResolveStepHook} source source
  15. * @param {string | ResolveStepHook} target target
  16. */
  17. constructor(source, target) {
  18. this.source = source;
  19. this.target = target;
  20. }
  21. /**
  22. * @param {Resolver} resolver the resolver
  23. * @returns {void}
  24. */
  25. apply(resolver) {
  26. const target = resolver.ensureHook(this.target);
  27. const fs = resolver.fileSystem;
  28. resolver
  29. .getHook(this.source)
  30. .tapAsync("SymlinkPlugin", (request, resolveContext, callback) => {
  31. if (request.ignoreSymlinks) return callback();
  32. const pathsResult = getPaths(/** @type {string} */ (request.path));
  33. const pathSegments = pathsResult.segments;
  34. const { paths } = pathsResult;
  35. let containsSymlink = false;
  36. let idx = -1;
  37. forEachBail(
  38. paths,
  39. /**
  40. * @param {string} path path
  41. * @param {(err?: null|Error, result?: null|number) => void} callback callback
  42. * @returns {void}
  43. */
  44. (path, callback) => {
  45. idx++;
  46. if (resolveContext.fileDependencies) {
  47. resolveContext.fileDependencies.add(path);
  48. }
  49. fs.readlink(path, (err, result) => {
  50. if (!err && result) {
  51. pathSegments[idx] = /** @type {string} */ (result);
  52. containsSymlink = true;
  53. // Shortcut when absolute symlink found
  54. const resultType = getType(result.toString());
  55. if (
  56. resultType === PathType.AbsoluteWin ||
  57. resultType === PathType.AbsolutePosix
  58. ) {
  59. return callback(null, idx);
  60. }
  61. }
  62. callback();
  63. });
  64. },
  65. /**
  66. * @param {(null | Error)=} err error
  67. * @param {(null|number)=} idx result
  68. * @returns {void}
  69. */
  70. (err, idx) => {
  71. if (!containsSymlink) return callback();
  72. const resultSegments =
  73. typeof idx === "number"
  74. ? pathSegments.slice(0, idx + 1)
  75. : [...pathSegments];
  76. const result = resultSegments.reduceRight((a, b) =>
  77. resolver.join(a, b),
  78. );
  79. /** @type {ResolveRequest} */
  80. const obj = {
  81. ...request,
  82. path: result,
  83. };
  84. resolver.doResolve(
  85. target,
  86. obj,
  87. `resolved symlink to ${result}`,
  88. resolveContext,
  89. callback,
  90. );
  91. },
  92. );
  93. });
  94. }
  95. };