'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, };