feat: support parsing dolby vision

scttcper · Apr 3, 2022 · 613804a · 613804a
1 parent 1738e44
commit 613804a
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/edition.ts b/src/edition.ts
@@ -16,6 +16,7 @@ const hsbs = /\b(Half-?SBS|HSBS)\b/i;
 const sbs = /\b((?<!H|HALF-)SBS)\b/i;
 const hou = /\b(HOU)\b/i;
 const uhd = /\b(UHD)\b/i;
+const dolbyVision = /\b(DV(\b(HDR10|HLG|SDR))?)\b/i;
 
 export interface Edition {
   internal?: boolean;
@@ -39,6 +40,7 @@ export interface Edition {
   hou?: boolean;
   /** most 2160p should be UHD but there might be some that aren't? */
   uhd?: boolean;
+  dolbyVision?: boolean;
 }
 
 export function parseEdition(title: string): Edition {
@@ -61,6 +63,7 @@ export function parseEdition(title: string): Edition {
     sbs: sbs.test(withoutTitle) || undefined,
     hou: hou.test(withoutTitle) || undefined,
     uhd: uhd.test(withoutTitle) || undefined,
+    dolbyVision: dolbyVision.test(withoutTitle) || undefined,
   };
 
   return removeEmpty(result);

diff --git a/test/edition.spec.ts b/test/edition.spec.ts
@@ -56,6 +56,11 @@ const cases: Array<[string, Partial<Edition>]> = [
     'Iron.Man.2008.INTERNAL.REMASTERED.2160p.UHD.BluRay.X265-IAMABLE',
     { uhd: true, remastered: true, internal: true },
   ],
+  ['Long Shot 2019 DV 2160p WEB H265-SLOT', { dolbyVision: true }],
+  [
+    'Sicario 2015 Hybrid 2160p UHD BluRay REMUX DV HDR10+ HEVC TrueHD 7.1 Atmos-WiLDCAT',
+    { uhd: true, dolbyVision: true },
+  ],
 ];
 
 for (const [title, result] of cases) {