traxxx/src/scrapers/mindgeek.js

'use strict';

/* eslint-disable newline-per-chained-call */
const Promise = require('bluebird');
const bhttp = require('bhttp');
const { CookieJar } = Promise.promisifyAll(require('tough-cookie'));
const moment = require('moment');

const { cookieToData } = require('../utils/cookies');

function getThumbs(scene) {
    if (scene.images.poster) {
        return scene.images.poster.map(image => image.xl.url);
    }

    if (scene.images.card_main_rect) {
        return scene.images.card_main_rect
            .concat(scene.images.card_secondary_rect || [])
            .map(image => image.xl.url.replace('.thumb', ''));
    }

    return [];
}

function scrapeLatestX(data, site) {
    const { id: entryId, title, description } = data;
    const hostname = site.parameters?.classic ? site.url : site.network.url;
    const url = `${hostname}/scene/${entryId}/`;
    const date = new Date(data.dateReleased);
    const actors = data.actors.map(actor => actor.name);

    const tags = data.tags.map(tag => tag.name);

    const [poster, ...photos] = getThumbs(data);
    const trailer = data.videos.mediabook && (data.videos.mediabook.files['720p'] || data.videos.mediabook.files['320p']);
    const duration = data.videos.mediabook && data.videos.mediabook.length;

    return {
        url,
        entryId,
        title,
        description,
        actors,
        tags,
        duration,
        poster,
        photos,
        trailer: trailer && {
            src: trailer.urls.view,
            quality: parseInt(trailer.format, 10),
        },
        date,
        site,
    };
}

async function scrapeLatest(items, site) {
    return Promise.all(items.map(async data => scrapeLatestX(data, site)));
}

async function scrapeScene(data, url, site) {
    const { id: entryId, title, description } = data;
    const date = new Date(data.dateReleased);
    const actors = data.actors.map(actor => actor.name);

    const tags = data.tags.map(tag => tag.name);

    const [poster, ...photos] = getThumbs(data);
    const trailer = data.videos.mediabook && (data.videos.mediabook.files['720p'] || data.videos.mediabook.files['320p']);

    const siteName = data.collections[0].name;
    const channel = siteName.replace(/\s+/g, '').toLowerCase();

    return {
        url,
        entryId,
        title,
        description,
        actors,
        tags,
        poster,
        photos,
        trailer: trailer && {
            src: trailer.urls.view,
            quality: parseInt(trailer.format, 10),
        },
        date,
        site,
        channel,
    };
}

function getUrl(site) {
    const { search } = new URL(site.url);

    if (search.match(/\?site=\d+/)) {
        return site.url;
    }

    if (site.parameters && site.parameters.siteId) {
        return `${site.network.url}/scenes?site=${site.parameters.siteId}`;
    }

    throw new Error(`Mind Geek site '${site.name}' (${site.url}) not supported`);
}

async function fetchLatest(site, page = 1) {
    const url = getUrl(site);
    const { search } = new URL(url);
    const siteId = new URLSearchParams(search).get('site');

    const cookieJar = new CookieJar();
    const session = bhttp.session({ cookieJar });

    await session.get(url);

    const cookieString = await cookieJar.getCookieStringAsync(url);
    const { instance_token: instanceToken } = cookieToData(cookieString);

    const beforeDate = moment().add('1', 'day').format('YYYY-MM-DD');
    const limit = 10;
    const apiUrl = `https://site-api.project1service.com/v2/releases?collectionId=${siteId}&dateReleased=<${beforeDate}&limit=${limit}&offset=${limit * (page - 1)}&orderBy=-dateReleased&type=scene`;
    const res = await session.get(apiUrl, {
        headers: {
            Instance: instanceToken,
        },
    });

    return scrapeLatest(res.body.result, site);
}

async function fetchScene(url, site) {
    const entryId = url.match(/\d+/)[0];

    const cookieJar = new CookieJar();
    const session = bhttp.session({ cookieJar });

    await session.get(url);

    const cookieString = await cookieJar.getCookieStringAsync(url);
    const { instance_token: instanceToken } = cookieToData(cookieString);

    const res = await session.get(`https://site-api.project1service.com/v2/releases/${entryId}`, {
        headers: {
            Instance: instanceToken,
        },
    });

    return scrapeScene(res.body.result, url, site);
}

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