astrabot 0.1.0

package/modes/setup.ts

@@ -0,0 +1,210 @@
+import chalk from "chalk";
+import { intro, outro, text, confirm, isCancel, autocomplete } from "@clack/prompts";
+import {
+  getEnv,
+  getConfigPath,
+  saveConfig,
+} from "../ai/config-loader";
+import { withSpinner } from "../tui/spinner"; // Adjust path as necessary
+
+// 1. Updated interface to include OpenRouter's exact pricing object structure
+interface OpenRouterModel {
+  id: string;
+  name: string;
+  architecture?: {
+    modality?: string;
+    output_modalities?: string[];
+  };
+  pricing?: {
+    prompt: string;      // Price per individual token
+    completion: string;  // Price per individual token
+  };
+}
+
+export async function runSetup(): Promise<void> {
+  intro(chalk.bold("astra setup"));
+
+  console.log(
+    chalk.dim(
+      `Config will be saved to ${getConfigPath()}\n` +
+        `Existing values will be preserved when possible.\n`
+    )
+  );
+
+  const currentKey = getEnv("OPENROUTER_API_KEY") ?? "";
+  const currentModel = getEnv("OPENROUTER_DEFAULT_MODEL") ?? "";
+  const currentFirecrawl = getEnv("FIRECRAWL_API_KEY") ?? "";
+  const currentSkillsDirs = getEnv("SKILLS_DIRS") ?? "";
+
+  // ── OpenRouter API Key ──────────────────────────────────────────
+  const setKey = await confirm({
+    message: "Set OpenRouter API key?",
+    initialValue: !currentKey,
+  });
+  if (isCancel(setKey)) return outro(chalk.dim("Setup cancelled."));
+
+  let apiKey = currentKey;
+  if (setKey) {
+    const val = await text({
+      message: "OpenRouter API key",
+      placeholder: "sk-or-...",
+      initialValue: currentKey,
+      validate: (v) =>
+        (v ?? "").trim() ? undefined : "API key is required",
+    });
+    if (isCancel(val)) return outro(chalk.dim("Setup cancelled."));
+    apiKey = val.trim();
+  }
+
+  // ── Fetching All OpenRouter Models Dynamically ───────────────────
+  let modelOptions: Array<{ value: string; label: string; hint?: string }> = [];
+
+  const setModel = await confirm({
+    message: "Set default OpenRouter model?",
+    initialValue: !currentModel,
+  });
+  if (isCancel(setModel)) return outro(chalk.dim("Setup cancelled."));
+
+  if (setModel) {
+    try {
+      modelOptions = await withSpinner(
+        {
+          message: "Fetching all available OpenRouter models & pricing...",
+          doneMessage: "Loaded models successfully.",
+          failMessage: "Failed to fetch dynamic list. Dropping back to fallback entry.",
+        },
+        async () => {
+          const response = await fetch("https://openrouter.ai/api/v1/models");
+          if (!response.ok) throw new Error("Failed to communicate with OpenRouter registry");
+          
+          const json = (await response.json()) as { data: OpenRouterModel[] };
+          
+          return json.data
+            .filter((model) => {
+              const outModalities = model.architecture?.output_modalities;
+              const modalityStr = model.architecture?.modality || "";
+              
+              if (outModalities) return outModalities.includes("text");
+              if (modalityStr) return modalityStr.endsWith("->text");
+              
+              return true;
+            })
+            .map((model) => {
+              const provider = model.id.split("/")[0];
+              
+              // Convert price-per-token strings to a clean price per 1 Million tokens
+              const promptPriceNum = parseFloat(model.pricing?.prompt || "0") * 1_000_000;
+              const completionPriceNum = parseFloat(model.pricing?.completion || "0") * 1_000_000;
+
+              let pricingHint = "Free";
+              if (promptPriceNum > 0 || completionPriceNum > 0) {
+                pricingHint = `In: $${promptPriceNum.toFixed(2)}, Out: $${completionPriceNum.toFixed(2)} /1M`;
+              }
+
+              return {
+                value: model.id,
+                label: model.name || model.id,
+                hint: `${provider} (${pricingHint})`,
+              };
+            });
+        }
+      );
+    } catch (error) {
+      modelOptions = []; 
+    }
+
+    let modelId = currentModel;
+
+    if (modelOptions.length > 0) {
+      const selectedModel = await autocomplete({
+        message: "Select an OpenRouter text model (Type to search & compare pricing)",
+        options: [
+          ...modelOptions,
+          { value: "custom", label: "Custom Entry...", hint: "Type manual ID" }
+        ],
+        placeholder: "Search e.g. 'claude', 'gpt', 'llama'...",
+      });
+      if (isCancel(selectedModel)) return outro(chalk.dim("Setup cancelled."));
+
+      if (selectedModel === "custom") {
+        const customVal = await text({
+          message: "Enter custom OpenRouter Model ID",
+          placeholder: "provider/model-name",
+          initialValue: currentModel,
+          validate: (v) => ((v ?? "").trim() ? undefined : "Model ID is required"),
+        });
+        if (isCancel(customVal)) return outro(chalk.dim("Setup cancelled."));
+        modelId = customVal.trim();
+      } else {
+        modelId = selectedModel as string;
+      }
+    } else {
+      const fallbackVal = await text({
+        message: "Enter OpenRouter Model ID",
+        placeholder: "anthropic/claude-3.5-sonnet",
+        initialValue: currentModel || "anthropic/claude-3.5-sonnet",
+        validate: (v) => ((v ?? "").trim() ? undefined : "Model ID is required"),
+      });
+      if (isCancel(fallbackVal)) return outro(chalk.dim("Setup cancelled."));
+      modelId = fallbackVal.trim();
+    }
+    
+    var finalModelId = modelId;
+  } else {
+    var finalModelId = currentModel;
+  }
+
+  // ── Firecrawl API Key (optional) ────────────────────────────────
+  const setFirecrawl = await confirm({
+    message: "Set Firecrawl API key? (optional — enables web search & crawl)",
+    initialValue: false,
+  });
+  if (isCancel(setFirecrawl)) return outro(chalk.dim("Setup cancelled."));
+
+  let firecrawlKey = currentFirecrawl;
+  if (setFirecrawl) {
+    const val = await text({
+      message: "Firecrawl API key",
+      placeholder: "fc-...",
+      initialValue: currentFirecrawl,
+      validate: (_v) => undefined,
+    });
+    if (isCancel(val)) return outro(chalk.dim("Setup cancelled."));
+    firecrawlKey = (val ?? "").trim();
+  }
+
+  // ── Skills Directories (optional) ───────────────────────────────
+  const setSkills = await confirm({
+    message: "Set custom skills directories? (optional)",
+    initialValue: false,
+  });
+  if (isCancel(setSkills)) return outro(chalk.dim("Setup cancelled."));
+
+  let skillsDirs = currentSkillsDirs;
+  if (setSkills) {
+    const val = await text({
+      message: "Skills directories (semicolon-separated)",
+      placeholder: "/path/to/skills;/another/dir",
+      initialValue: currentSkillsDirs,
+      validate: (_v) => undefined,
+    });
+    if (isCancel(val)) return outro(chalk.dim("Setup cancelled."));
+    skillsDirs = (val ?? "").trim();
+  }
+
+  // ── Save ────────────────────────────────────────────────────────
+  const entries: Record<string, string> = {};
+  if (apiKey) entries.OPENROUTER_API_KEY = apiKey;
+  if (finalModelId) entries.OPENROUTER_DEFAULT_MODEL = finalModelId;
+  if (firecrawlKey) entries.FIRECRAWL_API_KEY = firecrawlKey;
+  if (skillsDirs) entries.SKILLS_DIRS = skillsDirs;
+
+  saveConfig(entries);
+
+  outro(
+    chalk.green(
+      `\n✔ Configuration saved to ${getConfigPath()}\n` +
+        `    You can now run "astra wakeup" to get started.\n`
+    )
+  );
+}

