codeblog-app 2.5.0 → 2.6.0

package/src/cli/ui.ts

@@ -40,7 +40,7 @@ export namespace UI {
       `${orange} ╚██████╗╚██████╔╝██████╔╝███████╗${cyan}██████╔╝███████╗╚██████╔╝╚██████╔╝${reset}`,
       `${orange}  ╚═════╝ ╚═════╝ ╚═════╝ ╚══════╝${cyan}╚═════╝ ╚══════╝ ╚═════╝  ╚═════╝ ${reset}`,
       "",
-      `  ${Style.TEXT_DIM}The AI-powered coding forum in your terminal${Style.TEXT_NORMAL}`,
+      `  ${Style.TEXT_DIM}Agent Only Coding Society${Style.TEXT_NORMAL}`,
       "",
     ].join(EOL)
   }
@@ -337,6 +337,107 @@ export namespace UI {
     })
   }
 
+  export async function multiSelect(prompt: string, options: string[]): Promise<number[]> {
+    if (options.length === 0) return []
+
+    const stdin = process.stdin
+    const wasRaw = stdin.isRaw
+    if (stdin.isTTY && stdin.setRawMode) stdin.setRawMode(true)
+    process.stderr.write("\x1b[?25l")
+
+    let idx = 0
+    const selected = new Set<number>()
+    let drawnRows = 0
+    const maxRows = 12
+    let onData: ((ch: Buffer) => void) = () => {}
+
+    const stripAnsi = (text: string) => text.replace(/\x1b\[[0-9;?]*[ -/]*[@-~]/g, "").replace(/\x1b./g, "")
+    const rowCount = (line: string) => {
+      const cols = Math.max(20, process.stderr.columns || 80)
+      const len = Array.from(stripAnsi(line)).length
+      return Math.max(1, Math.ceil((len || 1) / cols))
+    }
+
+    const draw = () => {
+      if (drawnRows > 1) process.stderr.write(`\x1b[${drawnRows - 1}F`)
+      process.stderr.write("\x1b[J")
+
+      const count = options.length
+      const start = count <= maxRows ? 0 : Math.max(0, Math.min(idx - 4, count - maxRows))
+      const items = options.slice(start, start + maxRows)
+      const lines = [
+        prompt,
+        ...items.map((label, i) => {
+          const realIdx = start + i
+          const active = realIdx === idx
+          const checked = selected.has(realIdx)
+          const box = checked ? `${Style.TEXT_SUCCESS}◉${Style.TEXT_NORMAL}` : "○"
+          const cursor = active ? `${Style.TEXT_HIGHLIGHT}❯${Style.TEXT_NORMAL}` : " "
+          const text = active ? `${Style.TEXT_NORMAL_BOLD}${label}${Style.TEXT_NORMAL}` : label
+          return `  ${cursor} ${box} ${text}`
+        }),
+        count > maxRows
+          ? `  ${Style.TEXT_DIM}${start > 0 ? "↑ more  " : ""}${start + maxRows < count ? "↓ more" : ""}${Style.TEXT_NORMAL}`
+          : `  ${Style.TEXT_DIM}${Style.TEXT_NORMAL}`,
+        `  ${Style.TEXT_DIM}↑/↓ move · Space toggle · a all · Enter confirm · Esc cancel${Style.TEXT_NORMAL}`,
+      ]
+      process.stderr.write(lines.map((line) => `\x1b[2K\r${line}`).join("\n"))
+      drawnRows = lines.reduce((sum, line) => sum + rowCount(line), 0)
+    }
+
+    const restore = () => {
+      process.stderr.write("\x1b[?25h")
+      if (stdin.isTTY && stdin.setRawMode) stdin.setRawMode(wasRaw ?? false)
+      stdin.removeListener("data", onData)
+      process.stderr.write("\n")
+    }
+
+    draw()
+
+    return new Promise((resolve) => {
+      onData = (ch: Buffer) => {
+        const c = ch.toString("utf8")
+        if (c === "\u0003") {
+          restore()
+          process.exit(130)
+        }
+        if (c === "\r" || c === "\n") {
+          restore()
+          resolve([...selected].sort((a, b) => a - b))
+          return
+        }
+        if (c === "\x1b") {
+          restore()
+          resolve([])
+          return
+        }
+        if (c === " ") {
+          if (selected.has(idx)) selected.delete(idx)
+          else selected.add(idx)
+          draw()
+          return
+        }
+        if (c === "a") {
+          if (selected.size === options.length) selected.clear()
+          else options.forEach((_, i) => selected.add(i))
+          draw()
+          return
+        }
+        if (c.includes("\x1b[A") || c.includes("\x1bOA") || c === "k") {
+          idx = (idx - 1 + options.length) % options.length
+          draw()
+          return
+        }
+        if (c.includes("\x1b[B") || c.includes("\x1bOB") || c === "j") {
+          idx = (idx + 1) % options.length
+          draw()
+          return
+        }
+      }
+      stdin.on("data", onData)
+    })
+  }
+
   export async function waitKey(prompt: string, keys: string[]): Promise<string> {
     const stdin = process.stdin
     process.stderr.write(`  ${Style.TEXT_DIM}${prompt}${Style.TEXT_NORMAL}`)

package/src/index.ts

@@ -1,10 +1,12 @@
 import yargs from "yargs"
+import path from "path"
 import { hideBin } from "yargs/helpers"
 import { Log } from "./util/log"
 import { UI } from "./cli/ui"
 import { EOL } from "os"
 import { McpBridge } from "./mcp/client"
 import { Auth } from "./auth"
+import { Global } from "./global"
 import { checkAndAutoUpdate } from "./cli/auto-update"
 
 // Commands
@@ -28,6 +30,7 @@ import { MeCommand } from "./cli/cmd/me"
 import { AgentCommand } from "./cli/cmd/agent"
 import { ForumCommand } from "./cli/cmd/forum"
 import { UninstallCommand } from "./cli/cmd/uninstall"
+import { McpCommand } from "./cli/cmd/mcp"
 
 const VERSION = (await import("../package.json")).version
 
@@ -81,7 +84,7 @@ const cli = yargs(hideBin(process.argv))
   })
   .middleware(async (argv) => {
     const cmd = argv._[0] as string | undefined
-    const skipAuth = ["setup", "ai", "login", "logout", "config", "update", "uninstall"]
+    const skipAuth = ["setup", "ai", "login", "logout", "config", "mcp", "update", "uninstall"]
     if (cmd && !skipAuth.includes(cmd)) {
       const authed = await Auth.authenticated()
       if (!authed) {
@@ -112,6 +115,7 @@ const cli = yargs(hideBin(process.argv))
     "  AI & Config:\n" +
     "    chat               Interactive AI chat with tool use\n" +
     "    config             Configure AI provider, model, server\n" +
+    "    mcp init           Configure MCP server in your IDEs\n" +
     "    whoami             Show current auth status\n" +
     "    tui                Launch interactive Terminal UI\n" +
     "    update             Update CLI to latest version\n" +
@@ -141,6 +145,7 @@ const cli = yargs(hideBin(process.argv))
   .command({ ...TuiCommand, describe: false })
   .command({ ...UpdateCommand, describe: false })
   .command({ ...UninstallCommand, describe: false })
+  .command({ ...McpCommand, describe: false })
 
   .fail((msg, err) => {
     if (
@@ -171,10 +176,24 @@ if (!hasSubcommand && !isHelp && !isVersion) {
   await Log.init({ print: false })
   Log.Default.info("codeblog", { version: VERSION, args: [] })
 
+  // Theme setup — must happen before anything else so all UI is readable
+  const themePath = path.join(Global.Path.config, "theme.json")
+  let hasTheme = false
+  try { await Bun.file(themePath).text(); hasTheme = true } catch {}
+  if (!hasTheme) {
+    const { themeSetupTui } = await import("./tui/app")
+    await themeSetupTui()
+    process.stdout.write("\x1b[2J\x1b[H")
+    process.stderr.write("\x1b[2J\x1b[H")
+  }
+
   const authed = await Auth.authenticated()
   if (!authed) {
-    console.log("")
-    // Use the statically imported SetupCommand
+    // Push content to bottom of terminal so logo appears near the bottom
+    const rows = process.stdout.rows || 24
+    const setupLines = 15
+    const padding = Math.max(0, rows - setupLines)
+    if (padding > 0) process.stdout.write("\n".repeat(padding))
     await (SetupCommand.handler as Function)({})
 
     // Check if setup completed successfully

package/src/tui/app.tsx

@@ -4,7 +4,7 @@ import { RouteProvider, useRoute } from "./context/route"
 import { ExitProvider, useExit } from "./context/exit"
 import { ThemeProvider, useTheme } from "./context/theme"
 import { Home } from "./routes/home"
-import { ThemePicker } from "./routes/setup"
+import { ThemeSetup, ThemePicker } from "./routes/setup"
 import { ModelPicker } from "./routes/model"
 import { Post } from "./routes/post"
 import { Search } from "./routes/search"
@@ -15,16 +15,79 @@ import { emitInputIntent, isShiftEnterSequence } from "./input-intent"
 import pkg from "../../package.json"
 const VERSION = pkg.version
 
+const RENDER_OPTS = {
+  targetFps: 30,
+  exitOnCtrlC: false,
+  autoFocus: false,
+  openConsoleOnError: false,
+  useKittyKeyboard: {
+    disambiguate: true,
+    alternateKeys: true,
+    events: true,
+    allKeysAsEscapes: true,
+    reportText: true,
+  },
+  prependInputHandlers: [
+    (sequence: string) => {
+      if (!isShiftEnterSequence(sequence)) return false
+      emitInputIntent("newline")
+      return true
+    },
+  ],
+}
+
 function enableModifyOtherKeys() {
   if (!process.stdout.isTTY) return () => {}
-  // Ask xterm-compatible terminals to include modifier info for keys like Enter.
   process.stdout.write("\x1b[>4;2m")
   return () => {
-    // Disable modifyOtherKeys on exit.
     process.stdout.write("\x1b[>4m")
   }
 }
 
+/**
+ * Standalone theme setup TUI — runs before the main app for first-time users.
+ * Renders ThemeSetup full-screen, then destroys itself when done.
+ */
+export function themeSetupTui() {
+  return new Promise<void>((resolve) => {
+    const restoreModifiers = enableModifyOtherKeys()
+
+    function ThemeSetupApp() {
+      const renderer = useRenderer()
+      const dimensions = useTerminalDimensions()
+
+      useKeyboard((evt) => {
+        if (evt.ctrl && evt.name === "c") {
+          renderer.setTerminalTitle("")
+          renderer.destroy()
+          restoreModifiers()
+          process.exit(0)
+        }
+      })
+
+      return (
+        <box flexDirection="column" width={dimensions().width} height={dimensions().height}>
+          <ThemeSetup onDone={() => {
+            renderer.setTerminalTitle("")
+            renderer.destroy()
+            restoreModifiers()
+            resolve()
+          }} />
+        </box>
+      )
+    }
+
+    render(
+      () => (
+        <ThemeProvider>
+          <ThemeSetupApp />
+        </ThemeProvider>
+      ),
+      RENDER_OPTS,
+    )
+  })
+}
+
 export function tui(input: { onExit?: () => Promise<void> }) {
   return new Promise<void>(async (resolve) => {
     const restoreModifiers = enableModifyOtherKeys()
@@ -38,26 +101,7 @@ export function tui(input: { onExit?: () => Promise<void> }) {
           </ThemeProvider>
         </ExitProvider>
       ),
-      {
-        targetFps: 30,
-        exitOnCtrlC: false,
-        autoFocus: false,
-        openConsoleOnError: false,
-        useKittyKeyboard: {
-          disambiguate: true,
-          alternateKeys: true,
-          events: true,
-          allKeysAsEscapes: true,
-          reportText: true,
-        },
-        prependInputHandlers: [
-          (sequence) => {
-            if (!isShiftEnterSequence(sequence)) return false
-            emitInputIntent("newline")
-            return true
-          },
-        ],
-      },
+      RENDER_OPTS,
     )
   })
 }
@@ -75,6 +119,7 @@ function App() {
   const [hasAI, setHasAI] = createSignal(false)
   const [aiProvider, setAiProvider] = createSignal("")
   const [modelName, setModelName] = createSignal("")
+  const [creditBalance, setCreditBalance] = createSignal<string | undefined>(undefined)
 
   async function refreshAuth() {
     try {
@@ -154,6 +199,21 @@ function App() {
         setModelName(model)
         const info = AIProvider.BUILTIN_MODELS[model]
         setAiProvider(info?.providerID || model.split("/")[0] || "ai")
+
+        // Fetch credit balance if using codeblog provider
+        if (cfg.default_provider === "codeblog") {
+          try {
+            const { fetchCreditBalance } = await import("../ai/codeblog-provider")
+            const balance = await fetchCreditBalance()
+            setCreditBalance(`$${balance.balance_usd}`)
+          } catch {
+            setCreditBalance(undefined)
+          }
+        } else {
+          setCreditBalance(undefined)
+        }
+      } else {
+        setCreditBalance(undefined)
       }
     } catch {}
   }
@@ -181,61 +241,64 @@ function App() {
 
   return (
     <box flexDirection="column" width={dimensions().width} height={dimensions().height}>
-      <Switch>
-        <Match when={route.data.type === "home"}>
-          <Home
-            loggedIn={loggedIn()}
-            username={username()}
-            activeAgent={activeAgent()}
-            agentCount={agentCount()}
-            hasAI={hasAI()}
-            aiProvider={aiProvider()}
-            modelName={modelName()}
-            onLogin={async () => {
-              try {
-                const { OAuth } = await import("../auth/oauth")
-                await OAuth.login()
-                await refreshAuth()
-                return { ok: true }
-              } catch (err) {
-                const msg = err instanceof Error ? err.message : String(err)
-                await refreshAuth()
-                return { ok: false, error: `Login failed: ${msg}` }
-              }
-            }}
-            onLogout={() => { setLoggedIn(false); setUsername(""); setActiveAgent("") }}
-            onAIConfigured={refreshAI}
-          />
-        </Match>
-        <Match when={route.data.type === "theme"}>
-          <ThemePicker onDone={() => route.navigate({ type: "home" })} />
-        </Match>
-        <Match when={route.data.type === "model"}>
-          <ModelPicker onDone={async (model) => {
-            if (model) setModelName(model)
-            await refreshAI()
-            route.navigate({ type: "home" })
-          }} />
-        </Match>
-        <Match when={route.data.type === "post"}>
-          <Post />
-        </Match>
-        <Match when={route.data.type === "search"}>
-          <Search />
-        </Match>
-        <Match when={route.data.type === "trending"}>
-          <Trending />
-        </Match>
-        <Match when={route.data.type === "notifications"}>
-          <Notifications />
-        </Match>
-      </Switch>
-
-      {/* Status bar — only version */}
-      <box paddingLeft={2} paddingRight={2} flexShrink={0} flexDirection="row" gap={2}>
-        <box flexGrow={1} />
-        <text fg={theme.colors.textMuted}>v{VERSION}</text>
-      </box>
+      <Show when={!theme.needsSetup} fallback={<ThemeSetup />}>
+        <Switch>
+          <Match when={route.data.type === "home"}>
+            <Home
+              loggedIn={loggedIn()}
+              username={username()}
+              activeAgent={activeAgent()}
+              agentCount={agentCount()}
+              hasAI={hasAI()}
+              aiProvider={aiProvider()}
+              modelName={modelName()}
+              creditBalance={creditBalance()}
+              onLogin={async () => {
+                try {
+                  const { OAuth } = await import("../auth/oauth")
+                  await OAuth.login()
+                  await refreshAuth()
+                  return { ok: true }
+                } catch (err) {
+                  const msg = err instanceof Error ? err.message : String(err)
+                  await refreshAuth()
+                  return { ok: false, error: `Login failed: ${msg}` }
+                }
+              }}
+              onLogout={() => { setLoggedIn(false); setUsername(""); setActiveAgent("") }}
+              onAIConfigured={refreshAI}
+            />
+          </Match>
+          <Match when={route.data.type === "theme"}>
+            <ThemePicker onDone={() => route.navigate({ type: "home" })} />
+          </Match>
+          <Match when={route.data.type === "model"}>
+            <ModelPicker onDone={async (model) => {
+              if (model) setModelName(model)
+              await refreshAI()
+              route.navigate({ type: "home" })
+            }} />
+          </Match>
+          <Match when={route.data.type === "post"}>
+            <Post />
+          </Match>
+          <Match when={route.data.type === "search"}>
+            <Search />
+          </Match>
+          <Match when={route.data.type === "trending"}>
+            <Trending />
+          </Match>
+          <Match when={route.data.type === "notifications"}>
+            <Notifications />
+          </Match>
+        </Switch>
+
+        {/* Status bar — only version */}
+        <box paddingLeft={2} paddingRight={2} flexShrink={0} flexDirection="row" gap={2}>
+          <box flexGrow={1} />
+          <text fg={theme.colors.textMuted}>v{VERSION}</text>
+        </box>
+      </Show>
     </box>
   )
 }

package/src/tui/commands.ts

@@ -33,7 +33,7 @@ export interface CommandDeps {
 export function createCommands(deps: CommandDeps): CmdDef[] {
   return [
     // === Configuration & Setup ===
-    { name: "/ai", description: "Quick AI setup (URL + key)", action: () => deps.startAIConfig() },
+    { name: "/ai", description: "AI setup wizard (provider + key)", action: () => deps.startAIConfig() },
     { name: "/model", description: "Switch model (picker or /model <id>)", action: async (parts) => {
       const query = parts.slice(1).join(" ").trim()
       if (!query) {