@psychout98/tadaima 1.0.0

package/dist/updater.js

@@ -0,0 +1,191 @@
+import { createHash } from "node:crypto";
+import { createWriteStream, readFileSync, copyFileSync, chmodSync, renameSync } from "node:fs";
+import { pipeline } from "node:stream/promises";
+import { Readable } from "node:stream";
+import { dirname, join, extname } from "node:path";
+import { config } from "./config.js";
+import { GITHUB_RELEASES_API, getAssetNameForPlatform, } from "@tadaima/shared";
+let cachedETag = "";
+/**
+ * Try the relay's /api/version endpoint first for a quick version check.
+ * Falls back to the full GitHub API if the relay is unreachable.
+ */
+async function checkViaRelay(currentVersion) {
+    const relayUrl = config.get("relay");
+    if (!relayUrl)
+        return null;
+    try {
+        const res = await fetch(`${relayUrl}/api/version`, {
+            signal: AbortSignal.timeout(5000),
+        });
+        if (!res.ok)
+            return null;
+        const data = (await res.json());
+        if (data.latest && isNewer(data.latest, currentVersion)) {
+            return data.latest;
+        }
+        return null;
+    }
+    catch {
+        return null;
+    }
+}
+/**
+ * Check the GitHub Releases API for a newer version.
+ * Returns null if already up-to-date or unsupported platform.
+ */
+export async function checkForUpdate(currentVersion) {
+    const assetName = getAssetNameForPlatform();
+    if (!assetName)
+        return null;
+    // Quick check via relay first
+    const relayLatest = await checkViaRelay(currentVersion);
+    // If relay says we're up-to-date, skip the full GitHub API call
+    if (relayLatest === null) {
+        // Relay either unreachable or says up-to-date; still try GitHub for authoritative answer
+    }
+    const headers = {
+        Accept: "application/vnd.github+json",
+        "User-Agent": `tadaima-agent/${currentVersion}`,
+    };
+    if (cachedETag) {
+        headers["If-None-Match"] = cachedETag;
+    }
+    const res = await fetch(GITHUB_RELEASES_API, { headers });
+    if (res.status === 304)
+        return null; // not modified
+    if (!res.ok) {
+        throw new Error(`GitHub API returned ${res.status}: ${res.statusText}`);
+    }
+    const etag = res.headers.get("etag");
+    if (etag)
+        cachedETag = etag;
+    const data = (await res.json());
+    const latestVersion = data.tag_name.replace(/^v/, "");
+    if (!isNewer(latestVersion, currentVersion))
+        return null;
+    const binaryAsset = data.assets.find((a) => a.name === assetName);
+    const checksumAsset = data.assets.find((a) => a.name === "checksums.sha256");
+    if (!binaryAsset || !checksumAsset)
+        return null;
+    config.set("lastUpdateCheck", new Date().toISOString());
+    return {
+        version: latestVersion,
+        downloadUrl: binaryAsset.browser_download_url,
+        checksumUrl: checksumAsset.browser_download_url,
+    };
+}
+/**
+ * Download, verify, and replace the current binary with the update.
+ */
+export async function applyUpdate(update) {
+    const configDir = dirname(config.path);
+    const ext = extname(process.execPath);
+    const tmpPath = join(configDir, `tadaima-agent-update-${update.version}${ext}`);
+    const backupPath = join(configDir, `tadaima-agent-previous${ext}`);
+    // Download binary
+    const binRes = await fetch(update.downloadUrl);
+    if (!binRes.ok || !binRes.body) {
+        throw new Error(`Failed to download binary: ${binRes.status}`);
+    }
+    const ws = createWriteStream(tmpPath);
+    await pipeline(Readable.fromWeb(binRes.body), ws);
+    // Download and verify checksum
+    const csRes = await fetch(update.checksumUrl);
+    if (!csRes.ok) {
+        throw new Error(`Failed to download checksums: ${csRes.status}`);
+    }
+    const checksumText = await csRes.text();
+    const assetName = getAssetNameForPlatform();
+    const expectedLine = checksumText.split("\n").find((l) => l.includes(assetName));
+    if (!expectedLine) {
+        throw new Error(`No checksum found for ${assetName}`);
+    }
+    const expectedHash = expectedLine.trim().split(/\s+/)[0].toLowerCase();
+    const actualHash = createHash("sha256")
+        .update(readFileSync(tmpPath))
+        .digest("hex");
+    if (actualHash !== expectedHash) {
+        throw new Error(`Checksum mismatch: expected ${expectedHash}, got ${actualHash}`);
+    }
+    // Make executable
+    if (process.platform !== "win32") {
+        chmodSync(tmpPath, 0o755);
+    }
+    // Backup current binary
+    try {
+        copyFileSync(process.execPath, backupPath);
+        config.set("previousBinaryPath", backupPath);
+    }
+    catch {
+        // Backup may fail if execPath is read-only (npm global); non-fatal
+    }
+    // Atomic rename
+    renameSync(tmpPath, process.execPath);
+    // If running as a service, exit so the service manager restarts with the new binary
+    if (process.env.TADAIMA_DAEMON || isRunningAsService()) {
+        console.log(`Updated to v${update.version}. Exiting for service restart.`);
+        process.exit(0);
+    }
+    else {
+        console.log(`Update to v${update.version} downloaded. Restart to apply.`);
+    }
+}
+/**
+ * Check if enough time has passed since the last update check.
+ */
+export function shouldCheckNow(trigger) {
+    const last = config.get("lastUpdateCheck");
+    if (!last)
+        return true;
+    const elapsed = Date.now() - new Date(last).getTime();
+    const ONE_HOUR = 60 * 60 * 1000;
+    const TWENTY_FOUR_HOURS = 24 * ONE_HOUR;
+    if (trigger === "startup")
+        return elapsed > ONE_HOUR;
+    return elapsed > TWENTY_FOUR_HOURS;
+}
+/**
+ * Restore the previous binary from backup.
+ */
+export function rollback() {
+    const prev = config.get("previousBinaryPath");
+    if (!prev) {
+        console.log("No previous binary to roll back to.");
+        return;
+    }
+    try {
+        copyFileSync(prev, process.execPath);
+        console.log(`Rolled back to previous binary from ${prev}.`);
+        console.log("Restart the agent to use the restored version.");
+    }
+    catch (err) {
+        console.error("Rollback failed:", err instanceof Error ? err.message : err);
+    }
+}
+/**
+ * Log an update advisory for npm/Docker users who can't self-update.
+ */
+export function logUpdateAdvisory(currentVersion, latestVersion) {
+    console.log(`\nTadaima v${latestVersion} is available (you have v${currentVersion}).`);
+    console.log("  npm:    npm update -g @tadaima/agent");
+    console.log("  Docker: docker compose pull && docker compose up -d\n");
+}
+// ── Helpers ──────────────────────────────────────────────────────
+function isNewer(latest, current) {
+    const [lMaj, lMin, lPatch] = latest.split(".").map(Number);
+    const [cMaj, cMin, cPatch] = current.split(".").map(Number);
+    if (lMaj !== cMaj)
+        return lMaj > cMaj;
+    if (lMin !== cMin)
+        return lMin > cMin;
+    return lPatch > cPatch;
+}
+function isRunningAsService() {
+    // launchd/systemd/Windows Service detection heuristics
+    return (!!process.env.INVOCATION_ID || // systemd
+        process.ppid === 1 || // launchd (parent is launchd PID 1)
+        !!process.env.TADAIMA_SERVICE // explicit flag
+    );
+}
+//# sourceMappingURL=updater.js.map

