From 525329780990b50933360f3d0885fab0489185a3 Mon Sep 17 00:00:00 2001
From: neobooru <50623835+neobooru@users.noreply.github.com>
Date: Tue, 9 Jan 2024 23:15:22 +0100
Subject: [PATCH] Update danbooru tag category scraper

---
 package.json            |  2 +-
 src/engines/Danbooru.ts | 29 ++++++++++++++++-------------
 2 files changed, 17 insertions(+), 14 deletions(-)

diff --git a/package.json b/package.json
index a2b9794..ca4ea33 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "neo-scraper",
-  "version": "0.7.4",
+  "version": "0.7.5",
   "description": "Scrape a booru-like post object from the DOM.",
   "type": "module",
   "main": "dist/index.js",
diff --git a/src/engines/Danbooru.ts b/src/engines/Danbooru.ts
index d7395d0..398e41e 100644
--- a/src/engines/Danbooru.ts
+++ b/src/engines/Danbooru.ts
@@ -76,19 +76,22 @@ export default class Danbooru extends ScrapeEngineBase {
       }
 
       let category: TagCategory | undefined;
-      switch (tagElement.className) {
-        case "tag-type-3":
-          category = "copyright";
-          break;
-        case "tag-type-4":
-          category = "character";
-          break;
-        case "tag-type-1":
-          category = "artist";
-          break;
-        case "tag-type-5":
-          category = "meta";
-          break;
+      console.dir(tagElement.classList);
+      for (const className of tagElement.classList) {
+        switch (className) {
+          case "tag-type-3":
+            category = "copyright";
+            break;
+          case "tag-type-4":
+            category = "character";
+            break;
+          case "tag-type-1":
+            category = "artist";
+            break;
+          case "tag-type-5":
+            category = "meta";
+            break;
+        }
       }
 
       if (tagName) {