googlevideo 2.0.0 → 3.0.0

package/dist/protos/generated/video_streaming/video_playback_abr_request.js

@@ -1,7 +1,7 @@
 // Code generated by protoc-gen-ts_proto. DO NOT EDIT.
 // versions:
 //   protoc-gen-ts_proto  v2.2.0
-//   protoc               v5.28.0
+//   protoc               v3.21.12
 // source: video_streaming/video_playback_abr_request.proto
 /* eslint-disable */
 import { BinaryReader, BinaryWriter } from "@bufbuild/protobuf/wire";
@@ -15,6 +15,7 @@ function createBaseVideoPlaybackAbrRequest() {
         clientAbrState: undefined,
         selectedFormatIds: [],
         bufferedRanges: [],
+        playerTimeMs: 0,
         videoPlaybackUstreamerConfig: new Uint8Array(0),
         lo: undefined,
         selectedAudioFormatIds: [],
@@ -37,6 +38,9 @@ export const VideoPlaybackAbrRequest = {
         for (const v of message.bufferedRanges) {
             BufferedRange.encode(v, writer.uint32(26).fork()).join();
         }
+        if (message.playerTimeMs !== undefined && message.playerTimeMs !== 0) {
+            writer.uint32(32).int64(message.playerTimeMs);
+        }
         if (message.videoPlaybackUstreamerConfig !== undefined && message.videoPlaybackUstreamerConfig.length !== 0) {
             writer.uint32(42).bytes(message.videoPlaybackUstreamerConfig);
         }
@@ -91,6 +95,12 @@ export const VideoPlaybackAbrRequest = {
                     }
                     message.bufferedRanges.push(BufferedRange.decode(reader, reader.uint32()));
                     continue;
+                case 4:
+                    if (tag !== 32) {
+                        break;
+                    }
+                    message.playerTimeMs = longToNumber(reader.int64());
+                    continue;
                 case 5:
                     if (tag !== 42) {
                         break;
@@ -153,97 +163,6 @@ export const VideoPlaybackAbrRequest = {
         }
         return message;
     },
-    fromJSON(object) {
-        return {
-            clientAbrState: isSet(object.clientAbrState) ? ClientAbrState.fromJSON(object.clientAbrState) : undefined,
-            selectedFormatIds: globalThis.Array.isArray(object?.selectedFormatIds)
-                ? object.selectedFormatIds.map((e) => FormatId.fromJSON(e))
-                : [],
-            bufferedRanges: globalThis.Array.isArray(object?.bufferedRanges)
-                ? object.bufferedRanges.map((e) => BufferedRange.fromJSON(e))
-                : [],
-            videoPlaybackUstreamerConfig: isSet(object.videoPlaybackUstreamerConfig)
-                ? bytesFromBase64(object.videoPlaybackUstreamerConfig)
-                : new Uint8Array(0),
-            lo: isSet(object.lo) ? Lo.fromJSON(object.lo) : undefined,
-            selectedAudioFormatIds: globalThis.Array.isArray(object?.selectedAudioFormatIds)
-                ? object.selectedAudioFormatIds.map((e) => FormatId.fromJSON(e))
-                : [],
-            selectedVideoFormatIds: globalThis.Array.isArray(object?.selectedVideoFormatIds)
-                ? object.selectedVideoFormatIds.map((e) => FormatId.fromJSON(e))
-                : [],
-            streamerContext: isSet(object.streamerContext) ? StreamerContext.fromJSON(object.streamerContext) : undefined,
-            field21: isSet(object.field21) ? OQa.fromJSON(object.field21) : undefined,
-            field22: isSet(object.field22) ? globalThis.Number(object.field22) : 0,
-            field23: isSet(object.field23) ? globalThis.Number(object.field23) : 0,
-            field1000: globalThis.Array.isArray(object?.field1000) ? object.field1000.map((e) => Pqa.fromJSON(e)) : [],
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.clientAbrState !== undefined) {
-            obj.clientAbrState = ClientAbrState.toJSON(message.clientAbrState);
-        }
-        if (message.selectedFormatIds?.length) {
-            obj.selectedFormatIds = message.selectedFormatIds.map((e) => FormatId.toJSON(e));
-        }
-        if (message.bufferedRanges?.length) {
-            obj.bufferedRanges = message.bufferedRanges.map((e) => BufferedRange.toJSON(e));
-        }
-        if (message.videoPlaybackUstreamerConfig !== undefined && message.videoPlaybackUstreamerConfig.length !== 0) {
-            obj.videoPlaybackUstreamerConfig = base64FromBytes(message.videoPlaybackUstreamerConfig);
-        }
-        if (message.lo !== undefined) {
-            obj.lo = Lo.toJSON(message.lo);
-        }
-        if (message.selectedAudioFormatIds?.length) {
-            obj.selectedAudioFormatIds = message.selectedAudioFormatIds.map((e) => FormatId.toJSON(e));
-        }
-        if (message.selectedVideoFormatIds?.length) {
-            obj.selectedVideoFormatIds = message.selectedVideoFormatIds.map((e) => FormatId.toJSON(e));
-        }
-        if (message.streamerContext !== undefined) {
-            obj.streamerContext = StreamerContext.toJSON(message.streamerContext);
-        }
-        if (message.field21 !== undefined) {
-            obj.field21 = OQa.toJSON(message.field21);
-        }
-        if (message.field22 !== undefined && message.field22 !== 0) {
-            obj.field22 = Math.round(message.field22);
-        }
-        if (message.field23 !== undefined && message.field23 !== 0) {
-            obj.field23 = Math.round(message.field23);
-        }
-        if (message.field1000?.length) {
-            obj.field1000 = message.field1000.map((e) => Pqa.toJSON(e));
-        }
-        return obj;
-    },
-    create(base) {
-        return VideoPlaybackAbrRequest.fromPartial(base ?? {});
-    },
-    fromPartial(object) {
-        const message = createBaseVideoPlaybackAbrRequest();
-        message.clientAbrState = (object.clientAbrState !== undefined && object.clientAbrState !== null)
-            ? ClientAbrState.fromPartial(object.clientAbrState)
-            : undefined;
-        message.selectedFormatIds = object.selectedFormatIds?.map((e) => FormatId.fromPartial(e)) || [];
-        message.bufferedRanges = object.bufferedRanges?.map((e) => BufferedRange.fromPartial(e)) || [];
-        message.videoPlaybackUstreamerConfig = object.videoPlaybackUstreamerConfig ?? new Uint8Array(0);
-        message.lo = (object.lo !== undefined && object.lo !== null) ? Lo.fromPartial(object.lo) : undefined;
-        message.selectedAudioFormatIds = object.selectedAudioFormatIds?.map((e) => FormatId.fromPartial(e)) || [];
-        message.selectedVideoFormatIds = object.selectedVideoFormatIds?.map((e) => FormatId.fromPartial(e)) || [];
-        message.streamerContext = (object.streamerContext !== undefined && object.streamerContext !== null)
-            ? StreamerContext.fromPartial(object.streamerContext)
-            : undefined;
-        message.field21 = (object.field21 !== undefined && object.field21 !== null)
-            ? OQa.fromPartial(object.field21)
-            : undefined;
-        message.field22 = object.field22 ?? 0;
-        message.field23 = object.field23 ?? 0;
-        message.field1000 = object.field1000?.map((e) => Pqa.fromPartial(e)) || [];
-        return message;
-    },
 };
 function createBaseLo() {
     return { formatId: undefined, Lj: 0, sequenceNumber: 0, field4: undefined, MZ: 0 };
@@ -312,50 +231,6 @@ export const Lo = {
         }
         return message;
     },
-    fromJSON(object) {
-        return {
-            formatId: isSet(object.formatId) ? FormatId.fromJSON(object.formatId) : undefined,
-            Lj: isSet(object.Lj) ? globalThis.Number(object.Lj) : 0,
-            sequenceNumber: isSet(object.sequenceNumber) ? globalThis.Number(object.sequenceNumber) : 0,
-            field4: isSet(object.field4) ? Lo_Field4.fromJSON(object.field4) : undefined,
-            MZ: isSet(object.MZ) ? globalThis.Number(object.MZ) : 0,
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.formatId !== undefined) {
-            obj.formatId = FormatId.toJSON(message.formatId);
-        }
-        if (message.Lj !== undefined && message.Lj !== 0) {
-            obj.Lj = Math.round(message.Lj);
-        }
-        if (message.sequenceNumber !== undefined && message.sequenceNumber !== 0) {
-            obj.sequenceNumber = Math.round(message.sequenceNumber);
-        }
-        if (message.field4 !== undefined) {
-            obj.field4 = Lo_Field4.toJSON(message.field4);
-        }
-        if (message.MZ !== undefined && message.MZ !== 0) {
-            obj.MZ = Math.round(message.MZ);
-        }
-        return obj;
-    },
-    create(base) {
-        return Lo.fromPartial(base ?? {});
-    },
-    fromPartial(object) {
-        const message = createBaseLo();
-        message.formatId = (object.formatId !== undefined && object.formatId !== null)
-            ? FormatId.fromPartial(object.formatId)
-            : undefined;
-        message.Lj = object.Lj ?? 0;
-        message.sequenceNumber = object.sequenceNumber ?? 0;
-        message.field4 = (object.field4 !== undefined && object.field4 !== null)
-            ? Lo_Field4.fromPartial(object.field4)
-            : undefined;
-        message.MZ = object.MZ ?? 0;
-        return message;
-    },
 };
 function createBaseLo_Field4() {
     return { field1: 0, field2: 0, field3: 0 };
@@ -406,36 +281,6 @@ export const Lo_Field4 = {
         }
         return message;
     },
