49 lines
1.7 KiB
TypeScript
49 lines
1.7 KiB
TypeScript
import { ScrapeEngine, ScrapeResult, ScrapedPost, ScrapedTag } from "../ScrapeEngine";
|
|
|
|
export default class Shimmie2 implements ScrapeEngine {
|
|
name = "shimmie2";
|
|
|
|
canImport(url: Location): boolean {
|
|
return (
|
|
url.host == "rule34.paheal.net" ||
|
|
url.host == "rule34hentai.net"
|
|
);
|
|
}
|
|
|
|
scrapeDocument(document: Document): ScrapeResult {
|
|
let result = new ScrapeResult(this.name);
|
|
let post = new ScrapedPost();
|
|
post.source = document.location.href;
|
|
|
|
// Set image url
|
|
const originalImageElements = Array.from(document.querySelectorAll(".image_info > tbody > tr > td > a"))
|
|
.map(x => x as HTMLAnchorElement)
|
|
.filter(x => x.innerText == "Image Only");
|
|
|
|
if (originalImageElements.length > 0) {
|
|
post.imageUrl = originalImageElements[0].href;
|
|
} else {
|
|
const dowloadBtnEl = document.querySelector("a[download]") as HTMLAnchorElement;
|
|
if (dowloadBtnEl) {
|
|
post.imageUrl = dowloadBtnEl.href;
|
|
}
|
|
}
|
|
|
|
// Set safety
|
|
post.rating = "unsafe"; // Usually it is unsafe
|
|
|
|
// Set tags
|
|
// Shimmie2 doesn't support categories so we don't need to worry about those.
|
|
// They also use uppercase tags, so we toLowerCase() those.
|
|
// Tags might not match with other booru-likes, but that is for the user to figure out.
|
|
// For example by deleting 'wrong' tags or setting up some aliases in szurubooru.
|
|
post.tags = Array.from(document.querySelectorAll("a.tag_name"))
|
|
.map(x => (x as HTMLAnchorElement).innerText)
|
|
.map(x => new ScrapedTag(x));
|
|
|
|
result.tryAddPost(post);
|
|
|
|
return result;
|
|
}
|
|
}
|