traxxx/src/scrapers/vixen.js

'use strict';

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

const qu = require('../utils/qu');
const http = require('../utils/http');
const slugify = require('../utils/slugify');

const genderMap = {
	F: 'female',
	M: 'male',
	T: 'transsexual', // not yet observed
};

function getAvatarFallbacks(avatar) {
	return avatar
		.sort((imageA, imageB) => imageB.height - imageA.height)
		.map((image) => [image.highdpi?.['3x'], image.highdpi?.['2x'], image.src])
		.flat();
}

function curateSources(sources, type = 'image/jpeg') {
	if (!sources) {
		return null;
	}

	return sources
		.map((source) => ({
			src: source.src,
			width: source.width,
			height: source.height,
			type: source.type || type,
			expectType: {
				'binary/octet-stream': type,
			},
		}))
		.sort((resA, resB) => (resB.width * resB.height) - (resA.width * resA.height))			// number of pixels
		.sort((resA, resB) => Math.abs(1.8 - Number((resA.width / resA.height).toFixed(1)))		// approximation to 16:9
							- Math.abs(1.8 - Number((resB.width / resB.height).toFixed(1))));
}

async function getTrailer(scene, channel, url) {
	const res = await http.post(`${channel.url}/graphql`, {
		operationName: 'getToken',
		variables: {
			videoId: scene.newId,
			device: 'trailer',
		},
		query: `
			query getToken($videoId: ID!, $device: Device!) {
				generateVideoToken(input: {videoId: $videoId, device: $device}) {
					p270 {
						token
						cdn
						__typename
					}
					p360 {
						token
						cdn
						__typename
					}
					p480 {
						token
						cdn
						__typename
					}
					p480l {
						token
						cdn
						__typename
					}
					p720 {
						token
						cdn
						__typename
					}
					p1080 {
						token
						cdn
						__typename
					}
					p2160 {
						token
						cdn
						__typename
					}
					hls {
						token
						cdn
						__typename
					}
					__typename
				}
			}
		`,
	}, {
		headers: {
			referer: url,
			origin: channel.url,
		},
	});

	if (res.ok && res.body.data?.generateVideoToken) {
		return [
			{
				src: res.body.data.generateVideoToken.p2160?.token,
				quality: 2160,
			},
			{
				src: res.body.data.generateVideoToken.p1080?.token,
				quality: 1080,
			},
			{
				src: res.body.data.generateVideoToken.p720?.token,
				quality: 720,
			},
			{
				src: res.body.data.generateVideoToken.p480?.token,
				quality: 480,
			},
			{
				src: res.body.data.generateVideoToken.p360?.token,
				quality: 360,
			},
			{
				src: res.body.data.generateVideoToken.p270?.token,
				quality: 270,
			},
		];
	}

	return null;
}

function scrapeAll(scenes, channel) {
	return scenes.map((data) => {
		const release = {};

		release.entryId = data.videoId;
		release.url = `${channel.url}/videos/${data.slug}`;
		release.title = data.title;

		release.date = qu.extractDate(data.releaseDate);
		release.actors = data.modelsSlugged.map((model) => ({
			name: model.name,
			url: `${channel.url}/models/${model.slugged}`,
		}));

		release.poster = curateSources(data.images.listing);
		release.teaser = curateSources(data.previews.listing, 'video/mp4');

		release.stars = data.rating;

		return release;
	});
}

function scrapeUpcoming(scene, site) {
	if (!scene || scene.isPreReleasePeriod) {
		return null;
	}

	const release = {};

	release.entryId = scene.videoId;
	release.url = `${site.url}/videos/${scene.slug}`;

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

	release.date = moment.utc(scene.releaseDate).toDate();
	release.datePrecision = 'minute';

	release.actors = scene.models.map((model) => model.name);

	release.poster = curateSources(scene.images.poster);
	release.teaser = curateSources(scene.previews.poster);

	return [release];
}

async function fetchGraphqlDetails(release, channel, session) {
	const query = `
		query($query: String!, $site: Site!) {
		  searchVideos(input: {
			  query: $query
			  site: $site
		  }) {
			edges {
			  node {
				videoId
				title
				slug
				description
				releaseDate
				categories {
				  name
				}
				chapters {
				  video {
					title
					seconds
				  }
				}
				models {
				  name
				}
				images {
				  poster {
					...ImageInfo
				  }
				}
			  }
			}
		  }
		}

		fragment ImageInfo on Image {
		  src
		  highdpi {
			double
		  }
		}
	`;

	const variables = JSON.stringify({
		site: channel.slug.toUpperCase(),
		query: release.title,
	});

	const res = await http.get(`${channel.url}/graphql?query=${encodeURI(query)}&variables=${variables}`, {
		session,
		headers: {
			referer: channel.url,
			accept: '*/*',
		},
	});

	if (res.ok) {
		return res.body.data?.searchVideos?.edges?.find((edge) => edge.node.videoId === release.entryId)?.node || null;
	}

	return null;
}

