dasha 2.1.0 → 2.2.0

package/lib/manifest.js

@@ -5,6 +5,14 @@ const { processManifest } = require('./processor');
 const { getQualityLabel } = require('./utils');
 const { CONTENT_TYPE, KEY_SYSTEMS } = require('./constants');
 
+const isUrl = (value) => {
+  try {
+    return !!new URL(value);
+  } catch (e) {
+    return false;
+  }
+};
+
 class Manifest {
   constructor(manifest) {
     for (const key of Object.keys(manifest)) this[key] = manifest[key];
@@ -25,6 +33,7 @@ class Manifest {
     const representation = this.findBestRepresentation(matches.length ? matches : representations);
     const adaptationSet = adaptationSets.find((a) => a.representations.includes(representation));
     return {
+      id: 0,
       type: CONTENT_TYPE.video,
       pssh: this.getWidevinePssh(adaptationSet),
       licenseUrl: this.getWidevineLicenseUrl(adaptationSet),
@@ -39,10 +48,9 @@ class Manifest {
 
   getAudioTracks(languages) {
     const isAudioAdaptationSet = (adaptationSet) =>
-      adaptationSet.contentType === CONTENT_TYPE.audio ||
-      adaptationSet.mimeType === 'audio/mp4' ||
-      !adaptationSet.maxWidth ||
-      !adaptationSet.maxHeight;
+      (adaptationSet.contentType === CONTENT_TYPE.audio ||
+        adaptationSet.mimeType === 'audio/mp4') &&
+      !adaptationSet.maxWidth;
     const adaptationSets = this.periods
       .map((p) => p.adaptationSets)
       .flat(1)
@@ -50,16 +58,16 @@ class Manifest {
 
     const representations = adaptationSets.map((a) => a.representations).flat(2);
     const matches = representations.filter((r) => languages?.some((lang) => r.lang.includes(lang)));
-    // TODO: Select all audio tracks if they with different langs but same bandwidth
-    const selectedRepresentations = matches?.length
-      ? matches
-      : [this.findBestRepresentation(representations)];
+    const selectedRepresentations = matches?.length ? matches : representations;
 
     const tracks = [];
     for (const representation of selectedRepresentations) {
       const adaptationSet = adaptationSets.find((a) => a.representations.includes(representation));
       const track = {
+        id: tracks.length,
         type: CONTENT_TYPE.audio,
+        label: adaptationSet.label,
+        language: adaptationSet.lang,
         pssh: this.getWidevinePssh(adaptationSet),
         licenseUrl: this.getWidevineLicenseUrl(adaptationSet),
         segments: this.getSegments(adaptationSet, representation),
@@ -74,19 +82,53 @@ class Manifest {
   }
 
   getSubtitleTracks(languages) {
-    // TODO: Return subtitle tracks
+    const isSubtitleAdaptationSet = (adaptationSet) =>
+      (adaptationSet.contentType === CONTENT_TYPE.text ||
+        adaptationSet.mimeType.includes('text')) &&
+      !adaptationSet.maxWidth;
+    const adaptationSets = this.periods
+      .map((p) => p.adaptationSets)
+      .flat(1)
+      .filter(isSubtitleAdaptationSet);
+
+    const representations = adaptationSets.map((a) => a.representations).flat(2);
+    const matches = representations.filter((r) => languages?.some((lang) => r.lang.includes(lang)));
+    const selectedRepresentations = matches?.length ? matches : representations;
+
+    const tracks = [];
+    for (const representation of selectedRepresentations) {
+      const adaptationSet = adaptationSets.find((a) => a.representations.includes(representation));
+      const baseUrl = this.baseUrls?.[0]?.value;
+      const representationBaseUrl = representation.baseUrls?.[0]?.value;
+      const url = isUrl(representationBaseUrl)
+        ? representationBaseUrl
+        : `${baseUrl || ''}${representationBaseUrl || ''}`;
+      const track = {
+        id: tracks.length,
+        type: CONTENT_TYPE.text,
+        label: adaptationSet.label,
+        language: adaptationSet.lang,
+        format: representation.mimeType?.split('/')[1] || adaptationSet.mimeType?.split('/')[1],
+        segments: [{ url }],
+        bitrate: Math.round(representation.bandwidth / 1000), // Kbps
+        size: Math.ceil((representation.bandwidth / 8e6) * this.mediaPresentationDuration), // MB
+      };
+      tracks.push(track);
+    }
+
+    return tracks;
   }
 
   getWidevinePssh(adaptationSet) {
     const isWidevineProtection = (protection) =>
       protection.schemeIdUri === KEY_SYSTEMS['com.widevine.alpha'];
-    return adaptationSet.contentProtections.find(isWidevineProtection)?.cencPssh || null;
+    return adaptationSet?.contentProtections?.find(isWidevineProtection)?.cencPssh || null;
   }
 
   getWidevineLicenseUrl(adaptationSet) {
     const isWidevineProtection = (protection) =>
       protection.schemeIdUri === KEY_SYSTEMS['com.widevine.alpha'];
-    return adaptationSet.contentProtections.find(isWidevineProtection)?.licenseUrl || null;
+    return adaptationSet?.contentProtections?.find(isWidevineProtection)?.licenseUrl || null;
   }
 
   findBestRepresentation(representations) {

package/lib/processor.js

@@ -58,6 +58,8 @@ const processAdaptationSet = (adaptationSet) => {
     maxFrameRate: adaptationSet['maxFrameRate'],
     contentType: adaptationSet['contentType'],
     segmentTemplate: adaptationSet['SegmentTemplate'],
+    mimeType: adaptationSet['mimeType'],
+    label: adaptationSet['label'],
     contentProtections,
     representations,
   };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dasha",
-  "version": "2.1.0",
+  "version": "2.2.0",
   "author": "Vitaliy Gashkov <vitnore@gmail.com>",
   "description": "Simple MPEG-DASH MPD parser",
   "license": "Apache-2.0",