googlevideo 3.0.0 → 4.0.1

package/dist/src/utils/shared.js

@@ -0,0 +1,120 @@
+export const MAX_INT32_VALUE = 2147483647;
+export const EnabledTrackTypes = {
+    VIDEO_AND_AUDIO: 0, 0: "VIDEO_AND_AUDIO",
+    AUDIO_ONLY: 1, 1: "AUDIO_ONLY",
+    VIDEO_ONLY: 2, 2: "VIDEO_ONLY"
+};
+/**
+ * Determines if a given URL is a Google video URL, specifically for YouTube or SABR-related content.
+ *
+ * @param url - The URL to check.
+ *
+ * The function checks the following conditions:
+ * 1. If the URL starts with the `sabr://` protocol, it is considered a Google video URL.
+ * 2. If the URL ends with `/videoplayback`, it parses the query parameters to check for specific keys
+ *    (`source=youtube`, `sabr`, `lsig`, or `expire`) that indicate a Google video URL.
+ * 3. If the URL contains `/videoplayback/` (e.g., for live or post-live content), it checks the path
+ *    segments for specific keywords (`videoplayback`, `sabr`, `lsig`, or `expire`).
+ */
+export function isGoogleVideoURL(url) {
+    if (url.startsWith('sabr://')) {
+        return true;
+    }
+    const urlParts = url.split('?');
+    const urlPart = urlParts[0];
+    const queryPart = urlParts[1] || '';
+    if (urlPart.endsWith('/videoplayback')) {
+        const params = new URLSearchParams(queryPart);
+        if (params.get('source') === 'youtube' || params.has('sabr') || params.has('lsig') || params.has('expire')) {
+            return true;
+        }
+    }
+    else if (urlPart.includes('/videoplayback/')) { // For live content, post-live, etc.
+        const pathParts = urlPart.split('/');
+        return ['videoplayback', 'sabr', 'lsig', 'expire'].some((part) => pathParts.includes(part));
+    }
+    return false;
+}
+/**
+ * Parses the Range header value to extract the start and end byte positions.
+ * @param rangeHeaderValue
+ */
+export function parseRangeHeader(rangeHeaderValue) {
+    if (!rangeHeaderValue)
+        return undefined;
+    const parts = rangeHeaderValue.split('=')[1]?.split('-');
+    if (parts?.length) {
+        const start = Number(parts[0]);
+        const end = Number(parts[1]);
+        return { start, end };
+    }
+    return undefined;
+}
+/**
+ * Converts a Uint8Array to a Base64 string.
+ * @param u8
+ */
+export function u8ToBase64(u8) {
+    return btoa(String.fromCharCode.apply(null, Array.from(u8)));
+}
+/**
+ * Converts a Base64 string to a Uint8Array.
+ * @param base64
+ */
+export function base64ToU8(base64) {
+    const standard_base64 = base64.replace(/-/g, '+').replace(/_/g, '/');
+    const padded_base64 = standard_base64.padEnd(standard_base64.length + (4 - standard_base64.length % 4) % 4, '=');
+    return new Uint8Array(atob(padded_base64).split('').map((char) => char.charCodeAt(0)));
+}
+/**
+ * Concatenates multiple Uint8Array chunks into a single Uint8Array.
+ * @param chunks
+ */
+export function concatenateChunks(chunks) {
+    const totalLength = chunks.reduce((sum, chunk) => sum + chunk.length, 0);
+    const result = new Uint8Array(totalLength);
+    let offset = 0;
+    for (const chunk of chunks) {
+        result.set(chunk, offset);
+        offset += chunk.length;
+    }
+    return result;
+}
+/**
+ * Converts a FormatStream object to a SabrFormat object.
+ * @param formatStream
+ */
+export function buildSabrFormat(formatStream) {
+    return {
+        itag: formatStream.itag,
+        lastModified: parseInt(formatStream.last_modified_ms || formatStream.lastModified || '0'),
+        xtags: formatStream.xtags,
+        width: formatStream.width,
+        height: formatStream.height,
+        mimeType: formatStream.mime_type || formatStream.mimeType,
+        audioQuality: formatStream.audio_quality || formatStream.audioQuality,
+        bitrate: formatStream.bitrate,
+        averageBitrate: formatStream.average_bitrate || formatStream.averageBitrate,
+        quality: formatStream.quality,
+        qualityLabel: formatStream.quality_label || formatStream.qualityLabel,
+        audioTrackId: formatStream.audio_track?.id || formatStream.audioTrackId,
+        approxDurationMs: formatStream.approx_duration_ms || parseInt(formatStream.approxDurationMs || '0'),
+        contentLength: parseInt(formatStream.contentLength || '0') || formatStream.content_length,
+        // YouTube.js-specific properties.
+        isDrc: formatStream.is_drc,
+        isAutoDubbed: formatStream.is_auto_dubbed,
+        isDescriptive: formatStream.is_descriptive,
+        isDubbed: formatStream.is_dubbed,
+        language: formatStream.language,
+        isOriginal: formatStream.is_original,
+        isSecondary: formatStream.is_secondary
+    };
+}
+/**
+ * Returns a promise that resolves after a specified number of milliseconds.
+ * @param ms - The number of milliseconds to wait.
+ */
+export function wait(ms) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}
+//# sourceMappingURL=shared.js.map

