@protontech/drive-sdk 0.13.0 → 0.14.0

package/src/telemetry.test.ts

@@ -0,0 +1,40 @@
+import { reduceSizePrecision } from './telemetry';
+
+describe('reduceSizePrecision', () => {
+    it('returns 0 for size 0', () => {
+        expect(reduceSizePrecision(0)).toBe(0);
+    });
+
+    it('returns 4095 for very small files (size < 4096)', () => {
+        expect(reduceSizePrecision(1)).toBe(4095);
+        expect(reduceSizePrecision(100)).toBe(4095);
+        expect(reduceSizePrecision(4095)).toBe(4095);
+    });
+
+    it('returns precision (100_000) for sizes from 4096 to below precision', () => {
+        expect(reduceSizePrecision(4096)).toBe(100_000);
+        expect(reduceSizePrecision(50_000)).toBe(100_000);
+        expect(reduceSizePrecision(99_999)).toBe(100_000);
+    });
+
+    it('returns size unchanged when size equals precision', () => {
+        expect(reduceSizePrecision(100_000)).toBe(100_000);
+    });
+
+    it('rounds down to nearest 100_000 for sizes above precision', () => {
+        expect(reduceSizePrecision(100_001)).toBe(100_000);
+        expect(reduceSizePrecision(150_000)).toBe(100_000);
+        expect(reduceSizePrecision(199_999)).toBe(100_000);
+        expect(reduceSizePrecision(200_000)).toBe(200_000);
+        expect(reduceSizePrecision(250_000)).toBe(200_000);
+        expect(reduceSizePrecision(299_999)).toBe(200_000);
+        expect(reduceSizePrecision(300_000)).toBe(300_000);
+    });
+
+    it('handles large sizes', () => {
+        expect(reduceSizePrecision(1_000_000)).toBe(1_000_000);
+        expect(reduceSizePrecision(1_500_000)).toBe(1_500_000);
+        expect(reduceSizePrecision(1_999_999)).toBe(1_900_000);
+        expect(reduceSizePrecision(10_000_000)).toBe(10_000_000);
+    });
+});

package/src/telemetry.ts

@@ -352,3 +352,25 @@ class ConsoleMetricHandler<T extends MetricEvent> implements MetricHandler<T> {
         );
     }
 }
+
+export function reduceSizePrecision(size: number): number {
+    // The client shouldn't send the clear text size of the file.
+    // The intented upload size is needed only for early validation that
+    // the file can fit in the remaining quota to avoid data transfer when
+    // the upload would be rejected. The backend will still validate
+    // the quota during block upload and revision commit.
+    const precision = 100_000; // bytes
+
+    if (size === 0) {
+        return 0;
+    }
+    // We care about very small files in metrics, thus we handle explicitely
+    // the very small files so they appear correctly in metrics.
+    if (size < 4096) {
+        return 4095;
+    }
+    if (size < precision) {
+        return precision;
+    }
+    return Math.floor(size / precision) * precision;
+}