package/package.json

@@ -0,0 +1,62 @@
+{
+  "name": "astrabot",
+  "version": "0.1.0",
+  "description": "AI-native development companion — Agent, Ask, and Plan modes in your terminal.",
+  "module": "index.ts",
+  "type": "module",
+  "bin": {
+    "astra": "bin/astra"
+  },
+  "files": [
+    "bin/astra",
+    "index.ts",
+    "ai/",
+    "modes/",
+    "tui/",
+    "session/",
+    "core/",
+    "game/",
+    "README.md",
+    "tsconfig.json"
+  ],
+  "scripts": {
+    "dev": "bun run index.ts",
+    "setup": "bun run index.ts setup",
+    "wakeup": "bun run index.ts wakeup",
+    "test": "bun test",
+    "prepublishOnly": "bun test"
+  },
+  "devDependencies": {
+    "@types/bun": "latest"
+  },
+  "peerDependencies": {
+    "typescript": "^5"
+  },
+  "dependencies": {
+    "@clack/core": "^1.3.1",
+    "@clack/prompts": "^1.4.0",
+    "@mendable/firecrawl-js": "^4.25.1",
+    "@openrouter/ai-sdk-provider": "^2.9.0",
+    "@types/marked-terminal": "^6.1.1",
+    "@types/node": "^25.9.1",
+    "chalk": "^5.6.2",
+    "commander": "^15.0.0",
+    "diff": "^9.0.0",
+    "docx": "^9.7.1",
+    "dotenv": "^17.4.2",
+    "figlet": "^1.11.0",
+    "marked": "^18.0.4",
+    "marked-terminal": "^7.3.0"
+  },
+  "keywords": [
+    "ai",
+    "cli",
+    "agent",
+    "openrouter",
+    "coding-assistant"
+  ],
+  "license": "MIT",
+  "engines": {
+    "bun": ">=1.0.0"
+  }
+}