package/dist/updater.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"updater.js","sourceRoot":"","sources":["../src/updater.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,SAAS,CAAC;AAC/F,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACnD,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AACrC,OAAO,EACL,mBAAmB,EACnB,uBAAuB,GACxB,MAAM,iBAAiB,CAAC;AAQzB,IAAI,UAAU,GAAG,EAAE,CAAC;AAEpB;;;GAGG;AACH,KAAK,UAAU,aAAa,CAAC,cAAsB;IACjD,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACrC,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAE3B,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,GAAG,QAAQ,cAAc,EAAE;YACjD,MAAM,EAAE,WAAW,CAAC,OAAO,CAAC,IAAI,CAAC;SAClC,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,CAAC,EAAE;YAAE,OAAO,IAAI,CAAC;QACzB,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAA8B,CAAC;QAC7D,IAAI,IAAI,CAAC,MAAM,IAAI,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,cAAc,CAAC,EAAE,CAAC;YACxD,OAAO,IAAI,CAAC,MAAM,CAAC;QACrB,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,cAAsB;IACzD,MAAM,SAAS,GAAG,uBAAuB,EAAE,CAAC;IAC5C,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAE5B,8BAA8B;IAC9B,MAAM,WAAW,GAAG,MAAM,aAAa,CAAC,cAAc,CAAC,CAAC;IACxD,gEAAgE;IAChE,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;QACzB,yFAAyF;IAC3F,CAAC;IAED,MAAM,OAAO,GAA2B;QACtC,MAAM,EAAE,6BAA6B;QACrC,YAAY,EAAE,iBAAiB,cAAc,EAAE;KAChD,CAAC;IACF,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,CAAC;IACxC,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,mBAAmB,EAAE,EAAE,OAAO,EAAE,CAAC,CAAC;IAE1D,IAAI,GAAG,CAAC,MAAM,KAAK,GAAG;QAAE,OAAO,IAAI,CAAC,CAAC,eAAe;IACpD,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,uBAAuB,GAAG,CAAC,MAAM,KAAK,GAAG,CAAC,UAAU,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IACrC,IAAI,IAAI;QAAE,UAAU,GAAG,IAAI,CAAC;IAE5B,MAAM,IAAI,GAAG,CAAC,MAAM,GAAG,CAAC,IAAI,EAAE,CAG7B,CAAC;IAEF,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IACtD,IAAI,CAAC,OAAO,CAAC,aAAa,EAAE,cAAc,CAAC;QAAE,OAAO,IAAI,CAAC;IAEzD,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,kBAAkB,CAAC,CAAC;IAE7E,IAAI,CAAC,WAAW,IAAI,CAAC,aAAa;QAAE,OAAO,IAAI,CAAC;IAEhD,MAAM,CAAC,GAAG,CAAC,iBAAiB,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC,CAAC;IAExD,OAAO;QACL,OAAO,EAAE,aAAa;QACtB,WAAW,EAAE,WAAW,CAAC,oBAAoB;QAC7C,WAAW,EAAE,aAAa,CAAC,oBAAoB;KAChD,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,MAAoB;IACpD,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACvC,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACtC,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,EAAE,wBAAwB,MAAM,CAAC,OAAO,GAAG,GAAG,EAAE,CAAC,CAAC;IAChF,MAAM,UAAU,GAAG,IAAI,CAAC,SAAS,EAAE,yBAAyB,GAAG,EAAE,CAAC,CAAC;IAEnE,kBAAkB;IAClB,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC/C,IAAI,CAAC,MAAM,CAAC,EAAE,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;QAC/B,MAAM,IAAI,KAAK,CAAC,8BAA8B,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;IACjE,CAAC;IACD,MAAM,EAAE,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC;IACtC,MAAM,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,CAAC,IAA2C,CAAC,EAAE,EAAE,CAAC,CAAC;IAEzF,+BAA+B;IAC/B,MAAM,KAAK,GAAG,MAAM,KAAK,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC9C,IAAI,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;QACd,MAAM,IAAI,KAAK,CAAC,iCAAiC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IACnE,CAAC;IACD,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,IAAI,EAAE,CAAC;IACxC,MAAM,SAAS,GAAG,uBAAuB,EAAG,CAAC;IAC7C,MAAM,YAAY,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;IACjF,IAAI,CAAC,YAAY,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,yBAAyB,SAAS,EAAE,CAAC,CAAC;IACxD,CAAC;IACD,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IAEvE,MAAM,UAAU,GAAG,UAAU,CAAC,QAAQ,CAAC;SACpC,MAAM,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;SAC7B,MAAM,CAAC,KAAK,CAAC,CAAC;IAEjB,IAAI,UAAU,KAAK,YAAY,EAAE,CAAC;QAChC,MAAM,IAAI,KAAK,CACb,+BAA+B,YAAY,SAAS,UAAU,EAAE,CACjE,CAAC;IACJ,CAAC;IAED,kBAAkB;IAClB,IAAI,OAAO,CAAC,QAAQ,KAAK,OAAO,EAAE,CAAC;QACjC,SAAS,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,wBAAwB;IACxB,IAAI,CAAC;QACH,YAAY,CAAC,OAAO,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QAC3C,MAAM,CAAC,GAAG,CAAC,oBAAoB,EAAE,UAAU,CAAC,CAAC;IAC/C,CAAC;IAAC,MAAM,CAAC;QACP,mEAAmE;IACrE,CAAC;IAED,gBAAgB;IAChB,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAEtC,oFAAoF;IACpF,IAAI,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,kBAAkB,EAAE,EAAE,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,eAAe,MAAM,CAAC,OAAO,gCAAgC,CAAC,CAAC;QAC3E,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CACT,cAAc,MAAM,CAAC,OAAO,gCAAgC,CAC7D,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,cAAc,CAAC,OAA+B;IAC5D,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC;IAC3C,IAAI,CAAC,IAAI;QAAE,OAAO,IAAI,CAAC;IAEvB,MAAM,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;IACtD,MAAM,QAAQ,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;IAChC,MAAM,iBAAiB,GAAG,EAAE,GAAG,QAAQ,CAAC;IAExC,IAAI,OAAO,KAAK,SAAS;QAAE,OAAO,OAAO,GAAG,QAAQ,CAAC;IACrD,OAAO,OAAO,GAAG,iBAAiB,CAAC;AACrC,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,QAAQ;IACtB,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;IAC9C,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,OAAO,CAAC,GAAG,CAAC,qCAAqC,CAAC,CAAC;QACnD,OAAO;IACT,CAAC;IAED,IAAI,CAAC;QACH,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;QACrC,OAAO,CAAC,GAAG,CAAC,uCAAuC,IAAI,GAAG,CAAC,CAAC;QAC5D,OAAO,CAAC,GAAG,CAAC,gDAAgD,CAAC,CAAC;IAChE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CACX,kBAAkB,EAClB,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CACzC,CAAC;IACJ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,iBAAiB,CAAC,cAAsB,EAAE,aAAqB;IAC7E,OAAO,CAAC,GAAG,CAAC,cAAc,aAAa,4BAA4B,cAAc,IAAI,CAAC,CAAC;IACvF,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;IACtD,OAAO,CAAC,GAAG,CAAC,yDAAyD,CAAC,CAAC;AACzE,CAAC;AAED,oEAAoE;AAEpE,SAAS,OAAO,CAAC,MAAc,EAAE,OAAe;IAC9C,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC3D,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,MAAM,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAC5D,IAAI,IAAI,KAAK,IAAI;QAAE,OAAO,IAAI,GAAG,IAAI,CAAC;IACtC,IAAI,IAAI,KAAK,IAAI;QAAE,OAAO,IAAI,GAAG,IAAI,CAAC;IACtC,OAAO,MAAM,GAAG,MAAM,CAAC;AACzB,CAAC;AAED,SAAS,kBAAkB;IACzB,uDAAuD;IACvD,OAAO,CACL,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,aAAa,IAAI,UAAU;QACzC,OAAO,CAAC,IAAI,KAAK,CAAC,IAAI,oCAAoC;QAC1D,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,gBAAgB;KAC/C,CAAC;AACJ,CAAC"}

