feat: fix difficulty parsing

otohime-site · Nov 21, 2020 · 904693c · 904693c
1 parent dfffdc1
commit 904693c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/package-lock.json b/package-lock.json
diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@otohime-site/parser",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "license": "MIT",
   "main": "index.js",
   "types": "index.d.ts",

diff --git a/src/dx_intl/scores.ts b/src/dx_intl/scores.ts
@@ -41,7 +41,10 @@ const parseScores = (content: string | HTMLDocument, categoryTo: number = 6, wit
           currentCategory: prev.currentCategory + 1
         }
       }
-      const rawMusicDifficulty = (curr.querySelector('div')?.className?.match(/basic|advanced|expert|master|remaster/) ?? [''])[0]
+      const rawMusicDifficulty = (
+        curr.className?.match(/_(basic|advanced|expert|master|remaster)/) ??
+        curr.querySelector('div')?.className?.match(/_(basic|advanced|expert|master|remaster)/) ??
+        ['', ''])[1]
       const category = prev.currentCategory
       const title = curr.querySelector('.music_name_block')?.textContent ?? ''
       const difficulty = ['basic', 'advanced', 'expert', 'master', 'remaster'].indexOf(rawMusicDifficulty)