updating aichat accepting code as input

Author: gcclinux

internal/ui/aichat.go

@@ -12,6 +12,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/atotto/clipboard"
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/charmbracelet/lipgloss"
 	"github.com/user/terminal-intelligence/internal/ai"
@@ -45,46 +46,47 @@ import (
 //   - DisplayNotification: Shows fix results with distinct styling
 //   - GetLastAssistantResponse: Retrieves last AI response for insertion
 type AIChatPane struct {
-	messages         []types.ChatMessage        // Conversation history
-	inputBuffer      string                     // Current input being typed
-	aiClient         ai.AIClient                // AI service client
-	model            string                     // AI model to use
-	provider         string                     // "ollama" or "gemini"
-	scrollOffset     int                        // Vertical scroll offset for responses
-	width            int                        // Pane width
-	height           int                        // Pane height
-	focused          bool                       // Whether this pane is focused
-	streaming        bool                       // Whether AI is currently generating
-	copyMode         bool                       // Whether in code block selection mode
-	viewMode         bool                       // Whether viewing a code block
-	codeBlocks       []string                   // Extracted code blocks from responses
-	selectedBlock    int                        // Currently selected code block index
-	lastSelectedCode string                     // Last selected code block content
-	lastKeyPressed   string                     // Last key pressed (for debugging)
-	activeArea       int                        // 0: Input, 1: Response
-	viewModeScroll   int                        // Scroll offset for code block view mode
-	viewModeScrollX  int                        // Horizontal scroll offset for code block view mode
-	configMode       bool                       // Whether in config editor mode
-	configFields     []string                   // Config field names
-	configValues     []string                   // Config field values
-	selectedField    int                        // Currently selected config field
-	editingField     bool                       // Whether currently editing a field
-	editBuffer       string                     // Buffer for editing field value
-	editCursorPos    int                        // Cursor position within edit buffer
-	terminalMode     bool                       // Whether terminal execution is active
-	cmdRunning       bool                       // Whether the command is currently running
-	terminalOutput   []string                   // Output lines from terminal execution
-	stdinWriter      io.WriteCloser             // Stdin pipe for sending input to running command
-	terminalInput    string                     // Current input line being typed in terminal mode
-	aiAvailable      bool                       // Whether the AI service is reachable
-	aiChecked        bool                       // Whether the availability check has completed
-	suggestedFile    string                     // Filename suggested by AI for the current code block
-	workingDir       string                     // Directory from which to execute commands
-	codeBlockInfos   []dirtracker.CodeBlockInfo // Code blocks with language tags
-	blockDirMappings []string                   // Effective dir per code block index
-	workspaceRoot    string                     // From AppConfig.WorkspaceDir
-	runningCmd       *exec.Cmd                  // Currently running command process (for kill support)
-	processKilled    bool                       // Whether the process was killed by user (Ctrl+K)
+	messages          []types.ChatMessage        // Conversation history
+	inputBuffer       string                     // Current input being typed
+	aiClient          ai.AIClient                // AI service client
+	model             string                     // AI model to use
+	provider          string                     // "ollama" or "gemini"
+	scrollOffset      int                        // Vertical scroll offset for responses
+	width             int                        // Pane width
+	height            int                        // Pane height
+	focused           bool                       // Whether this pane is focused
+	streaming         bool                       // Whether AI is currently generating
+	copyMode          bool                       // Whether in code block selection mode
+	viewMode          bool                       // Whether viewing a code block
+	codeBlocks        []string                   // Extracted code blocks from responses
+	selectedBlock     int                        // Currently selected code block index
+	lastSelectedCode  string                     // Last selected code block content
+	lastKeyPressed    string                     // Last key pressed (for debugging)
+	activeArea        int                        // 0: Input, 1: Response
+	viewModeScroll    int                        // Scroll offset for code block view mode
+	viewModeScrollX   int                        // Horizontal scroll offset for code block view mode
+	configMode        bool                       // Whether in config editor mode
+	configFields      []string                   // Config field names
+	configValues      []string                   // Config field values
+	selectedField     int                        // Currently selected config field
+	editingField      bool                       // Whether currently editing a field
+	editBuffer        string                     // Buffer for editing field value
+	editCursorPos     int                        // Cursor position within edit buffer
+	terminalMode      bool                       // Whether terminal execution is active
+	cmdRunning        bool                       // Whether the command is currently running
+	terminalOutput    []string                   // Output lines from terminal execution
+	stdinWriter       io.WriteCloser             // Stdin pipe for sending input to running command
+	terminalInput     string                     // Current input line being typed in terminal mode
+	aiAvailable       bool                       // Whether the AI service is reachable
+	aiChecked         bool                       // Whether the availability check has completed
+	suggestedFile     string                     // Filename suggested by AI for the current code block
+	workingDir        string                     // Directory from which to execute commands
+	codeBlockInfos    []dirtracker.CodeBlockInfo // Code blocks with language tags
+	blockDirMappings  []string                   // Effective dir per code block index
+	workspaceRoot     string                     // From AppConfig.WorkspaceDir
+	runningCmd        *exec.Cmd                  // Currently running command process (for kill support)
+	processKilled     bool                       // Whether the process was killed by user (Ctrl+K)
+	lastKeystrokeTime time.Time                  // Last keypress timestamp to detect rapid/terminal paste
 }
 
 // AIResponseMsg is sent when AI response chunk is received.
@@ -201,20 +203,21 @@ func NewAIChatPane(client ai.AIClient, model string, provider string, workspaceR
 		workspaceRoot = cwd
 	}
 	return &AIChatPane{
-		messages:       []types.ChatMessage{},
-		inputBuffer:    "",
-		aiClient:       client,
-		model:          model,
-		provider:       provider,
-		scrollOffset:   0,
-		width:          0,
-		height:         0,
-		focused:        false,
-		streaming:      false,
-		activeArea:     0, // 0: Input, 1: Response
-		terminalOutput: []string{},
-		workingDir:     cwd,
-		workspaceRoot:  workspaceRoot,
+		messages:          []types.ChatMessage{},
+		inputBuffer:       "",
+		aiClient:          client,
+		model:             model,
+		provider:          provider,
+		scrollOffset:      0,
+		width:             0,
+		height:            0,
+		focused:           false,
+		streaming:         false,
+		activeArea:        0, // 0: Input, 1: Response
+		terminalOutput:    []string{},
+		workingDir:        cwd,
+		workspaceRoot:     workspaceRoot,
+		lastKeystrokeTime: time.Now(),
 	}
 }
 
