[r5n-86] [Hydra] update command

packages/hydra/src/cli.ts

@@ -1,6 +1,15 @@
 import { AbstractCLI, ConfigManager } from "@r5n/cli-core";
 import { version } from "../package.json";
-import { CreateCommand, InitCommand, ProfileCommand, RemoveCommand, StartCommand, StatusCommand, StopCommand } from "./commands";
+import {
+  CreateCommand,
+  InitCommand,
+  ProfileCommand,
+  RemoveCommand,
+  StartCommand,
+  StatusCommand,
+  StopCommand,
+  UpdateCommand,
+} from "./commands";
 import { CLI_BIN, DEFAULT_CONFIG, HYDRA_CONFIG_FILE } from "./constants";
 import type { HydraConfig } from "./types";
 
@@ -25,6 +34,7 @@ class HydraCLI extends AbstractCLI {
       new StopCommand(),
       new StatusCommand(),
       new RemoveCommand(),
+      new UpdateCommand(),
       new ProfileCommand(),
     ]);
   }

packages/hydra/src/commands/create.ts

@@ -1,5 +1,5 @@
 import { join } from "node:path";
-import { Exit, color, log, positionals, spinner, text } from "@r5n/cli-core";
+import { color, Exit, log, positionals, spinner, text } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../base-command";
 import { createProvider } from "../providers";
 import type { RunnerEntry } from "../types";
