tony
74deabd4b1
Move the Mac Tauri UI (pro.cleanflix.connector) into this repo under desktop/. Bundled Node runtime stays gitignored; package scripts fetch it when building.
198 lines
9.2 KiB
JavaScript
198 lines
9.2 KiB
JavaScript
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 firstText = value => Array.isArray(value)
|
|
? value.find(item => typeof item === 'string' && item.trim()) || ''
|
|
: typeof value === 'string' ? value : '';
|
|
const year = value => {
|
|
const parsed = Number(Array.isArray(value) ? value[0] : value);
|
|
return Number.isInteger(parsed) ? parsed : null;
|
|
};
|
|
const mapWithConcurrency = async (items, limit, mapper) => {
|
|
const results = new Array(items.length);
|
|
let next = 0;
|
|
const worker = async () => {
|
|
while (next < items.length) {
|
|
const index = next++;
|
|
results[index] = await mapper(items[index]);
|
|
}
|
|
};
|
|
await Promise.all(Array.from({ length: Math.min(limit, items.length) }, worker));
|
|
return results;
|
|
};
|
|
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();
|
|
let catalogCache = null;
|
|
let catalogRefresh = null;
|
|
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)
|
|
? (await fs.promises.readdir(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 = await mapWithConcurrency(files, 16, async file => {
|
|
const stat = await fs.promises.stat(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: firstText(mapping.title) || firstText(nfo.title) || firstText(parsed.title) || path.basename(file, path.extname(file)),
|
|
year: year(mapping.year) || year(nfo.year) || year(parsed.year),
|
|
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;
|
|
};
|
|
const refreshCatalog = async () => {
|
|
if (catalogRefresh) return catalogRefresh;
|
|
catalogRefresh = (async () => {
|
|
const items = await Promise.all((await scan()).map(metadataFor));
|
|
catalogCache = items;
|
|
return items;
|
|
})().finally(() => { catalogRefresh = null; });
|
|
return catalogRefresh;
|
|
};
|
|
return {
|
|
scan,
|
|
startReconciler(seconds = 900) {
|
|
const refresh = () => refreshCatalog().catch(error => console.error('Library reconciliation failed', error));
|
|
// Let the server start before scanning a large remote-backed library.
|
|
setImmediate(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 catalogCache || refreshCatalog();
|
|
},
|
|
async metadata(id) {
|
|
return (await this.catalog()).find(item => item.id === id) || null;
|
|
},
|
|
};
|
|
}
|