@milaboratories/pl-drivers 1.5.52 → 1.5.54

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@milaboratories/pl-drivers",
-  "version": "1.5.52",
+  "version": "1.5.54",
   "engines": {
     "node": ">=20"
   },
@@ -30,11 +30,11 @@
     "tar-fs": "^3.0.8",
     "undici": "~7.5.0",
     "zod": "~3.23.8",
-    "@milaboratories/ts-helpers": "^1.2.0",
     "@milaboratories/computable": "^2.4.7",
-    "@milaboratories/pl-client": "^2.9.0",
     "@milaboratories/pl-tree": "^1.6.1",
-    "@milaboratories/pl-model-common": "^1.14.0"
+    "@milaboratories/pl-client": "^2.9.0",
+    "@milaboratories/pl-model-common": "^1.14.0",
+    "@milaboratories/ts-helpers": "^1.2.0"
   },
   "devDependencies": {
     "eslint": "^9.25.1",
@@ -42,18 +42,15 @@
     "typescript": "~5.5.4",
     "vite": "^5.4.11",
     "@types/node": "~20.16.15",
-    "@types/jest": "^29.5.14",
+    "vitest": "^2.1.8",
     "@types/tar-fs": "^2.0.4",
-    "jest": "^29.7.0",
-    "@jest/globals": "^29.7.0",
-    "ts-jest": "^29.2.6",
-    "@milaboratories/eslint-config": "^1.0.4",
-    "@milaboratories/platforma-build-configs": "1.0.3"
+    "@milaboratories/platforma-build-configs": "1.0.3",
+    "@milaboratories/eslint-config": "^1.0.4"
   },
   "scripts": {
     "type-check": "tsc --noEmit --composite false",
     "build": "vite build",
-    "test": "jest --runInBand",
+    "test": "vitest",
     "lint": "eslint .",
     "do-pack": "rm -f *.tgz && pnpm pack && mv *.tgz package.tgz"
   }

package/src/clients/download.test.ts

@@ -8,7 +8,7 @@ import type { GrpcTransport } from '@protobuf-ts/grpc-transport';
 import type { Dispatcher } from 'undici';
 import { text } from 'node:stream/consumers';
 import { ClientDownload, getFullPath, parseLocalUrl } from '../clients/download';