package/dist/ws-client.d.ts

@@ -0,0 +1,26 @@
+type MessageHandler = (message: Record<string, unknown>) => void;
+export declare class AgentWebSocket {
+    private ws;
+    private backoff;
+    private heartbeatTimer;
+    private reconnectTimer;
+    private messageQueue;
+    private onMessage;
+    private stopped;
+    private startTime;
+    private activeJobs;
+    connect(): void;
+    stop(): void;
+    setMessageHandler(handler: MessageHandler): void;
+    send(message: Record<string, unknown>): void;
+    setActiveJobs(count: number): void;
+    private sendHello;
+    private sendHeartbeat;
+    private startHeartbeat;
+    private stopHeartbeat;
+    private drainQueue;
+    private cleanup;
+    private scheduleReconnect;
+}
+export {};
+//# sourceMappingURL=ws-client.d.ts.map

package/dist/ws-client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-client.d.ts","sourceRoot":"","sources":["../src/ws-client.ts"],"names":[],"mappings":"AAQA,KAAK,cAAc,GAAG,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;AAEjE,qBAAa,cAAc;IACzB,OAAO,CAAC,EAAE,CAA0B;IACpC,OAAO,CAAC,OAAO,CAAQ;IACvB,OAAO,CAAC,cAAc,CAA+C;IACrE,OAAO,CAAC,cAAc,CAA8C;IACpE,OAAO,CAAC,YAAY,CAAgB;IACpC,OAAO,CAAC,SAAS,CAA+B;IAChD,OAAO,CAAC,OAAO,CAAS;IACxB,OAAO,CAAC,SAAS,CAAc;IAC/B,OAAO,CAAC,UAAU,CAAK;IAEvB,OAAO,IAAI,IAAI;IAkDf,IAAI,IAAI,IAAI;IASZ,iBAAiB,CAAC,OAAO,EAAE,cAAc,GAAG,IAAI;IAIhD,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI;IAgB5C,aAAa,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAIlC,OAAO,CAAC,SAAS;IAcjB,OAAO,CAAC,aAAa;IAarB,OAAO,CAAC,cAAc;IAQtB,OAAO,CAAC,aAAa;IAOrB,OAAO,CAAC,UAAU;IAclB,OAAO,CAAC,OAAO;IAQf,OAAO,CAAC,iBAAiB;CAM1B"}

