Merge branch 'neo/fix-top-level-login-20260703' into 'main'

fix(cli): support top-level login command

Closes #263

See merge request postgres-ai/postgresai!343

Author: NikolayS

cli/bin/postgres-ai.ts

@@ -3931,13 +3931,9 @@ targets
 // Authentication and API key management
 const auth = program.command("auth").description("authentication and API key management");
 
-auth
-  .command("login", { isDefault: true })
-  .description("authenticate via browser (OAuth) or store API key directly")
-  .option("--set-key <key>", "store API key directly without OAuth flow")
-  .option("--port <port>", "local callback server port (default: random)", parseInt)
-  .option("--debug", "enable debug output")
-  .action(async (opts: { setKey?: string; port?: number; debug?: boolean }) => {
+type AuthLoginOptions = { setKey?: string; port?: number; debug?: boolean };
+
+async function runAuthLogin(opts: AuthLoginOptions) {
     // If --set-key is provided, store it directly without OAuth
     if (opts.setKey) {
       const trimmedKey = opts.setKey.trim();
@@ -4191,7 +4187,21 @@ auth
       console.error(`Authentication error: ${message}`);
       process.exit(1);
     }
-  });
+}
+
+function configureLoginCommand(command: Command): Command {
+  return command
+    .description("authenticate via browser (OAuth) or store API key directly")
+    .option("--set-key <key>", "store API key directly without OAuth flow")
+    .option("--port <port>", "local callback server port (default: random)", parseInt)
+    .option("--debug", "enable debug output");
+}
+
+configureLoginCommand(auth.command("login", { isDefault: true }))
+  .action(runAuthLogin);
+
+configureLoginCommand(program.command("login"))
+  .action(runAuthLogin);
 
 auth
   .command("show-key")

cli/test/auth.test.ts

@@ -1,5 +1,7 @@
 import { describe, test, expect } from "bun:test";
-import { resolve } from "path";
+import { mkdtempSync, readFileSync, rmSync } from "fs";
+import { tmpdir } from "os";
+import { join, resolve } from "path";
 
 import * as util from "../lib/util";
 import * as pkce from "../lib/pkce";
@@ -212,6 +214,33 @@ describe("Auth callback server", () => {
 });
 
 describe("CLI auth commands", () => {
+  test("cli: login --help shows all options", () => {
+    const r = runCli(["login", "--help"]);
+    expect(r.status).toBe(0);
+    expect(r.stdout).toMatch(/--set-key/);
+    expect(r.stdout).toMatch(/--debug/);
+  });
+
+  test("cli: login --set-key aliases auth login", () => {
+    const home = mkdtempSync(join(tmpdir(), "postgresai-login-"));
+
+    try {
+      const r = runCli(["login", "--set-key", "test-token"], {
+        HOME: home,
+        XDG_CONFIG_HOME: join(home, ".config"),
+      });
+      expect(r.status).toBe(0);
+      expect(r.stdout).toMatch(/API key saved/);
+
+      const saved = JSON.parse(
+        readFileSync(join(home, ".config", "postgresai", "config.json"), "utf8")
+      );
+      expect(saved.apiKey).toBe("test-token");
+    } finally {
+      rmSync(home, { recursive: true, force: true });
+    }
+  });
+
   test("cli: auth login --help shows all options", () => {
     const r = runCli(["auth", "login", "--help"]);
     expect(r.status).toBe(0);