Create buddy.js

lib/buddy.js

@@ -0,0 +1,117 @@
+const { Source, RunnerScraper, Hoster } = require("source-scraper-core");
+const { PuppeteerRunner } = require("source-scraper-puppeteer-runner");
+const cheerio = require("cheerio");
+const u = require("url");
+
+class NinexbuddySource extends Source {
+    constructor(source) {
+        super(source);
+        this.quality = source.quality;
+    }
+}
+
+function parseSourceQuality(raw) {
+    const ar1 = /^([a-zA-Z\d]+)\s*-\s*(\d+)([a-zA-Z]+)?$/i.exec(raw);
+    if (ar1 !== null && ar1.length >= 4)
+        return {
+            raw,
+            description: ar1[1],
+            value: Number(ar1[2]),
+            unit: ar1[3]
+        };
+    const ar2 = /^(\d+)([a-zA-Z]+)\s*\(?([a-zA-Z\s]+)\)?$/i.exec(raw);
+    if (ar2 !== null && ar2.length >= 3)
+        return {
+            raw,
+            description: ar2[3],
+            value: Number(ar2[1]),
+            unit: ar2[2]
+        };
+}
+
+function parseHosterName(raw) {
+    const ar = /^(?:embed\s*-)?\s*([a-zA-Z\d.]+)$/i.exec(raw.trim());
+    if (ar === null || ar.length < 2)
+        return;
+    return ar[ar.length - 1];
+}
+
+function isDefined(value) {
+    return typeof value !== 'undefined';
+}
+
+class NinexbuddyScraper extends RunnerScraper {
+    constructor() {
+        super();
+        this.name = 'ninexbuddy';
+        this.domains = [];
+        this.urlPattern = /.*/i;
+        this.runner = new PuppeteerRunner();
+        this.defaultOptions = {
+            runnerOptions: {
+                navigationOptions: {
+                    waitUntil: 'networkidle0'
+                }
+            }
+        };
+    }
+
+    transformUrl(url) {
+        return `https://9xbuddy.com/process?url=${url}`;
+    }
+
+    async execWithArgs({ page, url: ninexbuddyUrl }) {
+        const isHoster = !!(await page.$('.playlist_el').catch(() => false));
+        if (isHoster) {
+            const items = await page.$$('.playlist_el .text-blue');
+            const hosters = (await Promise.all(items.map(async (item) => {
+                const urlPromise = item.getProperty('href').then(async (e) => e.jsonValue()).catch(() => undefined);
+                const namePromise = item.getProperty('textContent').then(async (e) => e.jsonValue()).catch(() => undefined);
+                const [url, name] = await Promise.all([urlPromise, namePromise]);
+                const hosterUrl = u.parse(url, true).query.url;
+                if (!hosterUrl)
+                    return;
+                const parsed = name ? parseHosterName(name) : undefined;
+                if (!parsed)
+                    return;
+                return new Hoster({
+                    url: hosterUrl,
+                    name: parsed,
+                });
+            }))).filter(isDefined);
+            return {
+                ninexbuddyUrl,
+                hosters
+            };
+        } else {
+            const selector = '.tracking-wide.items-center.night-white';
+            const items = await page.$$(selector);
+            const sources = (await Promise.all(items.map(async (item) => {
+                const formatPromise = item.$eval('.w-24.text-blue', e => e.textContent).catch(() => undefined);
+                const qualityPromise = item.$eval('.w-1\\/2', e => e.textContent).catch(() => undefined);
+                const urlPromise = item.$eval('.leading-normal.sm\\:leading-none', e => e.getAttribute('href')).catch(() => undefined);
+                const [format, quality, url] = await Promise.all([formatPromise, qualityPromise, urlPromise]);
+                if (!url)
+                    return;
+                return new NinexbuddySource({
+                    url,
+                    quality: quality ? parseSourceQuality(quality) : undefined,
+                    type: format || undefined,
+                    codec: undefined
+                });
+            }))).filter(isDefined);
+            const poster = await page.$eval('img.shadow.border-2.border-grey-lighter.h-48', e => e.src).catch(() => undefined);
+            return {
+                ninexbuddyUrl,
+                title: await page.title(),
+                poster: poster && !poster.startsWith('data:image') ? poster : undefined,
+                sources
+            };
+        }
+    }
+}
+
+module.exports = {
+    NinexbuddySource,
+    NinexbuddyScraper
+}