feat: add desktop connector distributions
This commit is contained in:
+164
@@ -0,0 +1,164 @@
|
||||
import fs from 'node:fs';
|
||||
import path from 'node:path';
|
||||
import { guessit } from 'guessit-js';
|
||||
|
||||
const video = /\.(mkv|mp4|webm|m4v)$/i;
|
||||
const text = (source, tag) => source.match(new RegExp(`<${tag}[^>]*>([^<]+)</${tag}>`, 'i'))?.[1]?.trim();
|
||||
const nfoFor = (mediaRoot, file) => {
|
||||
const directory = path.dirname(path.join(mediaRoot, file));
|
||||
const stem = path.basename(file, path.extname(file));
|
||||
const nfoPath = [path.join(directory, `${stem}.nfo`), path.join(directory, 'movie.nfo')].find(fs.existsSync);
|
||||
if (!nfoPath) return {};
|
||||
try {
|
||||
const source = fs.readFileSync(nfoPath, 'utf8');
|
||||
return {
|
||||
title: text(source, 'title'),
|
||||
year: Number(text(source, 'year')) || null,
|
||||
imdbId: source.match(/<uniqueid[^>]*type=["']imdb["'][^>]*>(tt\d{7,10})<\/uniqueid>/i)?.[1] || text(source, 'id')?.match(/^tt\d{7,10}$/i)?.[0]?.toLowerCase(),
|
||||
};
|
||||
} catch { return {}; }
|
||||
};
|
||||
|
||||
export function createLibrary(mediaRoot, indexPath, tmdbApiKey) {
|
||||
const metadataCache = new Map();
|
||||
const load = () => {
|
||||
try { return JSON.parse(fs.readFileSync(indexPath, 'utf8')); } catch { return { mappings: {} }; }
|
||||
};
|
||||
const save = index => {
|
||||
fs.mkdirSync(path.dirname(indexPath), { recursive: true });
|
||||
fs.writeFileSync(indexPath, `${JSON.stringify(index, null, 2)}\n`);
|
||||
};
|
||||
const discover = async item => {
|
||||
if (item.imdbId || !tmdbApiKey || !item.title) return item;
|
||||
const type = item.type === 'episode' ? 'tv' : 'movie';
|
||||
const query = new URLSearchParams({ api_key: tmdbApiKey, query: item.title });
|
||||
if (item.year) query.set(type === 'movie' ? 'year' : 'first_air_date_year', item.year);
|
||||
const results = await fetch(`https://api.themoviedb.org/3/search/${type}?${query}`).then(res => res.ok ? res.json() : { results: [] });
|
||||
const candidate = results.results?.[0];
|
||||
if (!candidate) return item;
|
||||
const candidateTitle = candidate.title || candidate.name || '';
|
||||
const candidateYear = (candidate.release_date || candidate.first_air_date || '').slice(0, 4);
|
||||
const normalize = value => value.toLowerCase().replace(/[^a-z0-9]/g, '');
|
||||
const confident = Boolean(item.year) && normalize(candidateTitle) === normalize(item.title) && candidateYear === String(item.year);
|
||||
if (!confident) return { ...item, candidate: { title: candidateTitle, year: candidateYear, tmdbId: candidate.id } };
|
||||
const external = await fetch(`https://api.themoviedb.org/3/${type}/${candidate.id}/external_ids?api_key=${tmdbApiKey}`).then(res => res.ok ? res.json() : {});
|
||||
return { ...item, suggestedImdbId: external.imdb_id || null, autoMapped: Boolean(external.imdb_id), candidate: { title: candidateTitle, year: candidateYear, tmdbId: candidate.id } };
|
||||
};
|
||||
const tmdb = async (endpoint, query = {}) => {
|
||||
if (!tmdbApiKey) return null;
|
||||
const params = new URLSearchParams({ api_key: tmdbApiKey, ...query });
|
||||
const response = await fetch(`https://api.themoviedb.org/3${endpoint}?${params}`);
|
||||
return response.ok ? response.json() : null;
|
||||
};
|
||||
const imageUrl = (image, size) => image ? `https://image.tmdb.org/t/p/${size}${image}` : null;
|
||||
const metadataFor = async item => {
|
||||
const key = `${item.imdbId || ''}:${item.title}:${item.year || ''}:${item.type}`;
|
||||
if (metadataCache.has(key)) return metadataCache.get(key);
|
||||
const pending = (async () => {
|
||||
if (!tmdbApiKey) return item;
|
||||
const type = item.type === 'episode' ? 'tv' : 'movie';
|
||||
let result;
|
||||
if (item.imdbId) {
|
||||
const external = await tmdb(`/find/${item.imdbId}`, { external_source: 'imdb_id' });
|
||||
result = external?.[type === 'tv' ? 'tv_results' : 'movie_results']?.[0];
|
||||
}
|
||||
if (!result) {
|
||||
const query = { query: item.title };
|
||||
if (item.year) query[type === 'tv' ? 'first_air_date_year' : 'year'] = item.year;
|
||||
result = (await tmdb(`/search/${type}`, query))?.results?.[0];
|
||||
}
|
||||
if (!result?.id) return item;
|
||||
const detail = await tmdb(`/${type}/${result.id}`, { append_to_response: 'external_ids' });
|
||||
if (!detail) return item;
|
||||
const releaseDate = detail.release_date || detail.first_air_date || '';
|
||||
return {
|
||||
...item,
|
||||
imdbId: item.imdbId || detail.external_ids?.imdb_id || null,
|
||||
posterUrl: imageUrl(detail.poster_path, 'w500'),
|
||||
backgroundUrl: imageUrl(detail.backdrop_path, 'w1280'),
|
||||
overview: detail.overview || null,
|
||||
description: detail.overview || item.description || null,
|
||||
runtime: detail.runtime || detail.episode_run_time?.[0] || null,
|
||||
genres: detail.genres?.map(genre => genre.name).filter(Boolean) || [],
|
||||
releaseYear: Number(releaseDate.slice(0, 4)) || item.year || null,
|
||||
rating: typeof detail.vote_average === 'number' ? detail.vote_average : null,
|
||||
};
|
||||
})().catch(() => item);
|
||||
metadataCache.set(key, pending);
|
||||
return pending;
|
||||
};
|
||||
const scan = async () => {
|
||||
const index = load();
|
||||
const files = fs.existsSync(mediaRoot) ? fs.readdirSync(mediaRoot, { recursive: true }).filter(file => video.test(file)) : [];
|
||||
const present = new Set(files);
|
||||
let changed = false;
|
||||
for (const file of Object.keys(index.mappings)) {
|
||||
if (!present.has(file) && !index.mappings[file].missing) { index.mappings[file].missing = true; changed = true; }
|
||||
}
|
||||
const items = files.map(file => {
|
||||
const stat = fs.statSync(path.join(mediaRoot, file));
|
||||
const fingerprint = `${stat.size}:${Math.round(stat.mtimeMs)}`;
|
||||
const parsed = guessit(path.basename(file));
|
||||
const previous = Object.entries(index.mappings).find(([oldPath, value]) => oldPath !== file && value.missing && value.fingerprint === fingerprint)?.[1];
|
||||
const mapping = index.mappings[file] || previous || {};
|
||||
const nfo = nfoFor(mediaRoot, file);
|
||||
if (mapping.fingerprint !== fingerprint || mapping.missing) {
|
||||
index.mappings[file] = { ...mapping, fingerprint, missing: false, updatedAt: mapping.updatedAt || new Date().toISOString() };
|
||||
changed = true;
|
||||
}
|
||||
return {
|
||||
file,
|
||||
id: Buffer.from(file).toString('base64url'),
|
||||
title: mapping.title || nfo.title || parsed.title || path.basename(file, path.extname(file)),
|
||||
year: mapping.year || nfo.year || parsed.year || null,
|
||||
imdbId: mapping.imdbId || nfo.imdbId || parsed.imdb_id || file.match(/\btt\d{7,10}\b/i)?.[0]?.toLowerCase() || null,
|
||||
type: parsed.type === 'episode' ? 'episode' : 'video',
|
||||
season: mapping.season || parsed.season || null,
|
||||
episode: mapping.episode || parsed.episode || null,
|
||||
mapped: Boolean(mapping.imdbId),
|
||||
};
|
||||
});
|
||||
const discovered = await Promise.all(items.map(discover));
|
||||
for (const item of discovered) {
|
||||
if (!item.autoMapped || index.mappings[item.file]?.imdbId) continue;
|
||||
index.mappings[item.file] = {
|
||||
title: item.title,
|
||||
year: item.year,
|
||||
imdbId: item.suggestedImdbId,
|
||||
source: 'tmdb-auto',
|
||||
confidence: 'high',
|
||||
missing: false,
|
||||
fingerprint: index.mappings[item.file]?.fingerprint,
|
||||
updatedAt: new Date().toISOString(),
|
||||
};
|
||||
changed = true;
|
||||
}
|
||||
if (changed) save(index);
|
||||
return discovered;
|
||||
};
|
||||
return {
|
||||
scan,
|
||||
startReconciler(seconds = 900) {
|
||||
const refresh = () => scan().catch(error => console.error('Library reconciliation failed', error));
|
||||
refresh();
|
||||
setInterval(refresh, seconds * 1000).unref();
|
||||
try {
|
||||
let pending;
|
||||
fs.watch(mediaRoot, () => { clearTimeout(pending); pending = setTimeout(refresh, 1_000); });
|
||||
} catch (error) {
|
||||
console.warn('Library watcher unavailable; periodic reconciliation remains active', error.message);
|
||||
}
|
||||
},
|
||||
saveMapping(file, mapping) {
|
||||
const index = load();
|
||||
index.mappings[file] = { ...index.mappings[file], ...mapping, missing: false, updatedAt: new Date().toISOString() };
|
||||
save(index);
|
||||
},
|
||||
async catalog() {
|
||||
return Promise.all((await scan()).map(metadataFor));
|
||||
},
|
||||
async metadata(id) {
|
||||
return (await this.catalog()).find(item => item.id === id) || null;
|
||||
},
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user