package/dist/src/utils/shared.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"shared.js","sourceRoot":"","sources":["../../../src/utils/shared.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,eAAe,GAAG,UAAU,CAAC;AAE1C,aAAY,iBAAiB;IAC3B,eAAe,EAAG,CAAC,EAAD,CAAC,EAAnB,iBAAmB;IACnB,UAAU,EAAG,CAAC,EAAD,CAAC,EAAd,YAAc;IACd,UAAU,EAAG,CAAC,EAAD,CAAC,EAAd,YAAc;EACf;AAED;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,gBAAgB,CAAC,GAAW;IAC1C,IAAI,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;QAC9B,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,QAAQ,GAAG,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IAChC,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;IAC5B,MAAM,SAAS,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAEpC,IAAI,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,SAAS,CAAC,CAAC;QAC9C,IAAI,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,KAAK,SAAS,IAAI,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3G,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;SAAM,IAAI,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,oCAAoC;QACpF,MAAM,SAAS,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QACrC,OAAO,CAAE,eAAe,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,CAAE,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC;IAChG,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAOD;;;GAGG;AACH,MAAM,UAAU,gBAAgB,CAAC,gBAAoC;IACnE,IAAI,CAAC,gBAAgB;QAAE,OAAO,SAAS,CAAC;IAExC,MAAM,KAAK,GAAG,gBAAgB,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IACzD,IAAI,KAAK,EAAE,MAAM,EAAE,CAAC;QAClB,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/B,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7B,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC;IACxB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,UAAU,CAAC,EAAc;IACvC,OAAO,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;AAC/D,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,UAAU,CAAC,MAAc;IACvC,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IACrE,MAAM,aAAa,GAAG,eAAe,CAAC,MAAM,CAAC,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,eAAe,CAAC,MAAM,GAAG,CAAC,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,CAAC;IACjH,OAAO,IAAI,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AACzF,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,MAAoB;IACpD,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACzE,MAAM,MAAM,GAAG,IAAI,UAAU,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,eAAe,CAAC,YAA0B;IACxD,OAAO;QACL,IAAI,EAAE,YAAY,CAAC,IAAI;QACvB,YAAY,EAAE,QAAQ,CAAC,YAAY,CAAC,gBAAgB,IAAI,YAAY,CAAC,YAAY,IAAI,GAAG,CAAC;QACzF,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,KAAK,EAAE,YAAY,CAAC,KAAK;QACzB,MAAM,EAAE,YAAY,CAAC,MAAM;QAC3B,QAAQ,EAAE,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,QAAQ;QACzD,YAAY,EAAE,YAAY,CAAC,aAAa,IAAI,YAAY,CAAC,YAAY;QACrE,OAAO,EAAE,YAAY,CAAC,OAAO;QAC7B,cAAc,EAAE,YAAY,CAAC,eAAe,IAAI,YAAY,CAAC,cAAc;QAC3E,OAAO,EAAE,YAAY,CAAC,OAAO;QAC7B,YAAY,EAAE,YAAY,CAAC,aAAa,IAAI,YAAY,CAAC,YAAY;QACrE,YAAY,EAAE,YAAY,CAAC,WAAW,EAAE,EAAE,IAAI,YAAY,CAAC,YAAY;QACvE,gBAAgB,EAAE,YAAY,CAAC,kBAAkB,IAAI,QAAQ,CAAC,YAAY,CAAC,gBAAgB,IAAI,GAAG,CAAC;QACnG,aAAa,EAAE,QAAQ,CAAC,YAAY,CAAC,aAAa,IAAI,GAAG,CAAC,IAAI,YAAY,CAAC,cAAc;QAEzF,kCAAkC;QAClC,KAAK,EAAE,YAAY,CAAC,MAAM;QAC1B,YAAY,EAAE,YAAY,CAAC,cAAc;QACzC,aAAa,EAAE,YAAY,CAAC,cAAc;QAC1C,QAAQ,EAAE,YAAY,CAAC,SAAS;QAChC,QAAQ,EAAE,YAAY,CAAC,QAAQ;QAC/B,UAAU,EAAE,YAAY,CAAC,WAAW;QACpC,WAAW,EAAE,YAAY,CAAC,YAAY;KACvC,CAAC;AACJ,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,IAAI,CAAC,EAAU;IAC7B,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3D,CAAC"}