-    fromJSON(object) {
-        return {
-            field1: isSet(object.field1) ? globalThis.Number(object.field1) : 0,
-            field2: isSet(object.field2) ? globalThis.Number(object.field2) : 0,
-            field3: isSet(object.field3) ? globalThis.Number(object.field3) : 0,
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.field1 !== undefined && message.field1 !== 0) {
-            obj.field1 = Math.round(message.field1);
-        }
-        if (message.field2 !== undefined && message.field2 !== 0) {
-            obj.field2 = Math.round(message.field2);
-        }
-        if (message.field3 !== undefined && message.field3 !== 0) {
-            obj.field3 = Math.round(message.field3);
-        }
-        return obj;
-    },
-    create(base) {
-        return Lo_Field4.fromPartial(base ?? {});
-    },
-    fromPartial(object) {
-        const message = createBaseLo_Field4();
-        message.field1 = object.field1 ?? 0;
-        message.field2 = object.field2 ?? 0;
-        message.field3 = object.field3 ?? 0;
-        return message;
-    },
 };
 function createBaseOQa() {
     return { field1: [], field2: new Uint8Array(0), field3: "", field4: 0, field5: 0, field6: "" };
@@ -513,51 +358,6 @@ export const OQa = {
         }
         return message;
     },
-    fromJSON(object) {
-        return {
-            field1: globalThis.Array.isArray(object?.field1) ? object.field1.map((e) => globalThis.String(e)) : [],
-            field2: isSet(object.field2) ? bytesFromBase64(object.field2) : new Uint8Array(0),
-            field3: isSet(object.field3) ? globalThis.String(object.field3) : "",
-            field4: isSet(object.field4) ? globalThis.Number(object.field4) : 0,
-            field5: isSet(object.field5) ? globalThis.Number(object.field5) : 0,
-            field6: isSet(object.field6) ? globalThis.String(object.field6) : "",
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.field1?.length) {
-            obj.field1 = message.field1;
-        }
-        if (message.field2 !== undefined && message.field2.length !== 0) {
-            obj.field2 = base64FromBytes(message.field2);
-        }
-        if (message.field3 !== undefined && message.field3 !== "") {
-            obj.field3 = message.field3;
-        }
-        if (message.field4 !== undefined && message.field4 !== 0) {
-            obj.field4 = Math.round(message.field4);
-        }
-        if (message.field5 !== undefined && message.field5 !== 0) {
-            obj.field5 = Math.round(message.field5);
-        }
-        if (message.field6 !== undefined && message.field6 !== "") {
-            obj.field6 = message.field6;
-        }
-        return obj;
-    },
-    create(base) {
-        return OQa.fromPartial(base ?? {});
-    },
-    fromPartial(object) {
-        const message = createBaseOQa();
-        message.field1 = object.field1?.map((e) => e) || [];
-        message.field2 = object.field2 ?? new Uint8Array(0);
-        message.field3 = object.field3 ?? "";
-        message.field4 = object.field4 ?? 0;
-        message.field5 = object.field5 ?? 0;
-        message.field6 = object.field6 ?? "";
-        return message;
-    },
 };
 function createBasePqa() {
     return { formats: [], ud: [], clipId: "" };
@@ -608,52 +408,14 @@ export const Pqa = {
         }
         return message;
     },
