feat: add helper filters for hls manifest

lib/dash.js

@@ -1,7 +1,13 @@
 'use strict';
 
 const mpdParser = require('mpd-parser');
-const { parseBitrate, parseSize, getWidth, getBestTrack, getQualityLabel } = require('./util');
+const { parseBitrate, parseSize, getQualityLabel } = require('./util');
+const {
+  createResolutionFilter,
+  createVideoQualityFilter,
+  createAudioLanguageFilter,
+  createSubtitleLanguageFilter,
+} = require('./track');
 
 const segmentsDto = (data = []) => {
   const mapSegment = (item) => ({
@@ -115,37 +121,10 @@ const parseManifest = (manifestString, manifestUri) => {
       videos,
       audios,
       subtitles,
-      withResolution({ width, height }) {
-        return videos.filter(
-          (track) =>
-            (!width || track.resolution.width === width) &&
-            (!height || track.resolution.height === height)
-        );
-      },
-      withVideoQuality(quality) {
-        const height = parseInt(quality);
-        const matchHeight = (track) => {
-          if (!height) return false;
-          const trackWidth = track.resolution.width;
-          const targetWidth = getWidth(height);
-          if (!trackWidth || !targetWidth) return false;
-          return trackWidth === targetWidth;
-        };
-        const matches = videos.filter(matchHeight);
-        return matches.length ? matches : [getBestTrack(videos)];
-      },
-      withAudioLanguages(languages) {
-        if (!languages.length) return audios;
-        return audios.filter((track) =>
-          languages.some((language) => track.language.startsWith(language))
-        );
-      },
-      withSubtitleLanguages(languages) {
-        if (!languages.length) return subtitles;
-        return subtitles.filter((track) =>
-          languages.some((language) => track.language.startsWith(language))
-        );
-      },
+      withResolution: createResolutionFilter(videos),
+      withVideoQuality: createVideoQualityFilter(videos),
+      withAudioLanguages: createAudioLanguageFilter(audios),
+      withSubtitleLanguages: createSubtitleLanguageFilter(subtitles),
     },
   };
 

lib/hls.js

@@ -2,6 +2,12 @@
 
 const m3u8Parser = require('m3u8-parser');
 const { parseBitrate, getQualityLabel } = require('./util');
+const {
+  createResolutionFilter,
+  createVideoQualityFilter,
+  createAudioLanguageFilter,
+  createSubtitleLanguageFilter,
+} = require('./track');
 
 const parseM3u8 = (manifestString) => {
   const parser = new m3u8Parser.Parser();
@@ -123,6 +129,10 @@ const parseManifest = async (manifestString, manifestUri) => {
       videos,
       audios,
       subtitles,
+      withResolution: createResolutionFilter(videos),
+      withVideoQuality: createVideoQualityFilter(videos),
+      withAudioLanguages: createAudioLanguageFilter(audios),
+      withSubtitleLanguages: createSubtitleLanguageFilter(subtitles),
     },
   };
 

lib/track.js

@@ -0,0 +1,49 @@
+const createResolutionFilter = (videos) => {
+  return ({ width, height }) => {
+    return videos.filter(
+      (track) =>
+        (!width || track.resolution.width === width) &&
+        (!height || track.resolution.height === height)
+    );
+  };
+};
+
+const createVideoQualityFilter = (videos) => {
+  return (quality) => {
+    const height = parseInt(quality);
+    const matchHeight = (track) => {
+      if (!height) return false;
+      const trackWidth = track.resolution.width;
+      const targetWidth = getWidth(height);
+      if (!trackWidth || !targetWidth) return false;
+      return trackWidth === targetWidth;
+    };
+    const matches = videos.filter(matchHeight);
+    return matches.length ? matches : [getBestTrack(videos)];
+  };
+};
+
+const createAudioLanguageFilter = (audios) => {
+  return (languages) => {
+    if (!languages.length) return audios;
+    return audios.filter((track) =>
+      languages.some((language) => track.language.startsWith(language))
+    );
+  };
+};
+
+const createSubtitleLanguageFilter = (subtitles) => {
+  return (languages) => {
+    if (!languages.length) return subtitles;
+    return subtitles.filter((track) =>
+      languages.some((language) => track.language.startsWith(language))
+    );
+  };
+};
+
+module.exports = {
+  createResolutionFilter,
+  createVideoQualityFilter,
+  createAudioLanguageFilter,
+  createSubtitleLanguageFilter,
+};

package.json

@@ -1,6 +1,6 @@
 {
   "name": "dasha",
-  "version": "3.0.0.alpha.4",
+  "version": "3.0.0.alpha.5",
   "author": "Vitaly Gashkov <[email protected]>",
   "description": "Parser for MPEG-DASH & HLS manifests",
   "license": "AGPL-3.0",