traxxx/src/utils/posters.js

78 lines
2.7 KiB
JavaScript
Executable File

'use strict';
const config = require('config');
const path = require('path');
const fs = require('fs-extra');
const moment = require('moment');
const argv = require('../argv');
const knex = require('../knex');
async function actorPosters(actorNames) {
const posters = await knex('actors')
.select('actors.name as actor_name', 'releases.title', 'releases.date', 'media.path', 'media.index', 'sites.name as site_name', 'networks.name as network_name')
.whereIn('actors.name', actorNames)
.join('releases_actors', 'releases_actors.actor_id', 'actors.id')
.join('releases', 'releases_actors.release_id', 'releases.id')
.join('sites', 'sites.id', 'releases.site_id')
.join('networks', 'networks.id', 'sites.network_id')
.join('releases_posters', 'releases_posters.release_id', 'releases.id')
.join('media', 'releases_posters.media_id', 'media.id');
// .join('releases_photos', 'releases_photos.release_id', 'releases.id')
// .join('media', 'releases_photos.media_id', 'media.id');
await Promise.all(posters.map(async (poster) => {
const source = path.join(config.media.path, poster.path);
const directory = path.join(config.media.path, 'extracted', poster.actor_name);
const target = path.join(directory, `${poster.actor_name} - ${poster.network_name}: ${poster.site_name} - ${poster.title.replace(/[/.]/g, '_') || poster.actor_name} (${moment.utc(poster.date).format('YYYY-MM-DD')})-${poster.index}.jpeg`);
await fs.mkdir(path.join(directory), { recursive: true });
const file = await fs.readFile(source);
await fs.writeFile(target, file);
return file;
}));
knex.destroy();
}
async function sitePosters(siteSlugs) {
const posters = await knex('sites')
.select('sites.name as site_name', 'releases.title', 'releases.date', 'media.path')
.whereIn('sites.slug', siteSlugs)
.join('releases', 'releases.site_id', 'sites.id')
.join('releases_posters', 'releases_posters.release_id', 'releases.id')
.join('media', 'releases_posters.media_id', 'media.id');
// .where('releases.date', '<', '2020-01-01');
const files = await Promise.all(posters.map(async (poster) => {
const directory = path.join(config.media.path, 'extracted', poster.site_name);
const source = path.join(config.media.path, poster.path);
const target = path.join(directory, `${poster.site_name} - ${moment.utc(poster.date).format('YYYY-MM-DD')} - ${poster.title.replace(/[/.]/g, '_')}.jpeg`);
await fs.mkdir(directory, { recursive: true });
await fs.copyFile(source, target);
return target;
}));
console.log(files);
knex.destroy();
}
async function init() {
if (argv.actors) {
await actorPosters(argv.actors);
return;
}
if (argv.sites) {
await sitePosters(argv.sites);
}
}
init();