@@ -1159,6 +1162,11 @@ func (a *AIChatPane) handleKeyPress(msg tea.KeyMsg) tea.Cmd {
 		return nil
 	}
 
+	// Input area active - handle typing
+	now := time.Now()
+	isRapid := now.Sub(a.lastKeystrokeTime) < 25*time.Millisecond
+	a.lastKeystrokeTime = now
+
 	// Handle input based on active area
 	if a.activeArea == 1 {
 		// Response area active - handle scrolling
@@ -1192,7 +1200,21 @@ func (a *AIChatPane) handleKeyPress(msg tea.KeyMsg) tea.Cmd {
 	} else {
 		// Input area active - handle typing
 		switch msg.String() {
+		case "ctrl+v":
+			// Paste directly from clipboard into input buffer
+			content, err := clipboard.ReadAll()
+			if err == nil {
+				// Normalize line endings
+				content = strings.ReplaceAll(content, "\r\n", "\n")
+				a.inputBuffer += content
+			}
+			return nil
 		case "enter":
+			// Treat rapid enters (e.g., from native right-click paste) as newlines
+			if msg.Paste || isRapid {
+				a.inputBuffer += "\n"
+				return nil
+			}
 			// Send message when Enter is pressed
 			if a.inputBuffer != "" {
 				message := a.inputBuffer
@@ -1205,9 +1227,18 @@ func (a *AIChatPane) handleKeyPress(msg tea.KeyMsg) tea.Cmd {
 		case "backspace":
 			// Delete last character from input buffer
 			if len(a.inputBuffer) > 0 {
-				a.inputBuffer = a.inputBuffer[:len(a.inputBuffer)-1]
+				runes := []rune(a.inputBuffer)
+				if len(runes) > 0 {
+					a.inputBuffer = string(runes[:len(runes)-1])
+				}
 			}
 		default:
+			if msg.Paste {
+				if len(msg.Runes) > 0 {
+					a.inputBuffer += string(msg.Runes)
+				}
+				return nil
+			}
 			// Add character to input buffer
 			// Check for printable characters to avoid control chars
 			if len(msg.String()) == 1 {

internal/ui/app.go

@@ -184,7 +184,10 @@ func New(config *types.AppConfig, buildNumber string) *App {
 // This is part of the Bubble Tea Model interface.
 // Currently returns nil as no initial commands are needed.
 func (a *App) Init() tea.Cmd {
-	return a.aiPane.CheckAIAvailability()
+	return tea.Batch(
+		a.aiPane.CheckAIAvailability(),
+		tea.EnableBracketedPaste,
+	)
 }
 
 // Update handles messages and updates application state.

main.go

@@ -11,73 +11,66 @@ import (
 )
 
 var version = "0.0.2.3"
-
-// Build number (injected at compile time via -ldflags)
 var buildNumber = "dev"
 
 func main() {
-	// Load configuration
+	if len(os.Args) > 1 && (os.Args[1] == "-v" || os.Args[1] == "--version") {
+		fmt.Printf("Terminal Intelligence v%s (build %s)\n", version, buildNumber)
+		return
+	}
+
+	// 1. Initialize Default Config
 	appCfg := types.DefaultConfig()
 
-	// Check for config file and load if present
+	// 2. Load Configuration Logic
 	configPath, err := config.ConfigFilePath()
 	if err != nil {
-		fmt.Fprintf(os.Stderr, "Error determining config file path: %v\n", err)
+		// Fatal error: cannot even determine where config should be
+		fmt.Fprintf(os.Stderr, "Error determining config path: %v\n", err)
 		os.Exit(1)
 	}
 
 	if _, err := os.Stat(configPath); err == nil {
-		// Config file exists — load, validate, and apply
+		// File exists: Load and validate
 		jcfg, err := config.LoadFromFile(configPath)
 		if err != nil {
-			fmt.Fprintf(os.Stderr, "Error loading config file: %v\n", err)
+			fmt.Fprintf(os.Stderr, "Error loading config: %v\n", err)
 			os.Exit(1)
 		}
 		if err := config.Validate(jcfg); err != nil {
-			fmt.Fprintf(os.Stderr, "Invalid config file: %v\n", err)
+			fmt.Fprintf(os.Stderr, "Invalid config: %v\n", err)
 			os.Exit(1)
 		}
 		config.ApplyToAppConfig(jcfg, appCfg)
 	} else if os.IsNotExist(err) {
-		// Config file doesn't exist — create default config
-		createdPath, err := config.CreateDefaultConfig()
-		if err != nil {
-			fmt.Fprintf(os.Stderr, "Warning: Could not create default config file: %v\n", err)
-			fmt.Fprintf(os.Stderr, "Continuing with defaults...\n")
-		} else {
-			fmt.Fprintf(os.Stderr, "Created default config file at: %s\n", createdPath)
-			fmt.Fprintf(os.Stderr, "Edit this file to customize your settings.\n")
-			fmt.Fprintf(os.Stderr, "Continuing with defaults for this session...\n\n")
-		}
+		// File missing: Create it once and move on
+		_, _ = config.CreateDefaultConfig()
+		// Note: We don't exit here; we just continue with appCfg defaults
 	}
 
-	// Default workspace to current working directory if not set
+	// 3. Runtime Defaults (Do NOT save these back to the JSON file)
+	// This ensures the app respects the current folder it's opened in
+	// unless the user has explicitly hardcoded a path in their config.
 	if appCfg.WorkspaceDir == "" {
-		cwd, err := os.Getwd()
-		if err == nil {
+		if cwd, err := os.Getwd(); err == nil {
 			appCfg.WorkspaceDir = cwd
+		} else {
+			appCfg.WorkspaceDir = "." // Fallback
 		}
 	}
 
-	// Save updated workspace back to config.json
-	jcfgToSave := config.AppConfigToJSONConfig(appCfg)
-	if data, err := config.ToJSON(jcfgToSave); err == nil {
-		_ = os.WriteFile(configPath, data, 0644)
-	}
-
-	// Create workspace directory if it doesn't exist
+	// 4. Ensure Workspace exists
 	if err := os.MkdirAll(appCfg.WorkspaceDir, 0755); err != nil {
-		fmt.Fprintf(os.Stderr, "Error creating workspace directory: %v\n", err)
+		fmt.Fprintf(os.Stderr, "Error creating workspace: %v\n", err)
 		os.Exit(1)
 	}
 
-	// Create the application
+	// 5. Run Application
 	app := ui.New(appCfg, buildNumber)
-
-	// Start the Bubble Tea program
 	p := tea.NewProgram(app, tea.WithAltScreen())
+
 	if _, err := p.Run(); err != nil {
-		fmt.Fprintf(os.Stderr, "Error running application: %v\n", err)
+		fmt.Fprintf(os.Stderr, "Application error: %v\n", err)
 		os.Exit(1)
 	}
 }