Compare commits
21 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e9a376925d | |||
| b166403309 | |||
| 2046e5080b | |||
| 27a6656ac5 | |||
| fd5ef14999 | |||
| b30fe748b2 | |||
| 331298825e | |||
| 4719117504 | |||
| c0c33b8859 | |||
| 1922cf5b87 | |||
| fa5b7a3249 | |||
| 60f65152cb | |||
| d34b74e505 | |||
| 600173c959 | |||
| 49fd73aaf9 | |||
| ffa0f07daf | |||
| dc9f73961e | |||
| bbcd955f28 | |||
| c6cc529d54 | |||
| a4094cb795 | |||
| 1c7e2e794e |
@@ -17,12 +17,12 @@ const app = express();
|
||||
|
||||
// List of front end node modules to be served
|
||||
const frontEndModules = [
|
||||
'jquery', 'jquery-ui', 'moment', 'mustache',
|
||||
'jquery', 'jquery-ui', 'moment', 'mustache', 'jq-repeat',
|
||||
// 'bootstrap', '@fortawesome',
|
||||
];
|
||||
|
||||
// Hold list of functions to run when the server is ready
|
||||
app.onListen = [function(){console.log('hello')}];
|
||||
app.onListen = [];
|
||||
|
||||
// Allow the express app to be exported into other files.
|
||||
module.exports = app;
|
||||
@@ -104,7 +104,7 @@ app.use(function(err, req, res, next) {
|
||||
err.status = 422;
|
||||
}
|
||||
|
||||
if(![404, 422].includes(err.status || res.status)){
|
||||
if(![404, 422].includes(err.status || res.statusCode)){
|
||||
console.error(err.message);
|
||||
console.error(err.stack);
|
||||
console.error('=========================================');
|
||||
|
||||
@@ -1,6 +1,7 @@
|
||||
'use strict';
|
||||
|
||||
module.exports = {
|
||||
privateDownloadLocation: '/media/torrents-zfs',
|
||||
ldap: {
|
||||
url: 'ldap://10.1.0.55:389',
|
||||
bindDN: 'cn=ldapclient service,ou=people,dc=theta42,dc=com',
|
||||
@@ -21,5 +22,44 @@ module.exports = {
|
||||
port: 9091,
|
||||
username: 'william',
|
||||
password: '__IN SRECREST FILE__',
|
||||
statusUpdateInterval: 500,
|
||||
},
|
||||
tmdb: {
|
||||
apiKey: '__IN SRECREST FILE__',
|
||||
},
|
||||
ollama: {
|
||||
url: 'https://ollama.com',
|
||||
apiKey: '__IN SRECREST FILE__',
|
||||
model: 'gemma4:31b-cloud',
|
||||
},
|
||||
search: {
|
||||
// TPB HTML mirror we scrape search results from( same host the proxy targets).
|
||||
// apibay's JSON q.php is Cloudflare-gated, so we parse the HTML listing instead.
|
||||
tpbBase: 'https://piratebay.party',
|
||||
trackers: [
|
||||
'udp://tracker.opentrackr.org:1337/announce',
|
||||
'udp://open.stealth.si:80/announce',
|
||||
'udp://tracker.torrent.eu.org:451/announce',
|
||||
'udp://tracker.openbittorrent.com:6969/announce',
|
||||
'udp://tracker.coppersurfer.tk:6969/announce',
|
||||
'udp://open.demonii.com:1337/announce',
|
||||
'udp://exodus.desync.com:6969/announce',
|
||||
],
|
||||
},
|
||||
emby: {
|
||||
url: 'https://emby.718it.biz',
|
||||
apiKey: '__IN SRECREST FILE__',
|
||||
},
|
||||
library: {
|
||||
root: '/media/stuff',
|
||||
movie: 'movies',
|
||||
tv: 'tv',
|
||||
},
|
||||
organize: {
|
||||
includePrivate: false,
|
||||
interval: 10000,
|
||||
},
|
||||
wanted: {
|
||||
interval: 21600000, // re-check the wishlist every 6h
|
||||
}
|
||||
};
|
||||
|
||||
@@ -0,0 +1,4 @@
|
||||
'use strict';
|
||||
|
||||
module.exports = {
|
||||
};
|
||||
@@ -0,0 +1,19 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
|
||||
// Ask Emby to (re)scan its libraries so a newly filed item gets indexed.
|
||||
async function refresh(){
|
||||
let res = await fetch(`${conf.emby.url}/Library/Refresh?api_key=${conf.emby.apiKey}`, {
|
||||
method: 'POST',
|
||||
});
|
||||
if(!res.ok){
|
||||
let error = new Error('EmbyError');
|
||||
error.message = `Emby library refresh failed( ${res.status})`;
|
||||
error.status = 502;
|
||||
throw error;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
module.exports = { refresh };
|
||||
@@ -4,4 +4,6 @@ module.exports = {
|
||||
auth: require('./auth'),
|
||||
pubsub: require('./pubsub'),
|
||||
torrent: require('./torrent'),
|
||||
organizeWatcher: require('./organizeWatcher'),
|
||||
wantedWatcher: require('./wantedWatcher'),
|
||||
}
|
||||
@@ -0,0 +1,69 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
|
||||
// Resolution rank so we can compare "what you own" against a candidate release.
|
||||
function rankFromWidth(w){
|
||||
w = Number(w) || 0;
|
||||
if(w >= 3000) return 4; // 4k
|
||||
if(w >= 1700) return 3; // 1080p
|
||||
if(w >= 1000) return 2; // 720p
|
||||
if(w > 0) return 1; // sd
|
||||
return 0;
|
||||
}
|
||||
|
||||
function rankFromLabel(label){
|
||||
label = String(label || '').toLowerCase();
|
||||
if(/2160|4k|uhd/.test(label)) return 4;
|
||||
if(label.includes('1080')) return 3;
|
||||
if(label.includes('720')) return 2;
|
||||
if(label.includes('480')) return 1;
|
||||
return 0;
|
||||
}
|
||||
|
||||
function labelFromRank(rank){ return ['', '480p', '720p', '1080p', '4k'][rank] || ''; }
|
||||
|
||||
async function embyGet(pathAndQuery){
|
||||
let sep = pathAndQuery.includes('?') ? '&' : '?';
|
||||
let res = await fetch(`${conf.emby.url}${pathAndQuery}${sep}api_key=${conf.emby.apiKey}`);
|
||||
if(!res.ok) throw new Error(`Emby ${res.status}`);
|
||||
return await res.json();
|
||||
}
|
||||
|
||||
// Is this movie already in the library, and at what best quality? Never throws — an
|
||||
// unreachable Emby just means "unknown", so search still works.
|
||||
async function movieInLibrary(tmdbId){
|
||||
try{
|
||||
let data = await embyGet(`/Items?Recursive=true&IncludeItemTypes=Movie&AnyProviderIdEquals=tmdb.${tmdbId}&Fields=Width`);
|
||||
let items = data.Items || [];
|
||||
if(!items.length) return { owned: false, rank: 0, quality: '' };
|
||||
let rank = Math.max(...items.map(i => rankFromWidth(i.Width)));
|
||||
return { owned: true, rank, quality: labelFromRank(rank), count: items.length };
|
||||
}catch(error){
|
||||
return { owned: false, rank: 0, quality: '', unknown: true };
|
||||
}
|
||||
}
|
||||
|
||||
// Which episodes of a series are already present, grouped by season( season -> count).
|
||||
async function seriesInventory(tmdbId){
|
||||
try{
|
||||
let s = await embyGet(`/Items?Recursive=true&IncludeItemTypes=Series&AnyProviderIdEquals=tmdb.${tmdbId}&Fields=ProviderIds`);
|
||||
let series = (s.Items || [])[0];
|
||||
if(!series) return { owned: false, have: {} };
|
||||
|
||||
let ep = await embyGet(`/Shows/${series.Id}/Episodes`);
|
||||
let sets = {};
|
||||
for(let e of ep.Items || []){
|
||||
let sn = e.ParentIndexNumber, en = e.IndexNumber;
|
||||
if(sn == null || en == null || sn === 0) continue; // skip specials
|
||||
(sets[sn] = sets[sn] || new Set()).add(en);
|
||||
}
|
||||
let have = {};
|
||||
for(let sn of Object.keys(sets)) have[sn] = sets[sn].size;
|
||||
return { owned: true, seriesId: series.Id, have };
|
||||
}catch(error){
|
||||
return { owned: false, have: {}, unknown: true };
|
||||
}
|
||||
}
|
||||
|
||||
module.exports = { movieInLibrary, seriesInventory, rankFromWidth, rankFromLabel, labelFromRank };
|
||||
@@ -0,0 +1,42 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
|
||||
// Pull the first JSON object out of a model response( tolerates ```json fences / prose).
|
||||
function extractJSON(content){
|
||||
let text = String(content).replace(/```json/gi, '').replace(/```/g, '').trim();
|
||||
let start = text.indexOf('{');
|
||||
let end = text.lastIndexOf('}');
|
||||
if(start === -1 || end === -1) throw new Error('No JSON in model response');
|
||||
return JSON.parse(text.slice(start, end + 1));
|
||||
}
|
||||
|
||||
// One-shot JSON chat against the Ollama cloud model.
|
||||
async function chatJSON(system, user){
|
||||
let res = await fetch(`${conf.ollama.url}/api/chat`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': `Bearer ${conf.ollama.apiKey}`,
|
||||
},
|
||||
body: JSON.stringify({
|
||||
model: conf.ollama.model,
|
||||
stream: false,
|
||||
format: 'json',
|
||||
messages: [
|
||||
{ role: 'system', content: system },
|
||||
{ role: 'user', content: user },
|
||||
],
|
||||
}),
|
||||
});
|
||||
if(!res.ok){
|
||||
let error = new Error('OllamaError');
|
||||
error.message = `Ollama chat failed( ${res.status})`;
|
||||
error.status = 502;
|
||||
throw error;
|
||||
}
|
||||
let data = await res.json();
|
||||
return extractJSON(data.message && data.message.content);
|
||||
}
|
||||
|
||||
module.exports = { chatJSON, extractJSON };
|
||||
@@ -0,0 +1,286 @@
|
||||
'use strict';
|
||||
|
||||
const fs = require('fs');
|
||||
const path = require('path');
|
||||
const conf = require('>/conf');
|
||||
const ollama = require('>/controller/ollama');
|
||||
const tmdb = require('>/controller/tmdb');
|
||||
const emby = require('>/controller/emby');
|
||||
const ps = require('>/controller/pubsub');
|
||||
|
||||
const VIDEO_EXT = new Set(['.mkv', '.mp4', '.avi', '.m4v', '.ts', '.wmv', '.mov']);
|
||||
const SUB_EXT = new Set(['.srt', '.ass', '.ssa', '.sub', '.vtt', '.idx']);
|
||||
|
||||
function ext(name){ return path.extname(name).toLowerCase(); }
|
||||
function isVideo(name){ return VIDEO_EXT.has(ext(name)) && !/sample/i.test(name); }
|
||||
function isSub(name){ return SUB_EXT.has(ext(name)); }
|
||||
|
||||
// Strip characters that are illegal / awkward in file paths.
|
||||
function sanitize(text){
|
||||
return String(text || '').replace(/[\/\\:*?"<>|]/g, '').replace(/\s+/g, ' ').trim();
|
||||
}
|
||||
|
||||
function pad2(n){ return String(n).padStart(2, '0'); }
|
||||
|
||||
// Normalize resolution to the library's convention( 2160p/uhd -> 4k).
|
||||
function normalizeRes(res){
|
||||
res = String(res || '').toLowerCase();
|
||||
if(/2160|4k|uhd/.test(res)) return '4k';
|
||||
let m = res.match(/(480|720|1080)p?/);
|
||||
return m ? `${m[1]}p` : '';
|
||||
}
|
||||
|
||||
// "Unrated 1080p" style suffix from edition + quality.
|
||||
function qualitySuffix(meta){
|
||||
let res = normalizeRes(meta.quality && meta.quality.resolution);
|
||||
return [sanitize(meta.edition), res].filter(Boolean).join(' ');
|
||||
}
|
||||
|
||||
// --- LLM metadata extraction ----------------------------------------------
|
||||
|
||||
function buildExtractPrompt(today){
|
||||
return [
|
||||
`You are a media library organizer. Today's date is ${today}.`,
|
||||
`Given a torrent's raw release name, its coarse type (Movie or TV), and its list of`,
|
||||
`video file names, extract clean metadata for filing. Return STRICT JSON only:`,
|
||||
`{"mediaType":"movie"|"tv","title":"clean title, no year/quality/tags",`,
|
||||
`"year":"YYYY" or "","edition":"Extended|Director's Cut|Unrated|Uncut|Remastered|`,
|
||||
`Theatrical|IMAX|Redux|" (notable edition if present, else empty),`,
|
||||
`"quality":{"resolution":"2160p|1080p|720p|480p|","codec":"x265|x264|"},`,
|
||||
`"episodes":[{"file":"<exact name from the list>","season":<int>,"episode":<int>}]}`,
|
||||
`Movies: "episodes" is []. TV: one entry per video file, mapping each file to its`,
|
||||
`season/episode inferred from the file name (SxxExx, 1x02, etc.).`,
|
||||
`Do not invent data; leave a field empty or [] when unknown.`,
|
||||
].join('\n');
|
||||
}
|
||||
|
||||
async function llmExtract({ name, category, files }){
|
||||
let today = new Date().toISOString().slice(0, 10);
|
||||
let user = JSON.stringify({ name, type: category, files });
|
||||
return await ollama.chatJSON(buildExtractPrompt(today), user);
|
||||
}
|
||||
|
||||
// Parse + canonicalize. Returns a metadata object, or one carrying `organizeError`
|
||||
// (so the caller flags the torrent instead of mis-filing it).
|
||||
async function resolveMeta({ name, category, files }){
|
||||
let llm;
|
||||
try{
|
||||
llm = await llmExtract({ name, category, files });
|
||||
}catch(error){
|
||||
return { organizeError: `parse failed: ${error.message}` };
|
||||
}
|
||||
|
||||
let mediaType = llm.mediaType === 'tv' || llm.mediaType === 'movie'
|
||||
? llm.mediaType
|
||||
: (category === 'TV' ? 'tv' : 'movie');
|
||||
|
||||
if(!llm.title) return { organizeError: `could not determine a title for "${name}"` };
|
||||
|
||||
let best = await tmdb.tmdbFindBest(llm.title, llm.year, mediaType);
|
||||
if(!best) return { organizeError: `no TMDB match for "${llm.title}" (${llm.year || '?'})` };
|
||||
|
||||
return {
|
||||
mediaType,
|
||||
title: best.title,
|
||||
year: best.year,
|
||||
edition: llm.edition || '',
|
||||
quality: llm.quality || {},
|
||||
episodes: Array.isArray(llm.episodes) ? llm.episodes : [],
|
||||
tmdbId: best.tmdbId,
|
||||
imdbId: best.imdbId,
|
||||
};
|
||||
}
|
||||
|
||||
// --- Target path building --------------------------------------------------
|
||||
|
||||
// Build the destination path for one video file. `episode` is the matching entry from
|
||||
// meta.episodes (TV only).
|
||||
function buildTarget(meta, fileName, episode){
|
||||
let title = sanitize(meta.title);
|
||||
let folder = `${title} (${meta.year})`;
|
||||
let suffix = qualitySuffix(meta);
|
||||
let e = ext(fileName);
|
||||
|
||||
if(meta.mediaType === 'tv' && episode){
|
||||
let dir = path.join(conf.library.root, conf.library.tv, folder, `Season ${pad2(episode.season)}`);
|
||||
let se = `S${pad2(episode.season)}E${pad2(episode.episode)}`;
|
||||
let base = `${folder} - ${se}${suffix ? ' - ' + suffix : ''}${e}`;
|
||||
return path.join(dir, base);
|
||||
}
|
||||
|
||||
let dir = path.join(conf.library.root, conf.library.movie, folder);
|
||||
let base = `${folder}${suffix ? ' - ' + suffix : ''}${e}`;
|
||||
return path.join(dir, base);
|
||||
}
|
||||
|
||||
// If the exact target exists, disambiguate with codec then a counter.
|
||||
function resolveCollision(target, meta){
|
||||
if(!fs.existsSync(target)) return target;
|
||||
let dir = path.dirname(target);
|
||||
let e = path.extname(target);
|
||||
let baseNoExt = path.basename(target, e);
|
||||
let codec = meta.quality && meta.quality.codec ? ' ' + sanitize(meta.quality.codec) : '';
|
||||
|
||||
let candidate = path.join(dir, `${baseNoExt}${codec}${e}`);
|
||||
let n = 2;
|
||||
while(fs.existsSync(candidate)){
|
||||
candidate = path.join(dir, `${baseNoExt}${codec} (${n})${e}`);
|
||||
n++;
|
||||
}
|
||||
return candidate;
|
||||
}
|
||||
|
||||
// --- Filesystem move -------------------------------------------------------
|
||||
|
||||
async function moveFile(src, dst){
|
||||
await fs.promises.mkdir(path.dirname(dst), { recursive: true });
|
||||
try{
|
||||
await fs.promises.rename(src, dst);
|
||||
}catch(error){
|
||||
if(error.code === 'EXDEV'){
|
||||
await fs.promises.copyFile(src, dst);
|
||||
await fs.promises.unlink(src);
|
||||
}else{
|
||||
throw error;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// Subs that belong to a given video( same basename, ignoring a language suffix).
|
||||
function subsFor(videoName, subFiles, isMovie){
|
||||
if(isMovie) return subFiles;
|
||||
let stem = path.basename(videoName, ext(videoName)).toLowerCase();
|
||||
return subFiles.filter(s => path.basename(s, ext(s)).toLowerCase().startsWith(stem.slice(0, 20)));
|
||||
}
|
||||
|
||||
// --- Orchestration ---------------------------------------------------------
|
||||
|
||||
// File a finished torrent into the library, then remove it from Transmission and refresh
|
||||
// Emby. Returns { organized:true, ... } or { organized:false, error }.
|
||||
async function fileTorrent(torrent){
|
||||
const { Torrent } = require('>/models');
|
||||
|
||||
let info = (await Torrent.trClient.get(torrent.hashString, ['downloadDir', 'files', 'name'])).torrents[0];
|
||||
if(!info) return await flag(torrent, 'torrent not found in Transmission');
|
||||
|
||||
let downloadDir = info.downloadDir;
|
||||
let videoFiles = info.files.filter(f => isVideo(f.name));
|
||||
let subFiles = info.files.filter(f => isSub(f.name)).map(f => f.name);
|
||||
if(!videoFiles.length) return await flag(torrent, 'no video files in torrent');
|
||||
|
||||
let meta = await resolveMeta({
|
||||
name: torrent.name,
|
||||
category: torrent.category,
|
||||
files: videoFiles.map(f => f.name),
|
||||
});
|
||||
if(meta.organizeError) return await flag(torrent, meta.organizeError, meta);
|
||||
|
||||
let isMovie = meta.mediaType !== 'tv';
|
||||
// Movie: file only the largest video. TV: file every episode video.
|
||||
let toFile = isMovie
|
||||
? [videoFiles.reduce((a, b) => (b.length > a.length ? b : a))]
|
||||
: videoFiles;
|
||||
|
||||
let filed = [];
|
||||
let libraryPath = null;
|
||||
for(let vf of toFile){
|
||||
let episode = isMovie ? null : (meta.episodes.find(e => e.file === vf.name) || null);
|
||||
if(!isMovie && !episode) continue; // can't place an episode we couldn't map
|
||||
|
||||
let dst = resolveCollision(buildTarget(meta, vf.name, episode), meta);
|
||||
await moveFile(path.join(downloadDir, vf.name), dst);
|
||||
filed.push({ path: dst, season: episode ? episode.season : null, episode: episode ? episode.episode : null });
|
||||
libraryPath = path.dirname(dst);
|
||||
|
||||
// carry matching sidecar subtitles, renamed to the video's base name
|
||||
let dstBase = path.basename(dst, ext(dst));
|
||||
for(let sub of subsFor(vf.name, subFiles, isMovie)){
|
||||
try{
|
||||
await moveFile(path.join(downloadDir, sub), path.join(path.dirname(dst), dstBase + ext(sub)));
|
||||
}catch(error){ /* subs are best-effort */ }
|
||||
}
|
||||
}
|
||||
|
||||
if(!filed.length) return await flag(torrent, 'could not map any video files to episodes', meta);
|
||||
|
||||
// Move + stop seeding: drop the torrent and delete whatever download data is left.
|
||||
try{ await Torrent.trClient.remove(torrent.hashString, true); }catch(error){ /* already gone */ }
|
||||
|
||||
torrent.organizedAt = new Date();
|
||||
torrent.metadata = { ...meta, libraryPath, files: filed };
|
||||
await torrent.save();
|
||||
|
||||
try{ await emby.refresh(); }catch(error){ console.error('emby refresh failed:', error.message); }
|
||||
ps.publish('torrent:organized', { hashString: torrent.hashString, libraryPath });
|
||||
|
||||
return { organized: true, libraryPath, files: filed };
|
||||
}
|
||||
|
||||
// Re-file an already-organized torrent under a user-chosen TMDB match. Moves the files
|
||||
// already in the library to their new canonical paths.
|
||||
async function refileWithMatch(torrent, tmdbId, mediaType){
|
||||
if(!torrent.metadata || !Array.isArray(torrent.metadata.files) || !torrent.metadata.files.length){
|
||||
let error = new Error('NothingToRefile');
|
||||
error.message = 'This torrent has no filed media to re-match';
|
||||
error.status = 409;
|
||||
throw error;
|
||||
}
|
||||
|
||||
let details = await tmdb.tmdbDetails(tmdbId, mediaType);
|
||||
let meta = {
|
||||
...torrent.metadata,
|
||||
mediaType,
|
||||
title: details.title,
|
||||
year: details.year,
|
||||
tmdbId,
|
||||
imdbId: details.imdbId,
|
||||
};
|
||||
delete meta.organizeError;
|
||||
|
||||
let filed = [];
|
||||
let libraryPath = null;
|
||||
for(let entry of torrent.metadata.files){
|
||||
let old = typeof entry === 'string' ? entry : entry.path;
|
||||
let episode = (entry && entry.season != null) ? { season: entry.season, episode: entry.episode } : null;
|
||||
let dst = resolveCollision(buildTarget(meta, old, episode), meta);
|
||||
if(dst !== old) await moveFile(old, dst);
|
||||
filed.push({ path: dst, season: episode ? episode.season : null, episode: episode ? episode.episode : null });
|
||||
libraryPath = path.dirname(dst);
|
||||
|
||||
let oldDir = path.dirname(old);
|
||||
try{ if(oldDir !== libraryPath) await fs.promises.rmdir(oldDir); }catch(error){ /* not empty / gone */ }
|
||||
}
|
||||
|
||||
meta.files = filed;
|
||||
meta.libraryPath = libraryPath;
|
||||
torrent.metadata = meta;
|
||||
torrent.organizedAt = new Date();
|
||||
await torrent.save();
|
||||
|
||||
try{ await emby.refresh(); }catch(error){ console.error('emby refresh failed:', error.message); }
|
||||
ps.publish('torrent:organized', { hashString: torrent.hashString, libraryPath });
|
||||
|
||||
return { organized: true, libraryPath, files: filed };
|
||||
}
|
||||
|
||||
// Record a problem in metadata without setting organizedAt( leaves it flagged for a
|
||||
// manual "fix match").
|
||||
async function flag(torrent, message, meta){
|
||||
console.error(`organize: ${torrent.hashString} skipped — ${message}`);
|
||||
torrent.metadata = { ...(meta || {}), organizeError: message };
|
||||
await torrent.save();
|
||||
return { organized: false, error: message };
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
fileTorrent,
|
||||
refileWithMatch,
|
||||
resolveMeta,
|
||||
llmExtract,
|
||||
buildTarget,
|
||||
qualitySuffix,
|
||||
normalizeRes,
|
||||
sanitize,
|
||||
moveFile,
|
||||
resolveCollision,
|
||||
};
|
||||
@@ -0,0 +1,42 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
const organize = require('>/controller/organize');
|
||||
|
||||
let lock = false;
|
||||
|
||||
// Poll unorganized Movie/TV downloads; when Transmission reports one finished, file it.
|
||||
async function tick(){
|
||||
if(lock) return;
|
||||
lock = true;
|
||||
try{
|
||||
const { Torrent } = require('>/models');
|
||||
|
||||
let where = { organizedAt: null, category: ['Movie', 'TV'] };
|
||||
if(!conf.organize.includePrivate) where.isPrivate = false;
|
||||
|
||||
let pending = await Torrent.findAll({ where });
|
||||
if(!pending.length) return;
|
||||
|
||||
let byHash = {};
|
||||
for(let t of pending) byHash[t.hashString.toLowerCase()] = t;
|
||||
|
||||
let res = await Torrent.trClient.get(pending.map(t => t.hashString), ['hashString', 'percentDone', 'isFinished']);
|
||||
for(let tor of res.torrents){
|
||||
if(!(tor.isFinished || tor.percentDone === 1)) continue;
|
||||
let t = byHash[String(tor.hashString).toLowerCase()];
|
||||
if(!t) continue;
|
||||
if(t.metadata && t.metadata.organizeError) continue; // already tried; needs manual fix
|
||||
await organize.fileTorrent(t);
|
||||
}
|
||||
}catch(error){
|
||||
// Transmission down / transient — just try again next tick.
|
||||
}finally{
|
||||
lock = false;
|
||||
}
|
||||
}
|
||||
|
||||
setInterval(tick, conf.organize.interval || 10000);
|
||||
tick(); // reconcile once on boot
|
||||
|
||||
module.exports = { tick };
|
||||
@@ -1,6 +1,8 @@
|
||||
'use strict';
|
||||
|
||||
const {PubSub} = require('p2psub');
|
||||
|
||||
ps = new PubSub();
|
||||
const ps = new PubSub();
|
||||
|
||||
|
||||
module.exports = ps;
|
||||
|
||||
@@ -0,0 +1,363 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
const tmdb = require('>/controller/tmdb');
|
||||
const { tmdbSearch, tmdbDetails } = tmdb;
|
||||
const library = require('>/controller/library');
|
||||
const ollama = require('>/controller/ollama');
|
||||
|
||||
// TPB numeric categories that count as Movies / TV( used for the q.php search and to
|
||||
// keep the LLM from ever seeing non-video results).
|
||||
const MOVIE_CATS = [201, 202, 207, 209, 211];
|
||||
const TV_CATS = [205, 208, 212];
|
||||
|
||||
function humanSize(bytes){
|
||||
bytes = Number(bytes) || 0;
|
||||
let units = ['B', 'KB', 'MB', 'GB', 'TB'];
|
||||
let i = 0;
|
||||
while(bytes >= 1024 && i < units.length - 1){
|
||||
bytes /= 1024;
|
||||
i++;
|
||||
}
|
||||
return `${bytes.toFixed(bytes < 10 && i > 0 ? 1 : 0)} ${units[i]}`;
|
||||
}
|
||||
|
||||
async function fetchJSON(url, options){
|
||||
let res = await fetch(url, options);
|
||||
if(!res.ok){
|
||||
let error = new Error(`UpstreamError`);
|
||||
error.message = `Request to ${url.split('?')[0]} failed( ${res.status})`;
|
||||
error.status = 502;
|
||||
throw error;
|
||||
}
|
||||
return await res.json();
|
||||
}
|
||||
|
||||
// Turn a raw TPB category id into 'movie' | 'tv' | null.
|
||||
function videoKind(category){
|
||||
category = parseInt(category, 10);
|
||||
if(TV_CATS.includes(category)) return 'tv';
|
||||
if(MOVIE_CATS.includes(category) || (category >= 200 && category < 300)) return 'movie';
|
||||
return null;
|
||||
}
|
||||
|
||||
// --- TPB HTML search ------------------------------------------------------
|
||||
|
||||
const SIZE_UNITS = { B: 1, KIB: 1024, MIB: 1024 ** 2, GIB: 1024 ** 3, TIB: 1024 ** 4 };
|
||||
|
||||
function sizeToBytes(value, unit){
|
||||
return Math.round(parseFloat(value) * (SIZE_UNITS[String(unit).toUpperCase()] || 1));
|
||||
}
|
||||
|
||||
function decodeEntities(text){
|
||||
return String(text)
|
||||
.replace(/&/g, '&').replace(/</g, '<').replace(/>/g, '>')
|
||||
.replace(/"/g, '"').replace(/�?39;/g, "'").replace(/'/g, "'");
|
||||
}
|
||||
|
||||
// Parse the classic TPB HTML result table( category cell, details link name, magnet,
|
||||
// then right-aligned size / seeders / leechers cells) into apibay-shaped objects.
|
||||
function parseTPBHtml(html){
|
||||
let out = [];
|
||||
for(let row of html.split(/<tr/i)){
|
||||
let hash = row.match(/magnet:\?xt=urn:btih:([A-Fa-f0-9]+)/i);
|
||||
if(!hash) continue;
|
||||
|
||||
let cat = row.match(/\/browse\/(\d+)/);
|
||||
let name = row.match(/title="Details for ([^"]+)"/i);
|
||||
let size = row.match(/<td align="right">\s*([\d.]+)(?: |\s)+(GiB|MiB|KiB|TiB)\s*<\/td>/i);
|
||||
let nums = [...row.matchAll(/<td align="right">\s*([\d,]+)\s*<\/td>/gi)].map(m => Number(m[1].replace(/,/g, '')));
|
||||
|
||||
if(!name) continue;
|
||||
// nums are [size, seeders, leechers]; size cell also matched the regex above so
|
||||
// seeders/leechers are the last two numeric right-aligned cells.
|
||||
let seeders = nums.length >= 2 ? nums[nums.length - 2] : 0;
|
||||
let leechers = nums.length >= 1 ? nums[nums.length - 1] : 0;
|
||||
|
||||
out.push({
|
||||
name: decodeEntities(name[1]),
|
||||
info_hash: hash[1].toLowerCase(),
|
||||
category: cat ? Number(cat[1]) : 0,
|
||||
size: size ? sizeToBytes(size[1], size[2]) : 0,
|
||||
seeders: seeders,
|
||||
leechers: leechers,
|
||||
imdb: null,
|
||||
});
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
async function tpbSearch(title){
|
||||
// Category 200 = all Video; we narrow to movie/tv subcats in prefilter().
|
||||
let url = `${conf.search.tpbBase}/search/${encodeURIComponent(title)}/1/99/200`;
|
||||
let res = await fetch(url, { headers: { 'User-Agent': 'Mozilla/5.0' } });
|
||||
if(!res.ok){
|
||||
let error = new Error('UpstreamError');
|
||||
error.message = `TPB search failed( ${res.status})`;
|
||||
error.status = 502;
|
||||
throw error;
|
||||
}
|
||||
return parseTPBHtml(await res.text());
|
||||
}
|
||||
|
||||
// Low-quality source tags we never want to recommend( token match to avoid false hits
|
||||
// like "GHOSTS" matching "TS").
|
||||
const JUNK = new Set(['cam', 'hdcam', 'camrip', 'ts', 'hdts', 'telesync', 'tc', 'telecine', 'scr', 'screener', 'dvdscr', 'workprint']);
|
||||
function isJunk(name){
|
||||
return String(name).toLowerCase().split(/[.\s_\-\[\]()]+/).some(tok => JUNK.has(tok));
|
||||
}
|
||||
|
||||
// Deterministic pre-filter before the torrents ever reach the LLM.
|
||||
function prefilter(torrents, imdbId){
|
||||
let out = torrents.filter(t => videoKind(t.category) && Number(t.seeders) > 0 && !isJunk(t.name));
|
||||
|
||||
// If we can positively identify the title by IMDB id, trust it and drop the rest.
|
||||
if(imdbId){
|
||||
let matches = out.filter(t => t.imdb && t.imdb === imdbId);
|
||||
if(matches.length) out = matches;
|
||||
}
|
||||
|
||||
return out
|
||||
.sort((a, b) => Number(b.seeders) - Number(a.seeders))
|
||||
.slice(0, 40);
|
||||
}
|
||||
|
||||
function buildMagnet(infoHash, name){
|
||||
let magnet = `magnet:?xt=urn:btih:${infoHash}&dn=${encodeURIComponent(name)}`;
|
||||
for(let tracker of conf.search.trackers){
|
||||
magnet += `&tr=${encodeURIComponent(tracker)}`;
|
||||
}
|
||||
return magnet;
|
||||
}
|
||||
|
||||
// --- LLM ranking ----------------------------------------------------------
|
||||
|
||||
function buildSystemPrompt(title, year, today){
|
||||
return [
|
||||
`You are a torrent-selection assistant. Today's date is ${today}.`,
|
||||
`Recent releases are legitimate: NEVER reject a torrent for being too new or assume`,
|
||||
`a current-year release is fake.`,
|
||||
``,
|
||||
`The user wants "${title}" (${year || 'unknown year'}). From the candidate list,`,
|
||||
`select the best releases.`,
|
||||
year ? `IMPORTANT: only pick releases whose name contains the year ${year}; reject other years or remakes of a same-named title.` : ``,
|
||||
`Preferences, in order:`,
|
||||
`- video codec HEVC/x265 (over x264/AVC)`,
|
||||
`- 1080p resolution`,
|
||||
`- total size around 1.5 GB (avoid needlessly huge files)`,
|
||||
`- English audio and English subtitles present`,
|
||||
`- prefer EXTENDED / UNRATED / UNCUT / DIRECTOR'S CUT editions`,
|
||||
`Reject: CAM, TS, TC, TELESYNC, HDCAM, SCREENER/SCR, and non-English-only copies.`,
|
||||
``,
|
||||
`Return STRICT JSON only, shape:`,
|
||||
`{"options":[{"role":"recommended|uhd|other","label":"...","info_hash":"...",`,
|
||||
`"name":"...","resolution":"...","codec":"...","hasSubs":true,"cut":"...",`,
|
||||
`"warning":"...","why":"..."}]}`,
|
||||
`Rules: exactly one "recommended" (the 1080p sweet spot). Include one "uhd" ONLY`,
|
||||
`if a 4K/2160p release exists, and set its "warning" to note it is a much larger,`,
|
||||
`slower download. Add "other" entries for genuinely distinct useful releases`,
|
||||
`(e.g. a different cut or a much smaller copy). "why" is one short sentence.`,
|
||||
`Copy info_hash verbatim from the chosen candidate.`,
|
||||
].join('\n');
|
||||
}
|
||||
|
||||
// Pull the first JSON object out of a model response( tolerates ```json fences / prose).
|
||||
function extractJSON(content){
|
||||
let text = String(content).replace(/```json/gi, '').replace(/```/g, '').trim();
|
||||
let start = text.indexOf('{');
|
||||
let end = text.lastIndexOf('}');
|
||||
if(start === -1 || end === -1) throw new Error('No JSON in model response');
|
||||
return JSON.parse(text.slice(start, end + 1));
|
||||
}
|
||||
|
||||
async function rankReleases(candidates, meta){
|
||||
let compact = candidates.map(t => ({
|
||||
name: t.name,
|
||||
sizeHuman: humanSize(t.size),
|
||||
seeders: Number(t.seeders),
|
||||
info_hash: t.info_hash,
|
||||
imdb: t.imdb || null,
|
||||
}));
|
||||
|
||||
let body = {
|
||||
model: conf.ollama.model,
|
||||
stream: false,
|
||||
format: 'json',
|
||||
messages: [
|
||||
{ role: 'system', content: buildSystemPrompt(meta.title, meta.year, meta.today) },
|
||||
{ role: 'user', content: JSON.stringify(compact) },
|
||||
],
|
||||
};
|
||||
|
||||
let data = await fetchJSON(`${conf.ollama.url}/api/chat`, {
|
||||
method: 'POST',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': `Bearer ${conf.ollama.apiKey}`,
|
||||
},
|
||||
body: JSON.stringify(body),
|
||||
});
|
||||
|
||||
let parsed = extractJSON(data.message && data.message.content);
|
||||
let options = Array.isArray(parsed.options) ? parsed.options : [];
|
||||
if(!options.length) throw new Error('Model returned no options');
|
||||
return options;
|
||||
}
|
||||
|
||||
// Deterministic fallback if the LLM is unavailable or returns junk: pick the most-seeded
|
||||
// candidate, preferring x265 + 1080p, so the feature degrades instead of failing.
|
||||
function fallbackReleases(candidates){
|
||||
let score = t => (/x265|hevc/i.test(t.name) ? 2 : 0) + (/1080p/i.test(t.name) ? 1 : 0);
|
||||
let best = [...candidates].sort((a, b) =>
|
||||
(score(b) - score(a)) || (Number(b.seeders) - Number(a.seeders))
|
||||
)[0];
|
||||
if(!best) return [];
|
||||
return [{
|
||||
role: 'recommended',
|
||||
label: 'Most seeded',
|
||||
info_hash: best.info_hash,
|
||||
name: best.name,
|
||||
sizeHuman: humanSize(best.size),
|
||||
seeders: Number(best.seeders),
|
||||
why: 'Automatically chosen (AI ranking unavailable): most seeders.',
|
||||
}];
|
||||
}
|
||||
|
||||
// Attach the fields the front end / add flow need to each option and drop any the model
|
||||
// hallucinated( info_hash must exist in the candidate set).
|
||||
function decorateOptions(options, candidates){
|
||||
let byHash = {};
|
||||
for(let t of candidates) byHash[String(t.info_hash).toLowerCase()] = t;
|
||||
|
||||
let out = [];
|
||||
for(let opt of options){
|
||||
let src = byHash[String(opt.info_hash).toLowerCase()];
|
||||
if(!src) continue;
|
||||
out.push({
|
||||
...opt,
|
||||
info_hash: src.info_hash,
|
||||
name: opt.name || src.name,
|
||||
category: Number(src.category),
|
||||
sizeHuman: opt.sizeHuman || humanSize(src.size),
|
||||
seeders: Number(src.seeders),
|
||||
magnetLink: buildMagnet(src.info_hash, src.name),
|
||||
});
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
async function findReleases(tmdbId, mediaType){
|
||||
let meta = await tmdbDetails(tmdbId, mediaType);
|
||||
let torrents = await tpbSearch(meta.title);
|
||||
let candidates = prefilter(torrents, meta.imdbId);
|
||||
|
||||
// For movies, require the confirmed release year in the name. TPB search is fuzzy
|
||||
// (a "Toy Story 5" query returns Toy Story 4/3/…), and without this the fallback can
|
||||
// cross to the wrong movie. If nothing matches, we genuinely have no good copy yet.
|
||||
if(mediaType !== 'tv' && meta.year){
|
||||
candidates = candidates.filter(t => t.name.includes(meta.year));
|
||||
}
|
||||
|
||||
if(!candidates.length){
|
||||
return { title: meta.title, year: meta.year, tmdbId, mediaType, options: [] };
|
||||
}
|
||||
|
||||
let today = new Date().toISOString().slice(0, 10);
|
||||
let options;
|
||||
try{
|
||||
options = await rankReleases(candidates, { title: meta.title, year: meta.year, today });
|
||||
}catch(error){
|
||||
console.error('rankReleases failed, using fallback:', error.message);
|
||||
options = fallbackReleases(candidates);
|
||||
}
|
||||
|
||||
options = decorateOptions(options, candidates);
|
||||
if(!options.length) options = decorateOptions(fallbackReleases(candidates), candidates);
|
||||
|
||||
let result = { title: meta.title, year: meta.year, tmdbId, mediaType, options };
|
||||
|
||||
// Quality-aware dedup: flag options you already own at equal-or-better quality.
|
||||
if(mediaType !== 'tv'){
|
||||
let lib = await library.movieInLibrary(tmdbId);
|
||||
if(lib.owned){
|
||||
for(let o of options) o.alreadyOwned = library.rankFromLabel(o.resolution) <= lib.rank;
|
||||
result.library = { owned: true, quality: lib.quality };
|
||||
}
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
// --- TV: library-aware season fill -----------------------------------------
|
||||
|
||||
function buildSeasonPrompt(title, season, today){
|
||||
return [
|
||||
`You are a torrent selector. Today's date is ${today}.`,
|
||||
`Pick the single best COMPLETE season pack for "${title}" Season ${season}.`,
|
||||
`It MUST be the full season (an S${String(season).padStart(2, '0')} / "Season ${season}"`,
|
||||
`complete pack), NOT a single episode. Prefer HEVC/x265, 1080p, English audio +`,
|
||||
`English subtitles, reasonable size. Reject single episodes, wrong seasons,`,
|
||||
`CAM/TS/telesync, and non-English-only.`,
|
||||
`Return STRICT JSON {"info_hash":"<from candidates>","why":"one short sentence"};`,
|
||||
`if nothing suitable, {"info_hash":null}.`,
|
||||
].join('\n');
|
||||
}
|
||||
|
||||
async function bestSeasonPack(candidates, meta){
|
||||
let compact = candidates.map(t => ({ name: t.name, sizeHuman: humanSize(t.size), seeders: Number(t.seeders), info_hash: t.info_hash }));
|
||||
let parsed = await ollama.chatJSON(buildSeasonPrompt(meta.title, meta.season, meta.today), JSON.stringify(compact));
|
||||
return parsed && parsed.info_hash ? parsed : null;
|
||||
}
|
||||
|
||||
// Compare TMDB's season/episode counts against the Emby inventory to classify each season.
|
||||
async function seasonPlan(tmdbId){
|
||||
let [tv, inv] = await Promise.all([ tmdb.tmdbSeasons(tmdbId), library.seriesInventory(tmdbId) ]);
|
||||
let seasons = tv.seasons.map(s => {
|
||||
let have = Number(inv.have && inv.have[s.season]) || 0;
|
||||
let status = have === 0 ? 'missing' : (have >= s.episodeCount ? 'complete' : 'partial');
|
||||
return { season: s.season, episodeCount: s.episodeCount, haveCount: have, status };
|
||||
});
|
||||
return { title: tv.title, year: tv.year, seasons, missing: seasons.filter(s => s.status !== 'complete').map(s => s.season) };
|
||||
}
|
||||
|
||||
// For every missing/incomplete season, pick the best complete-season pack. Returns the
|
||||
// chosen packs( the route adds them to Transmission) plus any seasons it couldn't fill.
|
||||
async function fillMissing(tmdbId){
|
||||
let plan = await seasonPlan(tmdbId);
|
||||
let details = await tmdbDetails(tmdbId, 'tv');
|
||||
let title = details.title || plan.title;
|
||||
let today = new Date().toISOString().slice(0, 10);
|
||||
|
||||
let picks = [], skipped = [];
|
||||
for(let season of plan.missing){
|
||||
let cands = prefilter(await tpbSearch(`${title} S${String(season).padStart(2, '0')}`), null);
|
||||
if(!cands.length){ skipped.push({ season, reason: 'no results' }); continue; }
|
||||
|
||||
let pick = null;
|
||||
try{ pick = await bestSeasonPack(cands, { title, season, today }); }catch(error){ pick = null; }
|
||||
let src = pick && cands.find(c => c.info_hash.toLowerCase() === String(pick.info_hash).toLowerCase());
|
||||
if(!src){ skipped.push({ season, reason: 'no suitable season pack' }); continue; }
|
||||
|
||||
picks.push({ season, name: src.name, info_hash: src.info_hash, category: Number(src.category), magnetLink: buildMagnet(src.info_hash, src.name), why: pick.why });
|
||||
}
|
||||
return { title, year: plan.year, picks, skipped };
|
||||
}
|
||||
|
||||
module.exports = {
|
||||
tmdbSearch,
|
||||
findReleases,
|
||||
seasonPlan,
|
||||
fillMissing,
|
||||
// exported for unit tests
|
||||
prefilter,
|
||||
buildMagnet,
|
||||
humanSize,
|
||||
videoKind,
|
||||
extractJSON,
|
||||
decorateOptions,
|
||||
fallbackReleases,
|
||||
tpbSearch,
|
||||
parseTPBHtml,
|
||||
sizeToBytes,
|
||||
isJunk,
|
||||
};
|
||||
@@ -0,0 +1,85 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
|
||||
async function fetchJSON(url, options){
|
||||
let res = await fetch(url, options);
|
||||
if(!res.ok){
|
||||
let error = new Error('UpstreamError');
|
||||
error.message = `Request to ${url.split('?')[0]} failed( ${res.status})`;
|
||||
error.status = 502;
|
||||
throw error;
|
||||
}
|
||||
return await res.json();
|
||||
}
|
||||
|
||||
async function tmdbSearch(query){
|
||||
let url = `https://api.themoviedb.org/3/search/multi?include_adult=false`
|
||||
+ `&query=${encodeURIComponent(query)}&api_key=${conf.tmdb.apiKey}`;
|
||||
|
||||
let data = await fetchJSON(url);
|
||||
|
||||
return (data.results || [])
|
||||
.filter(item => item.media_type === 'movie' || item.media_type === 'tv')
|
||||
.slice(0, 4)
|
||||
.map(item => {
|
||||
let date = item.release_date || item.first_air_date || '';
|
||||
return {
|
||||
tmdbId: item.id,
|
||||
mediaType: item.media_type,
|
||||
title: item.title || item.name,
|
||||
year: date ? date.slice(0, 4) : '',
|
||||
posterUrl: item.poster_path ? `https://image.tmdb.org/t/p/w200${item.poster_path}` : null,
|
||||
};
|
||||
});
|
||||
}
|
||||
|
||||
async function tmdbDetails(tmdbId, mediaType){
|
||||
let url = `https://api.themoviedb.org/3/${mediaType}/${tmdbId}`
|
||||
+ `?append_to_response=external_ids&api_key=${conf.tmdb.apiKey}`;
|
||||
|
||||
let data = await fetchJSON(url);
|
||||
let date = data.release_date || data.first_air_date || '';
|
||||
|
||||
return {
|
||||
title: data.title || data.name,
|
||||
year: date ? date.slice(0, 4) : '',
|
||||
imdbId: data.imdb_id || (data.external_ids && data.external_ids.imdb_id) || null,
|
||||
};
|
||||
}
|
||||
|
||||
// Canonicalize a( possibly messy) title + year to the authoritative TMDB record.
|
||||
// Returns null when nothing matches so callers can flag instead of guessing.
|
||||
async function tmdbFindBest(title, year, mediaType){
|
||||
let results = await tmdbSearch(title);
|
||||
|
||||
let pool = results.filter(r => r.mediaType === mediaType);
|
||||
if(!pool.length) pool = results;
|
||||
if(!pool.length) return null;
|
||||
|
||||
let pick = (year && pool.find(r => r.year === String(year))) || pool[0];
|
||||
let details = await tmdbDetails(pick.tmdbId, pick.mediaType);
|
||||
|
||||
return {
|
||||
title: details.title,
|
||||
year: details.year,
|
||||
tmdbId: pick.tmdbId,
|
||||
mediaType: pick.mediaType,
|
||||
imdbId: details.imdbId,
|
||||
};
|
||||
}
|
||||
|
||||
// Per-season episode counts for a show( season 0 / specials excluded).
|
||||
async function tmdbSeasons(tmdbId){
|
||||
let data = await fetchJSON(`https://api.themoviedb.org/3/tv/${tmdbId}?api_key=${conf.tmdb.apiKey}`);
|
||||
let date = data.first_air_date || '';
|
||||
return {
|
||||
title: data.name,
|
||||
year: date ? date.slice(0, 4) : '',
|
||||
seasons: (data.seasons || [])
|
||||
.filter(s => s.season_number >= 1 && s.episode_count > 0)
|
||||
.map(s => ({ season: s.season_number, episodeCount: s.episode_count })),
|
||||
};
|
||||
}
|
||||
|
||||
module.exports = { fetchJSON, tmdbSearch, tmdbDetails, tmdbFindBest, tmdbSeasons };
|
||||
@@ -2,8 +2,7 @@
|
||||
|
||||
const ps = require('./pubsub.js');
|
||||
const {Torrent} = require('>/models');
|
||||
|
||||
console.log('here!!!!!!!!!!!!!')
|
||||
const conf = require('>/conf');
|
||||
|
||||
let statusLock = false;
|
||||
setInterval(async function(){
|
||||
@@ -17,9 +16,4 @@ setInterval(async function(){
|
||||
// console.error('status interval error', error)
|
||||
}
|
||||
statusLock = false
|
||||
}, 10000, statusLock);
|
||||
|
||||
|
||||
// ps.subscribe(/./g, function(...args){
|
||||
// console.log('event', args);
|
||||
// });
|
||||
}, conf.transmission.statusUpdateInterval, statusLock);
|
||||
|
||||
@@ -0,0 +1,47 @@
|
||||
'use strict';
|
||||
|
||||
const conf = require('>/conf');
|
||||
const search = require('>/controller/search');
|
||||
|
||||
let lock = false;
|
||||
|
||||
// Periodically re-run Smart Search for wishlisted movies; when a good copy finally
|
||||
// appears, queue it( which then flows through the normal download + organize pipeline).
|
||||
async function tick(){
|
||||
if(lock) return;
|
||||
lock = true;
|
||||
try{
|
||||
const { Wanted, Torrent } = require('>/models');
|
||||
|
||||
let wanted = await Wanted.findAll({ where: { status: 'wanted' } });
|
||||
for(let w of wanted){
|
||||
if(!w.requestedBy) continue;
|
||||
try{
|
||||
let result = await search.findReleases(w.tmdbId, w.mediaType);
|
||||
let pick = result.options.find(o => o.role === 'recommended' && !o.alreadyOwned)
|
||||
|| result.options.find(o => !o.alreadyOwned);
|
||||
if(!pick) continue;
|
||||
|
||||
await Torrent.create({
|
||||
magnetLink: pick.magnetLink,
|
||||
isPrivate: false,
|
||||
added_by: w.requestedBy,
|
||||
category: pick.category,
|
||||
});
|
||||
w.status = 'fulfilled';
|
||||
await w.save();
|
||||
}catch(error){
|
||||
// leave it wanted; try again next tick
|
||||
}
|
||||
}
|
||||
}catch(error){
|
||||
// DB/Transmission transient — retry next tick
|
||||
}finally{
|
||||
lock = false;
|
||||
}
|
||||
}
|
||||
|
||||
setInterval(tick, conf.wanted.interval || 6 * 60 * 60 * 1000);
|
||||
tick(); // check once on boot
|
||||
|
||||
module.exports = { tick };
|
||||
+2
-1
@@ -3,5 +3,6 @@
|
||||
module.exports = {
|
||||
User: require('./ldap/user').User,
|
||||
AuthToken: require('./sql').AuthToken,
|
||||
Torrent: require('./sql').Torrent
|
||||
Torrent: require('./sql').Torrent,
|
||||
Wanted: require('./sql').Wanted
|
||||
};
|
||||
|
||||
+30
-5
@@ -11,9 +11,11 @@ var userLUR = new LRUCache({
|
||||
maxAge: 60000,
|
||||
});
|
||||
|
||||
const client = new Client({
|
||||
url: conf.url,
|
||||
});
|
||||
const escapeFilterValue = function(value){
|
||||
return String(value).replace(/[\0()*\\]/g, function(char){
|
||||
return '\\' + char.charCodeAt(0).toString(16).padStart(2, '0');
|
||||
});
|
||||
}
|
||||
|
||||
const user_parse = function(data){
|
||||
if(data[conf.userNameAttribute]){
|
||||
@@ -31,6 +33,10 @@ User.backing = "LDAP";
|
||||
|
||||
User.list = async function(){
|
||||
try{
|
||||
const client = new Client({
|
||||
url: conf.url,
|
||||
});
|
||||
|
||||
await client.bind(conf.bindDN, conf.bindPassword);
|
||||
|
||||
const res = await client.search(conf.userBase, {
|
||||
@@ -49,6 +55,10 @@ User.list = async function(){
|
||||
|
||||
User.listDetail = async function(){
|
||||
try{
|
||||
const client = new Client({
|
||||
url: conf.url,
|
||||
});
|
||||
|
||||
await client.bind(conf.bindDN, conf.bindPassword);
|
||||
|
||||
const res = await client.search(conf.userBase, {
|
||||
@@ -85,8 +95,12 @@ User.get = async function(data, key){
|
||||
data.searchKey = data.searchKey || key || conf.userNameAttribute;
|
||||
data.searchValue = data.searchValue || data.uid;
|
||||
|
||||
let filter = `(&${conf.userFilter}(${data.searchKey}=${data.searchValue}))`;
|
||||
if(userLUR.get(filter)) return userLUR.get(filter)
|
||||
let filter = `(&${conf.userFilter}(${escapeFilterValue(data.searchKey)}=${escapeFilterValue(data.searchValue)}))`;
|
||||
if(userLUR.get(filter)) return userLUR.get(filter);
|
||||
|
||||
const client = new Client({
|
||||
url: conf.url,
|
||||
});
|
||||
|
||||
await client.bind(conf.bindDN, conf.bindPassword);
|
||||
const res = await client.search(conf.userBase, {
|
||||
@@ -128,9 +142,20 @@ User.exists = async function(data, key){
|
||||
|
||||
User.login = async function(data){
|
||||
try{
|
||||
if(!data.password){
|
||||
let error = new Error('LDAPLoginFailed');
|
||||
error.name = 'LDAPLoginFailed';
|
||||
error.message = 'Invalid Credentials, login failed.';
|
||||
error.status = 401;
|
||||
throw error;
|
||||
}
|
||||
|
||||
let user = await this.get(data.uid || data[conf.userNameAttribute] || data.username);
|
||||
|
||||
const client = new Client({
|
||||
url: conf.url,
|
||||
});
|
||||
|
||||
await client.bind(user.dn, data.password);
|
||||
|
||||
await client.unbind();
|
||||
|
||||
@@ -3,13 +3,9 @@
|
||||
module.exports = {
|
||||
async up(queryInterface, Sequelize) {
|
||||
await queryInterface.createTable('Torrents', {
|
||||
id: {
|
||||
allowNull: false,
|
||||
autoIncrement: true,
|
||||
primaryKey: true,
|
||||
type: Sequelize.INTEGER
|
||||
},
|
||||
hashString: {
|
||||
primaryKey: true,
|
||||
allowNull: false,
|
||||
type: Sequelize.STRING
|
||||
},
|
||||
magnetLink: {
|
||||
|
||||
@@ -0,0 +1,14 @@
|
||||
'use strict';
|
||||
/** @type {import('sequelize-cli').Migration} */
|
||||
module.exports = {
|
||||
async up(queryInterface, Sequelize) {
|
||||
await queryInterface.addColumn('Torrents', 'category', {
|
||||
type: Sequelize.STRING,
|
||||
allowNull: false,
|
||||
defaultValue: 'Other'
|
||||
});
|
||||
},
|
||||
async down(queryInterface, Sequelize) {
|
||||
await queryInterface.removeColumn('Torrents', 'category');
|
||||
}
|
||||
};
|
||||
@@ -0,0 +1,18 @@
|
||||
'use strict';
|
||||
/** @type {import('sequelize-cli').Migration} */
|
||||
module.exports = {
|
||||
async up(queryInterface, Sequelize) {
|
||||
await queryInterface.addColumn('Torrents', 'organizedAt', {
|
||||
type: Sequelize.DATE,
|
||||
allowNull: true,
|
||||
});
|
||||
await queryInterface.addColumn('Torrents', 'metadata', {
|
||||
type: Sequelize.JSON,
|
||||
allowNull: true,
|
||||
});
|
||||
},
|
||||
async down(queryInterface, Sequelize) {
|
||||
await queryInterface.removeColumn('Torrents', 'organizedAt');
|
||||
await queryInterface.removeColumn('Torrents', 'metadata');
|
||||
}
|
||||
};
|
||||
@@ -0,0 +1,23 @@
|
||||
'use strict';
|
||||
/** @type {import('sequelize-cli').Migration} */
|
||||
module.exports = {
|
||||
async up(queryInterface, Sequelize) {
|
||||
await queryInterface.createTable('Wanteds', {
|
||||
tmdbId: {
|
||||
type: Sequelize.STRING,
|
||||
primaryKey: true,
|
||||
allowNull: false,
|
||||
},
|
||||
mediaType: { type: Sequelize.STRING, allowNull: false, defaultValue: 'movie' },
|
||||
title: { type: Sequelize.STRING },
|
||||
year: { type: Sequelize.STRING },
|
||||
status: { type: Sequelize.STRING, allowNull: false, defaultValue: 'wanted' },
|
||||
requestedBy: { type: Sequelize.STRING },
|
||||
createdAt: { allowNull: false, type: Sequelize.DATE },
|
||||
updatedAt: { allowNull: false, type: Sequelize.DATE },
|
||||
});
|
||||
},
|
||||
async down(queryInterface, Sequelize) {
|
||||
await queryInterface.dropTable('Wanteds');
|
||||
}
|
||||
};
|
||||
+51
-6
@@ -29,6 +29,21 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
|
||||
static trClient = tr_client;
|
||||
|
||||
// Map a raw TPB category id( e.g. 207) to one of our buckets. TPB groups by the
|
||||
// hundreds digit; TV shows are the exception pulled out of the Video group.
|
||||
static categoryFromTPB(id){
|
||||
id = parseInt(id, 10);
|
||||
if([205, 208, 212].includes(id)) return 'TV';
|
||||
return {
|
||||
1: 'Music',
|
||||
2: 'Movie',
|
||||
3: 'App',
|
||||
4: 'Game',
|
||||
5: 'Adult',
|
||||
6: 'Other',
|
||||
}[Math.floor(id / 100)] || 'Other';
|
||||
}
|
||||
|
||||
static async create(data, ...args){
|
||||
try{
|
||||
|
||||
@@ -36,14 +51,21 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
// console.log('instance', instance)
|
||||
await this.build(data).validate();
|
||||
// console.log('validate', val);
|
||||
data.isPrivate = data.isPrivate === 'true' ? true : false;
|
||||
|
||||
let res = await tr_client.addUrl(data.magnetLink);
|
||||
let options = {
|
||||
'download-dir': data.isPrivate ? `${conf.privateDownloadLocation}/${data.added_by}` : undefined,
|
||||
};
|
||||
|
||||
let res = await tr_client.addUrl(data.magnetLink, options);
|
||||
|
||||
return await super.create({
|
||||
magnetLink: data.magnetLink,
|
||||
hashString: res.hashString,
|
||||
isPrivate: data.isPrivate,
|
||||
name: res.name,
|
||||
added_by: data.added_by,
|
||||
category: this.categoryFromTPB(data.category),
|
||||
status: 0,
|
||||
percentDone: 0,
|
||||
}, args);
|
||||
@@ -53,12 +75,16 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
}
|
||||
}
|
||||
|
||||
async getTorrentData(noUpdate){
|
||||
async getTorrentData(){
|
||||
try{
|
||||
|
||||
if(this.percentDone === 1) return this.dataValues
|
||||
|
||||
let res = ( await tr_client.get(this.hashString, [
|
||||
"eta", "percentDone", "status", "rateDownload",
|
||||
"errorString", "hashString", 'name',
|
||||
'downloadDir',
|
||||
'dateCreated',
|
||||
'files', //array of files
|
||||
'filesStats', // array of files with status
|
||||
'isFinished',
|
||||
@@ -68,9 +94,8 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
'sizeWhenDone',
|
||||
]) ).torrents[0];
|
||||
|
||||
if(this.percentDone === 1) return this.dataValues
|
||||
await this.update(res);
|
||||
if(noUpdate) await this.save();
|
||||
|
||||
return {...res, ...this.dataValues};
|
||||
}catch(error){
|
||||
if(error.code === 'ECONNREFUSED'){
|
||||
@@ -95,11 +120,16 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
}
|
||||
|
||||
async destroy(){
|
||||
await await this.constructor.trClient.remove(this.hashString, true);
|
||||
await this.constructor.trClient.remove(this.hashString, true);
|
||||
return await super.destroy()
|
||||
}
|
||||
}
|
||||
Torrent.init({
|
||||
hashString: {
|
||||
type: DataTypes.STRING,
|
||||
allowNull: false,
|
||||
primaryKey: true
|
||||
},
|
||||
magnetLink: {
|
||||
type: DataTypes.STRING,
|
||||
allowNull: false,
|
||||
@@ -112,8 +142,12 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
type: DataTypes.BOOLEAN,
|
||||
defaultValue: false,
|
||||
},
|
||||
hashString: DataTypes.STRING,
|
||||
name: DataTypes.STRING,
|
||||
category: {
|
||||
type: DataTypes.STRING,
|
||||
allowNull: false,
|
||||
defaultValue: 'Other',
|
||||
},
|
||||
added_by: {
|
||||
type: DataTypes.STRING,
|
||||
ldapModel: 'User',
|
||||
@@ -137,6 +171,17 @@ module.exports = (sequelize, DataTypes, Model) => {
|
||||
type: DataTypes.NUMBER,
|
||||
allowNull: true,
|
||||
},
|
||||
createdAt: {
|
||||
type: DataTypes.DATE
|
||||
},
|
||||
organizedAt: {
|
||||
type: DataTypes.DATE,
|
||||
allowNull: true,
|
||||
},
|
||||
metadata: {
|
||||
type: DataTypes.JSON,
|
||||
allowNull: true,
|
||||
},
|
||||
}, {
|
||||
sequelize,
|
||||
modelName: 'Torrent',
|
||||
|
||||
@@ -0,0 +1,32 @@
|
||||
'use strict';
|
||||
|
||||
module.exports = (sequelize, DataTypes, Model) => {
|
||||
class Wanted extends Model {
|
||||
static associate(models) {
|
||||
}
|
||||
}
|
||||
Wanted.init({
|
||||
tmdbId: {
|
||||
type: DataTypes.STRING,
|
||||
primaryKey: true,
|
||||
allowNull: false,
|
||||
},
|
||||
mediaType: {
|
||||
type: DataTypes.STRING,
|
||||
allowNull: false,
|
||||
defaultValue: 'movie',
|
||||
},
|
||||
title: DataTypes.STRING,
|
||||
year: DataTypes.STRING,
|
||||
status: {
|
||||
type: DataTypes.STRING,
|
||||
allowNull: false,
|
||||
defaultValue: 'wanted', // wanted | fulfilled
|
||||
},
|
||||
requestedBy: DataTypes.STRING,
|
||||
}, {
|
||||
sequelize,
|
||||
modelName: 'Wanted',
|
||||
});
|
||||
return Wanted;
|
||||
};
|
||||
Generated
+1328
-814
File diff suppressed because it is too large
Load Diff
@@ -16,6 +16,7 @@
|
||||
"express": "^4.18.2",
|
||||
"extend": "^3.0.2",
|
||||
"http-proxy-middleware": "^0.20.0",
|
||||
"jq-repeat": "^2.2.1",
|
||||
"jquery": "^3.7.1",
|
||||
"jquery-ui": "^1.13.2",
|
||||
"ldapts": "^7.0.7",
|
||||
|
||||
+420
-40
@@ -1,56 +1,436 @@
|
||||
// socket
|
||||
var app = {};
|
||||
|
||||
// api
|
||||
app.pubsub = (function(){
|
||||
app.topics = {};
|
||||
|
||||
// auth
|
||||
app.subscribe = function(topic, listener) {
|
||||
if(topic instanceof RegExp){
|
||||
listener.match = topic;
|
||||
topic = "__REGEX__";
|
||||
}
|
||||
|
||||
// user
|
||||
// create the topic if not yet created
|
||||
if(!app.topics[topic]) app.topics[topic] = [];
|
||||
|
||||
// group
|
||||
// add the listener
|
||||
app.topics[topic].push(listener);
|
||||
}
|
||||
|
||||
"use strict";
|
||||
/* globals jQuery, Mustache, XRegExp, $, app */
|
||||
var app = app || {};
|
||||
app.matchTopics = function(topic){
|
||||
topic = topic || '';
|
||||
var matches = [... app.topics[topic] ? app.topics[topic] : []];
|
||||
|
||||
( function($){
|
||||
if(!app.topics['__REGEX__']) return matches;
|
||||
|
||||
/*
|
||||
delay calling document.ready until app is loaded.
|
||||
*/
|
||||
jQuery.holdReady(true);
|
||||
for(var listener of app.topics['__REGEX__']){
|
||||
if(topic.match(listener.match)) matches.push(listener);
|
||||
}
|
||||
|
||||
app.includeScrip = function(path, base){
|
||||
path = (base || JQapp_BASE_PATH || '') + path;
|
||||
return matches;
|
||||
}
|
||||
|
||||
jQuery.ajax({
|
||||
url: path,
|
||||
dataType: 'script',
|
||||
async: false
|
||||
app.publish = function(topic, data) {
|
||||
|
||||
// send the event to all listeners
|
||||
app.matchTopics(topic).forEach(function(listener) {
|
||||
setTimeout(function(data, topic){
|
||||
listener(data || {}, topic);
|
||||
}, 0, data, topic);
|
||||
});
|
||||
};
|
||||
}
|
||||
|
||||
app.include = function(scripts, base){
|
||||
scripts = $.isArray(scripts) ? scripts : [scripts];
|
||||
for(let script of scripts){
|
||||
app.includeScrip(script, base)
|
||||
return this;
|
||||
})(app);
|
||||
|
||||
app.socket = (function(app){
|
||||
// $.getScript('/socket.io/socket.io.js')
|
||||
// <script type="text/javascript" src="/socket.io/socket.io.js"></script>
|
||||
|
||||
var socket;
|
||||
$(document).ready(function(){
|
||||
socket = io({
|
||||
auth: {
|
||||
token: app.auth.getToken()
|
||||
}
|
||||
});
|
||||
// socket.emit('chat message', $('#m').val());
|
||||
socket.on('P2PSub', function(msg){
|
||||
msg.data.__noSocket = true;
|
||||
app.publish(msg.topic, msg.data);
|
||||
});
|
||||
|
||||
app.subscribe(/./g, function(data, topic){
|
||||
// console.log('local_pubs', data, topic)
|
||||
if(data.__noSocket) return;
|
||||
// console.log('local_pubs 2', data, topic)
|
||||
|
||||
socket.emit('P2PSub', { topic, data });
|
||||
});
|
||||
})
|
||||
|
||||
return socket;
|
||||
|
||||
})(app);
|
||||
|
||||
app.api = (function(app){
|
||||
var baseURL = '/__api/'
|
||||
|
||||
|
||||
$( document ).on( "ajaxError", function(event, res, req) {
|
||||
console.log('bad!', `app:api:error:`, res.status);
|
||||
app.publish(`app:api:error:${res.status}`, {res, res});
|
||||
} );
|
||||
|
||||
function post(url, data, callback){
|
||||
callback = callback || app.util.emptyFuction;
|
||||
$.ajax({
|
||||
type: 'POST',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
data: JSON.stringify(data),
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function put(url, data, callback){
|
||||
$.ajax({
|
||||
type: 'PUT',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
data: JSON.stringify(data),
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function remove(url, callback, callback2){
|
||||
if(!$.isFunction(callback)) callback = callback2;
|
||||
$.ajax({
|
||||
type: 'delete',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function get(url, callback){
|
||||
$.ajax({
|
||||
type: 'GET',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
return {post: post, get: get, put: put, delete: remove}
|
||||
})(app)
|
||||
|
||||
app.auth = (function(app) {
|
||||
var user = {}
|
||||
function setToken(token){
|
||||
localStorage.setItem('APIToken', token);
|
||||
setCookieToken(token);
|
||||
}
|
||||
|
||||
function getToken(){
|
||||
return localStorage.getItem('APIToken');
|
||||
}
|
||||
|
||||
// The proxy gates page navigation on a cookie( the auth-token header can not ride
|
||||
// along on a navigation), so mirror the token into a cookie.
|
||||
function setCookieToken(token){
|
||||
document.cookie = 'auth-token=' + encodeURIComponent(token) + '; path=/; max-age=31536000; samesite=lax';
|
||||
}
|
||||
|
||||
function clearCookieToken(){
|
||||
document.cookie = 'auth-token=; path=/; max-age=0; samesite=lax';
|
||||
}
|
||||
|
||||
function isLoggedIn(callback){
|
||||
if(getToken()){
|
||||
return app.api.get('user/me', function(error, data){
|
||||
if(error === 'Unauthorized') logOut();
|
||||
if(!error) app.auth.user = data;
|
||||
return callback(error, data);
|
||||
});
|
||||
}else{
|
||||
callback(null, false);
|
||||
}
|
||||
}
|
||||
})(jQuery);
|
||||
|
||||
app.include([
|
||||
'/utils.js',
|
||||
'/pubsub.js',
|
||||
'/api.js',
|
||||
'/socket.js',
|
||||
'/user.js',
|
||||
'/group.js',
|
||||
'/auth.js',
|
||||
], JQapp_BASE_PATH)
|
||||
function logIn(args, callback){
|
||||
app.api.post('auth/login', args, function(error, data){
|
||||
if(data.login){
|
||||
setToken(data.token);
|
||||
}
|
||||
callback(error, !!data.token);
|
||||
});
|
||||
}
|
||||
|
||||
function logOut(callback){
|
||||
callback = callback || app.util.emptyFuction;
|
||||
localStorage.removeItem('APIToken');
|
||||
clearCookieToken();
|
||||
callback();
|
||||
}
|
||||
|
||||
function makeUserFromInvite(args, callback){
|
||||
app.api.post('auth/invite/'+ args.token, args, function(error, data){
|
||||
if(data.login){
|
||||
callback(null, data);
|
||||
setToken(data.token);
|
||||
}
|
||||
callback(error, !!data.token);
|
||||
});
|
||||
}
|
||||
|
||||
function forceLogin(){
|
||||
$.holdReady( true );
|
||||
app.auth.isLoggedIn(function(error, isLoggedIn){
|
||||
if(error || !isLoggedIn){
|
||||
app.auth.logOut(function(){})
|
||||
location.replace(`/login${location.href.replace(location.origin, '')}`);
|
||||
}else{
|
||||
$.holdReady( false );
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function logInRedirect(){
|
||||
window.location.href = location.href.replace(location.origin+'/login', '') || '/'
|
||||
}
|
||||
|
||||
|
||||
$( document ).ready( function(){
|
||||
isLoggedIn(function(error, isLoggedIn){
|
||||
if(!error && isLoggedIn){
|
||||
// Refresh the navigation cookie; if this is the logged out gate page,
|
||||
// reload now that the cookie is set so the real content is proxied.
|
||||
setCookieToken(getToken());
|
||||
if(window.__tbpLoginGate) return location.reload();
|
||||
$('.tbp_proxy_is_authed').show();
|
||||
$('.tbp_proxy_not_authed').hide();
|
||||
}else{
|
||||
clearCookieToken();
|
||||
$('.tbp_proxy_is_authed').hide();
|
||||
$('.tbp_proxy_not_authed').show();
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
return {
|
||||
getToken: getToken,
|
||||
setToken: setToken,
|
||||
isLoggedIn: isLoggedIn,
|
||||
logIn: logIn,
|
||||
logOut: logOut,
|
||||
makeUserFromInvite: makeUserFromInvite,
|
||||
forceLogin,
|
||||
logInRedirect,
|
||||
}
|
||||
|
||||
})(app);
|
||||
|
||||
app.util = (function(app){
|
||||
|
||||
function getUrlParameter(name) {
|
||||
name = name.replace(/[\[]/, '\\[').replace(/[\]]/, '\\]');
|
||||
var regex = new RegExp('[\\?&]' + name + '=([^&#]*)');
|
||||
var results = regex.exec(location.search);
|
||||
return results === null ? '' : decodeURIComponent(results[1].replace(/\+/g, ' '));
|
||||
};
|
||||
|
||||
function actionMessage(message, $target, type, callback){
|
||||
message = message || '';
|
||||
$target = $target.closest('div.card').find('.actionMessage');
|
||||
type = type || 'info';
|
||||
callback = callback || function(){};
|
||||
|
||||
if($target.html() === message) return;
|
||||
|
||||
if($target.html()){
|
||||
$target.slideUp('fast', function(){
|
||||
$target.html('')
|
||||
$target.removeClass (function (index, className) {
|
||||
return (className.match (/(^|\s)ui-\S+/g) || []).join(' ');
|
||||
});
|
||||
if(message) return actionMessage(message, $target, type, callback);
|
||||
$target.hide()
|
||||
})
|
||||
}else{
|
||||
if(type) $target.addClass('ui-' + type);
|
||||
$target.html(message).slideDown('fast');
|
||||
}
|
||||
setTimeout(callback,10)
|
||||
}
|
||||
|
||||
$.fn.serializeObject = function() {
|
||||
var
|
||||
arr = $(this).serializeArray(),
|
||||
obj = {};
|
||||
|
||||
for(var i = 0; i < arr.length; i++) {
|
||||
if(obj[arr[i].name] === undefined) {
|
||||
obj[arr[i].name] = arr[i].value;
|
||||
} else {
|
||||
if(!(obj[arr[i].name] instanceof Array)) {
|
||||
obj[arr[i].name] = [obj[arr[i].name]];
|
||||
}
|
||||
obj[arr[i].name].push(arr[i].value);
|
||||
}
|
||||
}
|
||||
return obj;
|
||||
};
|
||||
|
||||
return {
|
||||
getUrlParameter: getUrlParameter,
|
||||
actionMessage: actionMessage,
|
||||
emptyFuction: function(){},
|
||||
|
||||
}
|
||||
})(app);
|
||||
|
||||
|
||||
app.user = (function(app){
|
||||
function list(callback){
|
||||
app.api.get('user/?detail=true', function(error, data){
|
||||
callback(error, data);
|
||||
})
|
||||
}
|
||||
|
||||
function add(args, callback){
|
||||
app.api.post('user/', args, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function remove(args, callback){
|
||||
if(!confirm('Delete '+ args.uid+ 'user?')) return false;
|
||||
app.api.delete('user/'+ args.uid, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function changePassword(args, callback){
|
||||
app.api.put('users/'+ arg.uid || '', args, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function createInvite(callback){
|
||||
app.api.post('user/invite', {}, function(error, data, status){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function consumeInvite(args){
|
||||
app.api.post('/auth/invite/'+args.token, args, function(error, data){
|
||||
if(data.token){
|
||||
app.auth.setToken(data.token)
|
||||
return callback(null, true)
|
||||
}
|
||||
callback(error)
|
||||
});
|
||||
}
|
||||
|
||||
return {list, remove, createInvite};
|
||||
|
||||
})(app);
|
||||
|
||||
app.group = (function(app){
|
||||
function list(callback){
|
||||
app.api.get('group?detail=true', function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function remove(args, callback){
|
||||
app.api.delete('group/'+args.cn, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
return {list, remove}
|
||||
})(app);
|
||||
|
||||
$( document ).ready( function () {
|
||||
|
||||
$( 'div.row' ).fadeIn( 'slow' ); //show the page
|
||||
|
||||
//panel button's
|
||||
$( '.fa-arrows-v' ).click( function () {
|
||||
$( this ).closest( '.card' ).find( '.card-body' ).slideToggle( 'fast' );
|
||||
});
|
||||
|
||||
$('.actionMessage').on('click', 'button.action-close', function(event){
|
||||
app.util.actionMessage(null, $(this));
|
||||
})
|
||||
|
||||
/*
|
||||
when the DOM is finished, start the app
|
||||
*/
|
||||
jQuery(document).on("DOMContentLoaded", function(event) {
|
||||
app.publish("__dom-content-loaded-start");
|
||||
app.publish("__dom-content-loaded-end");
|
||||
});
|
||||
|
||||
//ajax form submit
|
||||
function formAJAX( btn, del ) {
|
||||
event.preventDefault(); // avoid to execute the actual submit of the form.
|
||||
var $form = $(btn).closest( '[action]' ); // gets the 'form' parent
|
||||
var formData = $form.find( '[name]' ).serializeObject(); // builds query formDataing
|
||||
var method = $form.attr('method') || 'post';
|
||||
|
||||
// if( !$form.validate()) {
|
||||
// app.util.actionMessage('Please fix the form errors.', $form, 'danger')
|
||||
// return false;
|
||||
// }
|
||||
|
||||
app.util.actionMessage(
|
||||
'<div class="spinner-border" role="status"><span class="sr-only">Loading...</span></div>',
|
||||
$form,
|
||||
'state-highlight'
|
||||
);
|
||||
|
||||
app.api[method]($form.attr('action'), formData, function(error, data){
|
||||
app.util.actionMessage(data.message, $form, error ? 'state-error' : 'priority-primary'); //re-populate table
|
||||
if(!error){
|
||||
$form.trigger("reset");
|
||||
eval($form.attr('evalAJAX')); //gets JS to run after completion
|
||||
}
|
||||
});
|
||||
|
||||
}
|
||||
|
||||
@@ -1,91 +0,0 @@
|
||||
app.api = (function(app){
|
||||
var baseURL = '/__api/'
|
||||
|
||||
|
||||
$( document ).on( "ajaxError", function(event, res, req) {
|
||||
console.log('bad!', `app:api:error:`, res.status);
|
||||
app.publish(`app:api:error:${res.status}`, {res, res});
|
||||
} );
|
||||
|
||||
function post(url, data, callback){
|
||||
callback = callback || app.util.emptyFuction;
|
||||
$.ajax({
|
||||
type: 'POST',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
data: JSON.stringify(data),
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function put(url, data, callback){
|
||||
$.ajax({
|
||||
type: 'PUT',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
data: JSON.stringify(data),
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function remove(url, callback, callback2){
|
||||
if(!$.isFunction(callback)) callback = callback2;
|
||||
$.ajax({
|
||||
type: 'delete',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function get(url, callback){
|
||||
$.ajax({
|
||||
type: 'GET',
|
||||
url: baseURL+url,
|
||||
headers:{
|
||||
'auth-token': app.auth.getToken()
|
||||
},
|
||||
contentType: "application/json; charset=utf-8",
|
||||
dataType: "json",
|
||||
complete: function(res, text){
|
||||
callback(
|
||||
text !== 'success' ? res.statusText : null,
|
||||
JSON.parse(res.responseText),
|
||||
res.status
|
||||
)
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
return {post: post, get: get, put: put, delete: remove}
|
||||
})(app);
|
||||
@@ -1,87 +0,0 @@
|
||||
app.auth = (function(app) {
|
||||
var user = {}
|
||||
function setToken(token){
|
||||
localStorage.setItem('APIToken', token);
|
||||
}
|
||||
|
||||
function getToken(){
|
||||
return localStorage.getItem('APIToken');
|
||||
}
|
||||
|
||||
function isLoggedIn(callback){
|
||||
if(getToken()){
|
||||
return app.api.get('user/me', function(error, data){
|
||||
if(error === 'Unauthorized') logOut();
|
||||
if(!error) app.auth.user = data;
|
||||
return callback(error, data);
|
||||
});
|
||||
}else{
|
||||
callback(null, false);
|
||||
}
|
||||
}
|
||||
|
||||
function logIn(args, callback){
|
||||
app.api.post('auth/login', args, function(error, data){
|
||||
if(data.login){
|
||||
setToken(data.token);
|
||||
}
|
||||
callback(error, !!data.token);
|
||||
});
|
||||
}
|
||||
|
||||
function logOut(callback){
|
||||
callback = callback || app.util.emptyFuction;
|
||||
localStorage.removeItem('APIToken');
|
||||
callback();
|
||||
}
|
||||
|
||||
function makeUserFromInvite(args, callback){
|
||||
app.api.post('auth/invite/'+ args.token, args, function(error, data){
|
||||
if(data.login){
|
||||
callback(null, data);
|
||||
setToken(data.token);
|
||||
}
|
||||
callback(error, !!data.token);
|
||||
});
|
||||
}
|
||||
|
||||
function forceLogin(){
|
||||
$.holdReady( true );
|
||||
app.auth.isLoggedIn(function(error, isLoggedIn){
|
||||
if(error || !isLoggedIn){
|
||||
app.auth.logOut(function(){})
|
||||
location.replace(`/login${location.href.replace(location.origin, '')}`);
|
||||
}else{
|
||||
$.holdReady( false );
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function logInRedirect(){
|
||||
window.location.href = location.href.replace(location.origin+'/login', '') || '/'
|
||||
}
|
||||
|
||||
|
||||
$( document ).ready( function(){
|
||||
isLoggedIn(function(error, isLoggedIn){
|
||||
if(!error && isLoggedIn){
|
||||
$('.tbp_proxy_is_authed').show();
|
||||
$('.tbp_proxy_not_authed').hide();
|
||||
}else{
|
||||
$('.tbp_proxy_is_authed').hide();
|
||||
$('.tbp_proxy_not_authed').show();
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
return {
|
||||
getToken: getToken,
|
||||
setToken: setToken,
|
||||
isLoggedIn: isLoggedIn,
|
||||
logIn: logIn,
|
||||
logOut: logOut,
|
||||
makeUserFromInvite: makeUserFromInvite,
|
||||
forceLogin,
|
||||
logInRedirect,
|
||||
}
|
||||
})(app);
|
||||
@@ -1,15 +0,0 @@
|
||||
app.group = (function(app){
|
||||
function list(callback){
|
||||
app.api.get('group?detail=true', function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function remove(args, callback){
|
||||
app.api.delete('group/'+args.cn, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
return {list, remove}
|
||||
})(app);
|
||||
@@ -1,41 +0,0 @@
|
||||
app.pubsub = (function(){
|
||||
app.topics = {};
|
||||
|
||||
app.subscribe = function(topic, listener) {
|
||||
if(topic instanceof RegExp){
|
||||
listener.match = topic;
|
||||
topic = "__REGEX__";
|
||||
}
|
||||
|
||||
// create the topic if not yet created
|
||||
if(!app.topics[topic]) app.topics[topic] = [];
|
||||
|
||||
// add the listener
|
||||
app.topics[topic].push(listener);
|
||||
}
|
||||
|
||||
app.matchTopics = function(topic){
|
||||
topic = topic || '';
|
||||
var matches = [... app.topics[topic] ? app.topics[topic] : []];
|
||||
|
||||
if(!app.topics['__REGEX__']) return matches;
|
||||
|
||||
for(var listener of app.topics['__REGEX__']){
|
||||
if(topic.match(listener.match)) matches.push(listener);
|
||||
}
|
||||
|
||||
return matches;
|
||||
}
|
||||
|
||||
app.publish = function(topic, data) {
|
||||
|
||||
// send the event to all listeners
|
||||
app.matchTopics(topic).forEach(function(listener) {
|
||||
setTimeout(function(data, topic){
|
||||
listener(data || {}, topic);
|
||||
}, 0, data, topic);
|
||||
});
|
||||
}
|
||||
|
||||
return this;
|
||||
})(app);
|
||||
@@ -1,28 +0,0 @@
|
||||
app.socket = (function(app){
|
||||
// $.getScript('/socket.io/socket.io.js')
|
||||
// <script type="text/javascript" src="/socket.io/socket.io.js"></script>
|
||||
|
||||
var socket;
|
||||
$(document).ready(function(){
|
||||
socket = io({
|
||||
auth: {
|
||||
token: app.auth.getToken()
|
||||
}
|
||||
});
|
||||
// socket.emit('chat message', $('#m').val());
|
||||
socket.on('P2PSub', function(msg){
|
||||
msg.data.__noSocket = true;
|
||||
app.publish(msg.topic, msg.data);
|
||||
});
|
||||
|
||||
app.subscribe(/./g, function(data, topic){
|
||||
// console.log('local_pubs', data, topic)
|
||||
if(data.__noSocket) return;
|
||||
// console.log('local_pubs 2', data, topic)
|
||||
|
||||
socket.emit('P2PSub', { topic, data });
|
||||
});
|
||||
})
|
||||
|
||||
return socket;
|
||||
})(app);
|
||||
@@ -1,122 +0,0 @@
|
||||
app.torrent = (function(app){
|
||||
let isDown = false;
|
||||
|
||||
$( document ).on( "ajaxSend", function(event, ajax, res, ...args) {
|
||||
console.log('right?', res.url, res.url.startsWith('/__api/torrent'), app.torrent.isDown)
|
||||
if(res.url.startsWith('/__api/torrent') && isDown){
|
||||
ajax.abort()
|
||||
}
|
||||
// $( ".log" ).text( "Triggered ajaxStart handler." );
|
||||
// throw new Error('go')
|
||||
} );
|
||||
|
||||
statusMap = [
|
||||
'Inactive', // 0
|
||||
'CHECK_WAIT', // 1
|
||||
'Verifying', // 2
|
||||
'DOWNLOAD_WAIT', // 3
|
||||
'Downloading', // 4
|
||||
'SEED_WAIT', // 5
|
||||
'Seeding', // 6
|
||||
'ISOLATED', // 7
|
||||
'Unknown', // 8
|
||||
];
|
||||
|
||||
function list(callback, username) {
|
||||
app.api.get('torrent', function(error, data){
|
||||
if(error) return;
|
||||
callback(null, data)
|
||||
});
|
||||
}
|
||||
|
||||
function get(callback, id, forceUpdate){
|
||||
app.api.get(`torrent/${id}?${forceUpdate ? 'latest': '' }`, function(error, data){
|
||||
if(error) return;
|
||||
callback(null, data)
|
||||
});
|
||||
}
|
||||
|
||||
function start(id, callback){
|
||||
app.api.post(`torrent/${id}/start`, function(error, data){
|
||||
if(error) return;
|
||||
callback(null, data)
|
||||
});
|
||||
}
|
||||
|
||||
function stop(id, callback){
|
||||
app.api.post(`torrent/${id}/stop`, function(error, data){
|
||||
if(error) return;
|
||||
callback(null, data)
|
||||
});
|
||||
}
|
||||
|
||||
function destroy(id, callback){
|
||||
app.api.delete(`torrent/${id}`, function(error, data){
|
||||
if(error) return;
|
||||
callback(null, data)
|
||||
});
|
||||
}
|
||||
|
||||
function parseServerStatus(data){
|
||||
return {
|
||||
...data,
|
||||
"downloadSpeed": humanFileSize(data.downloadSpeed)+'/s',
|
||||
"uploadSpeed": humanFileSize(data.uploadSpeed)+'/s',
|
||||
/* "activeTorrentCount": 11,
|
||||
"cumulative-stats": {
|
||||
"downloadedBytes": 2925927098021,
|
||||
"filesAdded": 80609,
|
||||
"secondsActive": 12136579,
|
||||
"sessionCount": 21,
|
||||
"uploadedBytes": 107123787853
|
||||
},
|
||||
"current-stats": {
|
||||
"downloadedBytes": 48440590262,
|
||||
"filesAdded": 544,
|
||||
"secondsActive": 111907,
|
||||
"sessionCount": 1,
|
||||
"uploadedBytes": 461874022
|
||||
},
|
||||
"pausedTorrentCount": 462,
|
||||
"torrentCount": 473,
|
||||
"__noSocket": true*/
|
||||
}
|
||||
}
|
||||
|
||||
function parseTorrnetItem(torrent){
|
||||
let percentDone = (torrent.percentDone || 0)*100;
|
||||
return {
|
||||
...torrent,
|
||||
"eta": torrent.eta > 0 ? moment().seconds(torrent.eta).fromNow() : 'Unknown',
|
||||
"rateDownload": `${humanFileSize(torrent.rateDownload)}/s`,
|
||||
"sizeWhenDone": humanFileSize(torrent.sizeWhenDone),
|
||||
"percentDone": percentDone,
|
||||
"statusText": statusMap[torrent.status],
|
||||
"isActive": [3, 4, 5, 6].includes(torrent.status), // DOWNLOAD_WAIT ,DOWNLOAD, SEED_WAIT, SEED
|
||||
"isFinished": torrent.isFinished || percentDone === 100,
|
||||
"createdAtString": moment(torrent.createdAt).fromNow(),
|
||||
|
||||
// "isFinished": false,
|
||||
// "isStalled": false,
|
||||
// "name": "reacher.s02e06.1080p.web.h264-successfulcrab[EZTVx.to].mkv",
|
||||
// "hashString": "4794a0915cada6c491eb5c910e1f4a0da727cac8",
|
||||
// "status": 4,
|
||||
// "id": 1,
|
||||
|
||||
// "peersConnected": 50,
|
||||
// "added_by": "wmantly",
|
||||
// "errorString": "",
|
||||
|
||||
// "downloadDir": "/media/torrents",
|
||||
// "files": [],
|
||||
// "peers": [],
|
||||
// "magnetLink": "magnet:?xt=urn:btih:4794A0915CADA6C491EB5C910E1F4A0DA727CAC8&dn=Reacher+S02E06+1080p+WEB+H264-SuccessfulCrab&tr=http%3A%2F%2Fp4p.arenabg.com%3A1337%2Fannounce&tr=udp%3A%2F%2F47.ip-51-68-199.eu%3A6969%2Fannounce&tr=udp%3A%2F%2F9.rarbg.me%3A2780%2Fannounce&tr=udp%3A%2F%2F9.rarbg.to%3A2710%2Fannounce&tr=udp%3A%2F%2F9.rarbg.to%3A2730%2Fannounce&tr=udp%3A%2F%2F9.rarbg.to%3A2920%2Fannounce&tr=udp%3A%2F%2Fopen.stealth.si%3A80%2Fannounce&tr=udp%3A%2F%2Fopentracker.i2p.rocks%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.coppersurfer.tk%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.cyberia.is%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.dler.org%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.internetwarriors.net%3A1337%2Fannounce&tr=udp%3A%2F%2Ftracker.leechers-paradise.org%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.openbittorrent.com%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.opentrackr.org%3A1337&tr=udp%3A%2F%2Ftracker.pirateparty.gr%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.tiny-vps.com%3A6969%2Fannounce&tr=udp%3A%2F%2Ftracker.torrent.eu.org%3A451%2Fannounce",
|
||||
// "isPrivate": false,
|
||||
// "createdAt": "2024-01-05T21:18:30.607Z",
|
||||
// "updatedAt": "2024-01-05T21:32:54.493Z"
|
||||
// "torrent_id": "454",
|
||||
}
|
||||
}
|
||||
|
||||
return {list, get, start, stop, destroy, parseTorrnetItem, parseServerStatus, isDown};
|
||||
})(app);
|
||||
@@ -1,44 +0,0 @@
|
||||
app.user = (function(app){
|
||||
function list(callback){
|
||||
app.api.get('user/?detail=true', function(error, data){
|
||||
callback(error, data);
|
||||
})
|
||||
}
|
||||
|
||||
function add(args, callback){
|
||||
app.api.post('user/', args, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function remove(args, callback){
|
||||
if(!confirm('Delete '+ args.uid+ 'user?')) return false;
|
||||
app.api.delete('user/'+ args.uid, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function changePassword(args, callback){
|
||||
app.api.put('users/'+ arg.uid || '', args, function(error, data){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function createInvite(callback){
|
||||
app.api.post('user/invite', {}, function(error, data, status){
|
||||
callback(error, data);
|
||||
});
|
||||
}
|
||||
|
||||
function consumeInvite(args){
|
||||
app.api.post('/auth/invite/'+args.token, args, function(error, data){
|
||||
if(data.token){
|
||||
app.auth.setToken(data.token)
|
||||
return callback(null, true)
|
||||
}
|
||||
callback(error)
|
||||
});
|
||||
}
|
||||
|
||||
return {list, remove, createInvite};
|
||||
})(app);
|
||||
@@ -1,96 +0,0 @@
|
||||
app.util = (function(app){
|
||||
|
||||
function getUrlParameter(name) {
|
||||
name = name.replace(/[\[]/, '\\[').replace(/[\]]/, '\\]');
|
||||
var regex = new RegExp('[\\?&]' + name + '=([^&#]*)');
|
||||
var results = regex.exec(location.search);
|
||||
return results === null ? '' : decodeURIComponent(results[1].replace(/\+/g, ' '));
|
||||
};
|
||||
|
||||
function actionMessage(message, $target, type, callback){
|
||||
message = message || '';
|
||||
$target = $target.closest('div.card').find('.actionMessage');
|
||||
type = type || 'info';
|
||||
callback = callback || function(){};
|
||||
|
||||
if($target.html() === message) return;
|
||||
|
||||
if($target.html()){
|
||||
$target.slideUp('fast', function(){
|
||||
$target.html('')
|
||||
$target.removeClass (function (index, className) {
|
||||
return (className.match (/(^|\s)ui-\S+/g) || []).join(' ');
|
||||
});
|
||||
if(message) return actionMessage(message, $target, type, callback);
|
||||
$target.hide()
|
||||
})
|
||||
}else{
|
||||
if(type) $target.addClass('ui-' + type);
|
||||
$target.html(message).slideDown('fast');
|
||||
}
|
||||
setTimeout(callback,10)
|
||||
}
|
||||
|
||||
$( document ).ready( function () {
|
||||
$('.actionMessage').on('click', 'button.action-close', function(event){
|
||||
app.util.actionMessage(null, $(this));
|
||||
});
|
||||
});
|
||||
|
||||
$.fn.serializeObject = function() {
|
||||
var
|
||||
arr = $(this).serializeArray(),
|
||||
obj = {};
|
||||
|
||||
for(var i = 0; i < arr.length; i++) {
|
||||
if(obj[arr[i].name] === undefined) {
|
||||
obj[arr[i].name] = arr[i].value;
|
||||
} else {
|
||||
if(!(obj[arr[i].name] instanceof Array)) {
|
||||
obj[arr[i].name] = [obj[arr[i].name]];
|
||||
}
|
||||
obj[arr[i].name].push(arr[i].value);
|
||||
}
|
||||
}
|
||||
return obj;
|
||||
};
|
||||
|
||||
function formAJAX( btn, del ) {
|
||||
event.preventDefault(); // avoid to execute the actual submit of the form.
|
||||
var $form = $(btn).closest( '[action]' ); // gets the 'form' parent
|
||||
var formData = $form.find( '[name]' ).serializeObject(); // builds query formDataing
|
||||
var method = $form.attr('method') || 'post';
|
||||
|
||||
// if( !$form.validate()) {
|
||||
// app.util.actionMessage('Please fix the form errors.', $form, 'danger')
|
||||
// return false;
|
||||
// }
|
||||
|
||||
app.util.actionMessage(
|
||||
'<div class="spinner-border" role="status"><span class="sr-only">Loading...</span></div>',
|
||||
$form,
|
||||
'state-highlight'
|
||||
);
|
||||
|
||||
app.api[method]($form.attr('action'), formData, function(error, data){
|
||||
app.util.actionMessage(data.message, $form, error ? 'state-error' : 'priority-primary'); //re-populate table
|
||||
if(!error){
|
||||
$form.trigger("reset");
|
||||
eval($form.attr('evalAJAX')); //gets JS to run after completion
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
function humanFileSize(size) {
|
||||
var i = size == 0 ? 0 : Math.floor(Math.log(size) / Math.log(1024));
|
||||
return (size / Math.pow(1024, i)).toFixed(2) * 1 + ' ' + ['B', 'kB', 'MB', 'GB', 'TB'][i];
|
||||
}
|
||||
|
||||
return {
|
||||
emptyFuction: function(){},
|
||||
getUrlParameter,
|
||||
actionMessage,
|
||||
humanFileSize,
|
||||
formAJAX,
|
||||
}
|
||||
})(app);
|
||||
@@ -1,283 +0,0 @@
|
||||
(function($, Mustache){
|
||||
'use strict';
|
||||
if (!$.scope) {
|
||||
$.scope = {};
|
||||
}
|
||||
|
||||
var make = function( element ){
|
||||
|
||||
//construct array
|
||||
function makeArray( input ){
|
||||
|
||||
var result = [];
|
||||
|
||||
Object.defineProperty( result, "__repeatId", {
|
||||
value: repeatId,
|
||||
writable: true,
|
||||
enumerable: false,
|
||||
configurable: true
|
||||
} );
|
||||
|
||||
Object.defineProperty( result, "__rq_template", {
|
||||
value: '',
|
||||
writable: true,
|
||||
enumerable: false,
|
||||
configurable: true
|
||||
} );
|
||||
|
||||
result.splice = function(inputValue, ...args){
|
||||
//splice does all the heavy lifting by interacting with the DOM elements.
|
||||
|
||||
var toProto = [...args]
|
||||
|
||||
var index;
|
||||
//if a string is submitted as the index, try to match it to index number
|
||||
if( typeof arguments[0] === 'string' ){
|
||||
index = this.indexOf( arguments[0] );//set where to start
|
||||
if ( index === -1 ) {
|
||||
return [];
|
||||
}
|
||||
}else{
|
||||
index = arguments[0]; //set where to start
|
||||
}
|
||||
|
||||
toProto.unshift(index)
|
||||
|
||||
|
||||
var howMany = arguments[1]; //sets the amount of fields to remove
|
||||
var args = Array.prototype.slice.call( arguments ); // coverts arguments into array
|
||||
var toAdd = args.slice(2); // only keeps fields to add to array
|
||||
|
||||
// if the starting point is higher then the total index count, start at the end
|
||||
if( index > this.length ) {
|
||||
index = this.length;
|
||||
}
|
||||
// if the starting point is negative, start form the end of the array, minus the start point
|
||||
if( index < 0 ) {
|
||||
index = this.length - Math.abs( index );
|
||||
}
|
||||
|
||||
// if there are things to add, figure out the how many new indexes we need
|
||||
if( !howMany && howMany !== 0 ) {
|
||||
howMany = this.length - index;
|
||||
}
|
||||
//not sure why i put this here... but it does matter!
|
||||
if( howMany > this.length - index ) {
|
||||
howMany = this.length - index;
|
||||
}
|
||||
|
||||
//figure out how many positions we need to shift the current elements
|
||||
var shift = toAdd.length - howMany;
|
||||
|
||||
// figure out how big the new array will be
|
||||
// var newLength = this.length + shift;
|
||||
|
||||
//removes fields from array based on howMany needs to be removed
|
||||
for( var i = index; i < +index+howMany; i++ ) {
|
||||
this.__take.apply( this[index].__jq_$el );
|
||||
// this.__take.apply( $( '.jq-repeat-'+ this.__repeatId +'[jq-repeat-index="'+ ( i + index ) +'"]' ) );
|
||||
}
|
||||
|
||||
//re-factor element index's
|
||||
for(var i = 0; i < this.length; i++){
|
||||
if( i >= index){
|
||||
|
||||
this[i].__jq_$el.attr( 'jq-repeat-index', i+shift );
|
||||
}
|
||||
}
|
||||
|
||||
//if there are fields to add to the array, add them
|
||||
if( toAdd.length > 0 ){
|
||||
|
||||
//$.each( toAdd, function( key, value ){
|
||||
for(var I = 0; I < toAdd.length; I++){
|
||||
|
||||
//figure out new elements index
|
||||
var key = I + index;
|
||||
// apply values to template
|
||||
var render = Mustache.render( this.__rq_template, {__id:I, ...toAdd[I]} );
|
||||
|
||||
//set call name and index keys to DOM element
|
||||
var $render = $( render ).addClass( 'jq-repeat-'+ this.__repeatId ).attr( 'jq-repeat-index', key );
|
||||
|
||||
|
||||
//if add new elements in proper stop, or after the place holder.
|
||||
if( key === 0 ){
|
||||
$( '.jq-repeat-'+ this.__repeatId +'[jq-repeat-index="holder"]' ).after( $render );
|
||||
}else{
|
||||
$( '.jq-repeat-'+ this.__repeatId +'[jq-repeat-index="' + ( key -1 ) + '"]' ).after( $render );
|
||||
}
|
||||
|
||||
Object.defineProperty( toAdd[I], "__jq_$el", {
|
||||
value: $render,
|
||||
writable: true,
|
||||
enumerable: false,
|
||||
configurable: true
|
||||
} );
|
||||
|
||||
//animate element
|
||||
this.__put.apply($render, [toAdd[I]]);
|
||||
}
|
||||
}
|
||||
|
||||
//set and return new array
|
||||
return Array.prototype.splice.apply(this, toProto);
|
||||
};
|
||||
result.push = function(){
|
||||
//add one or more objects to the array
|
||||
|
||||
//set the index value, if none is set make it zero
|
||||
var index = this.length || 0;
|
||||
|
||||
//loop each passed object and pass it to slice
|
||||
for (var i = 0 ; i < arguments.length; ++i) {
|
||||
this.splice( ( index + i ), 0, arguments[i] );
|
||||
}
|
||||
|
||||
//return new array length
|
||||
return this.length;
|
||||
};
|
||||
|
||||
result.unshift = function(item){
|
||||
return this.splice(0, 0, item);
|
||||
};
|
||||
|
||||
result.pop = function(){
|
||||
//remove and return array element
|
||||
|
||||
return this.splice( -1, 1 )[0];
|
||||
};
|
||||
result.reverse = function() {
|
||||
var temp = this.splice( 0 );
|
||||
Array.prototype.reverse.apply( temp );
|
||||
|
||||
for( var i = 0; i < temp.length; i++ ){
|
||||
this.push( temp[i] );
|
||||
}
|
||||
|
||||
return this;
|
||||
};
|
||||
result.shift = function() {
|
||||
return this.splice( 0, 1 )[0];
|
||||
};
|
||||
result.loop = function(){
|
||||
var temp = this[0];
|
||||
this.splice( 0,1 );
|
||||
this.push( temp );
|
||||
|
||||
return temp;
|
||||
};
|
||||
result.loopUp = function(){
|
||||
var temp = this[this.length-1];
|
||||
this.splice( -1, 1 );
|
||||
this.splice( 0, 0, temp );
|
||||
return temp;
|
||||
};
|
||||
result.indexOf = function( key, value ){
|
||||
if( !value ){
|
||||
value = arguments[0];
|
||||
key = this.__index;
|
||||
}
|
||||
for ( var index = 0; index < this.length; ++index ) {
|
||||
if( this[index][key] === value ){
|
||||
|
||||
return index;
|
||||
}
|
||||
}
|
||||
return -1;
|
||||
};
|
||||
result.update = function( key, value, update ){
|
||||
//set variables using sting for index
|
||||
|
||||
// If update is called with no index/key, assume its the 0
|
||||
if(typeof key === 'object'){
|
||||
if(this[0]){
|
||||
return this.update(0, key);
|
||||
}
|
||||
return this.splice(0, 1, key);
|
||||
}
|
||||
|
||||
if( !update ){
|
||||
update = arguments[1];
|
||||
value = arguments[0];
|
||||
key = this.__index;
|
||||
}
|
||||
|
||||
var index = this.indexOf( key, value );
|
||||
|
||||
if(index === -1) {
|
||||
return [];
|
||||
}
|
||||
var object = $.extend( true, {}, this[index], update );
|
||||
return this.splice( index, 1, object )[0];
|
||||
};
|
||||
result.__put = function(){
|
||||
this.show();
|
||||
};
|
||||
result.__take = function(){
|
||||
this.remove();
|
||||
};
|
||||
|
||||
if(!input) {
|
||||
return result;
|
||||
}
|
||||
$.each( input, function( key, value ){
|
||||
var type = typeof value;
|
||||
if( type === 'object' ){
|
||||
result.push( value );
|
||||
}else if( type === 'string' ){
|
||||
Object.defineProperty( result, "__index", {
|
||||
value: value,
|
||||
writable: true,
|
||||
enumerable: false,
|
||||
configurable: true
|
||||
} );
|
||||
} else if ( type === 'function'){
|
||||
Object.defineProperty( result, value.name, {
|
||||
value: value,
|
||||
writable: true,
|
||||
enumerable: false,
|
||||
configurable: true
|
||||
} );
|
||||
}
|
||||
} );
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
|
||||
var $this = $( element );
|
||||
var repeatId = $this.attr( 'jq-repeat' );
|
||||
var tempId = repeatId + 'Template';
|
||||
var templateId = $( '#' + tempId ).html();
|
||||
|
||||
$this.removeAttr( 'jq-repeat' );
|
||||
var template = element.outerHTML
|
||||
|
||||
$this.replaceWith( '<script type="x-tmpl-mustache" id="' + tempId + '" class="jq-repeat-' + repeatId + ' " jq-repeat-index="holder"><\/script>' );
|
||||
|
||||
Mustache.parse(templateId); // optional, speeds up future uses
|
||||
|
||||
|
||||
$.scope[repeatId] = makeArray($.scope[repeatId]);
|
||||
$.scope[repeatId].__rq_template = template
|
||||
};
|
||||
|
||||
$( document ).ready( function(){
|
||||
$( '[jq-repeat]' ).each(function(key, value){
|
||||
make(value);
|
||||
});
|
||||
|
||||
$(document).on('DOMNodeInserted', function(e) {
|
||||
if ( $(e.target).is('[jq-repeat]') ){
|
||||
make( e.target );
|
||||
}else{
|
||||
var t = $(e.target).find('[jq-repeat]');
|
||||
t.each(function(key, value){
|
||||
make(value);
|
||||
});
|
||||
}
|
||||
});
|
||||
} );
|
||||
|
||||
})(jQuery, Mustache);
|
||||
+638
-272
File diff suppressed because it is too large
Load Diff
+2
-1
@@ -4,8 +4,9 @@ const router = require('express').Router();
|
||||
const middleware = require('>/middleware/auth');
|
||||
|
||||
router.use('/auth', require('./auth'));
|
||||
router.use('/token/auth', require('./authtoken'));
|
||||
router.use('/token/auth', middleware.auth, require('./authtoken'));
|
||||
router.use('/torrent', middleware.auth, require('./transmission'));
|
||||
router.use('/search', middleware.auth, require('./search'));
|
||||
router.use('/user', middleware.auth, require('./user'));
|
||||
|
||||
module.exports = router;
|
||||
|
||||
+25
-5
@@ -3,10 +3,22 @@
|
||||
const router = require('express').Router();
|
||||
const {AuthToken} = require('>/models');
|
||||
|
||||
function ownToken(token, req){
|
||||
if(!token || token.username !== req.user.username){
|
||||
let error = new Error('AuthTokenNotFound');
|
||||
error.name = 'AuthTokenNotFound';
|
||||
error.message = 'Token not found';
|
||||
error.status = 404;
|
||||
throw error;
|
||||
}
|
||||
return token;
|
||||
}
|
||||
|
||||
router.get('/', async function(req, res, next){
|
||||
try{
|
||||
return res.json(await AuthToken.findAll());
|
||||
return res.json(await AuthToken.findAll({where:{
|
||||
username: req.user.username
|
||||
}}));
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
@@ -14,7 +26,7 @@ router.get('/', async function(req, res, next){
|
||||
|
||||
router.post('/', async function(req, res, next){
|
||||
try{
|
||||
return res.json(await AuthToken.create(req.body));
|
||||
return res.json(await AuthToken.create({...req.body, username: req.user.username}));
|
||||
}catch(error){
|
||||
console.error(error)
|
||||
next(error);
|
||||
@@ -23,6 +35,14 @@ router.post('/', async function(req, res, next){
|
||||
|
||||
router.get('/user/:username', async function(req, res, next){
|
||||
try{
|
||||
if(req.params.username !== req.user.username){
|
||||
let error = new Error('AuthTokenNotFound');
|
||||
error.name = 'AuthTokenNotFound';
|
||||
error.message = 'Token not found';
|
||||
error.status = 404;
|
||||
throw error;
|
||||
}
|
||||
|
||||
return res.json(await AuthToken.findAll({where:{
|
||||
username: req.params.username
|
||||
}}));
|
||||
@@ -33,7 +53,7 @@ router.get('/user/:username', async function(req, res, next){
|
||||
|
||||
router.get('/:token', async function(req, res, next){
|
||||
try{
|
||||
let token = await AuthToken.findByPk(req.params.token)
|
||||
let token = ownToken(await AuthToken.findByPk(req.params.token), req);
|
||||
token.dataValues.user = await token.getUser()
|
||||
|
||||
return res.json(token);
|
||||
@@ -44,7 +64,7 @@ router.get('/:token', async function(req, res, next){
|
||||
|
||||
router.put('/:token', async function(req, res, next){
|
||||
try{
|
||||
let token = await AuthToken.findByPk(req.params.token);
|
||||
let token = ownToken(await AuthToken.findByPk(req.params.token), req);
|
||||
await token.update(req.body);
|
||||
return res.json(token);
|
||||
}catch(error){
|
||||
@@ -54,7 +74,7 @@ router.put('/:token', async function(req, res, next){
|
||||
|
||||
router.delete('/:token', async function(req, res, next){
|
||||
try{
|
||||
let token = await AuthToken.findByPk(req.params.token);
|
||||
let token = ownToken(await AuthToken.findByPk(req.params.token), req);
|
||||
await token.destroy();
|
||||
|
||||
return res.json({'deleted': true});
|
||||
|
||||
+101
-56
@@ -1,28 +1,71 @@
|
||||
'use static';
|
||||
'use strict';
|
||||
|
||||
const router = require('express').Router();
|
||||
const path = require('path');
|
||||
const zlib = require('zlib');
|
||||
const fs = require('fs');
|
||||
const https = require('https');
|
||||
const http = require("http");
|
||||
const proxy = require('http-proxy-middleware');
|
||||
const { Auth } = require('>/controller/auth');
|
||||
|
||||
const inject = fs.readFileSync('./inject.html', 'utf8');
|
||||
const mainjs = fs.readFileSync('./static/main.js', 'utf8');
|
||||
const inject = fs.readFileSync(path.join(__dirname, '..', 'inject.html'), 'utf8');
|
||||
const mainjs = fs.readFileSync(path.join(__dirname, '..', 'static', 'main.js'), 'utf8');
|
||||
|
||||
// Page served to users who are not logged in. It reuses the injected front end
|
||||
// (jQuery + header partial) so the login dialog is available, but nothing is proxied.
|
||||
const loginPage = "<html><head><meta name='robots' content='noindex, nofollow'><title>Login</title></head><body>"
|
||||
+ inject
|
||||
+ "<script>window.__tbpLoginGate = true;</script>"
|
||||
+ "</body></html>";
|
||||
|
||||
function parseCookies(req){
|
||||
let out = {};
|
||||
let header = req.headers.cookie;
|
||||
if(!header) return out;
|
||||
for(let pair of header.split(';')){
|
||||
let idx = pair.indexOf('=');
|
||||
if(idx < 0) continue;
|
||||
out[pair.slice(0, idx).trim()] = decodeURIComponent(pair.slice(idx + 1).trim());
|
||||
}
|
||||
return out;
|
||||
}
|
||||
|
||||
// Keep the proxy( and everything it serves) out of search engines.
|
||||
router.get('/robots.txt', function(req, res){
|
||||
res.type('text/plain').send('User-agent: *\nDisallow: /\n');
|
||||
});
|
||||
|
||||
// Block all proxying for users who are not logged in. Page navigation can not send
|
||||
// the auth-token header, so the token is read from a cookie set by the front end.
|
||||
router.use(async function(req, res, next){
|
||||
res.set('X-Robots-Tag', 'noindex, nofollow');
|
||||
try{
|
||||
await Auth.checkToken(parseCookies(req)['auth-token']);
|
||||
return next();
|
||||
}catch(error){
|
||||
return res.status(401).send(loginPage);
|
||||
}
|
||||
});
|
||||
|
||||
// app.all("/*.js", function(req, res){res.send('')});
|
||||
|
||||
router.all('/static/main.js', function(req,res){
|
||||
res.write(mainjs);
|
||||
res.end(mainjs);
|
||||
});
|
||||
|
||||
const proxyTarget = {
|
||||
// target: "https://wtfismyip.com",
|
||||
// host: "wtfismyip.com",
|
||||
target: 'https://piratebay.party',
|
||||
host: 'piratebay.party',
|
||||
// target: 'http://172.16.0.1',
|
||||
// host: 'piratebayo3klnzokct3wt5yyxb2vpebbuyjl7m623iaxmqhsd52coid.onion'
|
||||
// target: "https://wtfismyip.com",
|
||||
// host: "wtfismyip.com",
|
||||
target: 'https://piratebay.party',
|
||||
host: 'piratebay.party',
|
||||
// target: 'http://172.16.0.1',
|
||||
// target: 'http://piratebayo3klnzokct3wt5yyxb2vpebbuyjl7m623iaxmqhsd52coid.onion',
|
||||
// host: 'piratebayo3klnzokct3wt5yyxb2vpebbuyjl7m623iaxmqhsd52coid.onion'
|
||||
// target: 'https://thepiratebay.org',
|
||||
// host: 'thepiratebay.org',
|
||||
// target: 'https://www2.thepiratebay3.to',
|
||||
// host: 'www2.thepiratebay3.to'
|
||||
}
|
||||
|
||||
function generateRegexForDomain(domain) {
|
||||
@@ -36,60 +79,62 @@ function generateRegexForDomain(domain) {
|
||||
}
|
||||
|
||||
router.all("/*", proxy({
|
||||
target: proxyTarget.target,
|
||||
agent: proxyTarget.target.startsWith('https') ? https.globalAgent : http.globalAgent,
|
||||
secure: false,
|
||||
autoRewrite: true,
|
||||
changeOrigin: true,
|
||||
target: proxyTarget.target,
|
||||
agent: proxyTarget.target.startsWith('https') ? https.globalAgent : http.globalAgent,
|
||||
secure: true,
|
||||
autoRewrite: true,
|
||||
changeOrigin: true,
|
||||
followRedirects: true,
|
||||
headers: {
|
||||
host: proxyTarget.host
|
||||
},
|
||||
selfHandleResponse: true, // so that the onProxyRes takes care of sending the response
|
||||
onProxyRes: function(proxyRes, req, res){
|
||||
headers: {
|
||||
host: proxyTarget.host,
|
||||
'Accept-Encoding': 'gzip',
|
||||
},
|
||||
selfHandleResponse: true, // so that the onProxyRes takes care of sending the response
|
||||
onProxyRes: function(proxyRes, req, res){
|
||||
|
||||
if(proxyRes.statusCode === 403 && proxyRes.headers['content-type'] &&
|
||||
proxyRes.headers['content-type'].match('html')
|
||||
){
|
||||
console.log('403')
|
||||
var url = (req.protocol + '://' + req.get('host') + req.originalUrl);
|
||||
proxyRes.headers['location'] = url.replace(/\??ckattempt\=\d+/, '');
|
||||
proxyRes.statusCode == 307
|
||||
if(proxyRes.statusCode === 403 && proxyRes.headers['content-type'] &&
|
||||
proxyRes.headers['content-type'].match('html')
|
||||
){
|
||||
console.log('403')
|
||||
var url = (req.protocol + '://' + req.get('host') + req.originalUrl);
|
||||
proxyRes.headers['location'] = url.replace(/\??ckattempt\=\d+/, '');
|
||||
proxyRes.statusCode = 307;
|
||||
|
||||
return res.end()
|
||||
}
|
||||
return res.end()
|
||||
}
|
||||
|
||||
for(let key of Object.keys(proxyRes.headers)){
|
||||
if(['content-encoding'].includes(key)) continue;
|
||||
// res.set(key, proxyRes.headers[key].toString().replace('http://', 'https://'))
|
||||
}
|
||||
for(let key of Object.keys(proxyRes.headers)){
|
||||
if(['content-encoding'].includes(key)) continue;
|
||||
// res.set(key, proxyRes.headers[key].toString().replace('http://', 'https://'))
|
||||
}
|
||||
|
||||
let body = new Buffer('');
|
||||
proxyRes.on('error', function(e){
|
||||
console.error('ERROR!', e)
|
||||
});
|
||||
let body = Buffer.alloc(0);
|
||||
proxyRes.on('error', function(e){
|
||||
console.error('ERROR!', e)
|
||||
});
|
||||
|
||||
proxyRes.on('data', function(data){
|
||||
body = Buffer.concat([body, data]);
|
||||
});
|
||||
proxyRes.on('data', function(data){
|
||||
body = Buffer.concat([body, data]);
|
||||
});
|
||||
|
||||
proxyRes.on('end', function(){
|
||||
body = proxyRes.headers['content-encoding'] === 'gzip' ? zlib.gunzipSync(body).toString('utf8') : body;
|
||||
body = proxyRes.headers['content-encoding'] === 'br' ? zlib.brotliDecompressSync(body).toString('utf8') : body;
|
||||
if(proxyRes.statusCode === 200 &&
|
||||
proxyRes.headers['content-type'] &&
|
||||
proxyRes.headers['content-type'].match('html')
|
||||
){
|
||||
body = body.toString().replace(/<\s*script[^]*?script>/igm, '');
|
||||
body = body.replace(generateRegexForDomain(proxyTarget.host), '');
|
||||
body = body.replace(/<\s*iframe[^]*?iframe>/igm, '');
|
||||
body = body.replace("</html>", '');
|
||||
body = body+inject+"</html>";
|
||||
}
|
||||
res.status(proxyRes.statusCode).end(body);
|
||||
});
|
||||
proxyRes.on('end', function(){
|
||||
// console.log("proxyRes.headers['content-encoding']", proxyRes.headers['content-encoding']);
|
||||
body = proxyRes.headers['content-encoding'] === 'gzip' ? zlib.gunzipSync(body).toString('utf8') : body;
|
||||
body = proxyRes.headers['content-encoding'] === 'br' ? zlib.brotliDecompressSync(body).toString('utf8') : body;
|
||||
if(proxyRes.statusCode === 200 &&
|
||||
proxyRes.headers['content-type'] &&
|
||||
proxyRes.headers['content-type'].match('html')
|
||||
){
|
||||
body = body.toString().replace(/<\s*script[^]*?script>/igm, '');
|
||||
body = body.replace(generateRegexForDomain(proxyTarget.host), '');
|
||||
body = body.replace(/<\s*iframe[^]*?iframe>/igm, '');
|
||||
body = body.replace("</html>", '');
|
||||
body = body+inject+"</html>";
|
||||
}
|
||||
res.status(proxyRes.statusCode).end(body);
|
||||
});
|
||||
|
||||
}
|
||||
}
|
||||
}));
|
||||
|
||||
module.exports = router;
|
||||
@@ -0,0 +1,92 @@
|
||||
'use strict';
|
||||
|
||||
const router = require('express').Router();
|
||||
const search = require('>/controller/search');
|
||||
const { Torrent, Wanted } = require('>/models');
|
||||
|
||||
// Step 1: fuzzy query -> a few TMDB title candidates( with posters) to confirm.
|
||||
router.get('/title', async function(req, res, next){
|
||||
try{
|
||||
res.json({results: await search.tmdbSearch(req.query.q)});
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
// Step 2: confirmed title -> TPB search + LLM-curated release options.
|
||||
router.post('/releases', async function(req, res, next){
|
||||
try{
|
||||
let result = await search.findReleases(req.body.tmdbId, req.body.mediaType);
|
||||
|
||||
// Nothing good yet( e.g. unreleased) -> remember it and grab it later.
|
||||
if(!result.options.length && result.mediaType !== 'tv' && result.tmdbId){
|
||||
await Wanted.upsert({
|
||||
tmdbId: String(result.tmdbId),
|
||||
mediaType: result.mediaType,
|
||||
title: result.title,
|
||||
year: result.year,
|
||||
status: 'wanted',
|
||||
requestedBy: req.user.username,
|
||||
});
|
||||
result.remembered = true;
|
||||
}
|
||||
|
||||
res.json(result);
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
// Wishlist: things we couldn't find yet and are watching for.
|
||||
router.get('/wanted', async function(req, res, next){
|
||||
try{
|
||||
res.json(await Wanted.findAll({ order: [['createdAt', 'DESC']] }));
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
router.delete('/wanted/:tmdbId', async function(req, res, next){
|
||||
try{
|
||||
let wanted = await Wanted.findByPk(req.params.tmdbId);
|
||||
if(wanted) await wanted.destroy();
|
||||
res.json({ deleted: true });
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
// TV: which seasons you have vs are missing (Emby inventory vs TMDB counts).
|
||||
router.get('/seasons', async function(req, res, next){
|
||||
try{
|
||||
res.json(await search.seasonPlan(req.query.tmdbId));
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
// TV: queue the best complete-season pack for every missing/incomplete season.
|
||||
router.post('/fill', async function(req, res, next){
|
||||
try{
|
||||
let plan = await search.fillMissing(req.body.tmdbId);
|
||||
let queued = [];
|
||||
for(let pick of plan.picks){
|
||||
try{
|
||||
await Torrent.create({
|
||||
magnetLink: pick.magnetLink,
|
||||
isPrivate: false,
|
||||
added_by: req.user.username,
|
||||
category: pick.category,
|
||||
});
|
||||
queued.push({ season: pick.season, name: pick.name });
|
||||
}catch(error){
|
||||
plan.skipped.push({ season: pick.season, reason: error.message });
|
||||
}
|
||||
}
|
||||
res.json({ title: plan.title, year: plan.year, queued, skipped: plan.skipped });
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
module.exports = router;
|
||||
+55
-14
@@ -1,16 +1,33 @@
|
||||
'use static';
|
||||
'use strict';
|
||||
|
||||
const router = require('express').Router();
|
||||
const {Torrent} = require('>/models');
|
||||
const organize = require('>/controller/organize');
|
||||
|
||||
function authTorrent(torrent, req){
|
||||
if(torrent && torrent.isPrivate && torrent.added_by !== req.user.username){
|
||||
let error = new Error('TorrentNotFound');
|
||||
error.name = 'TorrentNotFound';
|
||||
error.message = 'Torrent not found';
|
||||
error.status = 404;
|
||||
throw error;
|
||||
}
|
||||
return torrent;
|
||||
}
|
||||
|
||||
router.get('/', async function(req, res, next){
|
||||
try{
|
||||
let username = req.query.username || req.user.username;
|
||||
let where = {added_by: username};
|
||||
if(username !== req.user.username) where.isPrivate = false;
|
||||
|
||||
res.json({results: await Torrent.findAll({
|
||||
where:{added_by: req.query.username || req.user.username},
|
||||
// limit: req.query.limit,
|
||||
// offset: req.query.offset,
|
||||
// order: [[req.query.orderKey || "created_at", (req.query.asc && "ASC")]],
|
||||
// from: {$between: [startDate, endDate]}
|
||||
where,
|
||||
limit: req.query.limit,
|
||||
offset: req.query.offset,
|
||||
order: [
|
||||
['createdAt', 'DESC'],
|
||||
],
|
||||
})});
|
||||
}catch(error){
|
||||
next(error);
|
||||
@@ -33,9 +50,9 @@ router.get('/server', async function(req, res, next){
|
||||
}
|
||||
});
|
||||
|
||||
router.get("/:id", async function(req, res, next){
|
||||
router.get("/:hashString", async function(req, res, next){
|
||||
try{
|
||||
let torrent = await Torrent.findByPk(req.params.id);
|
||||
let torrent = authTorrent(await Torrent.findByPk(req.params.hashString), req);
|
||||
if('latest' in req.query){
|
||||
torrent = await torrent.getTorrentData();
|
||||
}
|
||||
@@ -45,9 +62,9 @@ router.get("/:id", async function(req, res, next){
|
||||
}
|
||||
});
|
||||
|
||||
router.delete("/:id", async function(req, res, next){
|
||||
router.delete("/:hashString", async function(req, res, next){
|
||||
try{
|
||||
let torrent = await Torrent.findByPk(req.params.id);
|
||||
let torrent = authTorrent(await Torrent.findByPk(req.params.hashString), req);
|
||||
|
||||
res.json({result: torrent, activity: await torrent.destroy()});
|
||||
}catch(error){
|
||||
@@ -55,9 +72,9 @@ router.delete("/:id", async function(req, res, next){
|
||||
}
|
||||
});
|
||||
|
||||
router.post("/:id/stop", async function(req, res, next){
|
||||
router.post("/:hashString/stop", async function(req, res, next){
|
||||
try{
|
||||
let torrent = await Torrent.findByPk(req.params.id);
|
||||
let torrent = authTorrent(await Torrent.findByPk(req.params.hashString), req);
|
||||
|
||||
res.json({result: torrent, activity: await torrent.stop()});
|
||||
}catch(error){
|
||||
@@ -65,9 +82,9 @@ router.post("/:id/stop", async function(req, res, next){
|
||||
}
|
||||
});
|
||||
|
||||
router.post("/:id/start", async function(req, res, next){
|
||||
router.post("/:hashString/start", async function(req, res, next){
|
||||
try{
|
||||
let torrent = await Torrent.findByPk(req.params.id);
|
||||
let torrent = authTorrent(await Torrent.findByPk(req.params.hashString), req);
|
||||
|
||||
res.json({result: torrent, activity: await torrent.start()});
|
||||
}catch(error){
|
||||
@@ -75,6 +92,30 @@ router.post("/:id/start", async function(req, res, next){
|
||||
}
|
||||
});
|
||||
|
||||
// Manually (re)run organization for a finished torrent.
|
||||
router.post("/:hashString/organize", async function(req, res, next){
|
||||
try{
|
||||
let torrent = authTorrent(await Torrent.findByPk(req.params.hashString), req);
|
||||
if(!torrent){ let e = new Error('TorrentNotFound'); e.status = 404; throw e; }
|
||||
|
||||
res.json(await organize.fileTorrent(torrent));
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
// Correct a wrong match: re-file an already-organized torrent under a chosen TMDB id.
|
||||
router.post("/:hashString/organize/match", async function(req, res, next){
|
||||
try{
|
||||
let torrent = authTorrent(await Torrent.findByPk(req.params.hashString), req);
|
||||
if(!torrent){ let e = new Error('TorrentNotFound'); e.status = 404; throw e; }
|
||||
|
||||
res.json(await organize.refileWithMatch(torrent, req.body.tmdbId, req.body.mediaType));
|
||||
}catch(error){
|
||||
next(error);
|
||||
}
|
||||
});
|
||||
|
||||
|
||||
|
||||
module.exports = router;
|
||||
|
||||
Reference in New Issue
Block a user