package/package.json

@@ -1,19 +1,22 @@
 {
   "name": "googlevideo",
-  "version": "3.0.0",
-  "description": "A set of utilities for working with Google Video APIs.",
+  "version": "4.0.1",
+  "description": "A collection of modules for working with YouTube's proprietary video streaming protocols (UMP/SABR).",
   "main": "dist/index.js",
   "type": "module",
   "types": "./dist/src/index.d.ts",
   "module": "./dist/src/index.js",
   "scripts": {
-    "watch": "npx tsc --watch",
+    "watch": "npx tspc --watch",
     "lint": "npx eslint ./src/**/*.ts",
     "lint:fix": "npx eslint --fix ./src/**/*.ts",
     "clean": "npx rimraf ./dist ./protos/generated",
     "build": "npm run clean && npm run lint && npm run build:proto && npm run build:esm",
-    "build:esm": "npx tsc",
+    "build:esm": "npx tspc",
     "build:proto": "node ./dev-scripts/generate-proto.mjs",
+    "build:docs": "typedoc",
+    "test": "vitest run --reporter verbose",
+    "test:watch": "vitest",
     "prepare": "npm run build"
   },
   "author": "LuanRT <luan.lrt4@gmail.com> (https://github.com/LuanRT)",
@@ -28,12 +31,33 @@
   ],
   "license": "MIT",
   "exports": {
-    ".": {
-      "node": {
-        "import": "./dist/src/index.js"
-      },
-      "types": "./dist/src/index.d.ts",
-      "default": "./dist/src/index.js"
+    "./ump": {
+      "types": "./dist/src/exports/ump.d.ts",
+      "import": "./dist/src/exports/ump.js",
+      "default": "./dist/src/exports/ump.js"
+    },
+    "./sabr-streaming-adapter": {
+      "types": "./dist/src/exports/sabr-streaming-adapter.d.ts",
+      "import": "./dist/src/exports/sabr-streaming-adapter.js",
+      "default": "./dist/src/exports/sabr-streaming-adapter.js"
+    },
+    "./sabr-stream": {
+      "types": "./dist/src/exports/sabr-stream.d.ts",
+      "import": "./dist/src/exports/sabr-stream.js",
+      "default": "./dist/src/exports/sabr-stream.js"
+    },
+    "./protos": {
+      "types": "./dist/src/exports/protos.d.ts",
+      "import": "./dist/src/exports/protos.js",
+      "default": "./dist/src/exports/protos.js"
+    },
+    "./utils": {
+      "types": "./dist/src/exports/utils.d.ts",
+      "import": "./dist/src/exports/utils.js",
+      "default": "./dist/src/exports/utils.js"
+    },
+    "./shared-types": {
+      "types": "./dist/src/types/shared.d.ts"
     }
   },
   "devDependencies": {
@@ -42,9 +66,13 @@
     "@types/eslint__js": "^8.42.3",
     "eslint": "^9.9.0",
     "globals": "^15.9.0",
+    "ts-patch": "^3.3.0",
     "ts-proto": "^2.2.0",
+    "typedoc": "^0.28.7",
+    "typedoc-plugin-markdown": "^4.7.1",
     "typescript": "^5.5.4",
-    "typescript-eslint": "^8.2.0"
+    "typescript-eslint": "^8.2.0",
+    "vitest": "^3.2.4"
   },
   "dependencies": {
     "@bufbuild/protobuf": "^2.0.0"
@@ -57,4 +85,4 @@
     "type": "git",
     "url": "git+https://github.com/LuanRT/GoogleVideo.git"
   }
-}
+}

