@coworker-jp/aidr 0.0.1

package/src/scheduled.mjs

@@ -0,0 +1,219 @@
+import fs from "fs/promises";
+import path from "path";
+import os from "os";
+import { execFile } from "child_process";
+import { promisify } from "util";
+
+const execFileP = promisify(execFile);
+
+function systemdDir(home) {
+  return path.join(home, ".config", "systemd", "user");
+}
+function systemdService(home) { return path.join(systemdDir(home), "aidr-scheduled.service"); }
+function systemdTimer(home)   { return path.join(systemdDir(home), "aidr-scheduled.timer"); }
+function systemdEnv(home)     { return path.join(systemdDir(home), "aidr-scheduled.env"); }
+function launchdPlist(home)   { return path.join(home, "Library", "LaunchAgents", "jp.coworker.aidr.scheduled.plist"); }
+
+async function exists(p) {
+  try { await fs.access(p); return true; } catch { return false; }
+}
+
+export async function isScheduledInstalled(home) {
+  return (
+    await exists(systemdTimer(home)) ||
+    await exists(launchdPlist(home))
+  );
+}
+
+export async function installScheduled(opts) {
+  const { home = os.homedir(), accessKey, binaryPath,
+          dryRun = false, interval = 4 } = opts;
+  const noActivate = opts.noActivate || process.env.AI_SCANNER_NO_SCHEDULED_ACTIVATE === "1";
+
+  // Schedule entry is agent-independent (cron points at ~/.aidr/bin/ai-scanner
+  // and the binary self-updates on each run). If a previous schedule entry
+  // exists, replace it unconditionally — the only thing that could differ is
+  // the binary path or interval, both of which the new install owns.
+  if (await isScheduledInstalled(home)) {
+    await uninstallScheduled({ home, dryRun, noActivate });
+  }
+
+  const platform = process.platform;
+  if (platform === "linux") {
+    await installSystemd({ home, accessKey, binaryPath, dryRun, noActivate, interval });
+  } else if (platform === "darwin") {
+    await installLaunchd({ home, accessKey, binaryPath, dryRun, noActivate, interval });
+  } else {
+    await installCrontab({ binaryPath, dryRun, interval });
+  }
+}
+
+async function installSystemd({ home, accessKey, binaryPath, dryRun, noActivate, interval = 4 }) {
+  const dir = systemdDir(home);
+
+  const serviceContent = `[Unit]
+Description=AIDR scheduled endpoint-info scan
+
+[Service]
+Type=oneshot
+EnvironmentFile=${systemdEnv(home)}
+ExecStart=${binaryPath} scan endpoint-info
+`;
+
+  const timerContent = `[Unit]
+Description=AIDR scheduled endpoint-info scan timer
+
+[Timer]
+OnBootSec=5min
+OnUnitActiveSec=${interval}h
+Persistent=true
+
+[Install]
+WantedBy=timers.target
+`;
+
+  const envContent = `AI_SCANNER_ACCESS_KEY=${accessKey}\n`;
+
+  if (dryRun) {
+    console.log("[dry-run] Would create:", systemdService(home), systemdTimer(home), systemdEnv(home));
+    return;
+  }
+
+  await fs.mkdir(dir, { recursive: true });
+  await fs.writeFile(systemdService(home), serviceContent, { mode: 0o644 });
+  await fs.writeFile(systemdTimer(home), timerContent, { mode: 0o644 });
+  await fs.writeFile(systemdEnv(home), envContent, { mode: 0o600 });
+
+  if (!noActivate) {
+    try {
+      await execFileP("systemctl", ["--user", "daemon-reload"]);
+      await execFileP("systemctl", ["--user", "enable", "--now", "aidr-scheduled.timer"]);
+      console.log(`Scheduled scan enabled via systemd user timer (every ${interval}h)`);
+    } catch (e) {
+      console.warn("Warning: systemctl --user failed:", e.message);
+      console.warn("Timer files were written. Enable manually: systemctl --user enable --now aidr-scheduled.timer");
+    }
+  }
+}
+
+async function installLaunchd({ home, accessKey, binaryPath, dryRun, noActivate, interval = 4 }) {
+  const dir = path.join(home, "Library", "LaunchAgents");
+
+  const plistContent = `<?xml version="1.0" encoding="UTF-8"?>
+<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
+<plist version="1.0">
+<dict>
+  <key>Label</key>
+  <string>jp.coworker.aidr.scheduled</string>
+  <key>ProgramArguments</key>
+  <array>
+    <string>${binaryPath}</string>
+    <string>scan</string>
+    <string>endpoint-info</string>
+  </array>
+  <key>EnvironmentVariables</key>
+  <dict>
+    <key>AI_SCANNER_ACCESS_KEY</key>
+    <string>${accessKey}</string>
+  </dict>
+  <key>StartInterval</key>
+  <integer>${interval * 3600}</integer>
+  <key>RunAtLoad</key>
+  <false/>
+</dict>
+</plist>
+`;
+
+  if (dryRun) {
+    console.log("[dry-run] Would create:", launchdPlist(home));
+    return;
+  }
+
+  await fs.mkdir(dir, { recursive: true });
+  await fs.writeFile(launchdPlist(home), plistContent, { mode: 0o644 });
+
+  if (!noActivate) {
+    try {
+      const uid = process.getuid?.() ?? (await execFileP("id", ["-u"])).stdout.trim();
+      await execFileP("launchctl", ["bootstrap", `gui/${uid}`, launchdPlist(home)]);
+      console.log(`Scheduled scan enabled via launchd (every ${interval}h)`);
+    } catch (e) {
+      console.warn("Warning: launchctl bootstrap failed:", e.message);
+      console.warn("Plist written. Enable manually: launchctl load ~/Library/LaunchAgents/jp.coworker.aidr.scheduled.plist");
+    }
+  }
+}
+
+async function installCrontab({ binaryPath, dryRun, interval = 4 }) {
+  const entry = `0 */${interval} * * * ${binaryPath} scan endpoint-info 2>/dev/null # aidr-scheduled\n`;
+
+  if (dryRun) {
+    console.log("[dry-run] Would add crontab entry:", entry.trim());
+    return;
+  }
+
+  let existing = "";
+  try {
+    const { stdout } = await execFileP("crontab", ["-l"]);
+    existing = stdout;
+  } catch { /* no crontab yet */ }
+
+  if (existing.includes("aidr-scheduled")) return;
+
+  const newCrontab = existing.endsWith("\n") || existing === ""
+    ? existing + entry
+    : existing + "\n" + entry;
+
+  const child = execFile("crontab", ["-"]);
+  child.stdin.write(newCrontab);
+  child.stdin.end();
+  await new Promise((resolve, reject) => child.on("close", c => c === 0 ? resolve() : reject(new Error(`crontab exit ${c}`))));
+  console.log(`Scheduled scan added to user crontab (every ${interval}h)`);
+}
+
+export async function uninstallScheduled({ home = os.homedir(), dryRun = false, noActivate = false } = {}) {
+  noActivate = noActivate || process.env.AI_SCANNER_NO_SCHEDULED_ACTIVATE === "1";
+
+  if (await exists(systemdTimer(home))) {
+    if (!dryRun && !noActivate) {
+      try { await execFileP("systemctl", ["--user", "disable", "--now", "aidr-scheduled.timer"]); } catch { /* ignore */ }
+    }
+    for (const f of [systemdService(home), systemdTimer(home), systemdEnv(home)]) {
+      if (!dryRun) { try { await fs.unlink(f); } catch { /* ignore */ } }
+      else { console.log("[dry-run] Would remove:", f); }
+    }
+  }
+
+  if (await exists(launchdPlist(home))) {
+    if (!dryRun && !noActivate) {
+      try {
+        const uid = process.getuid?.() ?? (await execFileP("id", ["-u"])).stdout.trim();
+        await execFileP("launchctl", ["bootout", `gui/${uid}`, launchdPlist(home)]);
+      } catch { /* ignore */ }
+    }
+    if (!dryRun) { try { await fs.unlink(launchdPlist(home)); } catch { /* ignore */ } }
+    else { console.log("[dry-run] Would remove:", launchdPlist(home)); }
+  }
+
+  try {
+    const { stdout } = await execFileP("crontab", ["-l"]);
+    const filtered = stdout.split("\n").filter(l => !l.includes("aidr-scheduled")).join("\n");
+    if (filtered !== stdout) {
+      if (!dryRun) {
+        const child = execFile("crontab", ["-"]);
+        child.stdin.write(filtered + (filtered.endsWith("\n") ? "" : "\n"));
+        child.stdin.end();
+        await new Promise(r => child.on("close", r));
+      } else {
+        console.log("[dry-run] Would remove aidr-scheduled from crontab");
+      }
+    }
+  } catch { /* no crontab */ }
+}
+
+function promptUser(question) {
+  return new Promise(resolve => {
+    process.stdout.write(question);
+    process.stdin.once("data", d => resolve(d.toString().trim()));
+  });
+}