feat: default download info

ido-pluto · Feb 26, 2024 · 24d6701 · 24d6701
1 parent 6abd0c3
commit 24d6701
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 8 deletions.
diff --git a/src/browser.ts b/src/browser.ts
@@ -11,7 +11,7 @@ import EmptyResponseError from "./download/download-engine/streams/download-engi
 import StatusCodeError from "./download/download-engine/streams/download-engine-fetch-stream/errors/status-code-error.js";
 import XhrError from "./download/download-engine/streams/download-engine-fetch-stream/errors/xhr-error.js";
 import BaseDownloadEngineFetchStream, {
-    DownloadEngineFetchStreamOptions
+    BaseDownloadEngineFetchStreamOptions
 } from "./download/download-engine/streams/download-engine-fetch-stream/base-download-engine-fetch-stream.js";
 import DownloadEngineFetchStreamFetch
     from "./download/download-engine/streams/download-engine-fetch-stream/download-engine-fetch-stream-fetch.js";
@@ -46,6 +46,6 @@ export type {
     DownloadEngineOptionsBrowser,
     DownloadEngineWriteStreamBrowserWriter,
     DownloadEngineFileOptionsWithDefaults as DownloadEngineFileOptions,
-    DownloadEngineFetchStreamOptions,
+    BaseDownloadEngineFetchStreamOptions,
     TransferProgressWithStatus
 };
diff --git a/src/download/download-engine/engine/base-download-engine.ts b/src/download/download-engine/engine/base-download-engine.ts
@@ -1,6 +1,8 @@
 import {DownloadEngineFileOptions, DownloadFile} from "../types.js";
 import DownloadEngineFile, {DownloadEngineFileEvents} from "../download-engine-file.js";
-import BaseDownloadEngineFetchStream from "../streams/download-engine-fetch-stream/base-download-engine-fetch-stream.js";
+import BaseDownloadEngineFetchStream, {
+    BaseDownloadEngineFetchStreamOptions
+} from "../streams/download-engine-fetch-stream/base-download-engine-fetch-stream.js";
 import UrlInputError from "./error/url-input-error.js";
 import Emittery from "emittery";
 import ProgressStatisticsBuilder, {TransferProgressWithStatus} from "../../transfer-visualize/progress-statistics-builder.js";
@@ -13,6 +15,7 @@ export type BaseDownloadEngineOptions<FetchStrategy = "xhr" | "fetch" | "localFi
         headers?: Record<string, string>;
         acceptRangeIsKnown?: boolean;
         fetchStrategy?: FetchStrategy;
+        defaultFetchDownloadInfo?: BaseDownloadEngineFetchStreamOptions["defaultFetchDownloadInfo"];
     };
 
 export interface BaseDownloadEngineEvents extends Omit<DownloadEngineFileEvents, "progress"> {

diff --git a/...download-engine/streams/download-engine-fetch-stream/base-download-engine-fetch-stream.ts b/...download-engine/streams/download-engine-fetch-stream/base-download-engine-fetch-stream.ts
@@ -1,16 +1,17 @@
 import retry from "async-retry";
 
-export type DownloadEngineFetchStreamOptions = {
+export type BaseDownloadEngineFetchStreamOptions = {
     retry?: retry.Options
     headers?: Record<string, string>,
     /**
      * If true, parallel download will be enabled even if the server does not return `accept-range` header, this is good when using cross-origin requests
      */
     acceptRangeAlwaysTrue?: boolean
+    defaultFetchDownloadInfo?: { length: number, acceptRange: boolean }
 };
 
 export default abstract class BaseDownloadEngineFetchStream {
-    constructor(public readonly options: Partial<DownloadEngineFetchStreamOptions> = {}) {
+    constructor(public readonly options: Partial<BaseDownloadEngineFetchStreamOptions> = {}) {
     }
 
     public async fetchBytes(url: string, start: number, end: number, onProgress?: (length: number) => void) {
@@ -22,7 +23,7 @@ export default abstract class BaseDownloadEngineFetchStream {
     protected abstract _fetchBytesWithoutRetry(url: string, start: number, end: number, onProgress?: (length: number) => void): Promise<Uint8Array>;
 
     public async fetchDownloadInfo(url: string): Promise<{ length: number, acceptRange: boolean }> {
-        return await retry(async () => {
+        return this.options.defaultFetchDownloadInfo ?? await retry(async () => {
             return await this._fetchDownloadInfoWithoutRetry(url);
         }, this.options.retry);
     }

diff --git a/src/index.ts b/src/index.ts
@@ -6,7 +6,7 @@ import DownloadEngineFile, {DownloadEngineFileOptionsWithDefaults} from "./downl
 import {downloadFile, downloadSequence} from "./download/node-download.js";
 import BaseDownloadEngine from "./download/download-engine/engine/base-download-engine.js";
 import BaseDownloadEngineFetchStream, {
-    DownloadEngineFetchStreamOptions
+    BaseDownloadEngineFetchStreamOptions
 } from "./download/download-engine/streams/download-engine-fetch-stream/base-download-engine-fetch-stream.js";
 import DownloadEngineFetchStreamFetch
     from "./download/download-engine/streams/download-engine-fetch-stream/download-engine-fetch-stream-fetch.js";
@@ -41,7 +41,7 @@ export type {
     TransferCliStatus,
     DownloadEngineOptionsNodejs,
     DownloadEngineFileOptionsWithDefaults as DownloadEngineFileOptions,
-    DownloadEngineFetchStreamOptions,
+    BaseDownloadEngineFetchStreamOptions,
     TransferProgressWithStatus
 };