package/dist/protos/generated/video_streaming/onesie_player_response.d.ts

@@ -1,31 +0,0 @@
-import { BinaryReader, BinaryWriter } from "@bufbuild/protobuf/wire";
-import { HttpHeader } from "../misc/common.js";
-export declare const protobufPackage = "video_streaming";
-export declare enum OnesieProxyStatus {
-    ONESIE_PROXY_STATUS_UNKNOWN = 0,
-    ONESIE_PROXY_STATUS_OK = 1,
-    ONESIE_PROXY_STATUS_DECRYPTION_FAILED = 2,
-    ONESIE_PROXY_STATUS_PARSING_FAILED = 3,
-    ONESIE_PROXY_STATUS_MISSING_X_FORWARDED_FOR = 4,
-    ONESIE_PROXY_STATUS_INVALID_X_FORWARDED_FOR = 5,
-    ONESIE_PROXY_STATUS_INVALID_CONTENT_TYPE = 6,
-    ONESIE_PROXY_STATUS_BACKEND_ERROR = 7,
-    ONESIE_PROXY_STATUS_CLIENT_ERROR = 8,
-    ONESIE_PROXY_STATUS_MISSING_CRYPTER = 9,
-    ONESIE_PROXY_STATUS_RESPONSE_JSON_SERIALIZATION_FAILED = 10,
-    ONESIE_PROXY_STATUS_DECOMPRESSION_FAILED = 11,
-    ONESIE_PROXY_STATUS_JSON_PARSING_FAILED = 12,
-    ONESIE_PROXY_STATUS_UNKNOWN_COMPRESSION_TYPE = 13,
-    UNRECOGNIZED = -1
-}
-export interface OnesiePlayerResponse {
-    onesieProxyStatus?: OnesieProxyStatus | undefined;
-    httpStatus?: number | undefined;
-    headers: HttpHeader[];
-    body?: Uint8Array | undefined;
-}
-export declare const OnesiePlayerResponse: MessageFns<OnesiePlayerResponse>;
-export interface MessageFns<T> {
-    encode(message: T, writer?: BinaryWriter): BinaryWriter;
-    decode(input: BinaryReader | Uint8Array, length?: number): T;
-}

package/dist/protos/generated/video_streaming/proxy_status.d.ts

@@ -1,14 +0,0 @@
-export declare const protobufPackage = "video_streaming";
-export declare enum ProxyStatus {
-    VAL_0 = 0,
-    OK = 1,
-    VAL_2 = 2,
-    VAL_3 = 3,
-    VAL_4 = 4,
-    VAL_5 = 5,
-    VAL_6 = 6,
-    VAL_7 = 7,
-    VAL_8 = 8,
-    VAL_9 = 9,
-    UNRECOGNIZED = -1
-}

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

