@ridit/lens 0.3.3 → 0.3.5

package/src/components/repo/RepoAnalysis.tsx

@@ -1,18 +1,26 @@
 import React from "react";
-import { Box, Text, useInput } from "ink";
+import { Box, Text, Static, useInput } from "ink";
 import Spinner from "ink-spinner";
 import figures from "figures";
-import { useState } from "react";
+import { useState, useRef } from "react";
 import { writeFileSync } from "fs";
 import path from "path";
 import { ACCENT } from "../../colors";
-import { requestFileList, analyzeRepo } from "../../utils/ai";
+import {
+  requestFileList,
+  analyzeRepo,
+  extractToolingPatch,
+} from "../../utils/ai";
 import { ProviderPicker } from "../provider/ProviderPicker";
 import { PreviewRunner } from "./PreviewRunner";
 import { IssueFixer } from "./IssueFixer";
-import { writeLensFile } from "../../utils/lensfile";
+import { writeLensFile, patchLensFile } from "../../utils/lensfile";
+import { callChat } from "../../utils/chat";
+import { StaticMessage } from "../chat/ChatMessage";
+import { InputBox, TypewriterText, ShortcutBar } from "../chat/ChatOverlays";
 import type { Provider } from "../../types/config";
 import type { AnalysisResult, ImportantFile } from "../../types/repo";
+import type { Message } from "../../types/chat";
 import { useThinkingPhrase } from "../../utils/thinking";
 
 type AnalysisStage =
@@ -24,12 +32,17 @@ type AnalysisStage =
   | { type: "written"; filePath: string }
   | { type: "previewing" }
   | { type: "fixing"; result: AnalysisResult }
+  | { type: "asking"; result: AnalysisResult }
   | { type: "error"; message: string };
 
 const OUTPUT_FILES = ["CLAUDE.md", "copilot-instructions.md"] as const;
 type OutputFile = (typeof OUTPUT_FILES)[number];
 
 function buildMarkdown(repoUrl: string, result: AnalysisResult): string {
+  const toolingLines = Object.entries(result.tooling ?? {})
+    .map(([k, v]) => `- **${k}**: ${v}`)
+    .join("\n");
+
   return `# Repository Analysis
 
 > ${repoUrl}
@@ -37,28 +50,56 @@ function buildMarkdown(repoUrl: string, result: AnalysisResult): string {
 ## Overview
 ${result.overview}
 
+## Architecture
+${result.architecture ?? ""}
+
+## Tooling
+${toolingLines || "- Not determined"}
+
 ## Important Folders
 ${result.importantFolders.map((f) => `- ${f}`).join("\n")}
 
-## Missing Configs
-${
-  result.missingConfigs.length > 0
-    ? result.missingConfigs.map((f) => `- ${f}`).join("\n")
-    : "- None detected"
-}
+## Key Files
+${(result.keyFiles ?? []).map((f) => `- ${f}`).join("\n")}
 
-## Security Issues
-${
-  result.securityIssues.length > 0
-    ? result.securityIssues.map((s) => `- ⚠️ ${s}`).join("\n")
-    : "- None detected"
-}
+## Patterns & Idioms
+${(result.patterns ?? []).map((p) => `- ${p}`).join("\n")}
 
 ## Suggestions
 ${result.suggestions.map((s) => `- ${s}`).join("\n")}
 `;
 }
 
