2019-11-16 02:33:36 +00:00
|
|
|
'use strict';
|
|
|
|
|
|
|
|
const config = require('config');
|
|
|
|
|
|
|
|
const argv = require('./argv');
|
|
|
|
const scrapers = require('./scrapers/scrapers');
|
2019-11-17 02:56:45 +00:00
|
|
|
const { storeReleases } = require('./releases');
|
2019-11-16 02:33:36 +00:00
|
|
|
const { findSiteByUrl } = require('./sites');
|
|
|
|
const { findNetworkByUrl } = require('./networks');
|
|
|
|
|
|
|
|
async function findSite(url, release) {
|
|
|
|
const site = (release && release.site) || await findSiteByUrl(url);
|
|
|
|
|
|
|
|
if (site) {
|
|
|
|
return site;
|
|
|
|
}
|
|
|
|
|
|
|
|
const network = await findNetworkByUrl(url);
|
|
|
|
|
|
|
|
if (network) {
|
|
|
|
return {
|
|
|
|
...network,
|
2019-11-16 22:37:33 +00:00
|
|
|
network,
|
2019-11-16 02:33:36 +00:00
|
|
|
isFallback: true,
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
return null;
|
|
|
|
}
|
|
|
|
|
2019-12-15 04:42:51 +00:00
|
|
|
async function scrapeRelease(url, release, deep = true, type = 'scene') {
|
2019-11-16 02:33:36 +00:00
|
|
|
const site = await findSite(url, release);
|
2019-12-06 23:42:47 +00:00
|
|
|
|
2019-11-16 02:33:36 +00:00
|
|
|
if (!site) {
|
|
|
|
throw new Error('Could not find site in database');
|
|
|
|
}
|
|
|
|
|
2019-12-06 23:42:47 +00:00
|
|
|
const scraper = scrapers.releases[site.slug] || scrapers.releases[site.network.slug];
|
|
|
|
|
2019-11-16 02:33:36 +00:00
|
|
|
if (!scraper) {
|
|
|
|
throw new Error('Could not find scraper for URL');
|
|
|
|
}
|
|
|
|
|
2019-12-15 04:42:51 +00:00
|
|
|
if (type === 'scene' && !scraper.fetchScene) {
|
2019-12-13 02:28:52 +00:00
|
|
|
throw new Error(`The '${site.name}'-scraper cannot fetch individual scenes`);
|
2019-11-16 02:33:36 +00:00
|
|
|
}
|
|
|
|
|
2019-12-15 04:42:51 +00:00
|
|
|
if (type === 'movie' && !scraper.fetchMovie) {
|
2019-12-13 02:28:52 +00:00
|
|
|
throw new Error(`The '${site.name}'-scraper cannot fetch individual movies`);
|
|
|
|
}
|
|
|
|
|
2019-12-15 04:42:51 +00:00
|
|
|
const scrapedRelease = type === 'scene'
|
|
|
|
? await scraper.fetchScene(url, site, release)
|
|
|
|
: await scraper.fetchMovie(url, site, release);
|
|
|
|
|
|
|
|
const curatedRelease = { ...scrapedRelease, type };
|
2019-11-16 02:33:36 +00:00
|
|
|
|
|
|
|
if (!deep && argv.save) {
|
|
|
|
// don't store release when called by site scraper
|
2019-12-15 04:42:51 +00:00
|
|
|
const movie = scrapedRelease.movie
|
|
|
|
? await scrapeRelease(scrapedRelease.movie, null, false, 'movie')
|
|
|
|
: null;
|
|
|
|
|
|
|
|
if (movie) {
|
|
|
|
const { releases: [storedMovie] } = await storeReleases([movie]);
|
|
|
|
curatedRelease.parentId = storedMovie.id;
|
|
|
|
}
|
|
|
|
|
|
|
|
const { releases: [storedRelease] } = await storeReleases([curatedRelease]);
|
2019-11-16 02:33:36 +00:00
|
|
|
|
2019-12-09 04:00:49 +00:00
|
|
|
if (storedRelease) {
|
|
|
|
console.log(`http://${config.web.host}:${config.web.port}/scene/${storedRelease.id}`);
|
|
|
|
}
|
2019-11-16 02:33:36 +00:00
|
|
|
}
|
|
|
|
|
2019-12-13 02:28:52 +00:00
|
|
|
return scrapedRelease;
|
2019-11-16 02:33:36 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
module.exports = scrapeRelease;
|