package/dist/ws-client.js

@@ -0,0 +1,155 @@
+import WebSocket from "ws";
+import { platform, freemem } from "node:os";
+import { createMessageId, createTimestamp } from "@tadaima/shared";
+import { config } from "./config.js";
+const HEARTBEAT_INTERVAL = 30_000;
+const MAX_BACKOFF = 30_000;
+export class AgentWebSocket {
+    ws = null;
+    backoff = 1000;
+    heartbeatTimer = null;
+    reconnectTimer = null;
+    messageQueue = [];
+    onMessage = null;
+    stopped = false;
+    startTime = Date.now();
+    activeJobs = 0;
+    connect() {
+        this.stopped = false;
+        const relayUrl = config.get("relay");
+        const deviceToken = config.get("deviceToken");
+        if (!relayUrl || !deviceToken) {
+            console.error("Not configured. Run `tadaima-agent setup` first.");
+            return;
+        }
+        const wsUrl = relayUrl
+            .replace(/^http/, "ws")
+            .replace(/\/$/, "");
+        const url = `${wsUrl}/ws/agent?token=${deviceToken}`;
+        this.ws = new WebSocket(url);
+        this.ws.on("open", () => {
+            console.log("Connected to relay");
+            this.backoff = 1000;
+            this.sendHello();
+            this.startHeartbeat();
+            this.drainQueue();
+        });
+        this.ws.on("message", (data) => {
+            try {
+                const msg = JSON.parse(data.toString());
+                if (this.onMessage) {
+                    this.onMessage(msg);
+                }
+            }
+            catch (err) {
+                const raw = data.toString().slice(0, 200);
+                console.error("Failed to parse WebSocket message:", err, "raw:", raw);
+            }
+        });
+        this.ws.on("close", () => {
+            console.log("Disconnected from relay");
+            this.cleanup();
+            this.scheduleReconnect();
+        });
+        this.ws.on("error", (err) => {
+            console.error("WebSocket error:", err.message);
+            this.cleanup();
+            this.scheduleReconnect();
+        });
+    }
+    stop() {
+        this.stopped = true;
+        this.cleanup();
+        if (this.ws) {
+            this.ws.close(1000, "Agent stopped");
+            this.ws = null;
+        }
+    }
+    setMessageHandler(handler) {
+        this.onMessage = handler;
+    }
+    send(message) {
+        const data = JSON.stringify(message);
+        if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+            try {
+                this.ws.send(data);
+            }
+            catch (err) {
+                console.error("Send failed, re-queuing message:", err);
+                this.messageQueue.push(data);
+                this.cleanup();
+                this.scheduleReconnect();
+            }
+        }
+        else {
+            this.messageQueue.push(data);
+        }
+    }
+    setActiveJobs(count) {
+        this.activeJobs = count;
+    }
+    sendHello() {
+        this.send({
+            id: createMessageId(),
+            type: "agent:hello",
+            timestamp: createTimestamp(),
+            payload: {
+                version: "0.0.0",
+                platform: platform(),
+                activeJobs: this.activeJobs,
+                diskFreeBytes: freemem(),
+            },
+        });
+    }
+    sendHeartbeat() {
+        this.send({
+            id: createMessageId(),
+            type: "agent:heartbeat",
+            timestamp: createTimestamp(),
+            payload: {
+                activeJobs: this.activeJobs,
+                diskFreeBytes: freemem(),
+                uptimeSeconds: Math.floor((Date.now() - this.startTime) / 1000),
+            },
+        });
+    }
+    startHeartbeat() {
+        this.stopHeartbeat();
+        this.heartbeatTimer = setInterval(() => this.sendHeartbeat(), HEARTBEAT_INTERVAL);
+    }
+    stopHeartbeat() {
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+            this.heartbeatTimer = null;
+        }
+    }
+    drainQueue() {
+        while (this.messageQueue.length > 0 && this.ws?.readyState === WebSocket.OPEN) {
+            try {
+                this.ws.send(this.messageQueue[0]);
+                this.messageQueue.shift();
+            }
+            catch (err) {
+                console.error("Drain failed, will retry on reconnect:", err);
+                this.cleanup();
+                this.scheduleReconnect();
+                return;
+            }
+        }
+    }
+    cleanup() {
+        this.stopHeartbeat();
+        if (this.reconnectTimer) {
+            clearTimeout(this.reconnectTimer);
+            this.reconnectTimer = null;
+        }
+    }
+    scheduleReconnect() {
+        if (this.stopped)
+            return;
+        this.backoff = Math.min(this.backoff * 2, MAX_BACKOFF);
+        console.log(`Reconnecting in ${this.backoff / 1000}s...`);
+        this.reconnectTimer = setTimeout(() => this.connect(), this.backoff);
+    }
+}
+//# sourceMappingURL=ws-client.js.map

