traxxx/src/scrapers/vixen.js

'use strict';

/* eslint-disable newline-per-chained-call */
const bhttp = require('bhttp');
const cheerio = require('cheerio');
const moment = require('moment');

function getPosterFallbacks(poster) {
    return poster
        .filter(image => /landscape/i.test(image.name))
        .sort((imageA, imageB) => imageB.height - imageA.height)
        .map((image) => {
            const sources = [image.src, image.highdpi?.['2x'], image.highdpi?.['3x']];
            // high DPI images for full HD source are huge, only prefer for smaller fallback sources
            return image.height === 1080 ? sources : sources.reverse();
        })
        .flat();
}

function scrapeLatest(html, site) {
    const $ = cheerio.load(html, { normalizeWhitespace: true });

    const stateScript = $('script:contains("INITIAL_STATE")').html();
    const { videos: scenes } = JSON.parse(stateScript.slice(stateScript.indexOf('{'), stateScript.indexOf('};') + 1));

    return scenes.map((scene) => {
        const entryId = String(scene.newId);

        const {
            title,
            models: actors,
        } = scene;

        const url = `${site.url}${scene.targetUrl}`;
        const date = moment.utc(scene.releaseDateFormatted, 'MMMM DD, YYYY').toDate();
        const stars = Number(scene.textRating) / 2;

        // largest thumbnail. poster is the same image but bigger, too large for storage space efficiency
        const poster = scene.images.listing.slice(-1)[0].src;
        const teaser = scene.previews.listing.slice(-1)[0];

        return {
            url,
            entryId,
            title,
            actors,
            date,
            poster,
            teaser: {
                src: teaser.src,
                type: teaser.type,
                quality: teaser.height,
            },
            rating: {
                stars,
            },
            site,
        };
    });
}

function scrapeUpcoming(html, site) {
    const statePrefix = html.indexOf('__INITIAL_STATE__');
    const stateString = html.slice(html.indexOf('{', statePrefix), html.indexOf('};', statePrefix) + 1);
    const data = JSON.parse(stateString);

    const scene = data.page.data['/'].data?.nextScene;
    if (!scene || scene.isPreReleasePeriod) return null;

    const release = {};

    release.title = scene.targetUrl
        .slice(1)
        .split('-')
        .map(component => `${component.charAt(0).toUpperCase()}${component.slice(1)}`)
        .join(' ');

    release.date = moment.utc(scene.releaseDate).toDate();
    release.url = `${site.url}${scene.targetUrl}`;

    release.actors = scene.models;

    release.poster = getPosterFallbacks(scene.images.poster);
    release.teaser = scene.previews.poster
        .filter(teaser => /landscape/i.test(teaser.name))
        .map(teaser => ({
            src: teaser.src,
            type: teaser.type,
            quality: Number(String(teaser.height).replace('353', '360')),
        }));

    release.entryId = (release.poster[0] || release.teaser[0])?.match(/\/(\d+)/)?.[1];

    return [release];
}

async function scrapeScene(html, url) {
    const $ = cheerio.load(html, { normalizeWhitespace: true });

    const stateObject = $('script:contains("INITIAL_STATE")');
    const data = JSON.parse(stateObject.html().trim().slice(27, -1));

    const pageData = data.page.data[data.location.pathname].data;
    const scene = data.videos.find(video => video.newId === pageData.video);

    const release = {
        url,
        title: scene.title,
        description: scene.description,
        actors: scene.models,
        director: scene.directorNames,
        duration: scene.runLength,
        stars: scene.totalRateVal,
        tags: scene.tags,
    };

    release.entryId = pageData.video;
    release.actors = scene.models;

    // release.poster = scene.rotatingThumbsUrlSizes[0]['1040w'];
    release.poster = getPosterFallbacks(scene.images.poster);
    release.photos = pageData.pictureset.map(photo => photo.main[0].src);

    const trailer = scene.previews.listing.find(preview => preview.height === 353);
    if (trailer) release.trailer = { src: trailer };

    // trailer must exist!

    release.teaser = scene.previews.poster
        .filter(teaser => /landscape/i.test(teaser.name))
        .map(teaser => ({
            src: teaser.src,
            type: teaser.type,
            quality: Number(String(teaser.height).replace('353', '360')),
        }));

    release.date = new Date(scene.releaseDate);

    return release;
}

async function fetchLatest(site, page = 1) {
    const url = `${site.url}/videos?page=${page}&size=7`;
    const res = await bhttp.get(url);

    if (res.statusCode === 200) {
        return scrapeLatest(res.body.toString(), site);
    }

    return res.statusCode;
}

async function fetchUpcoming(site) {
    const res = await bhttp.get(site.url);

    if (res.statusCode === 200) {
        return scrapeUpcoming(res.body.toString(), site);
    }

    return res.statusCode;
}

async function fetchScene(url, site) {
    const res = await bhttp.get(url);

    if (res.statusCode === 200) {
        return scrapeScene(res.body.toString(), url, site);
    }

    throw new Error(`Vixen response not OK for scene (${url}): ${res.statusCode}`);
}

module.exports = {
    fetchLatest,
    fetchUpcoming,
    fetchScene,
};