patchFs.js 3.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports["default"] = patchFs;
  6. var _lists = require("./util/lists");
  7. function _createForOfIteratorHelper(r, e) { var t = "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (!t) { if (Array.isArray(r) || (t = _unsupportedIterableToArray(r)) || e && r && "number" == typeof r.length) { t && (r = t); var _n = 0, F = function F() {}; return { s: F, n: function n() { return _n >= r.length ? { done: !0 } : { done: !1, value: r[_n++] }; }, e: function e(r) { throw r; }, f: F }; } throw new TypeError("Invalid attempt to iterate non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); } var o, a = !0, u = !1; return { s: function s() { t = t.call(r); }, n: function n() { var r = t.next(); return a = r.done, r; }, e: function e(r) { u = !0, o = r; }, f: function f() { try { a || null == t["return"] || t["return"](); } finally { if (u) throw o; } } }; }
  8. function _unsupportedIterableToArray(r, a) { if (r) { if ("string" == typeof r) return _arrayLikeToArray(r, a); var t = {}.toString.call(r).slice(8, -1); return "Object" === t && r.constructor && (t = r.constructor.name), "Map" === t || "Set" === t ? Array.from(r) : "Arguments" === t || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(t) ? _arrayLikeToArray(r, a) : void 0; } }
  9. function _arrayLikeToArray(r, a) { (null == a || a > r.length) && (a = r.length); for (var e = 0, n = Array(a); e < a; e++) n[e] = r[e]; return n; }
  10. function patchFs(vol) {
  11. var fs = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : require('fs');
  12. var bkp = {};
  13. var patch = function patch(key, newValue) {
  14. bkp[key] = fs[key];
  15. fs[key] = newValue;
  16. };
  17. var patchMethod = function patchMethod(key) {
  18. return patch(key, vol[key].bind(vol));
  19. };
  20. var _iterator = _createForOfIteratorHelper(_lists.fsProps),
  21. _step;
  22. try {
  23. for (_iterator.s(); !(_step = _iterator.n()).done;) {
  24. var prop = _step.value;
  25. if (typeof vol[prop] !== 'undefined') patch(prop, vol[prop]);
  26. }
  27. } catch (err) {
  28. _iterator.e(err);
  29. } finally {
  30. _iterator.f();
  31. }
  32. if (typeof vol.StatWatcher === 'function') {
  33. patch('StatWatcher', vol.FSWatcher.bind(null, vol));
  34. }
  35. if (typeof vol.FSWatcher === 'function') {
  36. patch('FSWatcher', vol.StatWatcher.bind(null, vol));
  37. }
  38. if (typeof vol.ReadStream === 'function') {
  39. patch('ReadStream', vol.ReadStream.bind(null, vol));
  40. }
  41. if (typeof vol.WriteStream === 'function') {
  42. patch('WriteStream', vol.WriteStream.bind(null, vol));
  43. }
  44. if (typeof vol._toUnixTimestamp === 'function') patchMethod('_toUnixTimestamp');
  45. var _iterator2 = _createForOfIteratorHelper(_lists.fsAsyncMethods),
  46. _step2;
  47. try {
  48. for (_iterator2.s(); !(_step2 = _iterator2.n()).done;) {
  49. var method = _step2.value;
  50. if (typeof vol[method] === 'function') patchMethod(method);
  51. }
  52. } catch (err) {
  53. _iterator2.e(err);
  54. } finally {
  55. _iterator2.f();
  56. }
  57. var _iterator3 = _createForOfIteratorHelper(_lists.fsSyncMethods),
  58. _step3;
  59. try {
  60. for (_iterator3.s(); !(_step3 = _iterator3.n()).done;) {
  61. var _method = _step3.value;
  62. if (typeof vol[_method] === 'function') patchMethod(_method);
  63. }
  64. } catch (err) {
  65. _iterator3.e(err);
  66. } finally {
  67. _iterator3.f();
  68. }
  69. var promisesBackup;
  70. try {
  71. promisesBackup = fs.promises;
  72. Object.defineProperty(fs, 'promises', {
  73. get: function get() {
  74. return vol.promises;
  75. }
  76. });
  77. } catch (_unused) {
  78. undefined;
  79. }
  80. return function unpatch() {
  81. for (var key in bkp) fs[key] = bkp[key];
  82. if (promisesBackup) {
  83. Object.defineProperty(fs, 'promises', {
  84. get: function get() {
  85. return promisesBackup;
  86. }
  87. });
  88. }
  89. };
  90. }
  91. ;