package/dist/ws-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ws-client.js","sourceRoot":"","sources":["../src/ws-client.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,IAAI,CAAC;AAC3B,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,eAAe,EAAE,eAAe,EAAE,MAAM,iBAAiB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,aAAa,CAAC;AAErC,MAAM,kBAAkB,GAAG,MAAM,CAAC;AAClC,MAAM,WAAW,GAAG,MAAM,CAAC;AAI3B,MAAM,OAAO,cAAc;IACjB,EAAE,GAAqB,IAAI,CAAC;IAC5B,OAAO,GAAG,IAAI,CAAC;IACf,cAAc,GAA0C,IAAI,CAAC;IAC7D,cAAc,GAAyC,IAAI,CAAC;IAC5D,YAAY,GAAa,EAAE,CAAC;IAC5B,SAAS,GAA0B,IAAI,CAAC;IACxC,OAAO,GAAG,KAAK,CAAC;IAChB,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,UAAU,GAAG,CAAC,CAAC;IAEvB,OAAO;QACL,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC;QACrB,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,WAAW,GAAG,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;QAE9C,IAAI,CAAC,QAAQ,IAAI,CAAC,WAAW,EAAE,CAAC;YAC9B,OAAO,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;YAClE,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,QAAQ;aACnB,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC;aACtB,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QACtB,MAAM,GAAG,GAAG,GAAG,KAAK,mBAAmB,WAAW,EAAE,CAAC;QAErD,IAAI,CAAC,EAAE,GAAG,IAAI,SAAS,CAAC,GAAG,CAAC,CAAC;QAE7B,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE;YACtB,OAAO,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;YAClC,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;YACpB,IAAI,CAAC,SAAS,EAAE,CAAC;YACjB,IAAI,CAAC,cAAc,EAAE,CAAC;YACtB,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAI,EAAE,EAAE;YAC7B,IAAI,CAAC;gBACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;gBACxC,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC;gBACtB,CAAC;YACH,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC1C,OAAO,CAAC,KAAK,CAAC,oCAAoC,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,CAAC,CAAC;YACxE,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,GAAG,EAAE;YACvB,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;YACvC,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QAEH,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE;YAC1B,OAAO,CAAC,KAAK,CAAC,kBAAkB,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI;QACF,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC;QACpB,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,IAAI,CAAC,EAAE,EAAE,CAAC;YACZ,IAAI,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;YACrC,IAAI,CAAC,EAAE,GAAG,IAAI,CAAC;QACjB,CAAC;IACH,CAAC;IAED,iBAAiB,CAAC,OAAuB;QACvC,IAAI,CAAC,SAAS,GAAG,OAAO,CAAC;IAC3B,CAAC;IAED,IAAI,CAAC,OAAgC;QACnC,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACrC,IAAI,IAAI,CAAC,EAAE,IAAI,IAAI,CAAC,EAAE,CAAC,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YACrD,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,GAAG,CAAC,CAAC;gBACvD,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC7B,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,CAAC,iBAAiB,EAAE,CAAC;YAC3B,CAAC;QACH,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAED,aAAa,CAAC,KAAa;QACzB,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC1B,CAAC;IAEO,SAAS;QACf,IAAI,CAAC,IAAI,CAAC;YACR,EAAE,EAAE,eAAe,EAAE;YACrB,IAAI,EAAE,aAAa;YACnB,SAAS,EAAE,eAAe,EAAE;YAC5B,OAAO,EAAE;gBACP,OAAO,EAAE,OAAO;gBAChB,QAAQ,EAAE,QAAQ,EAAE;gBACpB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,aAAa,EAAE,OAAO,EAAE;aACzB;SACF,CAAC,CAAC;IACL,CAAC;IAEO,aAAa;QACnB,IAAI,CAAC,IAAI,CAAC;YACR,EAAE,EAAE,eAAe,EAAE;YACrB,IAAI,EAAE,iBAAiB;YACvB,SAAS,EAAE,eAAe,EAAE;YAC5B,OAAO,EAAE;gBACP,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,aAAa,EAAE,OAAO,EAAE;gBACxB,aAAa,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,SAAS,CAAC,GAAG,IAAI,CAAC;aAChE;SACF,CAAC,CAAC;IACL,CAAC;IAEO,cAAc;QACpB,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,cAAc,GAAG,WAAW,CAC/B,GAAG,EAAE,CAAC,IAAI,CAAC,aAAa,EAAE,EAC1B,kBAAkB,CACnB,CAAC;IACJ,CAAC;IAEO,aAAa;QACnB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,aAAa,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACnC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,UAAU;QAChB,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,EAAE,EAAE,UAAU,KAAK,SAAS,CAAC,IAAI,EAAE,CAAC;YAC9E,IAAI,CAAC;gBACH,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC;gBACnC,IAAI,CAAC,YAAY,CAAC,KAAK,EAAE,CAAC;YAC5B,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,OAAO,CAAC,KAAK,CAAC,wCAAwC,EAAE,GAAG,CAAC,CAAC;gBAC7D,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBACzB,OAAO;YACT,CAAC;QACH,CAAC;IACH,CAAC;IAEO,OAAO;QACb,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,YAAY,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC7B,CAAC;IACH,CAAC;IAEO,iBAAiB;QACvB,IAAI,IAAI,CAAC,OAAO;YAAE,OAAO;QACzB,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,EAAE,WAAW,CAAC,CAAC;QACvD,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,CAAC,OAAO,GAAG,IAAI,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,cAAc,GAAG,UAAU,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;IACvE,CAAC;CACF"}