package/session/index.ts

@@ -0,0 +1,45 @@
+// ── Store (raw persistence layer) ─────────────────────────────────────────
+export {
+  listSessions,
+  getSession,
+  getMostRecentSession,
+  createSession,
+  updateSession,
+  deleteSession,
+  clearAllSessions,
+  appendTranscript,
+  readSessionActions,
+} from "./store";
+export type {
+  SessionMode,
+  SessionStatus,
+  SessionEntry,
+  TranscriptMessage,
+  SessionStoreIndex,
+} from "./store";
+
+// ── Context building ───────────────────────────────────────────────────────
+export {
+  captureSessionContext,
+  buildContextSummary,
+  buildSessionOneliner,
+} from "./session-context";
+export type { SessionContextData } from "./session-context";
+
+// ── Session lifecycle ──────────────────────────────────────────────────────
+export {
+  beginSession,
+  endSession,
+  endMultiSession,
+  markSessionInterrupted,
+  recordUserMessage,
+  recordAgentMessage,
+  getResumableSession,
+  getSessionHistory,
+  removeSession,
+  formatSessionLine,
+} from "./session-manager";
+export type { BeginSessionResult } from "./session-manager";
+
+// ── Agent tools ────────────────────────────────────────────────────────────
+export { createSessionTools } from "./session-tools";

package/session/session-context.ts