+function buildQASystemPrompt(repoUrl: string, result: AnalysisResult): string {
+  const toolingLines = Object.entries(result.tooling ?? {})
+    .map(([k, v]) => `- ${k}: ${v}`)
+    .join("\n");
+
+  return `You are a codebase assistant for the repository at ${repoUrl}.
+
+Here is what you know about this codebase:
+
+Overview:
+${result.overview}
+
+Architecture:
+${result.architecture ?? "Not determined"}
+
+Tooling:
+${toolingLines || "Not determined"}
+
+Important Folders:
+${result.importantFolders.map((f) => `- ${f}`).join("\n")}
+
+Key Files:
+${(result.keyFiles ?? []).map((f) => `- ${f}`).join("\n")}
+
+Patterns & Idioms:
+${(result.patterns ?? []).map((p) => `- ${p}`).join("\n")}
+
+Answer questions about this codebase concisely and accurately. If you're unsure about something not covered in the analysis, say so clearly rather than guessing.`;
+}
+
 function AskingFilesStep() {
   const phrase = useThinkingPhrase(true, "model");
   return (
@@ -83,6 +124,121 @@ function AnalyzingStep() {
   );
 }
 
+// ─── CodebaseQA ──────────────────────────────────────────────────────────────
+
+type QAStage = "idle" | "thinking";
+
+function CodebaseQA({
+  repoUrl,
+  result,
+  provider,
+  onExit,
+}: {
+  repoUrl: string;
+  result: AnalysisResult;
+  provider: Provider;
+  onExit: () => void;
+}) {
+  const [committed, setCommitted] = useState<Message[]>([]);
+  const [allMessages, setAllMessages] = useState<Message[]>([]);
+  const [inputValue, setInputValue] = useState("");
+  const [inputKey, setInputKey] = useState(0);
+  const [qaStage, setQaStage] = useState<QAStage>("idle");
+  const abortRef = useRef<AbortController | null>(null);
+  const systemPrompt = buildQASystemPrompt(repoUrl, result);
+  const thinkingPhrase = useThinkingPhrase(qaStage === "thinking");
+
+  useInput((_, key) => {
+    if (key.escape) {
+      if (qaStage === "thinking") {
+        abortRef.current?.abort();
+        abortRef.current = null;
+        setQaStage("idle");
+        return;
+      }
+      onExit();
+    }
+  });
+
+  const sendQuestion = (text: string) => {
+    const trimmed = text.trim();
+    if (!trimmed) return;
+
+    const userMsg: Message = { role: "user", type: "text", content: trimmed };
+    const nextAll = [...allMessages, userMsg];
+    setCommitted((prev) => [...prev, userMsg]);
+    setAllMessages(nextAll);
+    setQaStage("thinking");
+
+    const abort = new AbortController();
+    abortRef.current = abort;
+
+    callChat(provider, systemPrompt, nextAll, abort.signal)
+      .then((answer) => {
+        const assistantMsg: Message = {
+          role: "assistant",
+          type: "text",
+          content: answer,
+        };
+        setCommitted((prev) => [...prev, assistantMsg]);
+        setAllMessages([...nextAll, assistantMsg]);
+        setQaStage("idle");
+      })
+      .catch((err: unknown) => {
+        if (err instanceof Error && err.name === "AbortError") {
+          setQaStage("idle");
+          return;
+        }
+        const errMsg: Message = {
+          role: "assistant",
+          type: "text",
+          content: `Error: ${err instanceof Error ? err.message : "Request failed"}`,
+        };
+        setCommitted((prev) => [...prev, errMsg]);
+        setAllMessages([...nextAll, errMsg]);
+        setQaStage("idle");
+      });
+  };
+
+  return (
+    <Box flexDirection="column">
+      <Static items={committed}>
+        {(msg, i) => <StaticMessage key={i} msg={msg} />}
+      </Static>
+
+      {qaStage === "thinking" && (
+        <Box gap={1}>
+          <Text color={ACCENT}>●</Text>
+          <TypewriterText text={thinkingPhrase} />
+          <Text color="gray" dimColor>
+            · esc cancel
+          </Text>
+        </Box>
+      )}
+
+      {qaStage === "idle" && (
+        <Box flexDirection="column">
+          <InputBox
+            value={inputValue}
+            onChange={setInputValue}
+            onSubmit={(val) => {
+              if (val.trim()) sendQuestion(val.trim());
+              setInputValue("");
+              setInputKey((k) => k + 1);
+            }}
+            inputKey={inputKey}
+          />
+          <Text color="gray" dimColor>
+            enter send · esc back
+          </Text>
+        </Box>
+      )}
+    </Box>
+  );
+}
+
+// ─── RepoAnalysis ─────────────────────────────────────────────────────────────
+
 export const RepoAnalysis = ({
   repoUrl,
   repoPath,
@@ -103,18 +259,31 @@ export const RepoAnalysis = ({
       ? { type: "done", result: preloadedResult }
       : { type: "picking-provider" },
   );
-  const [selectedOutput, setSelectedOutput] = useState<0 | 1 | 2 | 3>(0);
+  const [selectedOutput, setSelectedOutput] = useState<0 | 1 | 2 | 3 | 4>(0);
   const [requestedFiles, setRequestedFiles] = useState<ImportantFile[]>([]);
   const [provider, setProvider] = useState<Provider | null>(null);
 
-  const OPTIONS = [...OUTPUT_FILES, "Preview repo", "Fix issues"] as const;
+  const OPTIONS = [
+    ...OUTPUT_FILES,
+    "Preview repo",
+    "Fix issues",
+    "Ask questions",
+  ] as const;
 
   const handleProviderDone = (p: Provider) => {
     setProvider(p);
     setStage({ type: "requesting-files" });
+
     requestFileList(repoUrl, repoPath, fileTree, p)
       .then((files) => {
         setRequestedFiles(files);
+
+        extractToolingPatch(repoUrl, files.length > 0 ? files : initialFiles, p)
+          .then((patch) => {
+            if (patch) patchLensFile(repoPath, patch);
+          })
+          .catch(() => {});
+
         setStage({ type: "analyzing" });
         return analyzeRepo(repoUrl, files.length > 0 ? files : initialFiles, p);
       })
@@ -133,10 +302,10 @@ export const RepoAnalysis = ({
   useInput((_, key) => {
     if (stage.type !== "done") return;
     if (key.leftArrow)
-      setSelectedOutput((i) => Math.max(0, i - 1) as 0 | 1 | 2 | 3);
+      setSelectedOutput((i) => Math.max(0, i - 1) as 0 | 1 | 2 | 3 | 4);
     if (key.rightArrow)
       setSelectedOutput(
-        (i) => Math.min(OPTIONS.length - 1, i + 1) as 0 | 1 | 2 | 3,
+        (i) => Math.min(OPTIONS.length - 1, i + 1) as 0 | 1 | 2 | 3 | 4,
       );
     if (key.return) {
       if (selectedOutput === 2) {
@@ -147,6 +316,10 @@ export const RepoAnalysis = ({
         setStage({ type: "fixing", result: stage.result });
         return;
       }
+      if (selectedOutput === 4) {
+        setStage({ type: "asking", result: stage.result });
+        return;
+      }
       const fileName = OUTPUT_FILES[selectedOutput] as OutputFile;
       setStage({ type: "writing" });
       try {
@@ -205,9 +378,7 @@ export const RepoAnalysis = ({
   if (stage.type === "written") {
     setTimeout(() => {
       if (onExit) onExit();
-      else {
-        process.exit(0);
-      }
+      else process.exit(0);
     }, 100);
     return (
       <Text color="green">
@@ -228,9 +399,7 @@ export const RepoAnalysis = ({
           onExit={() => {
             setTimeout(() => {
               if (onExit) onExit();
-              else {
-                process.exit(0);
-              }
+              else process.exit(0);
             }, 100);
           }}
         />
@@ -250,6 +419,17 @@ export const RepoAnalysis = ({
     );
   }
 
+  if (stage.type === "asking") {
+    return (
+      <CodebaseQA
+        repoUrl={repoUrl}
+        result={stage.result}
+        provider={provider!}
+        onExit={() => setStage({ type: "done", result: stage.result })}
+      />
+    );
+  }
+
   if (stage.type === "error") {
     return (
       <Text color="red">
@@ -269,6 +449,25 @@ export const RepoAnalysis = ({
         <Text color="white">{result.overview}</Text>
       </Box>
 
+      <Box flexDirection="column">
+        <Text bold color="cyan">
+          {figures.pointerSmall} Architecture
+        </Text>
+        <Text color="white">{result.architecture}</Text>
+      </Box>
+
+      <Box flexDirection="column">
+        <Text bold color="cyan">
+          {figures.pointerSmall} Tooling
+        </Text>
+        {Object.entries(result.tooling ?? {}).map(([k, v]) => (
+          <Text key={k} color="white">
+            {" "}
+            {figures.bullet} <Text bold>{k}</Text>: {v}
+          </Text>
+        ))}
+      </Box>
+
       <Box flexDirection="column">
         <Text bold color="cyan">
           {figures.pointerSmall} Important Folders
@@ -282,35 +481,27 @@ export const RepoAnalysis = ({
       </Box>
 
       <Box flexDirection="column">
-        <Text bold color="yellow">
-          {figures.warning} Missing Configs
+        <Text bold color="cyan">
+          {figures.pointerSmall} Key Files
         </Text>
-        {result.missingConfigs.length > 0 ? (
-          result.missingConfigs.map((f) => (
-            <Text key={f} color="yellow">
-              {" "}
-              {figures.bullet} {f}
-            </Text>
-          ))
-        ) : (
-          <Text color="gray"> None detected</Text>
-        )}
+        {(result.keyFiles ?? []).map((f) => (
+          <Text key={f} color="white">
+            {" "}
+            {figures.bullet} {f}
+          </Text>
+        ))}
       </Box>
 
       <Box flexDirection="column">
-        <Text bold color="red">
-          {figures.cross} Security Issues
+        <Text bold color="cyan">
+          {figures.pointerSmall} Patterns & Idioms
         </Text>
-        {result.securityIssues.length > 0 ? (
-          result.securityIssues.map((s) => (
-            <Text key={s} color="red">
-              {" "}
-              {figures.bullet} {s}
-            </Text>
-          ))
-        ) : (
-          <Text color="gray"> None detected</Text>
-        )}
+        {(result.patterns ?? []).map((p) => (
+          <Text key={p} color="white">
+            {" "}
+            {figures.bullet} {p}
+          </Text>
+        ))}
       </Box>
 
       <Box flexDirection="column">

package/src/components/watch/{WatchRunner.tsx → RunRunner.tsx}

@@ -354,7 +354,7 @@ type ActiveInvestigation = {
   startTime: number;
 };
 
-export function WatchRunner({
+export function RunRunner({
   cmd,
   repoPath,
   clean,
@@ -460,7 +460,6 @@ export function WatchRunner({
           lensContext = `Overview: ${lensFile.overview}
 
 Important folders: ${lensFile.importantFolders.join(", ")}
-${lensFile.securityIssues.length > 0 ? `\nKnown security issues:\n${lensFile.securityIssues.map((s) => `- ${s}`).join("\n")}` : ""}
 ${lensFile.suggestions.length > 0 ? `\nProject suggestions:\n${lensFile.suggestions.map((s) => `- ${s}`).join("\n")}` : ""}`;
         }
       }

package/src/index.tsx

@@ -7,7 +7,7 @@ import { InitCommand } from "./commands/provider";
 import { ReviewCommand } from "./commands/review";
 import { TaskCommand } from "./commands/task";
 import { ChatCommand } from "./commands/chat";
-import { WatchCommand } from "./commands/watch";
+import { RunCommand } from "./commands/run";
 import { TimelineCommand } from "./commands/timeline";
 import { CommitCommand } from "./commands/commit";
 import { registerBuiltins } from "./utils/tools/builtins";
@@ -19,8 +19,7 @@ await loadAddons();
 const program = new Command();
 
 program
-  .command("stalk <url>")
-  .alias("repo")
+  .command("repo <url>")
   .description("Analyze a remote repository")
   .action((url) => {
     render(<RepoCommand url={url} />);
@@ -34,16 +33,14 @@ program
   });
 
 program
-  .command("judge [path]")
-  .alias("review")
+  .command("review [path]")
   .description("Review a local codebase")
   .action((inputPath) => {
     render(<ReviewCommand path={inputPath ?? "."} />);
   });
 
 program
-  .command("cook <text>")
-  .alias("task")
+  .command("task <text>")
   .description("Apply a natural language change to the codebase")
   .option("-p, --path <path>", "Path to the repo", ".")
   .action((text: string, opts: { path: string }) => {
@@ -51,8 +48,7 @@ program
   });
 
 program
-  .command("vibe")
-  .alias("chat")
+  .command("chat")
   .description("Chat with your codebase — ask questions or make changes")
   .option("-p, --path <path>", "Path to the repo", ".")
   .action((opts: { path: string }) => {
@@ -60,8 +56,7 @@ program
   });
 
 program
-  .command("history")
-  .alias("timeline")
+  .command("timeline")
   .description(
     "Explore your code history — see commits, changes, and evolution",
   )
@@ -71,8 +66,7 @@ program
   });
 
 program
-  .command("crimes [files...]")
-  .alias("commit")
+  .command("commit [files...]")
   .description(
     "Generate a smart conventional commit message from staged changes or specific files",
   )
@@ -109,9 +103,10 @@ program
   );
 
 program
-  .command("watch <cmd>")
-  .alias("spy")
-  .description("Watch a dev command and get AI suggestions for errors")
+  .command("run <cmd>")
+  .description(
+    "Run your dev server. Lens detects and fixes errors automatically",
+  )
   .option("-p, --path <path>", "Path to the repo", ".")
   .option("--clean", "Only show AI suggestions, hide raw logs")
   .option("--fix-all", "Auto-apply fixes as errors are detected")
@@ -129,7 +124,7 @@ program
       },
     ) => {
       render(
-        <WatchCommand
+        <RunCommand
           cmd={cmd}
           path={opts.path}
           clean={opts.clean ?? false}

package/src/tools/chart.ts

@@ -1,9 +1,3 @@
-//
-
-//
-
-//
-
 import type { Tool } from "@ridit/lens-sdk";
 
 type ChartType = "bar" | "line" | "sparkline";
@@ -13,9 +7,7 @@ interface ChartInput {
   title?: string;
   labels?: string[];
   values: number[];
-  /** For line charts: height in rows. Default 10. */
   height?: number;
-  /** Bar fill character. Default "█" */
   fill?: string;
 }
 

package/src/types/repo.ts

@@ -27,12 +27,24 @@ export type AIProvider =
 export type AnalysisResult = {
   overview: string;
   importantFolders: string[];
-  missingConfigs: string[];
-  securityIssues: string[];
+
+  tooling: Record<string, string>;
+
+  keyFiles: string[];
+
+  patterns: string[];
+
+  architecture: string;
   suggestions: string[];
 };
 
-export type PackageManager = "npm" | "yarn" | "pnpm" | "pip" | "unknown";
+export type PackageManager =
+  | "npm"
+  | "yarn"
+  | "pnpm"
+  | "bun"
+  | "pip"
+  | "unknown";
 
 export type PreviewInfo = {
   packageManager: PackageManager;