From b1e76ddfd38e8c4bd3e3a954e85546bec784276b Mon Sep 17 00:00:00 2001
From: zkldi <notzkldi@zkldi.com>
Date: Thu, 20 May 2021 07:28:30 +0100
Subject: [PATCH] Create FindSDVXChartOnInGameIDVersion (lol)

---
 server/src/common/database-lookup/chart.ts | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/server/src/common/database-lookup/chart.ts b/server/src/common/database-lookup/chart.ts
index 74d3421f3..a83fdc293 100644
--- a/server/src/common/database-lookup/chart.ts
+++ b/server/src/common/database-lookup/chart.ts
@@ -170,7 +170,6 @@ export function FindIIDXChartWith2DXtraHash(hash: string) {
  */
 export function FindSDVXChartOnInGameID(
     inGameID: number,
-    playtype: Playtypes[Game],
     difficulty: "NOV" | "ADV" | "EXH" | "MXM" | "ANY_INF"
 ) {
     const diffQuery =
@@ -180,7 +179,23 @@ export function FindSDVXChartOnInGameID(
 
     return db.charts.sdvx.findOne({
         "data.inGameID": inGameID,
-        playtype,
         difficulty: diffQuery,
     });
 }
+
+export function FindSDVXChartOnInGameIDVersion(
+    inGameID: number,
+    difficulty: "NOV" | "ADV" | "EXH" | "MXM" | "ANY_INF",
+    version: string
+) {
+    const diffQuery =
+        difficulty === "ANY_INF"
+            ? { $in: ["INF", "GRV", "HVN", "VVD"] as Difficulties["sdvx:Single"][] }
+            : difficulty;
+
+    return db.charts.sdvx.findOne({
+        "data.inGameID": inGameID,
+        difficulty: diffQuery,
+        versions: version,
+    });
+}