-import { test, expect } from '@jest/globals';
+import { test, expect } from 'vitest';
 
 test('should parse local file url even on Windows', () => {
   const url

package/src/clients/download.ts

@@ -40,26 +40,35 @@ export class ClientDownload {
 
   close() {}
 
+  /** Gets a presign URL and downloads the file.
+   * An optional range with 2 numbers from what byte and to what byte to download can be provided. */
   async downloadBlob(
     info: ResourceInfo,
     options?: RpcOptions,
     signal?: AbortSignal,
+    fromBytes?: number,
+    toBytes?: number,
   ): Promise<DownloadResponse> {
     const { downloadUrl, headers } = await this.grpcGetDownloadUrl(info, options, signal);
 
-    this.logger.info(`download blob from url ${downloadUrl}`);
+    const remoteHeaders = toHeadersMap(headers, fromBytes, toBytes);
+    this.logger.info(`download blob from url ${downloadUrl}, headers: ${JSON.stringify(remoteHeaders)}`);
 
     return isLocal(downloadUrl)
-      ? await this.readLocalFile(downloadUrl)
-      : await this.remoteFileDownloader.download(downloadUrl, toHeadersMap(headers), signal);
+      ? await this.readLocalFile(downloadUrl, fromBytes, toBytes)
+      : await this.remoteFileDownloader.download(downloadUrl, remoteHeaders, signal);
   }
 
-  async readLocalFile(url: string): Promise<DownloadResponse> {
+  async readLocalFile(
+    url: string,
+    fromBytes?: number,
+    toBytes?: number,
+  ): Promise<DownloadResponse> {
     const { storageId, relativePath } = parseLocalUrl(url);
     const fullPath = getFullPath(storageId, this.localStorageIdsToRoot, relativePath);
 
     return {
-      content: Readable.toWeb(fs.createReadStream(fullPath)),
+      content: Readable.toWeb(fs.createReadStream(fullPath, { start: fromBytes, end: toBytes })),
       size: (await fsp.stat(fullPath)).size,
     };
   }

package/src/clients/helpers.ts

@@ -1,3 +1,12 @@
-export function toHeadersMap(headers: { name: string; value: string }[]): Record<string, string> {
-  return Object.fromEntries(headers.map(({ name, value }) => [name, value]));
+export function toHeadersMap(
+  headers: { name: string; value: string }[],
+  fromBytes?: number,
+  toBytes?: number,
+): Record<string, string> {
+  const result = Object.fromEntries(headers.map(({ name, value }) => [name, value]));
+  if (fromBytes !== undefined && toBytes !== undefined) {
+    result['Range'] = `bytes=${fromBytes}-${toBytes}`;
+  }
+
+  return result;
 }

package/src/clients/upload.test.ts

@@ -4,6 +4,7 @@ import { ClientUpload } from '../clients/upload';
 import type { Dispatcher } from 'undici';
 import type { GrpcTransport } from '@protobuf-ts/grpc-transport';
 import { ConsoleLoggerAdapter } from '@milaboratories/ts-helpers';
+import { test, expect } from 'vitest';
 
 test.skip('integration test, grpc upload blob should throw error on NOT_FOUND', async () => {
   await TestHelpers.withTempRoot(async (client) => {
@@ -19,7 +20,7 @@ test.skip('integration test, grpc upload blob should throw error on NOT_FOUND',
         id: 1n as ResourceId,
         type: { name: 'BlobUpload/main', version: '1' },
       });
-      fail('should throw NOT_FOUND');
+      expect(true).toBe(false);
     } catch (e) {
       const err = e as any;
       expect(err.code).toBe('NOT_FOUND');

package/src/drivers/{download_blob.test.ts → download_blob/download_blob.test.ts}

@@ -1,4 +1,4 @@
-import { expect, test } from '@jest/globals';
+import { expect, test } from 'vitest';
 import type {
   FieldId,
   FieldRef,
@@ -15,9 +15,9 @@ import * as fsp from 'node:fs/promises';
 import * as os from 'node:os';
 import * as path from 'node:path';
 import { scheduler } from 'node:timers/promises';
-import { createDownloadClient, createLogsClient } from '../clients/constructors';
+import { createDownloadClient, createLogsClient } from '../../clients/constructors';
 import { DownloadDriver } from './download_blob';
-import type { OnDemandBlobResourceSnapshot } from './types';
+import type { OnDemandBlobResourceSnapshot } from '../types';
 
 const fileName = 'answer_to_the_ultimate_question.txt';
 

package/src/drivers/{download_blob.ts → download_blob/download_blob.ts}

@@ -39,23 +39,23 @@ import * as path from 'node:path';
 import * as readline from 'node:readline/promises';
 import { Readable, Writable } from 'node:stream';
 import { buffer } from 'node:stream/consumers';
-import type { ClientDownload } from '../clients/download';
-import type { ClientLogs } from '../clients/logs';
+import type { ClientDownload } from '../../clients/download';
+import type { ClientLogs } from '../../clients/logs';
 import { DownloadBlobTask, nonRecoverableError } from './download_blob_task';
-import { FilesCache } from './helpers/files_cache';
+import { FilesCache } from '../helpers/files_cache';
 import {
   isLocalBlobHandle,
   newLocalHandle,
   parseLocalHandle,
-} from './helpers/download_local_handle';
-import { getSize, OnDemandBlobResourceSnapshot } from './types';
+} from '../helpers/download_local_handle';
+import { getSize, OnDemandBlobResourceSnapshot } from '../types';
 import {
   isRemoteBlobHandle,
   newRemoteHandle,
   parseRemoteHandle,
-} from './helpers/download_remote_handle';
-import { getResourceInfoFromLogHandle, newLogHandle } from './helpers/logs_handle';
-import { Updater, WrongResourceTypeError } from './helpers/helpers';
+} from '../helpers/download_remote_handle';
+import { getResourceInfoFromLogHandle, newLogHandle } from '../helpers/logs_handle';
+import { Updater, WrongResourceTypeError } from '../helpers/helpers';
 
 export type DownloadDriverOps = {
   /**
@@ -74,8 +74,8 @@ export type DownloadDriverOps = {
 /** DownloadDriver holds a queue of downloading tasks,
  * and notifies every watcher when a file were downloaded. */
 export class DownloadDriver implements BlobDriver {
-  /** Represents a Resource Id to the path of a blob as a map. */
-  private idToDownload: Map<ResourceId, DownloadBlobTask> = new Map();
+  /** Represents a unique key to the path of a blob as a map. */
+  private idToDownload: Map<string, DownloadBlobTask> = new Map();
 
   /** Writes and removes files to a hard drive and holds a counter for every
    * file that should be kept. */
@@ -84,10 +84,10 @@ export class DownloadDriver implements BlobDriver {
   /** Downloads files and writes them to the local dir. */
   private downloadQueue: TaskProcessor;
 
-  private idToOnDemand: Map<ResourceId, OnDemandBlobHolder> = new Map();
+  private idToOnDemand: Map<string, OnDemandBlobHolder> = new Map();
 
-  private idToLastLines: Map<ResourceId, LastLinesGetter> = new Map();
-  private idToProgressLog: Map<ResourceId, LastLinesGetter> = new Map();
+  private idToLastLines: Map<string, LastLinesGetter> = new Map();
+  private idToProgressLog: Map<string, LastLinesGetter> = new Map();
 
   private readonly saveDir: string;
 
@@ -105,26 +105,36 @@ export class DownloadDriver implements BlobDriver {
     this.saveDir = path.resolve(saveDir);
   }
 
-  /** Gets a blob by its resource id or downloads a blob and sets it in a cache. */
+  /** Gets a blob or part of the blob by its resource id or downloads a blob and sets it in a cache. */
   public getDownloadedBlob(
     res: ResourceInfo | PlTreeEntry,
-    ctx: ComputableCtx
+    ctx: ComputableCtx,
+    fromBytes?: number,
+    toBytes?: number,
   ): LocalBlobHandleAndSize | undefined;
   public getDownloadedBlob(
-    res: ResourceInfo | PlTreeEntry
+    res: ResourceInfo | PlTreeEntry,
+    ctx?: undefined,
+    fromBytes?: number,
+    toBytes?: number,
   ): ComputableStableDefined<LocalBlobHandleAndSize>;
   public getDownloadedBlob(
     res: ResourceInfo | PlTreeEntry,
     ctx?: ComputableCtx,
+    fromBytes?: number,
+    toBytes?: number,
   ): Computable<LocalBlobHandleAndSize | undefined> | LocalBlobHandleAndSize | undefined {
-    if (ctx === undefined) return Computable.make((ctx) => this.getDownloadedBlob(res, ctx));
+    if (ctx === undefined) {
+      return Computable.make((ctx) => this.getDownloadedBlob(res, ctx, fromBytes, toBytes));
+    }
 
     const rInfo = treeEntryToResourceInfo(res, ctx);
+    const key = blobKey(rInfo.id, fromBytes, toBytes);
 
     const callerId = randomUUID();
-    ctx.addOnDestroy(() => this.releaseBlob(rInfo.id, callerId));
+    ctx.addOnDestroy(() => this.releaseBlob(key, callerId));
 
-    const result = this.getDownloadedBlobNoCtx(ctx.watcher, rInfo as ResourceSnapshot, callerId);
+    const result = this.getDownloadedBlobNoCtx(ctx.watcher, rInfo as ResourceSnapshot, callerId, fromBytes, toBytes);
     if (result == undefined) ctx.markUnstable('download blob is still undefined');
 
     return result;
@@ -134,10 +144,12 @@ export class DownloadDriver implements BlobDriver {
     w: Watcher,
     rInfo: ResourceSnapshot,
     callerId: string,
+    fromBytes?: number,
+    toBytes?: number,
   ): LocalBlobHandleAndSize | undefined {
     validateDownloadableResourceType('getDownloadedBlob', rInfo.type);
 
-    let task = this.idToDownload.get(rInfo.id);
+    let task = this.idToDownload.get(blobKey(rInfo.id, fromBytes, toBytes));
 
     if (task === undefined) {
       // schedule the blob downloading
@@ -156,8 +168,8 @@ export class DownloadDriver implements BlobDriver {
     throw result.result.error;
   }
 
-  private setNewDownloadTask(rInfo: ResourceSnapshot) {
-    const fPath = this.getFilePath(rInfo.id);
+  private setNewDownloadTask(rInfo: ResourceSnapshot, fromBytes?: number, toBytes?: number) {
+    const fPath = path.resolve(this.saveDir, blobKey(rInfo.id, fromBytes, toBytes));
     const result = new DownloadBlobTask(
       this.logger,
       this.clientDownload,
@@ -165,7 +177,7 @@ export class DownloadDriver implements BlobDriver {
       fPath,
       newLocalHandle(fPath, this.signer),
     );
-    this.idToDownload.set(rInfo.id, result);
+    this.idToDownload.set(blobKey(rInfo.id, fromBytes, toBytes), result);
 
     return result;
   }
@@ -212,11 +224,11 @@ export class DownloadDriver implements BlobDriver {
   ): RemoteBlobHandleAndSize {
     validateDownloadableResourceType('getOnDemandBlob', info.type);
 
-    let blob = this.idToOnDemand.get(info.id);
+    let blob = this.idToOnDemand.get(blobKey(info.id));
 
     if (blob === undefined) {
       blob = new OnDemandBlobHolder(getSize(info), newRemoteHandle(info, this.signer));
-      this.idToOnDemand.set(info.id, blob);
+      this.idToOnDemand.set(blobKey(info.id), blob);
     }
 
     blob.attach(callerId);
@@ -279,7 +291,7 @@ export class DownloadDriver implements BlobDriver {
 
     const r = treeEntryToResourceInfo(res, ctx);
     const callerId = randomUUID();
-    ctx.addOnDestroy(() => this.releaseBlob(r.id, callerId));
+    ctx.addOnDestroy(() => this.releaseBlob(blobKey(r.id), callerId));
 
     const result = this.getLastLogsNoCtx(ctx.watcher, r as ResourceSnapshot, lines, callerId);
     if (result == undefined)
@@ -300,11 +312,11 @@ export class DownloadDriver implements BlobDriver {
 
     const { path } = parseLocalHandle(blob.handle, this.signer);
 
-    let logGetter = this.idToLastLines.get(rInfo.id);
+    let logGetter = this.idToLastLines.get(blobKey(rInfo.id));
 
     if (logGetter == undefined) {
       const newLogGetter = new LastLinesGetter(path, lines);
-      this.idToLastLines.set(rInfo.id, newLogGetter);
+      this.idToLastLines.set(blobKey(rInfo.id), newLogGetter);
       logGetter = newLogGetter;
     }
 
@@ -335,7 +347,7 @@ export class DownloadDriver implements BlobDriver {
 
     const r = treeEntryToResourceInfo(res, ctx);
     const callerId = randomUUID();
-    ctx.addOnDestroy(() => this.releaseBlob(r.id, callerId));
+    ctx.addOnDestroy(() => this.releaseBlob(blobKey(r.id), callerId));
 
     const result = this.getProgressLogNoCtx(
       ctx.watcher,
@@ -361,11 +373,11 @@ export class DownloadDriver implements BlobDriver {
     if (blob == undefined) return undefined;
     const { path } = parseLocalHandle(blob.handle, this.signer);
 
-    let logGetter = this.idToProgressLog.get(rInfo.id);
+    let logGetter = this.idToProgressLog.get(blobKey(rInfo.id));
 
     if (logGetter == undefined) {
       const newLogGetter = new LastLinesGetter(path, 1, patternToSearch);
-      this.idToProgressLog.set(rInfo.id, newLogGetter);
+      this.idToProgressLog.set(blobKey(rInfo.id), newLogGetter);
 
       logGetter = newLogGetter;
     }
@@ -440,8 +452,8 @@ export class DownloadDriver implements BlobDriver {
     };
   }
 
-  private async releaseBlob(blobId: ResourceId, callerId: string) {
-    const task = this.idToDownload.get(blobId);
+  private async releaseBlob(blobKey: string, callerId: string) {
+    const task = this.idToDownload.get(blobKey);
     if (task == undefined) return;
 
     if (this.cache.existsFile(task.path)) {
@@ -474,14 +486,14 @@ export class DownloadDriver implements BlobDriver {
   private removeTask(task: DownloadBlobTask, reason: string) {
     task.abort(reason);
     task.change.markChanged();
-    this.idToDownload.delete(task.rInfo.id);
-    this.idToLastLines.delete(task.rInfo.id);
-    this.idToProgressLog.delete(task.rInfo.id);
+    this.idToDownload.delete(blobKey(task.rInfo.id));
+    this.idToLastLines.delete(blobKey(task.rInfo.id));
+    this.idToProgressLog.delete(blobKey(task.rInfo.id));
   }
 
   private async releaseOnDemandBlob(blobId: ResourceId, callerId: string) {
-    const deleted = this.idToOnDemand.get(blobId)?.release(callerId) ?? false;
-    if (deleted) this.idToOnDemand.delete(blobId);
+    const deleted = this.idToOnDemand.get(blobKey(blobId))?.release(callerId) ?? false;
+    if (deleted) this.idToOnDemand.delete(blobKey(blobId));
   }
 
   /** Removes all files from a hard drive. */
@@ -493,10 +505,6 @@ export class DownloadDriver implements BlobDriver {
       task.change.markChanged();
     });
   }
-
-  private getFilePath(rId: ResourceId): string {
-    return path.resolve(this.saveDir, String(BigInt(rId)));
-  }
 }
 
 /** Keeps a counter to the on demand handle. */
@@ -610,3 +618,12 @@ function validateDownloadableResourceType(methodName: string, rType: ResourceTyp
     throw new WrongResourceTypeError(message);
   }
 }
+
+/** Returns a file name and the unique key of the file.*/
+function blobKey(rId: ResourceId, fromBytes?: number, toBytes?: number): string {
+  if (fromBytes !== undefined && toBytes !== undefined) {
+    return `${BigInt(rId)}_${fromBytes}-${toBytes}`;
+  }
+
+  return `${BigInt(rId)}`;
+}

package/src/drivers/{download_blob_task.ts → download_blob/download_blob_task.ts}

@@ -16,9 +16,9 @@ import fs from 'node:fs';
 import * as fsp from 'node:fs/promises';
 import * as path from 'node:path';
 import { Writable } from 'node:stream';
-import type { ClientDownload } from '../clients/download';
-import { UnknownStorageError, WrongLocalFileUrl } from '../clients/download';
-import { NetworkError400 } from '../helpers/download';
+import type { ClientDownload } from '../../clients/download';
+import { UnknownStorageError, WrongLocalFileUrl } from '../../clients/download';
+import { NetworkError400 } from '../../helpers/download';
 
 /** Downloads a blob. */
 export class DownloadBlobTask {

package/src/drivers/download_blob_url/driver.ts

@@ -21,7 +21,7 @@ import type { ClientDownload } from '../../clients/download';
 import { getPathForFolderURL, isFolderURL } from './url';
 import type { Id } from './driver_id';
 import { newId } from './driver_id';
-import { nonRecoverableError } from '../download_blob_task';
+import { nonRecoverableError } from '../download_blob/download_blob_task';
 
 export type DownloadBlobToURLDriverOps = {
   cacheSoftSizeBytes: number;

package/src/drivers/download_blob_url/url.test.ts

@@ -1,5 +1,5 @@
 import path from 'path';
-import { describe, test, expect } from '@jest/globals';
+import { describe, test, expect } from 'vitest';
 import { isFolderURL, getPathForFolderURL } from './url';
 import type { Signer } from '@milaboratories/ts-helpers';
 import { HmacSha256Signer } from '@milaboratories/ts-helpers';

package/src/drivers/download_url.test.ts

@@ -7,7 +7,7 @@ import * as fs from 'node:fs';
 import * as fsp from 'node:fs/promises';
 import * as path from 'node:path';
 import { DownloadUrlDriver } from './download_url';
-import { test, expect } from '@jest/globals';
+import { test, expect } from 'vitest';
 
 test('should download a tar archive and extracts its content and then deleted', async () => {
   await TestHelpers.withTempRoot(async (client) => {

package/src/drivers/helpers/files_cache.test.ts

@@ -1,4 +1,4 @@
-import { expect, test } from '@jest/globals';
+import { expect, test } from 'vitest';
 import { CallersCounter } from '@milaboratories/ts-helpers';
 import type { CachedFile } from './files_cache';
 import { FilesCache } from './files_cache';

package/src/drivers/helpers/files_cache.ts

@@ -50,7 +50,10 @@ export class FilesCache<T extends CachedFile> {
     mapEntries(this.cache)
       .filter(([_, file]: [string, T]) => file.counter.isZero())
       .forEach(([path, _]) => {
-        if (this.totalSizeBytes - freedBytes <= this.softSizeBytes) return;
+        if (this.totalSizeBytes - freedBytes <= this.softSizeBytes) {
+          return;
+        }
+
         const file = mapGet(this.cache, path);
         freedBytes += file.size;
         result.push(file);
@@ -64,9 +67,13 @@ export class FilesCache<T extends CachedFile> {
     this.cache.set(file.path, file);
     file.counter.inc(callerId);
 
-    if (file.size < 0) throw new Error(`empty sizeBytes: ${file}`);
+    if (file.size < 0) {
+      throw new Error(`empty sizeBytes: ${file}`);
+    }
 
-    if (created) this.totalSizeBytes += file.size;
+    if (created) {
+      this.totalSizeBytes += file.size;
+    }
   }
 
   removeCache(file: T) {

package/src/drivers/logs.test.ts

@@ -1,4 +1,4 @@
-import { expect, test } from '@jest/globals';
+import { expect, test } from 'vitest';
 import { Computable } from '@milaboratories/computable';
 import type {
   AnyFieldRef,
@@ -21,7 +21,7 @@ import * as os from 'node:os';
 import * as path from 'node:path';
 import { scheduler } from 'node:timers/promises';
 import { createDownloadClient, createLogsClient } from '../clients/constructors';
-import { DownloadDriver } from './download_blob';
+import { DownloadDriver } from './download_blob/download_blob';
 import { LogsDriver } from './logs';
 import { LogsStreamDriver } from './logs_stream';
 

package/src/drivers/logs.ts

@@ -4,7 +4,7 @@ import type { PlTreeEntry, ResourceInfo } from '@milaboratories/pl-tree';
 import type { LogsStreamDriver } from './logs_stream';
 import type * as sdk from '@milaboratories/pl-model-common';
 import type { MiLogger } from '@milaboratories/ts-helpers';
-import type { DownloadDriver } from './download_blob';
+import type { DownloadDriver } from './download_blob/download_blob';
 import { isLiveLogHandle } from './helpers/logs_handle';
 
 export class LogsDriver implements sdk.LogsDriver {

package/src/drivers/ls.test.ts

@@ -4,7 +4,7 @@ import { createLsFilesClient } from '../clients/constructors';
 import { TestHelpers } from '@milaboratories/pl-client';
 import * as os from 'node:os';
 import * as path from 'node:path';
-import { test, expect } from '@jest/globals';
+import { test, expect } from 'vitest';
 import { isImportFileHandleIndex, isImportFileHandleUpload } from '@milaboratories/pl-model-common';
 
 const assetsPath = path.resolve('../../../assets');

package/src/drivers/ls.ts

@@ -42,8 +42,19 @@ export interface InternalLsDriver extends sdk.LsDriver {
    * To be used in tests and in implementation of the native file selection UI API.
    * */
   getLocalFileHandle(localPath: string): Promise<sdk.LocalImportFileHandle>;
+
+  listRemoteFilesWithAdditionalInfo(storage: sdk.StorageHandle, fullPath: string): Promise<ListRemoteFilesResultWithAdditionalInfo>;
 }
 
+export type ListRemoteFilesResultWithAdditionalInfo = {
+  parent?: string;
+  entries: LsEntryWithAdditionalInfo[];
+};
+
+export type LsEntryWithAdditionalInfo = LsEntry & {
+  size: number;
+};
+
 export type OpenFileDialogCallback = (
   multipleFiles: boolean,
   ops?: OpenDialogOps
@@ -231,6 +242,28 @@ export class LsDriver implements InternalLsDriver {
     return { entries };
   }
 
+  public async listRemoteFilesWithAdditionalInfo(
+    storageHandle: sdk.StorageHandle,
+    fullPath: string,
+  ): Promise<ListRemoteFilesResultWithAdditionalInfo> {
+    const storageData = parseStorageHandle(storageHandle);
+    if (!storageData.isRemote) {
+      throw new Error(`Storage ${storageData.name} is not remote`);
+    }
+
+    const response = await this.lsClient.list(storageData, fullPath);
+
+    return {
+      entries: response.items.map((e) => ({
+        type: e.isDir ? 'dir' : 'file',
+        name: e.name,
+        fullPath: e.fullName,
+        handle: createIndexImportHandle(storageData.name, e.fullName),
+        size: Number(e.size),
+      })),
+    };
+  }
+
   public async fileToImportHandle(file: sdk.FileLike): Promise<sdk.ImportFileHandle> {
     throw new Error(
       'Not implemented. This method must be implemented and intercepted in desktop preload script.',

package/src/drivers/upload.test.ts

@@ -7,7 +7,7 @@ import * as os from 'node:os';
 import * as path from 'node:path';
 import { makeBlobImportSnapshot, UploadDriver } from './upload';
 import { createUploadBlobClient, createUploadProgressClient } from '../clients/constructors';
-import { expect, test } from '@jest/globals';
+import { expect, test } from 'vitest';
 import { Computable } from '@milaboratories/computable';
 import { SynchronizedTreeState } from '@milaboratories/pl-tree';
 import type { ImportResourceSnapshot } from './types';

package/src/helpers/download.ts

@@ -39,7 +39,7 @@ export class RemoteFileDownloader {
 }
 
 async function checkStatusCodeOk(statusCode: number, webBody: ReadableStream, url: string) {
-  if (statusCode != 200) {
+  if (statusCode != 200 && statusCode != 206 /* partial content from range request */) {
     const beginning = (await text(webBody)).substring(0, 1000);
 
     if (400 <= statusCode && statusCode < 500) {

package/src/index.ts

@@ -5,7 +5,7 @@ export * from './clients/ls_api';
 export * from './clients/logs';
 export * from './clients/constructors';
 
-export * from './drivers/download_blob';
+export * from './drivers/download_blob/download_blob';
 export * from './drivers/download_blob_url/driver';
 export * from './drivers/download_blob_url/snapshot';
 export * from './drivers/upload';

package/dist/drivers/download_blob.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"download_blob.d.ts","sourceRoot":"","sources":["../../src/drivers/download_blob.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,aAAa,EACb,uBAAuB,EAExB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAEL,UAAU,EACX,MAAM,4BAA4B,CAAC;AAGpC,OAAO,KAAK,EACV,YAAY,EACZ,UAAU,EACV,eAAe,EACf,sBAAsB,EACtB,cAAc,EACd,gBAAgB,EAChB,uBAAuB,EACvB,oBAAoB,EACrB,MAAM,iCAAiC,CAAC;AACzC,OAAO,KAAK,EACV,WAAW,EACX,YAAY,EACM,MAAM,yBAAyB,CAAC;AAMpD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,4BAA4B,CAAC;AAWnE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAQlD,OAAO,EAAW,4BAA4B,EAAE,MAAM,SAAS,CAAC;AAShE,MAAM,MAAM,iBAAiB,GAAG;IAC9B;;;;SAIK;IACL,kBAAkB,EAAE,MAAM,CAAC;IAC3B;;;SAGK;IACL,oBAAoB,EAAE,MAAM,CAAC;CAC9B,CAAC;AAEF;6DAC6D;AAC7D,qBAAa,cAAe,YAAW,UAAU;IAmB7C,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,cAAc;IAC/B,OAAO,CAAC,QAAQ,CAAC,UAAU;IAE3B,OAAO,CAAC,QAAQ,CAAC,MAAM;IAtBzB,+DAA+D;IAC/D,OAAO,CAAC,YAAY,CAAgD;IAEpE;mCAC+B;IAC/B,OAAO,CAAC,KAAK,CAA+B;IAE5C,wDAAwD;IACxD,OAAO,CAAC,aAAa,CAAgB;IAErC,OAAO,CAAC,YAAY,CAAkD;IAEtE,OAAO,CAAC,aAAa,CAA+C;IACpE,OAAO,CAAC,eAAe,CAA+C;IAEtE,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAS;gBAGd,MAAM,EAAE,QAAQ,EAChB,cAAc,EAAE,cAAc,EAC9B,UAAU,EAAE,UAAU,EACvC,OAAO,EAAE,MAAM,EACE,MAAM,EAAE,MAAM,EAC/B,GAAG,EAAE,iBAAiB;IAQxB,iFAAiF;IAC1E,iBAAiB,CACtB,GAAG,EAAE,YAAY,GAAG,WAAW,EAC/B,GAAG,EAAE,aAAa,GACjB,sBAAsB,GAAG,SAAS;IAC9B,iBAAiB,CACtB,GAAG,EAAE,YAAY,GAAG,WAAW,GAC9B,uBAAuB,CAAC,sBAAsB,CAAC;IAkBlD,OAAO,CAAC,sBAAsB;IA0B9B,OAAO,CAAC,kBAAkB;YAcZ,YAAY;IAQ1B,2BAA2B;IACpB,eAAe,CACpB,GAAG,EAAE,4BAA4B,GAAG,WAAW,GAC9C,UAAU,CAAC,uBAAuB,CAAC;IAC/B,eAAe,CACpB,GAAG,EAAE,4BAA4B,GAAG,WAAW,EAC/C,GAAG,EAAE,aAAa,GACjB,uBAAuB;IAqB1B,OAAO,CAAC,oBAAoB;IAkB5B,iCAAiC;IAC1B,YAAY,CAAC,MAAM,EAAE,eAAe,GAAG,MAAM;IAKpD,4CAA4C;IAC/B,UAAU,CAAC,MAAM,EAAE,eAAe,GAAG,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC;IAcxF;;;OAGG;IACI,oBAAoB,CACzB,GAAG,EAAE,YAAY,GAAG,WAAW,GAC9B,uBAAuB,CAAC,UAAU,CAAC;IAQtC;0DACsD;IAC/C,WAAW,CAChB,GAAG,EAAE,YAAY,GAAG,WAAW,EAC/B,KAAK,EAAE,MAAM,GACZ,UAAU,CAAC,MAAM,GAAG,SAAS,CAAC;IAC1B,WAAW,CAChB,GAAG,EAAE,YAAY,GAAG,WAAW,EAC/B,KAAK,EAAE,MAAM,EACb,GAAG,EAAE,aAAa,GACjB,UAAU,CAAC,MAAM,GAAG,SAAS,CAAC;IAmBjC,OAAO,CAAC,gBAAgB;IA0BxB;2DACuD;IAChD,cAAc,CACnB,GAAG,EAAE,YAAY,GAAG,WAAW,EAC/B,eAAe,EAAE,MAAM,GACtB,UAAU,CAAC,MAAM,GAAG,SAAS,CAAC;IAC1B,cAAc,CACnB,GAAG,EAAE,YAAY,GAAG,WAAW,EAC/B,eAAe,EAAE,MAAM,EACvB,GAAG,EAAE,aAAa,GACjB,MAAM,GAAG,SAAS;IAyBrB,OAAO,CAAC,mBAAmB;IA2B3B;uBACmB;IACZ,YAAY,CAAC,GAAG,EAAE,YAAY,GAAG,WAAW,GAAG,UAAU,CAAC,YAAY,CAAC;IACvE,YAAY,CAAC,GAAG,EAAE,YAAY,GAAG,WAAW,EAAE,GAAG,EAAE,aAAa,GAAG,YAAY;IAYtF,OAAO,CAAC,iBAAiB;IAKZ,SAAS,CACpB,MAAM,EAAE,cAAc,EACtB,SAAS,EAAE,MAAM,EACjB,WAAW,CAAC,EAAE,MAAM,EAAE,kCAAkC;IACxD,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,oBAAoB,CAAC;IAiBnB,QAAQ,CACnB,MAAM,EAAE,cAAc,EACtB,SAAS,EAAE,MAAM,EACjB,WAAW,CAAC,EAAE,MAAM,EACpB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,oBAAoB,CAAC;YAiBlB,WAAW;IA+BzB,OAAO,CAAC,UAAU;YAQJ,mBAAmB;IAKjC,2CAA2C;IACrC,UAAU;IAShB,OAAO,CAAC,WAAW;CAGpB"}

package/dist/drivers/download_blob_task.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"download_blob_task.d.ts","sourceRoot":"","sources":["../../src/drivers/download_blob_task.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,EAAE,YAAY,EAAE,MAAM,4BAA4B,CAAC;AAC1D,OAAO,KAAK,EAAE,eAAe,EAAE,sBAAsB,EAAE,MAAM,iCAAiC,CAAC;AAC/F,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,KAAK,EACV,YAAY,EACZ,QAAQ,EACT,MAAM,4BAA4B,CAAC;AACpC,OAAO,EACL,cAAc,EAIf,MAAM,4BAA4B,CAAC;AAKpC,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AAI1D,wBAAwB;AACxB,qBAAa,gBAAgB;IAUzB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,cAAc;IAC/B,QAAQ,CAAC,KAAK,EAAE,gBAAgB;IAChC,QAAQ,CAAC,IAAI,EAAE,MAAM;IACrB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAbzB,QAAQ,CAAC,OAAO,iBAAwB;IACxC,QAAQ,CAAC,MAAM,eAAsB;IACrC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAyB;IACnD,OAAO,CAAC,KAAK,CAAkB;IAC/B,OAAO,CAAC,IAAI,CAAS;IACrB,yDAAyD;IACzD,IAAI,SAAK;gBAGU,MAAM,EAAE,QAAQ,EAChB,cAAc,EAAE,cAAc,EACtC,KAAK,EAAE,gBAAgB,EACvB,IAAI,EAAE,MAAM,EACJ,MAAM,EAAE,eAAe;IAG1C,wDAAwD;IACjD,IAAI;;;;;;;IAUJ,MAAM,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM;IAK7B,QAAQ;YAiBP,gBAAgB;IAmBvB,KAAK,CAAC,MAAM,EAAE,MAAM;IAIpB,OAAO,IACV;QAAE,IAAI,EAAE,KAAK,CAAA;KAAE,GACf;QACA,IAAI,EAAE,IAAI,CAAC;QACX,MAAM,EAAE,YAAY,CAAC,sBAAsB,CAAC,CAAC;KAC9C;IAqBH,OAAO,CAAC,OAAO;IAKf,OAAO,CAAC,QAAQ;CAIjB;AAED,wBAAgB,mBAAmB,CAAC,CAAC,EAAE,GAAG,WAWzC"}