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) {