@@ -0,0 +1,188 @@
+import type { ActionTracker } from "../modes/agent/action-tracker";
+import type { SessionEntry, TranscriptMessage } from "./store";
+
+export type { TranscriptMessage };
+
+/**
+ * Rich context persisted alongside a session entry.
+ */
+export interface SessionContextData {
+  transcript: TranscriptMessage[];
+  activeFiles: string[];
+  pendingTasks: string[];
+  lastAgentResponse: string;
+}
+
+/**
+ * Build the full context payload from the current session state.
+ * This is called at session end so all fields are populated.
+ */
+export function captureSessionContext(
+  tracker: ActionTracker,
+  transcript: TranscriptMessage[],
+  lastAgentResponse: string,
+  pendingTasks: string[] = []
+): SessionContextData {
+  const actions = tracker.getActions();
+  const activeFiles = [
+    ...new Set(
+      actions
+        .filter(
+          (a) =>
+            a.type === "file_create" ||
+            a.type === "file_modify" ||
+            a.type === "file_delete" ||
+            a.type === "code_analysis"
+        )
+        .map((a) => a.path)
+        .filter(Boolean)
+    ),
+  ];
+
+  return {
+    transcript,
+    activeFiles,
+    pendingTasks,
+    lastAgentResponse,
+  };
+}
+
+// ── Context summary builders ────────────────────────────────────────────────
+
+/**
+ * Build the FULL resumption context block injected into the agent's system
+ * prompt when continuing an existing session.
+ *
+ * Includes:
+ *  - session metadata
+ *  - pending tasks
+ *  - recent transcript (last N turns, configurable)
+ *  - touched files
+ *  - action counts
+ *  - the agent's last response
+ */
+export function buildContextSummary(
+  entry: SessionEntry,
+  opts: { transcriptTurns?: number } = {}
+): string {
+  const { transcriptTurns = 10 } = opts;
+  const parts: string[] = [];
+
+  parts.push("╔══════════════════════════════════════╗");
+  parts.push("║       RESUMED SESSION CONTEXT        ║");
+  parts.push("╚══════════════════════════════════════╝");
+  parts.push("");
+
+  // ── Core metadata
+  parts.push(`Mode        : ${entry.mode}`);
+  parts.push(`Session ID  : ${entry.id}`);
+  parts.push(`Started     : ${new Date(entry.createdAt).toLocaleString()}`);
+  parts.push("");
+
+  // ── Goals (all goals across session, most recent last)
+  if (entry.allGoals && entry.allGoals.length > 1) {
+    parts.push("Goals in this session:");
+    entry.allGoals.forEach((g, i) => parts.push(`  ${i + 1}. ${g}`));
+  } else {
+    parts.push(`Goal: ${entry.lastGoal}`);
+  }
+  parts.push("");
+
+  // ── Summary (LLM-generated)
+  if (entry.summary) {
+    parts.push("What happened:");
+    parts.push(`  ${entry.summary}`);
+    parts.push("");
+  }
+
+  // ── Pending tasks
+  if (entry.pendingTasks && entry.pendingTasks.length > 0) {
+    parts.push("⚠ Pending / incomplete tasks:");
+    entry.pendingTasks.forEach((t) => parts.push(`  • ${t}`));
+    parts.push("");
+  }
+
+  // ── Files
+  if (entry.touchedFiles.length > 0) {
+    const shown = entry.touchedFiles.slice(0, 20);
+    parts.push(`Files touched (${entry.touchedFiles.length}):`);
+    shown.forEach((f) => parts.push(`  • ${f}`));
+    if (entry.touchedFiles.length > 20) {
+      parts.push(`  … and ${entry.touchedFiles.length - 20} more`);
+    }
+    parts.push("");
+  }
+
+  // ── Action counts
+  if (entry.appliedActions > 0 || entry.rejectedActions > 0) {
+    parts.push(
+      `Actions: ${entry.appliedActions} applied, ${entry.rejectedActions} rejected.`
+    );
+    parts.push("");
+  }
+
+  // ── Transcript (most recent N turns)
+  if (entry.transcript && entry.transcript.length > 0) {
+    const turns = entry.transcript.slice(-transcriptTurns * 2);
+    parts.push(`Recent conversation (last ${Math.floor(turns.length / 2)} turns):`);
+    parts.push("─".repeat(40));
+    for (const msg of turns) {
+      const label =
+        msg.role === "user"
+          ? "User"
+          : msg.role === "agent"
+            ? "Agent"
+            : "System";
+      const snippet = msg.content.length > 400
+        ? msg.content.slice(0, 400) + "…"
+        : msg.content;
+      parts.push(`[${label}] ${snippet}`);
+    }
+    parts.push("─".repeat(40));
+    parts.push("");
+  }
+
+  // ── Last agent response (in case transcript was trimmed)
+  if (entry.lastAgentResponse && entry.transcript.length === 0) {
+    parts.push("Agent's last response:");
+    parts.push(
+      entry.lastAgentResponse.length > 800
+        ? entry.lastAgentResponse.slice(0, 800) + "…"
+        : entry.lastAgentResponse
+    );
+    parts.push("");
+  }
+
+  parts.push("You are now resuming this session. Continue where you left off.");
+  parts.push(
+    entry.pendingTasks && entry.pendingTasks.length > 0
+      ? "Address the pending tasks above unless the user specifies otherwise."
+      : "Await the user's next instruction."
+  );
+
+  return parts.join("\n");
+}
+
+/**
+ * Lightweight one-line summary for display in the CLI session picker.
+ */
+export function buildSessionOneliner(entry: SessionEntry): string {
+  const age = humanAge(entry.updatedAt);
+  const pending = entry.pendingTasks?.length
+    ? ` [${entry.pendingTasks.length} pending]`
+    : "";
+  return `${age.padEnd(8)} [${entry.mode}] ${entry.lastGoal.slice(0, 60)}${pending}`;
+}
+
+// ── Helpers ────────────────────────────────────────────────────────────────
+
+function humanAge(isoString: string): string {
+  const diff = Date.now() - new Date(isoString).getTime();
+  const minutes = Math.floor(diff / 60_000);
+  if (minutes < 1) return "just now";
+  if (minutes < 60) return `${minutes}m ago`;
+  const hours = Math.floor(minutes / 60);
+  if (hours < 24) return `${hours}h ago`;
+  const days = Math.floor(hours / 24);
+  return `${days}d ago`;
+}