@jameslovespancakes/pi-plus 1.0.0 → 1.0.1

package/src/domains/subscriptions/accounts.ts

@@ -1,242 +1,242 @@
-import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import { accountRows, openAccountsPicker, type AccountState } from "./accounts-picker.ts";
-import {
-  accountProvider,
-  accountProviders,
-  type AccountContext,
-  type AccountProvider,
-  type ManagedAccount,
-} from "../../core/accounts/registry.ts";
-
-/**
- * `/accounts` is provider-agnostic subscription account management.
- *
- *   /accounts          every provider's accounts, toggled enabled/disabled
- *   /accounts add      pick a provider, name the account, authorize
- *   /accounts reauth   reauthorize an existing account
- *
- * No provider-specific logic lives here; adapters are registered in
- * core/accounts/registry.ts, so a new provider is an adapter rather than a
- * new command.
- */
-
-async function openBrowser(pi: ExtensionAPI, url: string): Promise<void> {
-  if (process.platform === "win32") {
-    await pi.exec("rundll32.exe", ["url.dll,FileProtocolHandler", url]);
-    return;
-  }
-  if (process.platform === "darwin") {
-    await pi.exec("open", [url]);
-    return;
-  }
-  await pi.exec("xdg-open", [url]);
-}
-
-function bridge(pi: ExtensionAPI, ctx: any): AccountContext {
-  return {
-    hasUI: ctx.hasUI,
-    ui: {
-      input: (title, placeholder) => ctx.ui.input(title, placeholder),
-      confirm: (title, message) => ctx.ui.confirm(title, message),
-      notify: (message, type) => ctx.ui.notify(message, type ?? "info"),
-    },
-    openBrowser: (url) => openBrowser(pi, url),
-  };
-}
-
-function describe(account: ManagedAccount): string {
-  const state = !account.enabled
-    ? "disabled"
-    : account.expiresAt !== undefined && account.expiresAt < Date.now() ? "expired" : "active";
-  return `${account.label.padEnd(20)} ${state}`;
-}
-
-async function listAll(ctx: any): Promise<void> {
-  const providers = accountProviders();
-  if (providers.length === 0) {
-    ctx.ui.notify("No account providers are registered.", "warning");
-    return;
-  }
-
-  const blocks: string[] = [];
-  for (const provider of providers) {
-    try {
-      const accounts = await provider.list();
-      const routing = provider.routing ? ` · routing: ${await provider.routing.get()}` : "";
-      blocks.push(
-        `${provider.label} (${provider.id})${routing}`,
-        ...(accounts.length > 0
-          ? accounts.map((account) => `  ${describe(account)}`)
-          : [`  no additional accounts, add one with /account ${provider.id} add`]),
-      );
-    } catch (error) {
-      blocks.push(`${provider.label} (${provider.id}): ${error instanceof Error ? error.message : String(error)}`);
-    }
-  }
-  ctx.ui.notify(blocks.join("\n"), "info");
-}
-
-
-
-async function pickAccount(ctx: any, provider: AccountProvider): Promise<string | undefined> {
-  const accounts = await provider.list();
-  if (accounts.length === 0) {
-    ctx.ui.notify(`No ${provider.label} accounts yet. Add one with /account ${provider.id} add.`, "info");
-    return undefined;
-  }
-  if (!ctx.hasUI) {
-    ctx.ui.notify(`Specify an account: /account ${provider.id} reauth <label>`, "warning");
-    return undefined;
-  }
-
-  const labels = accounts.map(describe);
-  const choice = await ctx.ui.select(`Reauthorize which ${provider.label} account?`, labels);
-  if (!choice) return undefined;
-  return accounts[labels.indexOf(choice)]?.id;
-}
-
-export function registerAccountCommands(pi: ExtensionAPI): void {
-  /** Toggles one account and returns the state it ended in. */
-  const toggle = (providerId: string, accountId: string): AccountState => {
-    const provider = accountProvider(providerId);
-    if (!provider?.setEnabled) throw new Error(`${providerId} accounts cannot be disabled.`);
-    const next = pendingState.get(`${providerId}:${accountId}`) === "enabled" ? "disabled" : "enabled";
-    pendingState.set(`${providerId}:${accountId}`, next);
-    // Adapters persist asynchronously; the picker needs the answer now, so the
-    // write is fired off and failures surface as a notification.
-    void provider.setEnabled(accountId, next === "enabled").catch(() => {});
-    return next;
-  };
-
-  /** Mirror of on-disk state, so the synchronous toggle can answer instantly. */
-  const pendingState = new Map<string, AccountState>();
-
-  const rows = async () => {
-    const list = await accountRows(accountProviders());
-    pendingState.clear();
-    for (const row of list) pendingState.set(row.id, row.state);
-    return list;
-  };
-
-  /** Adds an account: choose provider, name it, authorize. */
-  async function addAccount(ctx: any, providerId?: string): Promise<void> {
-    if (!ctx.hasUI) {
-      ctx.ui.notify("Adding an account requires interactive Pi mode.", "error");
-      return;
-    }
-
-    const providers = accountProviders();
-    if (providers.length === 0) {
-      ctx.ui.notify("No subscription providers are registered.", "error");
-      return;
-    }
-
-    let provider = providerId ? accountProvider(providerId) : undefined;
-    if (!provider) {
-      if (providers.length === 1) {
-        provider = providers[0];
-      } else {
-        const labels = providers.map((p) => `${p.label} (${p.id})`);
-        const picked = await ctx.ui.select("Add an account for which subscription?", labels);
-        if (!picked) return;
-        provider = providers[labels.indexOf(picked)];
-      }
-    }
-    if (!provider) return;
-
-    const label = await ctx.ui.input(`${provider.label} account name`, "Work, Personal, etc.");
-    if (!label?.trim()) return;
-
-    try {
-      const added = await provider.add(bridge(pi, ctx), label.trim());
-      if (added) ctx.ui.notify(`${provider.label} account “${added}” added.`, "info");
-    } catch (error) {
-      ctx.ui.notify(`Could not add account: ${error instanceof Error ? error.message : String(error)}`, "error");
-    }
-  }
-
-  /** Rename wizard: pick an account, type a new name. */
-  async function renameAccount(ctx: any): Promise<void> {
-    if (!ctx.hasUI) {
-      ctx.ui.notify("Renaming an account requires interactive Pi mode.", "error");
-      return;
-    }
-    const list = await accountRows(accountProviders());
-    const renameable = list.filter((row) => accountProvider(row.providerId)?.rename);
-    if (renameable.length === 0) {
-      ctx.ui.notify("No accounts can be renamed.", "info");
-      return;
-    }
-
-    const labels = renameable.map((row) => `${row.providerLabel}  ${row.label}`);
-    const picked = await ctx.ui.select("Rename which account?", labels);
-    if (!picked) return;
-    const row = renameable[labels.indexOf(picked)];
-    if (!row) return;
-
-    const next = await ctx.ui.input("New name", row.label);
-    if (!next?.trim() || next.trim() === row.label) return;
-
-    const provider = accountProvider(row.providerId);
-    try {
-      // Row ids are "<providerId>:<accountId>"; strip the prefix.
-      await provider!.rename!(row.id.slice(row.providerId.length + 1), next.trim());
-      ctx.ui.notify(`Renamed to “${next.trim()}”.`, "info");
-    } catch (error) {
-      ctx.ui.notify(`Could not rename: ${error instanceof Error ? error.message : String(error)}`, "error");
-    }
-  }
-
-  pi.registerCommand("accounts", {
-    description: "Subscription accounts (/accounts [add|rename|reauth])",
-    getArgumentCompletions: (prefix) => {
-      const parts = prefix.trim().split(/\s+/).filter(Boolean);
-      if (parts.length <= 1) {
-        return ["add", "rename", "reauth"]
-          .filter((option) => option.startsWith(parts[0] ?? ""))
-          .map((option) => ({ value: option, label: option }));
-      }
-      return accountProviders()
-        .map((p) => p.id)
-        .filter((id) => id.startsWith(parts[1] ?? ""))
-        .map((id) => ({ value: `${parts[0]} ${id}`, label: id }));
-    },
-    handler: async (args, ctx) => {
-      const [action, providerId] = args.trim().split(/\s+/).filter(Boolean);
-
-      if (action === "add") return addAccount(ctx, providerId);
-      if (action === "rename") return renameAccount(ctx);
-
-      if (action === "reauth") {
-        const provider = providerId ? accountProvider(providerId) : accountProviders()[0];
-        if (!provider) { ctx.ui.notify("No subscription providers are registered.", "error"); return; }
-        const target = await pickAccount(ctx, provider);
-        if (!target) return;
-        try {
-          const done = await provider.reauth(bridge(pi, ctx), target);
-          if (done) ctx.ui.notify(`${provider.label} account “${done}” reauthorized.`, "info");
-        } catch (error) {
-          ctx.ui.notify(`Could not reauthorize: ${error instanceof Error ? error.message : String(error)}`, "error");
-        }
-        return;
-      }
-
-      if (action) {
-        ctx.ui.notify(`Unknown action “${action}”. Use: /accounts [add|rename|reauth]`, "warning");
-        return;
-      }
-
-      // Bare /accounts: the picker interactively, plain text headless.
-      if (!ctx.hasUI) return listAll(ctx);
-
-      // The picker closes to run a wizard, then reopens so the result is
-      // visible immediately. Bounded so a misbehaving action cannot spin.
-      for (let step = 0; step < 24; step++) {
-        const requested = await openAccountsPicker(ctx, { rows, toggle });
-        if (!requested) return;
-        if (requested.kind === "add") await addAccount(ctx);
-        else if (requested.kind === "rename") await renameAccount(ctx);
-      }
-    },
-  });
-}
+import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
+import { accountRows, openAccountsPicker, type AccountState } from "./accounts-picker.ts";
+import {
+  accountProvider,
+  accountProviders,
+  type AccountContext,
+  type AccountProvider,
+  type ManagedAccount,
+} from "../../core/accounts/registry.ts";
+
+/**
+ * `/accounts` is provider-agnostic subscription account management.
+ *
+ *   /accounts          every provider's accounts, toggled enabled/disabled
+ *   /accounts add      pick a provider, name the account, authorize
+ *   /accounts reauth   reauthorize an existing account
+ *
+ * No provider-specific logic lives here; adapters are registered in
+ * core/accounts/registry.ts, so a new provider is an adapter rather than a
+ * new command.
+ */
+
+async function openBrowser(pi: ExtensionAPI, url: string): Promise<void> {
+  if (process.platform === "win32") {
+    await pi.exec("rundll32.exe", ["url.dll,FileProtocolHandler", url]);
+    return;
+  }
+  if (process.platform === "darwin") {
+    await pi.exec("open", [url]);
+    return;
+  }
+  await pi.exec("xdg-open", [url]);
+}
+
+function bridge(pi: ExtensionAPI, ctx: any): AccountContext {
+  return {
+    hasUI: ctx.hasUI,
+    ui: {
+      input: (title, placeholder) => ctx.ui.input(title, placeholder),
+      confirm: (title, message) => ctx.ui.confirm(title, message),
+      notify: (message, type) => ctx.ui.notify(message, type ?? "info"),
+    },
+    openBrowser: (url) => openBrowser(pi, url),
+  };
+}
+
+function describe(account: ManagedAccount): string {
+  const state = !account.enabled
+    ? "disabled"
+    : account.expiresAt !== undefined && account.expiresAt < Date.now() ? "expired" : "active";
+  return `${account.label.padEnd(20)} ${state}`;
+}
+
+async function listAll(ctx: any): Promise<void> {
+  const providers = accountProviders();
+  if (providers.length === 0) {
+    ctx.ui.notify("No account providers are registered.", "warning");
+    return;
+  }
+
+  const blocks: string[] = [];
+  for (const provider of providers) {
+    try {
+      const accounts = await provider.list();
+      const routing = provider.routing ? ` · routing: ${await provider.routing.get()}` : "";
+      blocks.push(
+        `${provider.label} (${provider.id})${routing}`,
+        ...(accounts.length > 0
+          ? accounts.map((account) => `  ${describe(account)}`)
+          : [`  no additional accounts, add one with /account ${provider.id} add`]),
+      );
+    } catch (error) {
+      blocks.push(`${provider.label} (${provider.id}): ${error instanceof Error ? error.message : String(error)}`);
+    }
+  }
+  ctx.ui.notify(blocks.join("\n"), "info");
+}
+
+
+
+async function pickAccount(ctx: any, provider: AccountProvider): Promise<string | undefined> {
+  const accounts = await provider.list();
+  if (accounts.length === 0) {
+    ctx.ui.notify(`No ${provider.label} accounts yet. Add one with /account ${provider.id} add.`, "info");
+    return undefined;
+  }
+  if (!ctx.hasUI) {
+    ctx.ui.notify(`Specify an account: /account ${provider.id} reauth <label>`, "warning");
+    return undefined;
+  }
+
+  const labels = accounts.map(describe);
+  const choice = await ctx.ui.select(`Reauthorize which ${provider.label} account?`, labels);
+  if (!choice) return undefined;
+  return accounts[labels.indexOf(choice)]?.id;
+}
+
+export function registerAccountCommands(pi: ExtensionAPI): void {
+  /** Toggles one account and returns the state it ended in. */
+  const toggle = (providerId: string, accountId: string): AccountState => {
+    const provider = accountProvider(providerId);
+    if (!provider?.setEnabled) throw new Error(`${providerId} accounts cannot be disabled.`);
+    const next = pendingState.get(`${providerId}:${accountId}`) === "enabled" ? "disabled" : "enabled";
+    pendingState.set(`${providerId}:${accountId}`, next);
+    // Adapters persist asynchronously; the picker needs the answer now, so the
+    // write is fired off and failures surface as a notification.
+    void provider.setEnabled(accountId, next === "enabled").catch(() => {});
+    return next;
+  };
+
+  /** Mirror of on-disk state, so the synchronous toggle can answer instantly. */
+  const pendingState = new Map<string, AccountState>();
+
+  const rows = async () => {
+    const list = await accountRows(accountProviders());
+    pendingState.clear();
+    for (const row of list) pendingState.set(row.id, row.state);
+    return list;
+  };
+
+  /** Adds an account: choose provider, name it, authorize. */
+  async function addAccount(ctx: any, providerId?: string): Promise<void> {
+    if (!ctx.hasUI) {
+      ctx.ui.notify("Adding an account requires interactive Pi mode.", "error");
+      return;
+    }
+
+    const providers = accountProviders();
+    if (providers.length === 0) {
+      ctx.ui.notify("No subscription providers are registered.", "error");
+      return;
+    }
+
+    let provider = providerId ? accountProvider(providerId) : undefined;
+    if (!provider) {
+      if (providers.length === 1) {
+        provider = providers[0];
+      } else {
+        const labels = providers.map((p) => `${p.label} (${p.id})`);
+        const picked = await ctx.ui.select("Add an account for which subscription?", labels);
+        if (!picked) return;
+        provider = providers[labels.indexOf(picked)];
+      }
+    }
+    if (!provider) return;
+
+    const label = await ctx.ui.input(`${provider.label} account name`, "Work, Personal, etc.");
+    if (!label?.trim()) return;
+
+    try {
+      const added = await provider.add(bridge(pi, ctx), label.trim());
+      if (added) ctx.ui.notify(`${provider.label} account “${added}” added.`, "info");
+    } catch (error) {
+      ctx.ui.notify(`Could not add account: ${error instanceof Error ? error.message : String(error)}`, "error");
+    }
+  }
+
+  /** Rename wizard: pick an account, type a new name. */
+  async function renameAccount(ctx: any): Promise<void> {
+    if (!ctx.hasUI) {
+      ctx.ui.notify("Renaming an account requires interactive Pi mode.", "error");
+      return;
+    }
+    const list = await accountRows(accountProviders());
+    const renameable = list.filter((row) => accountProvider(row.providerId)?.rename);
+    if (renameable.length === 0) {
+      ctx.ui.notify("No accounts can be renamed.", "info");
+      return;
+    }
+
+    const labels = renameable.map((row) => `${row.providerLabel}  ${row.label}`);
+    const picked = await ctx.ui.select("Rename which account?", labels);
+    if (!picked) return;
+    const row = renameable[labels.indexOf(picked)];
+    if (!row) return;
+
+    const next = await ctx.ui.input("New name", row.label);
+    if (!next?.trim() || next.trim() === row.label) return;
+
+    const provider = accountProvider(row.providerId);
+    try {
+      // Row ids are "<providerId>:<accountId>"; strip the prefix.
+      await provider!.rename!(row.id.slice(row.providerId.length + 1), next.trim());
+      ctx.ui.notify(`Renamed to “${next.trim()}”.`, "info");
+    } catch (error) {
+      ctx.ui.notify(`Could not rename: ${error instanceof Error ? error.message : String(error)}`, "error");
+    }
+  }
+
+  pi.registerCommand("accounts", {
+    description: "Subscription accounts (/accounts [add|rename|reauth])",
+    getArgumentCompletions: (prefix) => {
+      const parts = prefix.trim().split(/\s+/).filter(Boolean);
+      if (parts.length <= 1) {
+        return ["add", "rename", "reauth"]
+          .filter((option) => option.startsWith(parts[0] ?? ""))
+          .map((option) => ({ value: option, label: option }));
+      }
+      return accountProviders()
+        .map((p) => p.id)
+        .filter((id) => id.startsWith(parts[1] ?? ""))
+        .map((id) => ({ value: `${parts[0]} ${id}`, label: id }));
+    },
+    handler: async (args, ctx) => {
+      const [action, providerId] = args.trim().split(/\s+/).filter(Boolean);
+
+      if (action === "add") return addAccount(ctx, providerId);
+      if (action === "rename") return renameAccount(ctx);
+
+      if (action === "reauth") {
+        const provider = providerId ? accountProvider(providerId) : accountProviders()[0];
+        if (!provider) { ctx.ui.notify("No subscription providers are registered.", "error"); return; }
+        const target = await pickAccount(ctx, provider);
+        if (!target) return;
+        try {
+          const done = await provider.reauth(bridge(pi, ctx), target);
+          if (done) ctx.ui.notify(`${provider.label} account “${done}” reauthorized.`, "info");
+        } catch (error) {
+          ctx.ui.notify(`Could not reauthorize: ${error instanceof Error ? error.message : String(error)}`, "error");
+        }
+        return;
+      }
+
+      if (action) {
+        ctx.ui.notify(`Unknown action “${action}”. Use: /accounts [add|rename|reauth]`, "warning");
+        return;
+      }
+
+      // Bare /accounts: the picker interactively, plain text headless.
+      if (!ctx.hasUI) return listAll(ctx);
+
+      // The picker closes to run a wizard, then reopens so the result is
+      // visible immediately. Bounded so a misbehaving action cannot spin.
+      for (let step = 0; step < 24; step++) {
+        const requested = await openAccountsPicker(ctx, { rows, toggle });
+        if (!requested) return;
+        if (requested.kind === "add") await addAccount(ctx);
+        else if (requested.kind === "rename") await renameAccount(ctx);
+      }
+    },
+  });
+}