Added stash transfer tool.

2023-06-08 01:16:44 +02:00
parent f7708e0740
commit 56534800d8
6 changed files with 242 additions and 5 deletions
--- a/src/tools/stashes-load.js
+++ b/src/tools/stashes-load.js
@@ -0,0 +1,133 @@
+const fs = require('fs');
+const knex = require('../knex');
+const args = require('../argv');
+
+async function getStashId(stash, user) {
+	const existingStash = await knex('stashes')
+		.select('id')
+		.where('user_id', user.id)
+		.where((builder) => {
+			builder
+				.where('slug', stash.slug)
+				.orWhere('primary', stash.primary);
+		})
+		.first();
+
+	if (existingStash) {
+		return existingStash.id;
+	}
+
+	const [stashId] = await knex('stashes')
+		.insert({
+			user_id: user.id,
+			name: stash.name,
+			slug: stash.slug,
+			public: stash.public,
+			created_at: stash.createdAt,
+		})
+		.returning('id');
+
+	return stashId;
+}
+
+async function importReleases(type, stash, user, filename) {
+	const curatedType = type === 'release' ? 'scene' : type;
+
+	await stash[`${curatedType}s`].reduce(async (chain, scene) => {
+		await chain;
+
+		const release = await knex(`${type}s`)
+			.select(`${type}s.id`, 'entities.id as entity_id', 'entities.name as entity_name')
+			.leftJoin('entities', 'entities.id', `${type}s.entity_id`)
+			.where(`${type}s.entry_id`, scene.entryId)
+			.where('entities.slug', scene.entitySlug)
+			.where('entities.type', scene.entityType)
+			.first();
+
+		if (!release) {
+			throw new Error(`${curatedType.slice(0, 1).toUpperCase}${curatedType.slice(1)} ${scene.title} in ${scene.entityType} ${scene.entitySlug} does not exist`);
+		}
+
+		await knex(`stashes_${curatedType}s`)
+			.insert({
+				stash_id: stash.id,
+				[`${curatedType}_id`]: release.id,
+				comment: `import ${filename}`,
+				created_at: scene.createdAt,
+			})
+			.onConflict(['stash_id', `${curatedType}_id`])
+			.ignore();
+
+		console.log(`Imported ${stash.username} stash ${release.entity_name} ${curatedType} "${scene.title}"`);
+	}, Promise.resolve());
+}
+
+async function importActors(stash, user, filename) {
+	await stash.actors.reduce(async (chain, actor) => {
+		await chain;
+
+		const actorEntry = await knex('actors')
+			.select('actors.*')
+			.leftJoin('entities', 'entities.id', 'actors.entity_id')
+			.where('actors.slug', actor.slug)
+			.where((builder) => {
+				if (actor.entitySlug) {
+					builder
+						.where('entities.slug', actor.entitySlug)
+						.where('entities.type', actor.entityType);
+				}
+			})
+			.first();
+
+		if (!actorEntry) {
+			throw new Error(`Actor ${actor.slug} in ${user.username} stash ${stash.name} does not exist`);
+		}
+
+		await knex('stashes_actors')
+			.insert({
+				stash_id: stash.id,
+				actor_id: actorEntry.id,
+				comment: `import ${filename}`,
+				created_at: actor.createdAt,
+			})
+			.onConflict(['stash_id', 'actor_id'])
+			.ignore();
+
+		console.log(`Imported ${stash.username} stash actor "${actorEntry.name}"`);
+	}, Promise.resolve());
+}
+
+async function load() {
+	const filename = process.argv[2];
+	const file = await fs.promises.readFile(filename, 'utf8');
+
+	const stashes = file.split('\n')
+		.filter(Boolean)
+		.map((data) => JSON.parse(data))
+		.filter((stash) => !args.username || stash.username === args.username);
+
+	await stashes.reduce(async (stashChain, stash, index) => {
+		await stashChain;
+
+		const user = await knex('users')
+			.select('id')
+			.where('username', stash.username)
+			.first();
+
+		if (!user) {
+			throw new Error(`No user '${stash.username}'`);
+		}
+
+		const stashId = await getStashId(stash, user);
+
+		await importReleases('release', { ...stash, id: stashId }, user, filename);
+		await importReleases('movie', { ...stash, id: stashId }, user, filename);
+		await importActors({ ...stash, id: stashId }, user, filename);
+
+		console.log(`Imported ${index + 1}/${stashes.length} stash ${stash.name} from ${stash.username}`);
+	}, Promise.resolve());
+
+	process.exit();
+}
+
+load();