@@ -1,21 +0,0 @@
-// Code generated by protoc-gen-ts_proto. DO NOT EDIT.
-// versions:
-//   protoc-gen-ts_proto  v2.2.0
-//   protoc               v3.21.12
-// source: video_streaming/proxy_status.proto
-/* eslint-disable */
-export const protobufPackage = "video_streaming";
-export var ProxyStatus;
-(function (ProxyStatus) {
-    ProxyStatus[ProxyStatus["VAL_0"] = 0] = "VAL_0";
-    ProxyStatus[ProxyStatus["OK"] = 1] = "OK";
-    ProxyStatus[ProxyStatus["VAL_2"] = 2] = "VAL_2";
-    ProxyStatus[ProxyStatus["VAL_3"] = 3] = "VAL_3";
-    ProxyStatus[ProxyStatus["VAL_4"] = 4] = "VAL_4";
-    ProxyStatus[ProxyStatus["VAL_5"] = 5] = "VAL_5";
-    ProxyStatus[ProxyStatus["VAL_6"] = 6] = "VAL_6";
-    ProxyStatus[ProxyStatus["VAL_7"] = 7] = "VAL_7";
-    ProxyStatus[ProxyStatus["VAL_8"] = 8] = "VAL_8";
-    ProxyStatus[ProxyStatus["VAL_9"] = 9] = "VAL_9";
-    ProxyStatus[ProxyStatus["UNRECOGNIZED"] = -1] = "UNRECOGNIZED";
-})(ProxyStatus || (ProxyStatus = {}));

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

@@ -1,39 +0,0 @@
-import { EventEmitterLike } from '../utils/index.js';
-import type { InitOptions, ServerAbrResponse, ServerAbrStreamOptions } from '../utils/types.js';
-export declare class ServerAbrStream extends EventEmitterLike {
-    private fetchFunction;
-    private serverAbrStreamingUrl;
-    private videoPlaybackUstreamerConfig;
-    private poToken?;
-    private playbackCookie?;
-    private totalDurationMs;
-    private initializedFormats;
-    private formatsByKey;
-    private headerIdToFormatKeyMap;
-    private previousSequences;
-    constructor(args: ServerAbrStreamOptions);
-    on(event: 'end', listener: (streamData: ServerAbrResponse) => void): void;
-    on(event: 'data', listener: (streamData: ServerAbrResponse) => void): void;
-    on(event: 'error', listener: (error: Error) => void): void;
-    once(event: 'end', listener: (streamData: ServerAbrResponse) => void): void;
-    once(event: 'data', listener: (streamData: ServerAbrResponse) => void): void;
-    once(event: 'error', listener: (error: Error) => void): void;
-    /**
-     * Initializes the server ABR stream with the provided options.
-     * @param args - The initialization options.
-     */
-    init(args: InitOptions): Promise<void>;
-    private fetchMedia;
-    /**
-     * Parses the UMP response data and updates the initialized formats.
-     * @param response - The UMP response data as a byte array.
-     */
-    parseUMPResponse(response: Uint8Array): Promise<ServerAbrResponse>;
-    private processMediaHeader;
-    private processMediaData;
-    private processEndOfMedia;
-    private processNextRequestPolicy;
-    private processFormatInitialization;
-    private processSabrRedirect;
-    private registerFormat;
-}

package/dist/src/core/ServerAbrStream.js

