traxxx/assets/js/curate.js

import dayjs from 'dayjs';

function curateActor(actor, release) {
	if (!actor) {
		return null;
	}

	const curatedActor = {
		...actor,
		height: actor.heightMetric && {
			metric: actor.heightMetric,
			imperial: actor.heightImperial,
		},
		weight: actor.weightMetric && {
			metric: actor.weightMetric,
			imperial: actor.weightImperial,
		},
		origin: actor.birthCountry && {
			city: actor.birthCity,
			state: actor.birthState,
			country: actor.birthCountry,
		},
		residence: actor.residenceCountry && {
			city: actor.residenceCity,
			state: actor.residenceState,
			country: actor.residenceCountry,
		},
		scrapedAt: new Date(actor.createdAt),
		updatedAt: new Date(actor.updatedAt),
	};

	if (actor.profiles) {
		const photos = actor.profiles
			.map((profile) => ({ entity: profile.entity, ...profile.avatar }))
			.filter((avatar) => avatar.id && (!curatedActor.avatar || avatar.hash !== curatedActor.avatar.hash));

		const descriptions = actor.profiles.reduce((acc, profile) => ({
			...acc,
			...(profile.description && {
				[profile.descriptionHash]: {
					text: profile.description,
					entity: profile.entity,
				},
			}),
		}), {});

		curatedActor.photos = Object.values(photos.reduce((acc, photo) => ({ ...acc, [photo.hash]: photo }), {}));
		curatedActor.descriptions = Object.values(descriptions);
	}

	if (release && release.date && curatedActor.birthdate) {
		curatedActor.ageThen = dayjs(release.date).diff(actor.birthdate, 'year');
	}

	if (actor.aliasFor) {
		curatedActor.aliasFor = curateActor(curatedActor.aliasFor);
	}

	if (actor.stashes) {
		curatedActor.stashes = actor.stashes.filter(Boolean).map((stash) => curateStash(stash.stash || stash)); // eslint-disable-line no-use-before-define
	}

	curatedActor.stashes = actor.stashes?.map((stash) => stash.stash || stash) || [];

	return curatedActor;
}

function curateRelease(release, type = 'scene') {
	const curatedRelease = {
		...release,
		type: release.type || type,
		actors: [],
		poster: release.poster && release.poster.media,
		tags: release.tags ? release.tags.map((tag) => tag.tag || tag) : [],
	};

	if (release.scenes) curatedRelease.scenes = release.scenes.filter(Boolean).map(({ scene }) => curateRelease(scene, 'scene'));
	if (release.movies) curatedRelease.movies = release.movies.filter(Boolean).map(({ movie }) => curateRelease(movie, 'movie'));
	if (release.series) curatedRelease.series = release.series.filter(Boolean).map(({ serie }) => curateRelease(serie, 'serie'));
	if (release.chapters) curatedRelease.chapters = release.chapters.filter(Boolean).map((chapter) => curateRelease(chapter));
	if (release.photos) curatedRelease.photos = release.photos.filter(Boolean).map((photo) => photo.media || photo);
	if (release.covers) curatedRelease.covers = release.covers.filter(Boolean).map(({ media }) => media);
	if (release.trailer) curatedRelease.trailer = release.trailer.media;
	if (release.teaser) curatedRelease.teaser = release.teaser.media;
	if (release.actors) curatedRelease.actors = release.actors.filter(Boolean).map((actor) => curateActor(actor.actor || actor, curatedRelease));
	if (release.directors) curatedRelease.directors = release.directors.filter(Boolean).map((director) => curateActor(director.director || director, curatedRelease));
	if (release.movieTags && release.movieTags.length > 0) curatedRelease.tags = release.movieTags.filter(Boolean).map(({ tag }) => tag);
	if (release.movieActors && release.movieActors.length > 0) curatedRelease.actors = release.movieActors.filter(Boolean).map(({ actor }) => curateActor(actor, curatedRelease));
	if (release.stashes) curatedRelease.stashes = release.stashes.filter(Boolean).map((stash) => curateStash(stash.stash || stash)); // eslint-disable-line no-use-before-define

	if (release.productionLocation) {
		curatedRelease.productionLocation = {
			raw: release.productionLocation,
			city: release.productionCity,
			state: release.productionState,
			country: release.productionCountry,
		};
	}

	return curatedRelease;
}

