@milaboratories/pl-drivers 1.5.57 → 1.5.59

package/src/drivers/types.ts

@@ -1,6 +1,7 @@
 import { z } from 'zod';
 import type { InferSnapshot } from '@milaboratories/pl-tree';
 import { rsSchema } from '@milaboratories/pl-tree';
+import { RangeBytes } from '@milaboratories/pl-model-common';
 
 //
 // download
@@ -16,8 +17,18 @@ export const OnDemandBlobResourceSnapshot = rsSchema({
 
 export type OnDemandBlobResourceSnapshot = InferSnapshot<typeof OnDemandBlobResourceSnapshot>;
 
-export function getSize(bs: OnDemandBlobResourceSnapshot): number {
-  return bs.kv['ctl/file/blobInfo'].sizeBytes;
+export function getSize(bs: OnDemandBlobResourceSnapshot, range?: RangeBytes): number {
+  const size = bs.kv['ctl/file/blobInfo'].sizeBytes;
+  if (range) {
+    const newSize = range.to - range.from;
+    if (newSize > size) {
+      throw new Error(`getSize: range (${JSON.stringify(range)}, newSize: ${newSize}) is greater than size (${size})`);
+    }
+
+    return newSize;
+  }
+
+  return size;
 }
 
 //

package/src/drivers/upload.test.ts

@@ -11,6 +11,7 @@ import { expect, test } from 'vitest';
 import { Computable } from '@milaboratories/computable';
 import { SynchronizedTreeState } from '@milaboratories/pl-tree';
 import type { ImportResourceSnapshot } from './types';
+import * as env from '../test_env';
 
 test('upload a blob', async () => {
   await withTest(async ({ client, uploader, signer }: TestArg) => {
@@ -226,8 +227,8 @@ test('index a blob', async () => {
   await withTest(async ({ client, uploader }: TestArg) => {
     const uploadId = await createBlobIndex(
       client,
-      './another_answer_to_the_ultimate_question.txt',
-      'library',
+      'another_answer_to_the_ultimate_question.txt',
+      env.libraryStorage,
     );
     const handleRes = await getSnapshot(client, uploadId);
 

package/src/test_env.ts

@@ -0,0 +1,7 @@
+/** A storage where we keep all the assets.
+ * If you run tests locally, the storage should be named `library` and point
+ * to the `assets` directory in this repository,
+ * but in CI the storage is defined in env and it points to S3 bucket. */
+export const libraryStorage = process.env.PL_TEST_STORAGE_ID
+  ? process.env.PL_TEST_STORAGE_ID
+  : 'library';