From 2a7f3571a1157417192e5550ec5802c366422821 Mon Sep 17 00:00:00 2001
From: Viren070 <viren070@protonmail.com>
Date: Sat, 14 Dec 2024 18:52:38 +0000
Subject: [PATCH] feat: add considerHdrTagsAsEqual config option

---
 src/index.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/index.js b/src/index.js
index b96b3cc..ad27f63 100644
--- a/src/index.js
+++ b/src/index.js
@@ -22,7 +22,8 @@ const CONFIG = {
         "Unknown",
     ],
     visualTags: ["HDR10+", "HDR10", "HDR", "DV", "IMAX", "AI"],
-    sortBy: ["resolution", "visualTag",  "quality", "size"],
+    sortBy: ["resolution", "visualTag", "size", "quality"],
+    considerHdrTagsAsEqual: true,
     addonName: "GDrive",
     prioritiseLanguage: null,
     proxiedPlayback: true,
@@ -182,15 +183,20 @@ function compareByField(a, b, field) {
         );
     } else if (field === "visualTag") {
         // Find the highest priority visual tag in each file
+        const getIndexOfTag = (tag) => (
+            CONFIG.considerHdrTagsAsEqual && tag.startsWith("HDR")
+                ? CONFIG.visualTags.indexOf("HDR10+")
+                : CONFIG.visualTags.indexOf(tag)
+        )
         const aVisualTagIndex = a.visualTags.reduce(
             (minIndex, tag) =>
-                Math.min(minIndex, CONFIG.visualTags.indexOf(tag)),
+                Math.min(minIndex, getIndexOfTag(tag)),
             CONFIG.visualTags.length
         );
 
         const bVisualTagIndex = b.visualTags.reduce(
             (minIndex, tag) =>
-                Math.min(minIndex, CONFIG.visualTags.indexOf(tag)),
+                Math.min(minIndex, getIndexOfTag(tag)),
             CONFIG.visualTags.length
         );
         // Sort by the visual tag index