traxxx/src/scrapers/kink.js

'use strict';

const bhttp = require('bhttp');
const cheerio = require('cheerio');
const moment = require('moment');

const knex = require('../knex');
const { matchTags } = require('../tags');

function scrapeLatest(html, site) {
    const $ = cheerio.load(html, { normalizeWhitespace: true });
    const sceneElements = $('.shoot-list .shoot').toArray();

    return sceneElements.map((element) => {
        const sceneLinkElement = $(element).find('.shoot-thumb-title a');
        const href = sceneLinkElement.attr('href');
        const url = `https://kink.com${href}`;
        const shootId = href.split('/')[2];
        const title = sceneLinkElement.text();

        const date = moment.utc($(element).find('.date').text(), 'MMM DD, YYYY').toDate();
        const actors = $(element).find('.shoot-thumb-models a').map((actorIndex, actorElement) => $(actorElement).text()).toArray();
        const stars = $(element).find('.average-rating').attr('data-rating') / 10;

        const timestamp = $(element).find('.video span').text();
        const timestampComponents = timestamp.split(':'); // fix mixed hh:mm:ss and mm:ss format
        const duration = moment.duration(timestampComponents.length > 2 ? timestamp : `0:${timestamp}`).asSeconds();

        return {
            url,
            shootId,
            title,
            actors,
            date,
            rating: {
                stars,
            },
            duration,
            site,
        };
    });
}

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

    // const title = $('h1.shoot-title').text().replace(/\ue800/, ''); // fallback, special character is 'like'-heart
    const title = $('h1.shoot-title span.favorite-button').attr('data-title');
    const actorsRaw = $('.shoot-info p.starring');

    const sitename = $('.shoot-logo a').attr('href').split('/')[2];

    const date = moment.utc($(actorsRaw)
        .prev()
        .text()
        .trim()
        .replace('Date: ', ''),
    'MMMM DD, YYYY')
        .toDate();

    const actors = $(actorsRaw).find('span.names a').map((actorIndex, actorElement) => $(actorElement).text()).toArray();
    const description = $('.shoot-info .description').text();

    const { average: stars } = ratingRes.body;

    const rawTags = $('.tag-list > a[href*="/tag"]').map((tagIndex, tagElement) => $(tagElement).text()).toArray();

    const [channelSite, tags] = await Promise.all([
        knex('sites').where({ id: sitename }).first(),
        matchTags(rawTags),
    ]);

    return {
        url,
        shootId,
        title,
        date,
        actors,
        description,
        rating: {
            stars,
        },
        tags,
        site: channelSite || site,
    };
}

async function fetchLatest(site) {
    const res = await bhttp.get(`${site.url}/latest`);

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

async function fetchScene(url, site) {
    const shootId = new URL(url).pathname.split('/')[2];

    const [res, ratingRes] = await Promise.all([
        bhttp.get(url),
        bhttp.get(`https://kink.com/api/ratings/${shootId}`),
    ]);

    return scrapeScene(res.body.toString(), url, shootId, ratingRes, site);
}

module.exports = {
    fetchLatest,
    fetchScene,
};