56 lines
2.3 KiB
TypeScript
56 lines
2.3 KiB
TypeScript
// lidarr's events. The tool runtime imports this once the broker is bound. It watches the download
|
|
// queue and turns its comings and goings into mesh events.
|
|
//
|
|
// Emits (novox/hq ADR 0041/0042):
|
|
// module.lidarr.album.grabbed — a release entered the queue (Lidarr grabbed it)
|
|
// module.lidarr.download.completed — a release left the queue, imported. The download.completed
|
|
// routing key matches what a media consumer subscribes to
|
|
// (module.*.download.completed) to rescan its library.
|
|
// Consumes: none.
|
|
//
|
|
// The queue is polled and diffed, primed silently on the first look (like plex's index.ts) so a
|
|
// restart mid-download does not re-announce everything already in flight as freshly grabbed.
|
|
|
|
import { emit } from "@novox/mesh-sdk/events";
|
|
import { LidarrClient, type LidarrQueueItem } from "./client.js";
|
|
|
|
const lidarr = LidarrClient.fromEnv();
|
|
|
|
// Lidarr removes an item from the queue once it has been imported; a "warning"/"failed" status is
|
|
// how a stuck or broken grab shows itself, so we do not call those a completion when they vanish.
|
|
const FAILED_STATUSES = new Set(["failed", "warning"]);
|
|
|
|
const inQueue = new Map<number, LidarrQueueItem>();
|
|
let primed = false;
|
|
|
|
async function pollQueue(): Promise<void> {
|
|
const { items } = await lidarr.getQueue();
|
|
const now = new Map(items.map((i) => [i.id, i]));
|
|
|
|
if (primed) {
|
|
// Entered the queue since last look — Lidarr grabbed a release.
|
|
for (const [id, item] of now) {
|
|
if (!inQueue.has(id)) await emit("module.lidarr.album.grabbed", { title: item.title, status: item.status });
|
|
}
|
|
// Left the queue — imported and done, unless it was last seen failing.
|
|
for (const [id, item] of inQueue) {
|
|
if (!now.has(id) && !FAILED_STATUSES.has(item.status)) {
|
|
await emit("module.lidarr.download.completed", { title: item.title });
|
|
}
|
|
}
|
|
}
|
|
|
|
inQueue.clear();
|
|
for (const [id, item] of now) inQueue.set(id, item);
|
|
primed = true;
|
|
}
|
|
|
|
const tick = (fn: () => Promise<void>, everyMs: number): void => {
|
|
const run = (): void => void fn().catch((err) => console.error(`[lidarr] ${err}`));
|
|
setInterval(run, everyMs);
|
|
run();
|
|
};
|
|
tick(pollQueue, 30_000);
|
|
|
|
console.log("[lidarr] watching the download queue, emitting grabs and completions");
|