@@ -1,259 +0,0 @@
-import { UMP } from './UMP.js';
-import { ChunkedDataBuffer } from './ChunkedDataBuffer.js';
-import { EventEmitterLike, PART, QUALITY, base64ToU8, getFormatKey } from '../utils/index.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';
-import { FormatInitializationMetadata } from '../../protos/generated/video_streaming/format_initialization_metadata.js';
-import { SabrRedirect } from '../../protos/generated/video_streaming/sabr_redirect.js';
-import { SabrError } from '../../protos/generated/video_streaming/sabr_error.js';
-import { StreamProtectionStatus } from '../../protos/generated/video_streaming/stream_protection_status.js';
-import { PlaybackCookie } from '../../protos/generated/video_streaming/playback_cookie.js';
-const DEFAULT_QUALITY = QUALITY.HD720;
-export class ServerAbrStream extends EventEmitterLike {
-    constructor(args) {
-        super();
-        this.initializedFormats = [];
-        this.formatsByKey = new Map();
-        this.headerIdToFormatKeyMap = new Map();
-        this.previousSequences = new Map();
-        this.fetchFunction = args.fetch || fetch;
-        this.serverAbrStreamingUrl = args.serverAbrStreamingUrl;
-        this.videoPlaybackUstreamerConfig = args.videoPlaybackUstreamerConfig;
-        this.poToken = args.poToken;
-        this.totalDurationMs = args.durationMs;
-    }
-    on(event, listener) {
-        super.on(event, listener);
-    }
-    once(event, listener) {
-        super.once(event, listener);
-    }
-    /**
-     * Initializes the server ABR stream with the provided options.
-     * @param args - The initialization options.
-     */
-    async init(args) {
-        const { audioFormats, videoFormats, clientAbrState: initialState } = args;
-        const firstVideoFormat = videoFormats ? videoFormats[0] : undefined;
-        const clientAbrState = {
-            lastManualDirection: 0,
-            timeSinceLastManualFormatSelectionMs: 0,
-            lastManualSelectedResolution: videoFormats.length === 1 ? firstVideoFormat?.height : DEFAULT_QUALITY,
-            stickyResolution: videoFormats.length === 1 ? firstVideoFormat?.height : DEFAULT_QUALITY,
-            playerTimeMs: 0,
-            visibility: 0,
-            enabledTrackTypesBitfield: 0,
-            ...initialState
-        };
-        const audioFormatIds = audioFormats.map((fmt) => ({
-            itag: fmt.itag,
-            lastModified: parseInt(fmt.lastModified),
-            xtags: fmt.xtags
-        }));
-        const videoFormatIds = videoFormats.map((fmt) => ({
-            itag: fmt.itag,
-            lastModified: parseInt(fmt.lastModified),
-            xtags: fmt.xtags
-        }));
-        if (typeof clientAbrState.playerTimeMs !== 'number')
-            throw new Error('Invalid media start time');
-        try {
-            while (clientAbrState.playerTimeMs < this.totalDurationMs) {
-                const data = await this.fetchMedia({ clientAbrState, audioFormatIds, videoFormatIds });
-                this.emit('data', data);
-                if (data.sabrError)
-                    break;
-                const mainFormat = clientAbrState.enabledTrackTypesBitfield === 0
-                    ? data.initializedFormats.find((fmt) => fmt.mimeType?.includes('video'))
-                    : data.initializedFormats[0];
-                for (const fmt of data.initializedFormats) {
-                    this.previousSequences.set(fmt.formatKey, fmt.sequenceList.map((seq) => seq.sequenceNumber || 0));
-                }
-                if (!mainFormat ||
-                    mainFormat.sequenceCount ===
-                        mainFormat.sequenceList[mainFormat.sequenceList.length - 1]?.sequenceNumber) {
-                    this.emit('end', data);
-                    break;
-                }
-                clientAbrState.playerTimeMs += mainFormat.sequenceList.reduce((acc, seq) => acc + (seq.durationMs || 0), 0);
-            }
-        }
-        catch (error) {
-            this.emit('error', error);
-            clientAbrState.playerTimeMs = Infinity;
-        }
-    }
-    async fetchMedia(args) {
-        const { clientAbrState, audioFormatIds, videoFormatIds } = args;
-        const body = VideoPlaybackAbrRequest.encode({
-            clientAbrState: clientAbrState,
-            selectedAudioFormatIds: audioFormatIds,
-            selectedVideoFormatIds: videoFormatIds,
-            selectedFormatIds: this.initializedFormats.map((fmt) => fmt.formatId),
-            videoPlaybackUstreamerConfig: base64ToU8(this.videoPlaybackUstreamerConfig),
-            streamerContext: {
-                field5: [],
-                field6: [],
-                poToken: this.poToken ? base64ToU8(this.poToken) : undefined,
-                playbackCookie: this.playbackCookie ? PlaybackCookie.encode(this.playbackCookie).finish() : undefined,
-                clientInfo: {
-                    clientName: 1,
-                    clientVersion: '2.2040620.05.00',
-                    osName: 'Windows',
-                    osVersion: '10.0'
-                }
-            },
-            bufferedRanges: this.initializedFormats.map((fmt) => fmt._state),
-            field1000: []
-        }).finish();
-        const response = await this.fetchFunction(this.serverAbrStreamingUrl, { method: 'POST', body });
-        const data = await response.arrayBuffer();
-        if (response.status !== 200 || !data.byteLength)
-            throw new Error(`Received an invalid response from the server: ${response.status}`);
-        return this.parseUMPResponse(new Uint8Array(data));
-    }
-    /**
-     * Parses the UMP response data and updates the initialized formats.
-     * @param response - The UMP response data as a byte array.
-     */
-    async parseUMPResponse(response) {
-        this.headerIdToFormatKeyMap.clear();
-        this.initializedFormats.forEach((format) => {
-            format.sequenceList = [];
-            format.mediaChunks = [];
-        });
-        let sabrError;
-        let sabrRedirect;
-        let streamProtectionStatus;
-        const ump = new UMP(new ChunkedDataBuffer([response]));
-        ump.parse((part) => {
-            const data = part.data.chunks[0];
-            switch (part.type) {
-                case PART.MEDIA_HEADER:
-                    this.processMediaHeader(data);
-                    break;
-                case PART.MEDIA:
-                    this.processMediaData(part.data);
-                    break;
-                case PART.MEDIA_END:
-                    this.processEndOfMedia(part.data);
-                    break;
-                case PART.NEXT_REQUEST_POLICY:
-                    this.processNextRequestPolicy(data);
-                    break;
-                case PART.FORMAT_INITIALIZATION_METADATA:
-                    this.processFormatInitialization(data);
-                    break;
-                case PART.SABR_ERROR:
-                    sabrError = SabrError.decode(data);
-                    break;
-                case PART.SABR_REDIRECT:
-                    sabrRedirect = this.processSabrRedirect(data);
-                    break;
-                case PART.STREAM_PROTECTION_STATUS:
-                    streamProtectionStatus = StreamProtectionStatus.decode(data);
-                    break;
-                default:
-                    break;
-            }
-        });
-        return {
-            initializedFormats: this.initializedFormats,
-            streamProtectionStatus,
-            sabrRedirect,
-            sabrError
-        };
-    }
-    processMediaHeader(data) {
-        const mediaHeader = MediaHeader.decode(data);
-        if (!mediaHeader.formatId)
-            return;
-        const formatKey = getFormatKey(mediaHeader.formatId);
-        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;
-        // Save the header's ID so we can identify its stream data later.
-        if (mediaHeader.headerId !== undefined) {
-            if (!this.headerIdToFormatKeyMap.has(mediaHeader.headerId)) {
-                this.headerIdToFormatKeyMap.set(mediaHeader.headerId, formatKey);
-            }
-        }
-        if (!currentFormat.sequenceList.some((seq) => seq.sequenceNumber === (mediaHeader.sequenceNumber || 0))) {
-            currentFormat.sequenceList.push({
-                itag: mediaHeader.itag,
-                formatId: mediaHeader.formatId,
-                isInitSegment: mediaHeader.isInitSeg,
-                durationMs: mediaHeader.durationMs,
-                startMs: mediaHeader.startMs,
-                startDataRange: mediaHeader.startDataRange,
-                sequenceNumber: mediaHeader.sequenceNumber,
-                contentLength: mediaHeader.contentLength,
-                timeRange: mediaHeader.timeRange
-            });
-            if (typeof mediaHeader.sequenceNumber === 'number') {
-                currentFormat._state.durationMs += mediaHeader.durationMs || 0;
-                currentFormat._state.endSegmentIndex += 1;
-            }
-        }
-    }
-    processMediaData(data) {
-        const headerId = data.getUint8(0);
-        const streamData = data.split(1).remainingBuffer;
-        const formatKey = this.headerIdToFormatKeyMap.get(headerId);
-        if (!formatKey)
-            return;
-        const currentFormat = this.formatsByKey.get(formatKey);
-        if (!currentFormat)
-            return;
-        currentFormat.mediaChunks.push(streamData.chunks[0]);
-    }
-    processEndOfMedia(data) {
-        const headerId = data.getUint8(0);
-        this.headerIdToFormatKeyMap.delete(headerId);
-    }
-    processNextRequestPolicy(data) {
-        const nextRequestPolicy = NextRequestPolicy.decode(data);
-        this.playbackCookie = nextRequestPolicy.playbackCookie;
-    }
-    processFormatInitialization(data) {
-        const formatInitializationMetadata = FormatInitializationMetadata.decode(data);
-        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(data.formatId);
-        if (!this.formatsByKey.has(formatKey)) {
-            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: 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;
-        }
-    }
-}

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