function curateEntity(entity, parent, releases) {
	const curatedEntity = {
		...entity,
		children: [],
	};

	if (entity.tags) curatedEntity.tags = entity.tags.map(({ tag }) => tag);
	if (entity.sceneTags) curatedEntity.sceneTags = entity.sceneTags;

	if (entity.children) {
		if (entity.children.nodes) {
			curatedEntity.children = entity.children.nodes.map((childEntity) => curateEntity(childEntity, curatedEntity));
		}

		curatedEntity.childrenTotal = entity.children.totalCount;
	}

	if (entity.parent || parent) curatedEntity.parent = curateEntity(entity.parent || parent);
	if (releases) curatedEntity.releases = releases.map((release) => curateRelease(release));

	curatedEntity.sceneTotal = entity.sceneTotal;

	return curatedEntity;
}

function curateTag(tag) {
	const curatedTag = {
		...tag,
	};

	if (tag.releases) curatedTag.releases = tag.releases.map(({ release }) => curateRelease(release));
	if (tag.banners) curatedTag.banners = tag.banners.map(({ banner }) => banner);
	if (tag.photos) curatedTag.photos = tag.photos.map(({ media }) => media);
	if (tag.poster) curatedTag.poster = tag.poster.media;

	return curatedTag;
}

function curateStash(stash) {
	const curatedStash = stash;

	if (stash.scenes || stash.scenesConnection?.scenes) {
		curatedStash.sceneTotal = stash.scenesConnection?.totalCount || null;
		curatedStash.scenes = (stash.scenesConnection?.scenes || stash.scenes).map((item) => ({
			...item,
			scene: curateRelease(item.scene),
		}));
	}

	if (stash.actors || stash.actorsConnection?.actors) {
		curatedStash.actorTotal = stash.actorsConnection?.totalCount || null;
		curatedStash.actors = (stash.actorsConnection?.actors || stash.actors).map((item) => ({
			...item,
			actor: curateActor(item.actor),
		}));
	}

	if (stash.movies || stash.moviesConnection?.movies) {
		curatedStash.movieTotal = stash.moviesConnection?.totalCount || null;
		curatedStash.movies = (stash.moviesConnection?.movies || stash.movies).map((item) => ({
			...item,
			movie: curateRelease(item.movie),
		}));
	}

	return curatedStash;
}

function curateAlert(alert) {
	if (!alert) {
		return null;
	}

	const curatedAlert = alert;

	if (alert.actors) {
		curatedAlert.actors = alert.actors.map((actor) => curateActor(actor.actor || actor));
	}

	if (alert.tags) {
		curatedAlert.tags = alert.tags.map((tag) => curateTag(tag.tag || tag));
	}

	if (alert.entity) {
		curatedAlert.entity = curateEntity(alert.entity.entity || alert.entity);
	}

	if (alert.stashes) {
		curatedAlert.stashes = alert.stashes.map((stash) => curateStash(stash.stash || stash));
	}

	return curatedAlert;
}

function curateUser(user) {
	if (!user) {
		return null;
	}

	const curatedUser = user;

	if (user.stashes) {
		curatedUser.stashes = user.stashes.map((stash) => curateStash(stash.stash || stash));
	}

	if (user.alerts) {
		curatedUser.alerts = user.alerts.map((alert) => curateAlert(alert.alert || alert));
	}

	return curatedUser;
}

function curateNotification(notification) {
	const curatedNotification = notification;

	curatedNotification.scene = curateRelease(notification.scene);

	if (notification.alert) {
		curatedNotification.alert = curateAlert(notification.alert.alert || notification.alert);
	}

	return curatedNotification;
}

export {
	curateActor,
	curateEntity,
	curateRelease,
	curateNotification,
	curateTag,
	curateStash,
	curateUser,
};