async function scrapeScene(data, url, channel, options, session) {
	const release = {
		url,
		entryId: data.video.videoId || data.video.newId,
		title: data.video.title,
		description: data.video.description,
		actors: data.video.models,
		director: data.video.directorNames,
		duration: qu.durationToSeconds(data.video.runLength),
		stars: data.video.rating,
	};

	release.entryId = data.video.newId;
	release.date = qu.extractDate(data.video.releaseDate);

	release.actors = data.video.modelsSlugged.map((model) => ({
		name: model.name,
		url: `${channel.url}/models/${model.slugged}`,
	}));

	release.poster = curateSources(data.video.images?.poster) || data.video.videoImage?.src;
	release.photos = data.galleryImages?.length > 0
		? data.galleryImages.map((image) => image.src)
		: data.video.carousel?.map((photo) => photo.main[0]?.src).filter(Boolean);

	if (options.includeTrailers) {
		const trailer = await getTrailer(data.video, channel, url);

		if (trailer) {
			release.trailer = trailer;
		}
	}

	release.qualities = data.video?.downloadResolutions.map((quality) => Number(quality.width)).filter(Boolean); // width property is actually the height

	const graphqlDetails = await fetchGraphqlDetails(release, channel, session);

	if (graphqlDetails) {
		release.tags = graphqlDetails.categories?.map((category) => category.name);
		release.chapters = graphqlDetails.chapters?.video?.map((chapter) => ({
			time: chapter.seconds,
			tags: [chapter.title],
		}));
	}

	release.channel = data.video?.id.split(':')[0];

	return release;
}

async function fetchActorReleases(pages, model, origin) {
	const releasesPerPage = await Promise.map(pages, async (page) => {
		const url = `${origin}/api${model.targetUrl}?page=${page}`;
		const res = await http.get(url);

		if (res.status === 200) {
			return scrapeAll(res.body.data.videos.videos, null, origin);
		}

		return [];
	}, { concurrency: 3 });

	return releasesPerPage.flat();
}

async function scrapeProfile(data, origin, withReleases) {
	const model = data.model;
	const profile = {};

	profile.birthdate = new Date(model.dateOfBirth);
	profile.gender = genderMap[model.sex];

	profile.hair = model.hairColour;
	profile.nationality = model.nationality;

	if (model.biography.trim().length > 0) profile.description = model.biography;

	if (model.cupSize && model.bustMeasurment) profile.bust = `${model.bustMeasurment}${model.cupSize}`;
	if (model.waistMeasurment) profile.waist = model.waistMeasurment;
	if (model.hipMeasurment) profile.hip = model.hipMeasurment;

	profile.avatar = getAvatarFallbacks(model.images.listing);
	profile.poster = getAvatarFallbacks(model.images.profile);
	profile.banner = getAvatarFallbacks(model.images.poster);

	const releases = scrapeAll(data.videos.videos, null, origin);

	if (withReleases) {
		const pageCount = Math.ceil(data.videos.count / 6);
		const otherReleases = await fetchActorReleases((Array.from({ length: pageCount - 1 }, (value, index) => index + 2)), model, origin);

		profile.releases = [...releases, ...otherReleases];
	} else {
		profile.releases = releases;
	}

	return profile;
}

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

	if (res.ok) {
		const dataString = res.item.query.html('#__NEXT_DATA__');
		const data = dataString && JSON.parse(dataString);

		if (data?.props.pageProps.edges) {
			return scrapeAll(data.props.pageProps.edges.map((edge) => edge.node), site);
		}

		return [];
	}

	return res.status;
}

async function fetchUpcoming(channel) {
	const query = `
		query getNextScene($site: Site!) {
		  nextScene: findNextReleaseVideo(input: { site: $site }) {
			videoId
			slug
			isPreReleasePeriod
			releaseDate
			models {
			  name
			  __typename
			}
			images {
			  countdown {
				...ImageInfo
				__typename
			  }
			  poster {
				...ImageInfo
				__typename
			  }
			  __typename
			}
			previews {
			  countdown {
				...PreviewInfo
				__typename
			  }
			  poster {
				...PreviewInfo
				__typename
			  }
			  __typename
			}
			__typename
		  }
		}

		fragment ImageInfo on Image {
		  src
		  placeholder
		  width
		  height
		  highdpi {
			double
			triple
			__typename
		  }
		  webp {
			src
			placeholder
			highdpi {
			  double
			  triple
			  __typename
			}
			__typename
		  }
		}

		fragment PreviewInfo on Preview {
		  src
		  width
		  height
		  type
		}
	`;

	const res = await http.post(`${channel.url}/graphql`, {
		operationName: 'getNextScene',
		query,
		variables: {
			site: channel.slug.toUpperCase(),
		},
	});

	if (res.ok) {
		if (res.body.data.nextScene) {
			return scrapeUpcoming(res.body.data.nextScene, channel);
		}

		return [];
	}

	return res.status;
}

async function fetchScene(url, channel, baseRelease, options) {
	const session = qu.session();
	const res = await qu.get(url, null, null, { session });

	if (res.ok) {
		const dataString = res.item.query.html('#__NEXT_DATA__');
		const data = dataString && JSON.parse(dataString);

		return scrapeScene(data.props.pageProps, url, channel, options, session);
	}

	return res.status;
}

async function fetchProfile({ name: actorName }, { site }, include) {
	const origin = site.url;
	const actorSlug = slugify(actorName);
	const url = `${origin}/api/${actorSlug}`;
	const res = await http.get(url);

	if (res.ok) {
		if (res.body.data) {
			return scrapeProfile(res.body.data, origin, include.scenes);
		}

		return null;
	}

	return res.status;
}

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