@@ -1,3 +0,0 @@
-export * from './ChunkedDataBuffer.js';
-export * from './UMP.js';
-export * from './ServerAbrStream.js';

package/dist/src/core/index.js

@@ -1,3 +0,0 @@
-export * from './ChunkedDataBuffer.js';
-export * from './UMP.js';
-export * from './ServerAbrStream.js';

package/dist/src/index.d.ts

@@ -1,4 +0,0 @@
-import * as GoogleVideo from './core/index.js';
-export { GoogleVideo };
-export default GoogleVideo;
-export * from './utils/index.js';

package/dist/src/index.js

@@ -1,4 +0,0 @@
-import * as GoogleVideo from './core/index.js';
-export { GoogleVideo };
-export default GoogleVideo;
-export * from './utils/index.js';

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

@@ -1,63 +0,0 @@
-import type { FormatId } from '../../protos/generated/misc/common.js';
-export declare enum QUALITY {
-    AUTO = 0,
-    TINY = 144,
-    SMALL = 240,
-    MEDIUM = 360,
-    LIGHT = 144,
-    LARGE = 480,
-    HD720 = 720,
-    HD1080 = 1080,
-    HD1440 = 1440,
-    HD2160 = 2160,
-    HD2880 = 2880,
-    HIGHRES = 4320
-}
-export declare enum PART {
-    ONESIE_HEADER = 10,
-    ONESIE_DATA = 11,
-    MEDIA_HEADER = 20,
-    MEDIA = 21,
-    MEDIA_END = 22,
-    LIVE_METADATA = 31,
-    HOSTNAME_CHANGE_HINT = 32,
-    LIVE_METADATA_PROMISE = 33,
-    LIVE_METADATA_PROMISE_CANCELLATION = 34,
-    NEXT_REQUEST_POLICY = 35,
-    USTREAMER_VIDEO_AND_FORMAT_DATA = 36,
-    FORMAT_SELECTION_CONFIG = 37,
-    USTREAMER_SELECTED_MEDIA_STREAM = 38,
-    FORMAT_INITIALIZATION_METADATA = 42,
-    SABR_REDIRECT = 43,
-    SABR_ERROR = 44,
-    SABR_SEEK = 45,
-    RELOAD_PLAYER_RESPONSE = 46,
-    PLAYBACK_START_POLICY = 47,
-    ALLOWED_CACHED_FORMATS = 48,
-    START_BW_SAMPLING_HINT = 49,
-    PAUSE_BW_SAMPLING_HINT = 50,
-    SELECTABLE_FORMATS = 51,
-    REQUEST_IDENTIFIER = 52,
-    REQUEST_CANCELLATION_POLICY = 53,
-    ONESIE_PREFETCH_REJECTION = 54,
-    TIMELINE_CONTEXT = 55,
-    REQUEST_PIPELINING = 56,
-    SABR_CONTEXT_UPDATE = 57,
-    STREAM_PROTECTION_STATUS = 58,
-    SABR_CONTEXT_SENDING_POLICY = 59,
-    LAWNMOWER_POLICY = 60,
-    SABR_ACK = 61,
-    END_OF_TRACK = 62,
-    CACHE_LOAD_POLICY = 63,
-    LAWNMOWER_MESSAGING_POLICY = 64,
-    PREWARM_CONNECTION = 65
-}
-export declare function u8ToBase64(u8: Uint8Array): string;
-export declare function base64ToU8(base64: string): Uint8Array;
-export declare function getFormatKey(formatId: FormatId): string;
-export declare function concatenateChunks(chunks: Uint8Array[]): Uint8Array;
-export declare class CustomEvent extends Event {
-    #private;
-    constructor(type: string, options?: CustomEventInit<any[]>);
-    get detail(): any[] | null;
-}