forked from DebaucheryLibrarian/traxxx
78 lines
2.7 KiB
JavaScript
78 lines
2.7 KiB
JavaScript
'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();
|