UNPKG

rx-player

Version:
239 lines (238 loc) 10.3 kB
/** * Copyright 2015 CANAL+ Group * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ import { CustomLoaderError } from "../../errors"; import assert from "../../utils/assert"; import request from "../../utils/request"; import addQueryString from "../utils/add_query_string"; import byteRange from "../utils/byte_range"; import checkISOBMFFIntegrity from "../utils/check_isobmff_integrity"; import isMP4EmbeddedTrack from "./is_mp4_embedded_track"; import { createAudioInitSegment, createVideoInitSegment } from "./isobmff"; /** * Segment loader triggered if there was no custom-defined one in the API. * @param {string} initialUrl * @param {Object} context * @param {Object} loaderOptions * @param {Object} callbacks * @param {Object} cancelSignal * @param {boolean} checkMediaSegmentIntegrity * @returns {Promise} */ async function regularSegmentLoader(initialUrl, context, callbacks, loaderOptions, cancelSignal, checkMediaSegmentIntegrity) { var _a, _b; const cmcdHeaders = ((_a = loaderOptions.cmcdPayload) === null || _a === void 0 ? void 0 : _a.type) === "headers" ? loaderOptions.cmcdPayload.value : undefined; const range = context.segment.range; let headers; if (Array.isArray(range)) { headers = Object.assign(Object.assign({}, cmcdHeaders), { Range: byteRange(range) }); } else if (cmcdHeaders !== undefined) { headers = cmcdHeaders; } const url = ((_b = loaderOptions.cmcdPayload) === null || _b === void 0 ? void 0 : _b.type) === "query" ? addQueryString(initialUrl, loaderOptions.cmcdPayload.value) : initialUrl; const data = await request({ url, responseType: "arraybuffer", headers, timeout: loaderOptions.timeout, connectionTimeout: loaderOptions.connectionTimeout, cancelSignal, onProgress: callbacks.onProgress, }); const isMP4 = isMP4EmbeddedTrack(context.mimeType); if (!isMP4 || checkMediaSegmentIntegrity !== true) { return { resultType: "segment-loaded", resultData: data }; } const dataU8 = new Uint8Array(data.responseData); checkISOBMFFIntegrity(dataU8, context.segment.isInit); return { resultType: "segment-loaded", resultData: Object.assign(Object.assign({}, data), { responseData: dataU8 }), }; } /** * Defines the url for the request, load the right loader (custom/default * one). */ const generateSegmentLoader = ({ checkMediaSegmentIntegrity, segmentLoader, }) => (url, context, loaderOptions, cancelSignal, callbacks) => { const { segment } = context; if (segment.isInit) { if (segment.privateInfos === undefined || segment.privateInfos.smoothInitSegment === undefined) { throw new Error("Smooth: Invalid segment format"); } const smoothInitPrivateInfos = segment.privateInfos.smoothInitSegment; let responseData; const { codecPrivateData, timescale, height, width, protection = { keyId: undefined, keySystems: undefined }, } = smoothInitPrivateInfos; if (codecPrivateData === undefined) { throw new Error("Smooth: no codec private data."); } switch (context.type) { case "video": { responseData = createVideoInitSegment(timescale, width !== null && width !== void 0 ? width : 0, height !== null && height !== void 0 ? height : 0, 72, 72, 4, // vRes, hRes, nal codecPrivateData, protection.keyId); break; } case "audio": { const { channels = 0, bitsPerSample = 0, packetSize = 0, samplingRate = 0, } = smoothInitPrivateInfos; responseData = createAudioInitSegment(timescale, channels, bitsPerSample, packetSize, samplingRate, codecPrivateData, protection.keyId); break; } default: if (0 /* __ENVIRONMENT__.CURRENT_ENV */ === 1 /* __ENVIRONMENT__.DEV */) { assert(false, "responseData should have been set"); } responseData = new Uint8Array(0); } return Promise.resolve({ resultType: "segment-created", resultData: responseData, }); } else if (url === null) { return Promise.resolve({ resultType: "segment-created", resultData: null, }); } else { if (typeof segmentLoader !== "function") { return regularSegmentLoader(url, context, callbacks, loaderOptions, cancelSignal, checkMediaSegmentIntegrity); } return new Promise((res, rej) => { /** `true` when the custom segmentLoader should not be active anymore. */ let hasFinished = false; /** * Callback triggered when the custom segment loader has a response. * @param {Object} _args */ const resolve = (_args) => { if (hasFinished || cancelSignal.isCancelled()) { return; } hasFinished = true; cancelSignal.deregister(abortCustomLoader); let data; if (_args.data instanceof Uint8Array) { if (_args.data.buffer instanceof ArrayBuffer) { // Typescript is not so smart here for now data = _args.data; } else { data = _args.data.slice(); } } else { data = _args.data; } const isMP4 = isMP4EmbeddedTrack(context.mimeType); if (!isMP4 || checkMediaSegmentIntegrity !== true) { res({ resultType: "segment-loaded", resultData: { responseData: data, size: _args.size, requestDuration: _args.duration, }, }); } const dataU8 = data instanceof Uint8Array ? data : new Uint8Array(data); checkISOBMFFIntegrity(dataU8, context.segment.isInit); res({ resultType: "segment-loaded", resultData: { responseData: dataU8, size: _args.size, requestDuration: _args.duration, }, }); }; /** * Callback triggered when the custom segment loader fails * @param {*} err - The corresponding error encountered */ const reject = (err) => { var _a, _b; if (hasFinished || cancelSignal.isCancelled()) { return; } hasFinished = true; cancelSignal.deregister(abortCustomLoader); // Format error and send it const castedErr = err; const message = (_a = castedErr === null || castedErr === void 0 ? void 0 : castedErr.message) !== null && _a !== void 0 ? _a : "Unknown error when fetching a Smooth segment through a " + "custom segmentLoader."; const emittedErr = new CustomLoaderError(message, (_b = castedErr === null || castedErr === void 0 ? void 0 : castedErr.canRetry) !== null && _b !== void 0 ? _b : false, castedErr === null || castedErr === void 0 ? void 0 : castedErr.xhr); rej(emittedErr); }; const progress = (_args) => { if (hasFinished || cancelSignal.isCancelled()) { return; } callbacks.onProgress({ duration: _args.duration, size: _args.size, totalSize: _args.totalSize, }); }; const fallback = () => { if (hasFinished || cancelSignal.isCancelled()) { return; } hasFinished = true; cancelSignal.deregister(abortCustomLoader); regularSegmentLoader(url, context, callbacks, loaderOptions, cancelSignal, checkMediaSegmentIntegrity).then(res, rej); }; const customCallbacks = { reject, resolve, fallback, progress }; let byteRanges; if (context.segment.range !== undefined) { byteRanges = [context.segment.range]; if (context.segment.indexRange !== undefined) { byteRanges.push(context.segment.indexRange); } } const args = { isInit: context.segment.isInit, timeout: loaderOptions.timeout, byteRanges, trackType: context.type, url, cmcdPayload: loaderOptions.cmcdPayload, }; const abort = segmentLoader(args, customCallbacks); cancelSignal.register(abortCustomLoader); /** * The logic to run when the custom loader is cancelled while pending. * @param {Error} err */ function abortCustomLoader(err) { if (hasFinished) { return; } hasFinished = true; if (!hasFinished && typeof abort === "function") { abort(); } rej(err); } }); } }; export default generateSegmentLoader;