preloader.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150
  1. const Preloader = /** @constructor */ function () { // eslint-disable-line no-unused-vars
  2. function getTrackedResponse(response, load_status) {
  3. let clen = 0;
  4. let compressed = false;
  5. response.headers.forEach(function (value, header) {
  6. const h = header.toLowerCase().trim();
  7. // We can't accurately compute compressed stream length.
  8. if (h === 'content-encoding') {
  9. compressed = true;
  10. } else if (h === 'content-length') {
  11. const length = parseInt(value, 10);
  12. if (!Number.isNaN(length) && length > 0) {
  13. clen = length;
  14. }
  15. }
  16. });
  17. if (!compressed && clen) {
  18. load_status.total = clen;
  19. }
  20. function onloadprogress(reader, controller) {
  21. return reader.read().then(function (result) {
  22. if (load_status.done) {
  23. return Promise.resolve();
  24. }
  25. if (result.value) {
  26. controller.enqueue(result.value);
  27. load_status.loaded += result.value.length;
  28. }
  29. if (!result.done) {
  30. return onloadprogress(reader, controller);
  31. }
  32. load_status.done = true;
  33. return Promise.resolve();
  34. });
  35. }
  36. const reader = response.body.getReader();
  37. return new Response(new ReadableStream({
  38. start: function (controller) {
  39. onloadprogress(reader, controller).then(function () {
  40. controller.close();
  41. });
  42. },
  43. }), { headers: response.headers });
  44. }
  45. function loadFetch(file, tracker, raw) {
  46. tracker[file] = {
  47. total: 0,
  48. loaded: 0,
  49. done: false,
  50. };
  51. return fetch(file).then(function (response) {
  52. if (!response.ok) {
  53. return Promise.reject(new Error(`Failed loading file '${file}'`));
  54. }
  55. const tr = getTrackedResponse(response, tracker[file]);
  56. if (raw) {
  57. return Promise.resolve(tr);
  58. }
  59. return tr.arrayBuffer();
  60. });
  61. }
  62. function retry(func, attempts = 1) {
  63. function onerror(err) {
  64. if (attempts <= 1) {
  65. return Promise.reject(err);
  66. }
  67. return new Promise(function (resolve, reject) {
  68. setTimeout(function () {
  69. retry(func, attempts - 1).then(resolve).catch(reject);
  70. }, 1000);
  71. });
  72. }
  73. return func().catch(onerror);
  74. }
  75. const DOWNLOAD_ATTEMPTS_MAX = 4;
  76. const loadingFiles = {};
  77. const lastProgress = { loaded: 0, total: 0 };
  78. let progressFunc = null;
  79. const animateProgress = function () {
  80. let loaded = 0;
  81. let total = 0;
  82. let totalIsValid = true;
  83. let progressIsFinal = true;
  84. Object.keys(loadingFiles).forEach(function (file) {
  85. const stat = loadingFiles[file];
  86. if (!stat.done) {
  87. progressIsFinal = false;
  88. }
  89. if (!totalIsValid || stat.total === 0) {
  90. totalIsValid = false;
  91. total = 0;
  92. } else {
  93. total += stat.total;
  94. }
  95. loaded += stat.loaded;
  96. });
  97. if (loaded !== lastProgress.loaded || total !== lastProgress.total) {
  98. lastProgress.loaded = loaded;
  99. lastProgress.total = total;
  100. if (typeof progressFunc === 'function') {
  101. progressFunc(loaded, total);
  102. }
  103. }
  104. if (!progressIsFinal) {
  105. requestAnimationFrame(animateProgress);
  106. }
  107. };
  108. this.animateProgress = animateProgress;
  109. this.setProgressFunc = function (callback) {
  110. progressFunc = callback;
  111. };
  112. this.loadPromise = function (file, raw = false) {
  113. return retry(loadFetch.bind(null, file, loadingFiles, raw), DOWNLOAD_ATTEMPTS_MAX);
  114. };
  115. this.preloadedFiles = [];
  116. this.preload = function (pathOrBuffer, destPath) {
  117. let buffer = null;
  118. if (typeof pathOrBuffer === 'string') {
  119. const me = this;
  120. return this.loadPromise(pathOrBuffer).then(function (buf) {
  121. me.preloadedFiles.push({
  122. path: destPath || pathOrBuffer,
  123. buffer: buf,
  124. });
  125. return Promise.resolve();
  126. });
  127. } else if (pathOrBuffer instanceof ArrayBuffer) {
  128. buffer = new Uint8Array(pathOrBuffer);
  129. } else if (ArrayBuffer.isView(pathOrBuffer)) {
  130. buffer = new Uint8Array(pathOrBuffer.buffer);
  131. }
  132. if (buffer) {
  133. this.preloadedFiles.push({
  134. path: destPath,
  135. buffer: pathOrBuffer,
  136. });
  137. return Promise.resolve();
  138. }
  139. return Promise.reject(new Error('Invalid object for preloading'));
  140. };
  141. };