package/package.json

@@ -0,0 +1,62 @@
+{
+  "name": "@psychout98/tadaima",
+  "version": "1.0.0",
+  "description": "CLI download agent for Tadaima — a self-hosted media download orchestrator",
+  "type": "module",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/psychout98/tadaima.git",
+    "directory": "packages/agent"
+  },
+  "keywords": ["tadaima", "media", "download", "plex", "real-debrid", "agent"],
+  "engines": {
+    "node": ">=18"
+  },
+  "bin": {
+    "tadaima": "dist/index.js"
+  },
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js",
+      "default": "./dist/index.js"
+    }
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsx watch src/index.ts",
+    "lint": "eslint src/ --ignore-pattern src/__tests__/",
+    "test": "vitest run",
+    "typecheck": "tsc --noEmit",
+    "start": "node dist/index.js",
+    "compile": "pnpm run compile:all",
+    "compile:darwin-arm64": "bun build src/index.ts --compile --target=bun-darwin-arm64 --outfile=dist/tadaima-agent-darwin-arm64",
+    "compile:darwin-x64": "bun build src/index.ts --compile --target=bun-darwin-x64 --outfile=dist/tadaima-agent-darwin-x64",
+    "compile:win-x64": "bun build src/index.ts --compile --target=bun-windows-x64 --outfile=dist/tadaima-agent-win-x64.exe",
+    "compile:linux-x64": "bun build src/index.ts --compile --target=bun-linux-x64 --outfile=dist/tadaima-agent-linux-x64",
+    "compile:all": "pnpm run compile:darwin-arm64 && pnpm run compile:darwin-x64 && pnpm run compile:win-x64 && pnpm run compile:linux-x64",
+    "compile:checksums": "cd dist && shasum -a 256 tadaima-agent-* > checksums.sha256"
+  },
+  "dependencies": {
+    "@tadaima/shared": "workspace:*",
+    "conf": "^15.1.0",
+    "prompts": "^2.4.2",
+    "ws": "^8.20.0"
+  },
+  "devDependencies": {
+    "@types/node": "^25.5.2",
+    "@types/prompts": "^2.4.9",
+    "@types/ws": "^8.18.1",
+    "tsx": "^4.19.0",
+    "typescript": "^5.7.0",
+    "vitest": "^4.1.2"
+  }
+}