codebase-rag-tui 0.1.0 → 0.3.0

package/dist/api/chat.d.ts

@@ -1,4 +1,8 @@
-export declare function sendMessage(question: string, socketId: string, sessionId?: string): Promise<{
+export declare function sendMessage(question: string, socketId: string, mode: string, sessionId?: string): Promise<{
     session_id: string;
     response: string;
+    edit: boolean;
+}>;
+export declare function rejectChange(socketId: string, sessionId: string): Promise<{
+    response: string;
 }>;

package/dist/api/chat.js

@@ -1,14 +1,21 @@
-export async function sendMessage(question, socketId, sessionId) {
+export async function sendMessage(question, socketId, mode, sessionId) {
     const res = await fetch(`${process.env['BACKEND_URI']}/remote/repo/query`, {
         method: "POST",
         headers: {
-            "Content-Type": "application/json", // Add this!
+            "Content-Type": "application/json",
         },
         body: JSON.stringify({
             question: question,
             socket_id: socketId,
+            mode: mode,
             session_id: sessionId,
         })
     });
     return res.json();
 }
+export async function rejectChange(socketId, sessionId) {
+    const res = await fetch(`${process.env['BACKEND_URI']}/remote/repo/reject/sessions/${sessionId}?socket_id=${encodeURIComponent(socketId)}`, {
+        method: "DELETE",
+    });
+    return res.json();
+}

package/dist/app.js

@@ -1,12 +1,14 @@
-import React, { useState, useEffect } from 'react';
+import React, { useState, useEffect, useCallback } from 'react';
 import { Box, Text, useApp } from 'ink';
 import Header from './components/Header.js';
 import Message from './components/Message.js';
 import Input from './components/Input.js';
-import { sendMessage } from './api/chat.js';
+import TreeAnimation from './components/TreeAnimation.js';
+import { sendMessage, rejectChange } from './api/chat.js';
 import { useSocket } from './contexts/SocketContext.js';
 import { getWorkspaceInfo } from './utils/workspace.js';
 import { useWorkspace } from './contexts/WorkspaceContext.js';
+import SelectInput from 'ink-select-input';
 export default function App() {
     const { exit } = useApp();
     const { socket, isConnected, isConnecting, connectionError } = useSocket();
@@ -15,9 +17,51 @@ export default function App() {
     const [isLoading, setIsLoading] = useState(false);
     const [error, setError] = useState(null);
     const [sessionId, setSessionId] = useState(null);
+    const [mode, setMode] = useState(null);
+    const [pendingReview, setPendingReview] = useState(false);
+    const modes = [
+        {
+            label: 'Chat',
+            value: 'chat'
+        },
+        {
+            label: 'Agent',
+            value: 'agent'
+        }
+    ];
+    const [showAnimation, setShowAnimation] = useState(true);
+    const handleAnimationComplete = useCallback(() => {
+        setShowAnimation(false);
+    }, []);
+    const reviewChoices = [
+        { label: '✅ Accept changes', value: 'accept' },
+        { label: '❌ Reject changes', value: 'reject' },
+    ];
+    const handleReviewSelect = async (item) => {
+        if (item.value === 'accept') {
+            setMessages((prev) => [...prev, { role: 'model', text: 'Changes accepted.', edit: false }]);
+            setPendingReview(false);
+        }
+        else if (item.value === 'reject') {
+            if (!socket || !isConnected || !socket.id || !sessionId) {
+                setMessages((prev) => [...prev, { role: 'model', text: 'Cannot reject: not connected or no active session.', edit: false }]);
+                setPendingReview(false);
+                return;
+            }
+            try {
+                const res = await rejectChange(socket.id, sessionId);
+                setMessages((prev) => [...prev, { role: 'model', text: res.response, edit: false }]);
+            }
+            catch (err) {
+                const message = err instanceof Error ? err.message : 'Unknown error occurred';
+                setMessages((prev) => [...prev, { role: 'model', text: `Error rejecting changes: ${message}`, edit: false }]);
+            }
+            setPendingReview(false);
+        }
+    };
     const welcomeMessage = 'Start you session by entering the path to your repository';
     useEffect(() => {
-        setMessages([{ role: 'model', text: welcomeMessage }]);
+        setMessages([{ role: 'model', text: welcomeMessage, edit: false }]);
     }, []); // Only run once on mount
     const handleSubmit = async (text) => {
         if (text === '/help') {
@@ -25,8 +69,10 @@ export default function App() {
 • /help - Show this help message
 • /clear - Clear conversation
 • /quit - Leave current session and reset workspace
-• /exit - Exit the application`;
-            setMessages((prev) => [...prev, { role: 'model', text: helpText }]);
+• /exit - Exit the application
+• /agent - Switch to agent mode
+• /chat - Switch to chat mode`;
+            setMessages((prev) => [...prev, { role: 'model', text: helpText, edit: false }]);
         }
         if (text === '/exit') {
             exit();
@@ -40,45 +86,59 @@ export default function App() {
         if (text === '/quit') {
             setSessionId(null);
             setWorkspace(process.cwd());
-            setMessages([{ role: 'model', text: welcomeMessage }]);
+            setMessages([{ role: 'model', text: welcomeMessage, edit: false }]);
             return;
         }
-        setMessages((prev) => [...prev, { role: 'user', text }]);
+        // if (text === '/agent') {
+        // 	setMode('agent');
+        // 	setMessages((prev) => [...prev, {role: 'model', text: 'Switched to agent mode'}]);
+        // 	return;
+        // }
+        // if (text === '/chat') {
+        // 	setMode('chat');
+        // 	setMessages((prev) => [...prev, {role: 'model', text: 'Switched to chat mode'}]);
+        // 	return;
+        // }
+        setMessages((prev) => [...prev, { role: 'user', text, edit: false }]);
         if (workspace == process.cwd()) {
             try {
                 const info = await getWorkspaceInfo(text);
                 if (info.exists && info.isDirectory) {
                     setWorkspace(info.absolutePath);
-                    setMessages((prev) => [...prev, { role: 'model', text: `Workspace is set to: ${info.absolutePath}` }]);
+                    setMessages((prev) => [...prev, { role: 'model', text: `Workspace is set to: ${info.absolutePath}`, edit: false }]);
                 }
                 else if (info.exists) {
-                    setMessages((prev) => [...prev, { role: 'model', text: 'Path exists but is not a directory. Please enter a valid directory path.' }]);
+                    setMessages((prev) => [...prev, { role: 'model', text: 'Path exists but is not a directory. Please enter a valid directory path.', edit: false }]);
                 }
                 else {
-                    setMessages((prev) => [...prev, { role: 'model', text: 'Directory does not exist. Please enter a valid project directory path.' }]);
+                    setMessages((prev) => [...prev, { role: 'model', text: 'Directory does not exist. Please enter a valid project directory path.', edit: false }]);
                 }
             }
             catch (err) {
                 const message = err instanceof Error ? err.message : 'Unknown error occurred';
-                setMessages((prev) => [...prev, { role: 'model', text: `Error validating path: ${message}` }]);
+                setMessages((prev) => [...prev, { role: 'model', text: `Error validating path: ${message}`, edit: false }]);
             }
             return;
         }
         setIsLoading(true);
         setError(null);
         if (!socket || !isConnected || !socket.id) {
-            setMessages((prev) => [...prev, { role: 'model', text: 'Failed to connect to server' }]);
+            setMessages((prev) => [...prev, { role: 'model', text: 'Failed to connect to server', edit: false }]);
             return;
         }
         try {
             if (sessionId) {
-                const res = await sendMessage(text, socket.id, sessionId);
-                setMessages((prev) => [...prev, { role: 'model', text: res.response }]);
+                const res = await sendMessage(text, socket.id, mode, sessionId);
+                setMessages((prev) => [...prev, { role: 'model', text: res.response, edit: res.edit }]);
+                if (res.edit)
+                    setPendingReview(true);
             }
             else {
-                const res = await sendMessage(text, socket.id);
+                const res = await sendMessage(text, socket.id, mode);
                 setSessionId(res.session_id);
-                setMessages((prev) => [...prev, { role: 'model', text: res.response }]);
+                setMessages((prev) => [...prev, { role: 'model', text: res.response, edit: res.edit }]);
+                if (res.edit)
+                    setPendingReview(true);
             }
         }
         catch (err) {
@@ -89,17 +149,22 @@ export default function App() {
             setIsLoading(false);
         }
     };
+    if (showAnimation) {
+        return React.createElement(TreeAnimation, { onComplete: handleAnimationComplete });
+    }
     return (React.createElement(Box, { flexDirection: "column", padding: 1 },
         React.createElement(Header, null),
-        React.createElement(Box, { flexDirection: "column", marginTop: 1 }, messages.map((msg, i) => (
-        // eslint-disable-next-line react/no-array-index-key
-        React.createElement(Message, { key: i, role: msg.role, text: msg.text })))),
+        (!mode) && (React.createElement(SelectInput, { items: modes, onSelect: m => setMode(m.value) })),
+        (mode) && (React.createElement(Box, { flexDirection: "column", marginTop: 1 }, messages.map((msg, i) => (React.createElement(Message, { key: i, role: msg.role, text: msg.text, edit: msg.edit }))))),
         (error || connectionError) && (React.createElement(Box, { marginTop: 1 },
             React.createElement(Text, { color: "red" }, error || connectionError))),
         isConnecting && (React.createElement(Box, { marginTop: 1 },
             React.createElement(Text, { color: "yellow" }, "Connecting to server..."))),
         !isConnected && !isConnecting && !connectionError && (React.createElement(Box, { marginTop: 1 },
             React.createElement(Text, { color: "gray" }, "Disconnected from server"))),
-        React.createElement(Box, { marginTop: 1 },
-            React.createElement(Input, { onSubmit: handleSubmit, isLoading: isLoading }))));
+        mode && pendingReview && (React.createElement(Box, { flexDirection: "column", marginTop: 1 },
+            React.createElement(Text, { color: "yellow", bold: true }, "Review changes \u2014 accept or reject?"),
+            React.createElement(SelectInput, { items: reviewChoices, onSelect: handleReviewSelect }))),
+        mode && !pendingReview && (React.createElement(Box, { marginTop: 1 },
+            React.createElement(Input, { onSubmit: handleSubmit, isLoading: isLoading, mode: mode })))));
 }

package/dist/components/Input.d.ts

@@ -2,6 +2,7 @@ import React from 'react';
 type Props = {
     onSubmit: (value: string) => void;
     isLoading: boolean;
+    mode: string;
 };
-export default function Input({ onSubmit, isLoading }: Props): React.JSX.Element;
+export default function Input({ onSubmit, isLoading, mode }: Props): React.JSX.Element;
 export {};

package/dist/components/Input.js

@@ -1,7 +1,7 @@
 import React, { useState } from 'react';
 import { Box, Text } from 'ink';
 import TextInput from 'ink-text-input';
-export default function Input({ onSubmit, isLoading }) {
+export default function Input({ onSubmit, isLoading, mode }) {
     const [value, setValue] = useState('');
     const handleSubmit = (text) => {
         if (text.trim().length === 0) {
@@ -15,6 +15,10 @@ export default function Input({ onSubmit, isLoading }) {
             React.createElement(Text, { dimColor: true }, "Thinking...")));
     }
     return (React.createElement(Box, null,
-        React.createElement(Text, { bold: true, color: "green" }, '> '),
+        React.createElement(Text, { color: "cyan" },
+            "[",
+            mode,
+            "]"),
+        React.createElement(Text, { bold: true, color: "green" }, ' > '),
         React.createElement(TextInput, { value: value, onChange: setValue, onSubmit: handleSubmit, placeholder: "Type a message...", focus: !isLoading })));
 }

package/dist/components/Message.d.ts

@@ -2,6 +2,7 @@ import React from 'react';
 type Props = {
     role: 'user' | 'model';
     text: string;
+    edit?: boolean;
 };
-export default function Message({ role, text }: Props): React.JSX.Element;
+export default function Message({ role, text, edit }: Props): React.JSX.Element;
 export {};

package/dist/components/Message.js

@@ -1,14 +1,16 @@
 import React from 'react';
 import { Box, Text } from 'ink';
-export default function Message({ role, text }) {
+import Markdown from 'ink-markdown-es';
+export default function Message({ role, text, edit }) {
     if (role === 'user') {
         return (React.createElement(Box, null,
             React.createElement(Text, { color: "green" },
                 '> ',
                 text)));
     }
-    return (React.createElement(Box, null,
-        React.createElement(Text, { color: "blue" },
-            "Codebase Rag Agent: ",
-            text)));
+    return (React.createElement(Box, { flexDirection: "column" },
+        React.createElement(Text, { color: "blue", bold: true }, "Codebase Rag Agent:"),
+        React.createElement(Markdown, null, text),
+        edit && (React.createElement(Box, { marginLeft: 2, marginTop: 0 },
+            React.createElement(Text, { color: "yellow" }, "\u26A1 Code changes were applied")))));
 }

package/dist/components/TreeAnimation.d.ts

@@ -0,0 +1,6 @@
+import React from 'react';
+type Props = {
+    onComplete: () => void;
+};
+export default function TreeAnimation({ onComplete }: Props): React.JSX.Element;
+export {};

package/dist/components/TreeAnimation.js

@@ -0,0 +1,100 @@
+import React, { useState, useEffect } from 'react';
+import { Box, Text } from 'ink';
+const treeFrames = [
+    // Frame 1: Ground only
+    String.raw `
+                     
+                     
+                     
+                     
+                     
+                     
+                     
+                     
+                     
+                     |   =|
+                     |    |
+--------------------/ ,  . \--------._
+`,
+    // Frame 2: Trunk grows
+    String.raw `
+                     
+                     
+                     
+                     
+                     
+                     
+                     
+                     |    |//
+                     |_    /
+                     |-   |
+                     |   =|
+                     |    |
+--------------------/ ,  . \--------._
+`,
+    // Frame 3: Lower branches
+    String.raw `
+                     
+                     
+                     
+                     
+           '7-,--.   ||  / / ,
+           /'       . / / |/_.'
+                     |    |//
+                     |_    /
+                     |-   |
+                     |   =|
+                     |    |
+--------------------/ ,  . \--------._
+`,
+    // Frame 4: More branches
+    String.raw `
+                     
+                     
+               \\     y |  //
+         _ _.___\\,  / -. ||
+           '7-,--.'._||  / / ,
+           /'     '-. './ / |/_.'
+                     |    |//
+                     |_    /
+                     |-   |
+                     |   =|
+                     |    |
+--------------------/ ,  . \--------._
+`,
+    // Frame 5: Full tree
+    String.raw `
+              v .   ._, |_  .,
+           '-._\/  .  \ /    |/_
+               \\  _\, y | \//
+         _\_.___\\, \\/ -.\||
+           '7-,--.'. ||  / / ,
+           /'     '-. './ / |/_.'
+                     |    |//
+                     |_    /
+                     |-   |
+                     |   =|
+                     |    |
+--------------------/ ,  . \--------._
+`,
+];
+export default function TreeAnimation({ onComplete }) {
+    const [frameIndex, setFrameIndex] = useState(0);
+    useEffect(() => {
+        if (frameIndex < treeFrames.length - 1) {
+            const timer = setTimeout(() => {
+                setFrameIndex((prev) => prev + 1);
+            }, 200);
+            return () => clearTimeout(timer);
+        }
+        else {
+            const timer = setTimeout(() => {
+                onComplete();
+            }, 800);
+            return () => clearTimeout(timer);
+        }
+    }, [frameIndex, onComplete]);
+    return (React.createElement(Box, { flexDirection: "column", alignItems: "center", justifyContent: "center", padding: 2 },
+        React.createElement(Text, { color: "green" }, treeFrames[frameIndex]),
+        React.createElement(Text, { color: "cyan", bold: true }, "Codebase RAG Agent")));
+}

package/dist/contexts/SocketContext.js

@@ -55,27 +55,23 @@ export const SocketProvider = ({ children, serverUrl = process.env['BACKEND_URI'
         });
         // File system event handlers
         newSocket.on('dir:list', async (payload, ack) => {
-            console.log('dir:list event heard');
             const result = await listDirectory(payload.dir_path, workspaceRef.current);
             ack(result);
         });
         newSocket.on('bytes:read', async (payload, ack) => {
-            console.log('bytes:read event heard');
             const result = await readFileBytes(payload.file_path, workspaceRef.current);
             ack(result);
         });
         newSocket.on('file:read', async (payload, ack) => {
-            console.log('file:read event heard');
             const result = await readFileText(payload.file_path, workspaceRef.current);
             ack(result);
         });
         newSocket.on('file:write', async (payload, ack) => {
-            console.log('file:write event heard');
             const result = await writeFile(payload.file_path, payload.content, workspaceRef.current);
             ack(result);
         });
         newSocket.on("command:run", async (payload, ack) => {
-            const res = await runCommand(payload.cmd_parts, workspace, payload.timeout);
+            const res = await runCommand(payload.cmd_parts, workspaceRef.current, payload.timeout);
             ack(res);
         });
         setSocket(newSocket);

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "codebase-rag-tui",
-	"version": "0.1.0",
+	"version": "0.3.0",
 	"description": "Terminal-based AI agent for codebase RAG (Retrieval-Augmented Generation)",
 	"license": "MIT",
 	"author": "johnsonafool",
@@ -42,6 +42,8 @@
 		"@google/generative-ai": "^0.24.1",
 		"dotenv": "^17.2.4",
 		"ink": "^4.1.0",
+		"ink-markdown-es": "^1.1.0",
+		"ink-select-input": "^6.2.0",
 		"ink-text-input": "^6.0.0",
 		"meow": "^11.0.0",
 		"react": "^18.2.0",