feat: temporary ?disableFuzzy parameter

Sv443 · Sep 19, 2023 · ab6b55c · ab6b55c
1 parent 102c234
commit ab6b55c
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/src/routes/search.ts b/src/routes/search.ts
@@ -5,7 +5,7 @@ import { getMeta } from "../songData";
 export function initSearchRoutes(router: Router) {
   router.get("/search", async (req, res) => {
     try {
-      const { q, artist, song, format: fmt, threshold: thr } = req.query;
+      const { q, artist, song, format: fmt, threshold: thr, disableFuzzy } = req.query;
 
       const format: string = fmt ? String(fmt) : "json";
       const threshold = isNaN(Number(thr)) ? undefined : Number(thr);
@@ -19,6 +19,7 @@ export function initSearchRoutes(router: Router) {
             song: String(song),
           }),
           threshold,
+          disableFuzzy: typeof disableFuzzy === "string",
         });
 
         if(!meta || meta.all.length < 1)
@@ -39,7 +40,7 @@ export function initSearchRoutes(router: Router) {
 
   router.get("/search/top", async (req, res) => {
     try {
-      const { q, artist, song, format: fmt, threshold: thr } = req.query;
+      const { q, artist, song, format: fmt, threshold: thr, disableFuzzy } = req.query;
 
       const format: string = fmt ? String(fmt) : "json";
       const threshold = isNaN(Number(thr)) ? undefined : Number(thr);
@@ -53,6 +54,7 @@ export function initSearchRoutes(router: Router) {
             song: String(song),
           }),
           threshold,
+          disableFuzzy: typeof disableFuzzy === "string",
         });
 
         if(!meta || !meta.top)

diff --git a/src/songData.ts b/src/songData.ts
@@ -19,6 +19,7 @@ export async function getMeta({
   artist,
   song,
   threshold,
+  disableFuzzy,
 }: GetMetaArgs): Promise<GetMetaResult | null>
 {
   const query = q ? q : `${artist} ${song}`;
@@ -73,6 +74,16 @@ export async function getMeta({
         id: result.id ?? null,
       }));
 
+    if(disableFuzzy === true) {
+      if(hits.length === 0)
+        return null;
+
+      return {
+        top: hits[0]!,
+        all: hits.slice(0, 10),
+      };
+    }
+
     const scoreMap: Record<string, number> = {};
 
     hits = hits.map(h => {

diff --git a/src/types.ts b/src/types.ts
@@ -53,6 +53,7 @@ export interface GetMetaArgs {
   artist?: string;
   song?: string;
   threshold?: number;
+  disableFuzzy?: boolean;
 }
 
 export type SearchFilterArgs = GetMetaArgs;