@@ -24,7 +24,9 @@ export class CreateCommand extends BaseCommand {
 
     const count = ctx.interactive
       ? await this.promptCount(profile.numberOfMachines)
-      : (ctx.positionals.count ? Number.parseInt(ctx.positionals.count, 10) : profile.numberOfMachines);
+      : ctx.positionals.count
+        ? Number.parseInt(ctx.positionals.count, 10)
+        : profile.numberOfMachines;
 
     if (!count || count < 1) {
       throw new Exit("Runner count must be at least 1");
@@ -73,7 +75,9 @@ export class CreateCommand extends BaseCommand {
 
     const isDefault = profileName === ctx.config.get("defaultProfile");
     const profileSuffix = isDefault ? "" : ` ${profileName}`;
-    log.info(`${color.green("Created")} ${toCreate} runner(s) for "${profileName}". Run ${color.green(`hydra start${profileSuffix}`)} to start them.`);
+    log.info(
+      `${color.green("Created")} ${toCreate} runner(s) for "${profileName}". Run ${color.green(`hydra start${profileSuffix}`)} to start them.`,
+    );
   }
 
   private async promptCount(defaultCount: number): Promise<number> {

packages/hydra/src/commands/index.ts

@@ -1,7 +1,8 @@
 export { CreateCommand } from "./create";
 export { InitCommand } from "./init";
-export { StatusCommand } from "./status";
 export { ProfileCommand } from "./profile";
 export { RemoveCommand } from "./remove";
 export { StartCommand } from "./start";
+export { StatusCommand } from "./status";
 export { StopCommand } from "./stop";
+export { UpdateCommand } from "./update";

packages/hydra/src/commands/init.ts

@@ -1,13 +1,11 @@
 import { existsSync } from "node:fs";
 import { mkdir } from "node:fs/promises";
-import { Cancel, Exit, args, color, confirm, group, note, positionals, text } from "@r5n/cli-core";
+import { args, Cancel, color, confirm, Exit, group, note, positionals, text } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../base-command";
 import { CLI_BIN, DEFAULT_PROFILE, RUNNERS_DIR, SHARED_DIR } from "../constants";
 import type { Profile } from "../types";
 
-const initPositionals = positionals({
-  url: { description: "GitHub repository or organization URL" },
-});
+const initPositionals = positionals({ url: { description: "GitHub repository or organization URL" } });
 
 const initArgs = args({
   force: { alias: "f", default: false, description: "Overwrite existing config", type: "boolean" },
@@ -27,9 +25,7 @@ export class InitCommand extends BaseCommand {
   prompts = true;
 
   async execute(ctx: InitCtx) {
-    const profileName = ctx.interactive
-      ? await this.promptProfileName(ctx)
-      : (ctx.args.profile ?? DEFAULT_PROFILE);
+    const profileName = ctx.interactive ? await this.promptProfileName(ctx) : (ctx.args.profile ?? DEFAULT_PROFILE);
 
     const isNewInit = !ctx.config.exists();
 
@@ -38,7 +34,10 @@ export class InitCommand extends BaseCommand {
       if (!ctx.interactive) {
         throw new Exit(`Profile "${profileName}" already exists. Use --force to overwrite.`);
       }
-      const overwrite = await confirm({ initialValue: false, message: `Profile "${profileName}" already exists. Overwrite?` });
+      const overwrite = await confirm({
+        initialValue: false,
+        message: `Profile "${profileName}" already exists. Overwrite?`,
+      });
       if (!overwrite) return;
     }
 
@@ -70,10 +69,7 @@ export class InitCommand extends BaseCommand {
   }
 
   private async promptProfileName(ctx: InitCtx): Promise<string> {
-    return text({
-      initialValue: ctx.args.profile ?? DEFAULT_PROFILE,
-      message: "Profile name",
-    });
+    return text({ initialValue: ctx.args.profile ?? DEFAULT_PROFILE, message: "Profile name" });
   }
 
   private async ensureDirectories() {
@@ -102,12 +98,7 @@ export class InitCommand extends BaseCommand {
               return undefined;
             },
           }),
-        name: () =>
-          text({
-            initialValue: "runner",
-            message: "Base name for runners",
-            placeholder: "runner",
-          }),
+        name: () => text({ initialValue: "runner", message: "Base name for runners", placeholder: "runner" }),
         runners: () =>
           text({
             initialValue: "1",
@@ -121,12 +112,13 @@ export class InitCommand extends BaseCommand {
             },
           }),
         labels: () =>
-          text({
-            message: "Additional labels (comma-separated, optional)",
-            placeholder: "self-hosted,macOS,ARM64",
-          }),
+          text({ message: "Additional labels (comma-separated, optional)", placeholder: "self-hosted,macOS,ARM64" }),
+      },
+      {
+        onCancel: () => {
+          throw new Cancel();
+        },
       },
-      { onCancel: () => { throw new Cancel(); } },
     );
 
     return {

packages/hydra/src/commands/profile/default.ts

@@ -1,10 +1,8 @@
-import { Exit, color, log, positionals } from "@r5n/cli-core";
+import { color, Exit, log, positionals } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../../base-command";
 import { resolveProfile, selectProfile } from "../../utils";
 
-const defaultPositionals = positionals({
-  name: { description: "Profile name to set as default" },
-});
+const defaultPositionals = positionals({ name: { description: "Profile name to set as default" } });
 
 type DefaultCtx = Ctx<Record<string, never>, typeof defaultPositionals>;
 
@@ -14,9 +12,7 @@ export class ProfileDefaultCommand extends BaseCommand {
   positionals = defaultPositionals;
 
   async execute(ctx: DefaultCtx) {
-    const profileName = ctx.interactive
-      ? await selectProfile(ctx.config)
-      : ctx.positionals.name;
+    const profileName = ctx.interactive ? await selectProfile(ctx.config) : ctx.positionals.name;
 
     if (!profileName) {
       throw new Exit("Profile name is required", "Usage: hydra profile default <name>");

packages/hydra/src/commands/profile/list.ts

@@ -1,4 +1,4 @@
-import { Exit, color, log, note } from "@r5n/cli-core";
+import { color, Exit, log, note } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../../base-command";
 import { createProvider } from "../../providers";
 import type { Profile } from "../../types";

packages/hydra/src/commands/profile/remove.ts

@@ -1,14 +1,12 @@
 import { rm } from "node:fs/promises";
 import { join } from "node:path";
-import { Exit, color, confirm, log, positionals, spinner } from "@r5n/cli-core";
+import { color, confirm, Exit, log, positionals, spinner } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../../base-command";
 import { createProvider } from "../../providers";
 import type { Profile } from "../../types";
 import { resolveProfile, selectProfile } from "../../utils";
 
-const removePositionals = positionals({
-  name: { description: "Profile name to remove" },
-});
+const removePositionals = positionals({ name: { description: "Profile name to remove" } });
 
 type RemoveCtx = Ctx<Record<string, never>, typeof removePositionals>;
 
@@ -18,9 +16,7 @@ export class ProfileRemoveCommand extends BaseCommand {
   positionals = removePositionals;
 
   async execute(ctx: RemoveCtx) {
-    const profileName = ctx.interactive
-      ? await selectProfile(ctx.config)
-      : ctx.positionals.name;
+    const profileName = ctx.interactive ? await selectProfile(ctx.config) : ctx.positionals.name;
 
     if (!profileName) {
       throw new Exit("Profile name is required", "Usage: hydra profile remove <name>");
@@ -38,7 +34,10 @@ export class ProfileRemoveCommand extends BaseCommand {
       });
       if (!proceed) return;
 
-      await this.removeRunners(profile, profileRunners.map((e) => e.id));
+      await this.removeRunners(
+        profile,
+        profileRunners.map((e) => e.id),
+      );
       const remainingEntries = entries.filter((e) => e.profile !== profileName);
       ctx.config.set("runners", remainingEntries);
     }

packages/hydra/src/commands/remove.ts

@@ -1,6 +1,6 @@
 import { rm } from "node:fs/promises";
 import { join } from "node:path";
-import { Exit, color, log, multiselect, positionals, spinner } from "@r5n/cli-core";
+import { color, Exit, log, multiselect, positionals, spinner } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../base-command";
 import { createProvider } from "../providers";
 import { resolveProfile, resolveRunnerIds, selectProfile } from "../utils";
@@ -30,7 +30,10 @@ export class RemoveCommand extends BaseCommand {
 
     const ids = ctx.interactive
       ? await this.promptRunnerSelection(profileEntries.map((e) => e.id))
-      : resolveRunnerIds(ctx.positionals.ids ?? [], profileEntries.map((e) => e.id));
+      : resolveRunnerIds(
+          ctx.positionals.ids ?? [],
+          profileEntries.map((e) => e.id),
+        );
 
     const provider = createProvider(profile);
     const s = spinner();

packages/hydra/src/commands/start.ts

@@ -1,4 +1,4 @@
-import { Exit, color, log, multiselect, positionals, spinner } from "@r5n/cli-core";
+import { color, Exit, log, multiselect, positionals, spinner } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../base-command";
 import { createProvider } from "../providers";
 import { resolveProfile, resolveRunnerIds, selectProfile } from "../utils";
@@ -39,7 +39,10 @@ export class StartCommand extends BaseCommand {
 
     const ids = ctx.interactive
       ? await this.promptRunnerSelection(stoppedIds)
-      : resolveRunnerIds(ctx.positionals.ids ?? [], profileEntries.map((e) => e.id));
+      : resolveRunnerIds(
+          ctx.positionals.ids ?? [],
+          profileEntries.map((e) => e.id),
+        );
 
     let started = 0;
     for (const id of ids) {

packages/hydra/src/commands/status.ts

@@ -1,7 +1,7 @@
-import { Exit, color, log } from "@r5n/cli-core";
+import { color, Exit, log } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../base-command";
-import { createProvider } from "../providers";
 import type { RunnerInfo } from "../providers";
+import { createProvider } from "../providers";
 import type { Profile } from "../types";
 
 type ListCtx = Ctx;

packages/hydra/src/commands/stop.ts

@@ -1,4 +1,4 @@
-import { Exit, color, log, multiselect, positionals, spinner } from "@r5n/cli-core";
+import { color, Exit, log, multiselect, positionals, spinner } from "@r5n/cli-core";
 import { BaseCommand, type Ctx } from "../base-command";
 import { createProvider } from "../providers";
 import { resolveProfile, resolveRunnerIds, selectProfile } from "../utils";
@@ -39,7 +39,10 @@ export class StopCommand extends BaseCommand {
 
     const ids = ctx.interactive
       ? await this.promptRunnerSelection(activeIds)
-      : resolveRunnerIds(ctx.positionals.ids ?? [], profileEntries.map((e) => e.id));
+      : resolveRunnerIds(
+          ctx.positionals.ids ?? [],
+          profileEntries.map((e) => e.id),
+        );
 
     let stopped = 0;
     for (const id of ids) {

packages/hydra/src/commands/update.ts

@@ -0,0 +1,66 @@
+import { color, Exit, log, spinner } from "@r5n/cli-core";
+import { BaseCommand, type Ctx } from "../base-command";
+import { createProvider } from "../providers";
+import type { Profile } from "../types";
+
+type UpdateCtx = Ctx;
+
+export class UpdateCommand extends BaseCommand {
+  name = "update";
+  description = "Update runners to the latest version";
+
+  async execute(ctx: UpdateCtx) {
+    const profiles = ctx.config.get("profiles");
+    const entries = ctx.config.get("runners") ?? [];
+
+    if (entries.length === 0) {
+      throw new Exit("No runners found", "Run hydra create to provision runners");
+    }
+
+    const s = spinner();
+    const byProfile = Map.groupBy(entries, (e) => e.profile);
+    let currentVersion: string | null = null;
+    let latestVersion = "";
+    let versionChecked = false;
+
+    for (const [profileName, profileEntries] of byProfile) {
+      const profile = profiles[profileName] as Profile | undefined;
+      const firstRunner = profileEntries[0];
+      if (!profile || !firstRunner) continue;
+
+      const provider = createProvider(profile);
+
+      if (!versionChecked) {
+        versionChecked = true;
+        s.start("Checking for updates...");
+        const result = await provider.download();
+        latestVersion = result.version;
+        currentVersion = await provider.currentVersion(firstRunner.id);
+        s.stop(`Current: v${currentVersion ?? "unknown"} | Latest: v${latestVersion}`);
+
+        if (currentVersion === latestVersion) {
+          log.info(`Already on latest version ${color.dim(`(v${latestVersion})`)}`);
+          return;
+        }
+      }
+
+      const statuses = await provider.list();
+      const runningIds = new Set(statuses.filter((r) => r.status === "running").map((r) => r.id));
+
+      for (const entry of profileEntries) {
+        const wasRunning = runningIds.has(entry.id);
+
+        s.start(`Updating ${entry.id}...`);
+
+        if (wasRunning) await provider.stop([entry.id]);
+        await provider.update([entry.id]);
+        if (wasRunning) await provider.start([entry.id]);
+
+        const suffix = wasRunning ? ` ${color.dim("(restarted)")}` : "";
+        s.stop(`${color.green("✓")} ${entry.id}${suffix}`);
+      }
+    }
+
+    log.info(`${color.green("Updated")} ${entries.length} runner(s) from v${currentVersion} to v${latestVersion}`);
+  }
+}

packages/hydra/src/constants.ts

@@ -9,6 +9,4 @@ export const RUNNERS_DIR = `${HYDRA_DIR}/runners`;
 
 export const DEFAULT_PROFILE = "default";
 
-export const DEFAULT_CONFIG: HydraConfig = {
-  profiles: {},
-};
+export const DEFAULT_CONFIG: HydraConfig = { profiles: {} };