-    fromJSON(object) {
-        return {
-            formats: globalThis.Array.isArray(object?.formats) ? object.formats.map((e) => FormatId.fromJSON(e)) : [],
-            ud: globalThis.Array.isArray(object?.ud) ? object.ud.map((e) => BufferedRange.fromJSON(e)) : [],
-            clipId: isSet(object.clipId) ? globalThis.String(object.clipId) : "",
-        };
-    },
-    toJSON(message) {
-        const obj = {};
-        if (message.formats?.length) {
-            obj.formats = message.formats.map((e) => FormatId.toJSON(e));
-        }
-        if (message.ud?.length) {
-            obj.ud = message.ud.map((e) => BufferedRange.toJSON(e));
-        }
-        if (message.clipId !== undefined && message.clipId !== "") {
-            obj.clipId = message.clipId;
-        }
-        return obj;
-    },
-    create(base) {
-        return Pqa.fromPartial(base ?? {});
-    },
-    fromPartial(object) {
-        const message = createBasePqa();
-        message.formats = object.formats?.map((e) => FormatId.fromPartial(e)) || [];
-        message.ud = object.ud?.map((e) => BufferedRange.fromPartial(e)) || [];
-        message.clipId = object.clipId ?? "";
-        return message;
-    },
 };
-function bytesFromBase64(b64) {
-    const bin = globalThis.atob(b64);
-    const arr = new Uint8Array(bin.length);
-    for (let i = 0; i < bin.length; ++i) {
-        arr[i] = bin.charCodeAt(i);
+function longToNumber(int64) {
+    const num = globalThis.Number(int64.toString());
+    if (num > globalThis.Number.MAX_SAFE_INTEGER) {
+        throw new globalThis.Error("Value is larger than Number.MAX_SAFE_INTEGER");
     }
-    return arr;
-}
-function base64FromBytes(arr) {
-    const bin = [];
-    arr.forEach((byte) => {
-        bin.push(globalThis.String.fromCharCode(byte));
-    });
-    return globalThis.btoa(bin.join(""));
-}
-function isSet(value) {
-    return value !== null && value !== undefined;
+    if (num < globalThis.Number.MIN_SAFE_INTEGER) {
+        throw new globalThis.Error("Value is smaller than Number.MIN_SAFE_INTEGER");
+    }
+    return num;
 }

package/dist/src/core/ServerAbrStream.d.ts

@@ -35,4 +35,5 @@ export declare class ServerAbrStream extends EventEmitterLike {
     private processNextRequestPolicy;
     private processFormatInitialization;
     private processSabrRedirect;
+    private registerFormat;
 }

package/dist/src/core/ServerAbrStream.js

@@ -1,7 +1,6 @@
 import { UMP } from './UMP.js';
 import { ChunkedDataBuffer } from './ChunkedDataBuffer.js';
 import { EventEmitterLike, PART, QUALITY, base64ToU8, getFormatKey } from '../utils/index.js';
-import { ClientAbrState_MediaType } from '../../protos/generated/video_streaming/client_abr_state.js';
 import { VideoPlaybackAbrRequest } from '../../protos/generated/video_streaming/video_playback_abr_request.js';
 import { MediaHeader } from '../../protos/generated/video_streaming/media_header.js';
 import { NextRequestPolicy } from '../../protos/generated/video_streaming/next_request_policy.js';
@@ -40,11 +39,11 @@ export class ServerAbrStream extends EventEmitterLike {
         const clientAbrState = {
             lastManualDirection: 0,
             timeSinceLastManualFormatSelectionMs: 0,
-            quality: videoFormats.length === 1 ? firstVideoFormat?.width : DEFAULT_QUALITY,
-            selectedQualityHeight: videoFormats.length === 1 ? firstVideoFormat?.width : DEFAULT_QUALITY,
-            startTimeMs: 0,
+            lastManualSelectedResolution: videoFormats.length === 1 ? firstVideoFormat?.height : DEFAULT_QUALITY,
+            stickyResolution: videoFormats.length === 1 ? firstVideoFormat?.height : DEFAULT_QUALITY,
+            playerTimeMs: 0,
             visibility: 0,
-            mediaType: ClientAbrState_MediaType.MEDIA_TYPE_DEFAULT,
+            enabledTrackTypesBitfield: 0,
             ...initialState
         };
         const audioFormatIds = audioFormats.map((fmt) => ({
@@ -57,15 +56,15 @@ export class ServerAbrStream extends EventEmitterLike {
             lastModified: parseInt(fmt.lastModified),
             xtags: fmt.xtags
         }));
-        if (typeof clientAbrState.startTimeMs !== 'number')
+        if (typeof clientAbrState.playerTimeMs !== 'number')
             throw new Error('Invalid media start time');
         try {
-            while (clientAbrState.startTimeMs < this.totalDurationMs) {
+            while (clientAbrState.playerTimeMs < this.totalDurationMs) {
                 const data = await this.fetchMedia({ clientAbrState, audioFormatIds, videoFormatIds });
                 this.emit('data', data);
                 if (data.sabrError)
                     break;
-                const mainFormat = clientAbrState.mediaType === ClientAbrState_MediaType.MEDIA_TYPE_DEFAULT
+                const mainFormat = clientAbrState.enabledTrackTypesBitfield === 0
                     ? data.initializedFormats.find((fmt) => fmt.mimeType?.includes('video'))
                     : data.initializedFormats[0];
                 for (const fmt of data.initializedFormats) {
@@ -77,11 +76,12 @@ export class ServerAbrStream extends EventEmitterLike {
                     this.emit('end', data);
                     break;
                 }
-                clientAbrState.startTimeMs += mainFormat.sequenceList.reduce((acc, seq) => acc + (seq.durationMs || 0), 0);
+                clientAbrState.playerTimeMs += mainFormat.sequenceList.reduce((acc, seq) => acc + (seq.durationMs || 0), 0);
             }
         }
         catch (error) {
             this.emit('error', error);
+            clientAbrState.playerTimeMs = Infinity;
         }
     }
     async fetchMedia(args) {
@@ -170,27 +170,9 @@ export class ServerAbrStream extends EventEmitterLike {
         if (!mediaHeader.formatId)
             return;
         const formatKey = getFormatKey(mediaHeader.formatId);
-        let currentFormat = this.formatsByKey.get(formatKey);
-        if (!currentFormat) {
-            this.initializedFormats.push({
-                formatId: mediaHeader.formatId,
-                formatKey,
-                durationMs: mediaHeader.durationMs,
-                mimeType: undefined,
-                sequenceCount: undefined,
-                sequenceList: [],
-                mediaChunks: [],
-                _state: {
-                    formatId: mediaHeader.formatId,
-                    startTimeMs: 0,
-                    durationMs: 0,
-                    startSegmentIndex: 1,
-                    endSegmentIndex: 0
-                }
-            });
-            this.formatsByKey.set(formatKey, this.initializedFormats[this.initializedFormats.length - 1]);
-            currentFormat = this.formatsByKey.get(formatKey);
-        }
+        const currentFormat = this.formatsByKey.get(formatKey) || this.registerFormat(mediaHeader);
+        if (!currentFormat)
+            return;
         // FIXME: This is a hacky workaround to prevent duplicate sequences from being added. This should be fixed in the future (preferably by figuring out how to make the server not send duplicates).
         if (mediaHeader.sequenceNumber !== undefined && this.previousSequences.get(formatKey)?.includes(mediaHeader.sequenceNumber))
             return;
@@ -239,34 +221,39 @@ export class ServerAbrStream extends EventEmitterLike {
     }
     processFormatInitialization(data) {
         const formatInitializationMetadata = FormatInitializationMetadata.decode(data);
-        if (!formatInitializationMetadata.formatId)
+        this.registerFormat(formatInitializationMetadata);
+    }
+    processSabrRedirect(data) {
+        const sabrRedirect = SabrRedirect.decode(data);
+        if (!sabrRedirect.url)
+            throw new Error('Invalid SABR redirect');
+        this.serverAbrStreamingUrl = sabrRedirect.url;
+        return sabrRedirect;
+    }
+    registerFormat(data) {
+        if (!data.formatId)
             return;
-        const formatKey = getFormatKey(formatInitializationMetadata.formatId);
+        const formatKey = getFormatKey(data.formatId);
         if (!this.formatsByKey.has(formatKey)) {
-            this.initializedFormats.push({
-                formatId: formatInitializationMetadata.formatId,
-                formatKey: getFormatKey(formatInitializationMetadata.formatId),
-                durationMs: formatInitializationMetadata.durationMs,
-                mimeType: formatInitializationMetadata.mimeType,
-                sequenceCount: formatInitializationMetadata.field4,
+            const format = {
+                formatId: data.formatId,
+                formatKey: formatKey,
+                durationMs: data.durationMs,
+                mimeType: 'mimeType' in data ? data.mimeType : undefined,
+                sequenceCount: 'field4' in data ? data.field4 : undefined,
                 sequenceList: [],
                 mediaChunks: [],
                 _state: {
-                    formatId: formatInitializationMetadata.formatId,
+                    formatId: data.formatId,
                     startTimeMs: 0,
                     durationMs: 0,
                     startSegmentIndex: 1,
                     endSegmentIndex: 0
                 }
-            });
+            };
+            this.initializedFormats.push(format);
             this.formatsByKey.set(formatKey, this.initializedFormats[this.initializedFormats.length - 1]);
+            return format;
         }
     }
-    processSabrRedirect(data) {
-        const sabrRedirect = SabrRedirect.decode(data);
-        if (!sabrRedirect.url)
-            throw new Error('Invalid SABR redirect');
-        this.serverAbrStreamingUrl = sabrRedirect.url;
-        return sabrRedirect;
-    }
 }

package/dist/src/core/UMP.d.ts

@@ -2,9 +2,33 @@ import type { Part } from '../index.js';
 import type { ChunkedDataBuffer } from './ChunkedDataBuffer.js';
 export declare class UMP {
     private chunkedDataBuffer;
+    /**
+     * Creates a new UMP parser.
+     * @param chunkedDataBuffer - Buffer containing UMP format data.
+     */
     constructor(chunkedDataBuffer: ChunkedDataBuffer);
-    parse(handlePart: (part: Part) => void): void;
+    /**
+     * Parses parts from the buffer and calls the handler for each complete part.
+     * @param handlePart - Function called with each complete part.
+     * @returns Partial part if parsing is incomplete, undefined otherwise.
+     */
+    parse(handlePart: (part: Part) => void): Part | undefined;
+    /**
+     * Reads a variable-length integer from the buffer.
+     * @param offset - Position to start reading from.
+     * @returns Tuple of [value, new offset] or [-1, offset] if incomplete.
+     */
     readVarInt(offset: number): [number, number];
+    /**
+     * Checks if the specified bytes can be read from the current chunk.
+     * @param offset - Position to start reading from.
+     * @param length - Number of bytes to read.
+     * @returns True if bytes can be read from current chunk, false otherwise.
+     */
     canReadFromCurrentChunk(offset: number, length: number): boolean;
+    /**
+     * Gets a DataView of the current chunk, creating it if necessary.
+     * @returns DataView for the current chunk.
+     */
     getCurrentDataView(): DataView;
 }

package/dist/src/core/UMP.js

@@ -1,7 +1,16 @@
 export class UMP {
+    /**
+     * Creates a new UMP parser.
+     * @param chunkedDataBuffer - Buffer containing UMP format data.
+     */
     constructor(chunkedDataBuffer) {
         this.chunkedDataBuffer = chunkedDataBuffer;
     }
+    /**
+     * Parses parts from the buffer and calls the handler for each complete part.
+     * @param handlePart - Function called with each complete part.
+     * @returns Partial part if parsing is incomplete, undefined otherwise.
+     */
     parse(handlePart) {
         while (true) {
             let offset = 0;
@@ -11,9 +20,15 @@ export class UMP {
             offset = finalOffset;
             if (partType < 0 || partSize < 0)
                 break;
-            // Note that we don't handle cases like this YET..
-            if (!this.chunkedDataBuffer.canReadBytes(offset, partSize))
-                break;
+            if (!this.chunkedDataBuffer.canReadBytes(offset, partSize)) {
+                if (!this.chunkedDataBuffer.canReadBytes(offset, 1))
+                    break;
+                return {
+                    type: partType,
+                    size: partSize,
+                    data: this.chunkedDataBuffer
+                };
+            }
             const splitResult = this.chunkedDataBuffer.split(offset).remainingBuffer.split(partSize);
             offset = 0;
             handlePart({
@@ -24,26 +39,17 @@ export class UMP {
             this.chunkedDataBuffer = splitResult.remainingBuffer;
         }
     }
+    /**
+     * Reads a variable-length integer from the buffer.
+     * @param offset - Position to start reading from.
+     * @returns Tuple of [value, new offset] or [-1, offset] if incomplete.
+     */
     readVarInt(offset) {
         let byteLength;
+        // Determine the length of the val
         if (this.chunkedDataBuffer.canReadBytes(offset, 1)) {
             const firstByte = this.chunkedDataBuffer.getUint8(offset);
-            // Determine the length of the val
-            if (firstByte < 128) {
-                byteLength = 1;
-            }
-            else if (firstByte < 192) {
-                byteLength = 2;
-            }
-            else if (firstByte < 224) {
-                byteLength = 3;
-            }
-            else if (firstByte < 240) {
-                byteLength = 4;
-            }
-            else {
-                byteLength = 5;
-            }
+            byteLength = firstByte < 128 ? 1 : firstByte < 192 ? 2 : firstByte < 224 ? 3 : firstByte < 240 ? 4 : 5;
         }
         else {
             byteLength = 0;
@@ -96,9 +102,19 @@ export class UMP {
         }
         return [value, offset];
     }
+    /**
+     * Checks if the specified bytes can be read from the current chunk.
+     * @param offset - Position to start reading from.
+     * @param length - Number of bytes to read.
+     * @returns True if bytes can be read from current chunk, false otherwise.
+     */
     canReadFromCurrentChunk(offset, length) {
         return offset - this.chunkedDataBuffer.currentChunkOffset + length <= this.chunkedDataBuffer.chunks[this.chunkedDataBuffer.currentChunkIndex].length;
     }
+    /**
+     * Gets a DataView of the current chunk, creating it if necessary.
+     * @returns DataView for the current chunk.
+     */
     getCurrentDataView() {
         if (!this.chunkedDataBuffer.currentDataView) {
             const currentChunk = this.chunkedDataBuffer.chunks[this.chunkedDataBuffer.currentChunkIndex];

package/dist/src/utils/Protos.d.ts

@@ -14,10 +14,11 @@ export { EncryptedPlayerRequest } from '../../protos/generated/video_streaming/e
 export { OnesieHeader } from '../../protos/generated/video_streaming/onesie_header.js';
 export { OnesieHeaderType } from '../../protos/generated/video_streaming/onesie_header_type.js';
 export { OnesiePlayerRequest } from '../../protos/generated/video_streaming/onesie_player_request.js';
-export { OnesiePlayerResponse } from '../../protos/generated/video_streaming/onesie_player_response.js';
+export { OnesiePlayerResponse, OnesieProxyStatus } from '../../protos/generated/video_streaming/onesie_player_response.js';
 export { ClientAbrState } from '../../protos/generated/video_streaming/client_abr_state.js';
 export { StreamerContext } from '../../protos/generated/video_streaming/streamer_context.js';
 export { ProxyStatus } from '../../protos/generated/video_streaming/proxy_status.js';
 export { MediaCapabilities } from '../../protos/generated/video_streaming/media_capabilities.js';
 export { CryptoParams } from '../../protos/generated/video_streaming/crypto_params.js';
-export { ClientAbrState_MediaType as MediaType } from '../../protos/generated/video_streaming/client_abr_state.js';
+export { LiveMetadata } from '../../protos/generated/video_streaming/live_metadata.js';
+export { FormatId, KeyValuePair, InitRange, IndexRange, HttpHeader } from '../../protos/generated/misc/common.js';

package/dist/src/utils/Protos.js

@@ -14,10 +14,11 @@ export { EncryptedPlayerRequest } from '../../protos/generated/video_streaming/e
 export { OnesieHeader } from '../../protos/generated/video_streaming/onesie_header.js';
 export { OnesieHeaderType } from '../../protos/generated/video_streaming/onesie_header_type.js';
 export { OnesiePlayerRequest } from '../../protos/generated/video_streaming/onesie_player_request.js';
-export { OnesiePlayerResponse } from '../../protos/generated/video_streaming/onesie_player_response.js';
+export { OnesiePlayerResponse, OnesieProxyStatus } from '../../protos/generated/video_streaming/onesie_player_response.js';
 export { ClientAbrState } from '../../protos/generated/video_streaming/client_abr_state.js';
 export { StreamerContext } from '../../protos/generated/video_streaming/streamer_context.js';
 export { ProxyStatus } from '../../protos/generated/video_streaming/proxy_status.js';
 export { MediaCapabilities } from '../../protos/generated/video_streaming/media_capabilities.js';
 export { CryptoParams } from '../../protos/generated/video_streaming/crypto_params.js';
-export { ClientAbrState_MediaType as MediaType } from '../../protos/generated/video_streaming/client_abr_state.js';
+export { LiveMetadata } from '../../protos/generated/video_streaming/live_metadata.js';
+export { FormatId, KeyValuePair, InitRange, IndexRange, HttpHeader } from '../../protos/generated/misc/common.js';