gf-lt/helpfuncs.go

package main

import (
	"fmt"
	"gf-lt/models"
	"gf-lt/pngmeta"
	"image"
	"net/url"
	"os"
	"os/exec"
	"path"
	"path/filepath"
	"slices"
	"strconv"
	"strings"
	"time"
	"unicode"

	"github.com/rivo/tview"
)

// Cached model color - updated by background goroutine
var cachedModelColor string = "orange"

// startModelColorUpdater starts a background goroutine that periodically updates
// the cached model color. Only runs HTTP requests for local llama.cpp APIs.
func startModelColorUpdater() {
	go func() {
		ticker := time.NewTicker(5 * time.Second)
		defer ticker.Stop()
		// Initial check
		updateCachedModelColor()
		for range ticker.C {
			updateCachedModelColor()
		}
	}()
}

// updateCachedModelColor updates the global cachedModelColor variable
func updateCachedModelColor() {
	if !isLocalLlamacpp() {
		cachedModelColor = "orange"
		return
	}
	// Check if model is loaded
	loaded, err := isModelLoaded(chatBody.Model)
	if err != nil {
		// On error, assume not loaded (red)
		cachedModelColor = "red"
		return
	}
	if loaded {
		cachedModelColor = "green"
	} else {
		cachedModelColor = "red"
	}
}

func isASCII(s string) bool {
	for i := 0; i < len(s); i++ {
		if s[i] > unicode.MaxASCII {
			return false
		}
	}
	return true
}

func mapToString[V any](m map[string]V) string {
	rs := strings.Builder{}
	for k, v := range m {
		fmt.Fprintf(&rs, "%v: %v\n", k, v)
	}
	return rs.String()
}

// stripThinkingFromMsg removes thinking blocks from assistant messages.
// Skips user, tool, and system messages as they may contain thinking examples.
func stripThinkingFromMsg(msg *models.RoleMsg) *models.RoleMsg {
	if !cfg.StripThinkingFromAPI {
		return msg
	}
	// Skip user, tool, they might contain thinking and system messages - examples
	if msg.Role == cfg.UserRole || msg.Role == cfg.ToolRole || msg.Role == "system" {
		return msg
	}
	// Strip thinking from assistant messages
	msgText := msg.GetText()
	if thinkRE.MatchString(msgText) {
		cleanedText := thinkRE.ReplaceAllString(msgText, "")
		cleanedText = strings.TrimSpace(cleanedText)
		msg.SetText(cleanedText)
	}
	return msg
}

// refreshChatDisplay updates the chat display based on current character view
// It filters messages for the character the user is currently "writing as"
// and updates the textView with the filtered conversation
func refreshChatDisplay() {
	// Determine which character's view to show
	viewingAs := cfg.UserRole
	if cfg.WriteNextMsgAs != "" {
		viewingAs = cfg.WriteNextMsgAs
	}
	// Filter messages for this character
	filteredMessages := filterMessagesForCharacter(chatBody.Messages, viewingAs)
	displayText := chatToText(filteredMessages, cfg.ShowSys)
	textView.SetText(displayText)
	colorText()
	updateStatusLine()
	if scrollToEndEnabled {
		textView.ScrollToEnd()
	}
}

// stopTTSIfNotForUser: character specific context, not meant fot the human to hear
func stopTTSIfNotForUser(msg *models.RoleMsg) {
	if strings.Contains(cfg.CurrentAPI, "/chat") || !cfg.CharSpecificContextEnabled {
		return
	}
	viewingAs := cfg.UserRole
	if cfg.WriteNextMsgAs != "" {
		viewingAs = cfg.WriteNextMsgAs
	}
	// stop tts if msg is not for user
	if !slices.Contains(msg.KnownTo, viewingAs) && cfg.TTS_ENABLED {
		TTSDoneChan <- true
	}
}

func colorText() {
	text := textView.GetText(false)
	quoteReplacer := strings.NewReplacer(
		`”`, `"`,
		`“`, `"`,
		`“`, `"`,
		`”`, `"`,
		`**`, `*`,
	)
	text = quoteReplacer.Replace(text)
	// Step 1: Extract code blocks and replace them with unique placeholders
	var codeBlocks []string
	placeholder := "__CODE_BLOCK_%d__"
	counter := 0
	// thinking
	var thinkBlocks []string
	placeholderThink := "__THINK_BLOCK_%d__"
	counterThink := 0
	// Replace code blocks with placeholders and store their styled versions
	text = codeBlockRE.ReplaceAllStringFunc(text, func(match string) string {
		// Style the code block and store it
		styled := fmt.Sprintf("[red::i]%s[-:-:-]", match)
		codeBlocks = append(codeBlocks, styled)
		// Generate a unique placeholder (e.g., "__CODE_BLOCK_0__")
		id := fmt.Sprintf(placeholder, counter)
		counter++
		return id
	})
	text = thinkRE.ReplaceAllStringFunc(text, func(match string) string {
		// Style the code block and store it
		styled := fmt.Sprintf("[red::i]%s[-:-:-]", match)
		thinkBlocks = append(thinkBlocks, styled)
		// Generate a unique placeholder (e.g., "__CODE_BLOCK_0__")
		id := fmt.Sprintf(placeholderThink, counterThink)
		counterThink++
		return id
	})
	// Step 2: Apply other regex styles to the non-code parts
	text = quotesRE.ReplaceAllString(text, `[orange::-]$1[-:-:-]`)
	text = starRE.ReplaceAllString(text, `[turquoise::i]$1[-:-:-]`)
	text = singleBacktickRE.ReplaceAllString(text, "`[pink::i]$1[-:-:-]`")
	// text = thinkRE.ReplaceAllString(text, `[yellow::i]$1[-:-:-]`)
	// Step 3: Restore the styled code blocks from placeholders
	for i, cb := range codeBlocks {
		text = strings.Replace(text, fmt.Sprintf(placeholder, i), cb, 1)
	}
	for i, tb := range thinkBlocks {
		text = strings.Replace(text, fmt.Sprintf(placeholderThink, i), tb, 1)
	}
	textView.SetText(text)
}

func updateStatusLine() {
	status := makeStatusLine()
	statusLineWidget.SetText(status)
}

func initSysCards() ([]string, error) {
	labels := []string{}
	labels = append(labels, sysLabels...)
	cards, err := pngmeta.ReadDirCards(cfg.SysDir, cfg.UserRole, logger)
	if err != nil {
		logger.Error("failed to read sys dir", "error", err)
		return nil, err
	}
	for _, cc := range cards {
		if cc.Role == "" {
			logger.Warn("empty role", "file", cc.FilePath)
			continue
		}
		if cc.ID == "" {
			cc.ID = models.ComputeCardID(cc.Role, cc.FilePath)
		}
		sysMap[cc.ID] = cc
		roleToID[cc.Role] = cc.ID
		labels = append(labels, cc.Role)
	}
	return labels, nil
}

func startNewChat(keepSysP bool) {
	id, err := store.ChatGetMaxID()
	if err != nil {
		logger.Error("failed to get chat id", "error", err)
	}
	if ok := charToStart(cfg.AssistantRole, keepSysP); !ok {
		logger.Warn("no such sys msg", "name", cfg.AssistantRole)
	}
	// set chat body
	chatBody.Messages = chatBody.Messages[:2]
	textView.SetText(chatToText(chatBody.Messages, cfg.ShowSys))
	newChat := &models.Chat{
		ID:        id + 1,
		Name:      fmt.Sprintf("%d_%s", id+1, cfg.AssistantRole),
		CreatedAt: time.Now(),
		UpdatedAt: time.Now(),
		// chat is written to db when we get first llm response (or any)
		// actual chat history (messages) would be parsed then
		Msgs:  "",
		Agent: cfg.AssistantRole,
	}
	activeChatName = newChat.Name
	chatMap[newChat.Name] = newChat
	updateStatusLine()
	colorText()
}

func renameUser(oldname, newname string) {
	if oldname == "" {
		// not provided; deduce who user is
		// INFO: if user not yet spoke, it is hard to replace mentions in sysprompt and first message about thme
		roles := chatBody.ListRoles()
		for _, role := range roles {
			if role == cfg.AssistantRole {
				continue
			}
			if role == "tool" {
				continue
			}
			if role == "system" {
				continue
			}
			oldname = role
			break
		}
		if oldname == "" {
			// still
			logger.Warn("fn: renameUser; failed to find old name", "newname", newname)
			return
		}
	}
	viewText := textView.GetText(false)
	viewText = strings.ReplaceAll(viewText, oldname, newname)
	chatBody.Rename(oldname, newname)
	textView.SetText(viewText)
}

func setLogLevel(sl string) {
	switch sl {
	case "Debug":
		logLevel.Set(-4)
	case "Info":
		logLevel.Set(0)
	case "Warn":
		logLevel.Set(4)
	}
}

func listRolesWithUser() []string {
	roles := listChatRoles()
	// Remove user role if it exists in the list (to avoid duplicates and ensure it's at position 0)
	filteredRoles := make([]string, 0, len(roles))
	for _, role := range roles {
		if role != cfg.UserRole {
			filteredRoles = append(filteredRoles, role)
		}
	}
	// Prepend user role to the beginning of the list
	result := append([]string{cfg.UserRole}, filteredRoles...)
	slices.Sort(result)
	return result
}

func loadImage() error {
	filepath := defaultImage
	cc := GetCardByRole(cfg.AssistantRole)
	if cc != nil {
		if strings.HasSuffix(cc.FilePath, ".png") {
			filepath = cc.FilePath
		}
	}
	file, err := os.Open(filepath)
	if err != nil {
		return fmt.Errorf("failed to open image: %w", err)
	}
	defer file.Close()
	img, _, err := image.Decode(file)
	if err != nil {
		return fmt.Errorf("failed to decode image: %w", err)
	}
	imgView.SetImage(img)
	return nil
}

func strInSlice(s string, sl []string) bool {
	for _, el := range sl {
		if strings.EqualFold(s, el) {
			return true
		}
	}
	return false
}

// isLocalLlamacpp checks if the current API is a local llama.cpp instance.
func isLocalLlamacpp() bool {
	u, err := url.Parse(cfg.CurrentAPI)
	if err != nil {
		return false
	}
	host := u.Hostname()
	return host == "localhost" || host == "127.0.0.1" || host == "::1"
}

// getModelColor returns the cached color tag for the model name.
// The cached value is updated by a background goroutine every 5 seconds.
// For non-local models, returns orange. For local llama.cpp models, returns green if loaded, red if not.
func getModelColor() string {
	return cachedModelColor
}

func makeStatusLine() string {
	isRecording := false
	if asr != nil {
		isRecording = asr.IsRecording()
	}
	persona := cfg.UserRole
	if cfg.WriteNextMsgAs != "" {
		persona = cfg.WriteNextMsgAs
	}
	botPersona := cfg.AssistantRole
	if cfg.WriteNextMsgAsCompletionAgent != "" {
		botPersona = cfg.WriteNextMsgAsCompletionAgent
	}
	// Add image attachment info to status line
	var imageInfo string
	if imageAttachmentPath != "" {
		// Get just the filename from the path
		imageName := path.Base(imageAttachmentPath)
		imageInfo = fmt.Sprintf(" | attached img: [orange:-:b]%s[-:-:-]", imageName)
	} else {
		imageInfo = ""
	}
	// Add shell mode status to status line
	var shellModeInfo string
	if shellMode {
		shellModeInfo = " | [green:-:b]SHELL MODE[-:-:-]"
	} else {
		shellModeInfo = ""
	}
	// Get model color based on load status for local llama.cpp models
	modelColor := getModelColor()
	statusLine := fmt.Sprintf(statusLineTempl, activeChatName,
		boolColors[cfg.ToolUse], modelColor, chatBody.Model, boolColors[cfg.SkipLLMResp],
		cfg.CurrentAPI, persona, botPersona)
	if cfg.STT_ENABLED {
		recordingS := fmt.Sprintf(" | [%s:-:b]voice recording[-:-:-] (ctrl+r)",
			boolColors[isRecording])
		statusLine += recordingS
	}
	// completion endpoint
	if !strings.Contains(cfg.CurrentAPI, "chat") {
		roleInject := fmt.Sprintf(" | [%s:-:b]role injection[-:-:-] (alt+7)", boolColors[injectRole])
		statusLine += roleInject
	}
	// context tokens
	contextTokens := getContextTokens()
	maxCtx := getMaxContextTokens()
	if maxCtx == 0 {
		maxCtx = 16384
	}
	if contextTokens > 0 {
		contextInfo := fmt.Sprintf(" | context-estim: [orange:-:b]%d/%d[-:-:-]", contextTokens, maxCtx)
		statusLine += contextInfo
	}
	return statusLine + imageInfo + shellModeInfo
}

func getContextTokens() int {
	if chatBody == nil || chatBody.Messages == nil {
		return 0
	}
	total := 0
	messages := chatBody.Messages
	for i := range messages {
		msg := &messages[i]
		if msg.Stats != nil && msg.Stats.Tokens > 0 {
			total += msg.Stats.Tokens
		} else if msg.GetText() != "" {
			total += len(msg.GetText()) / 4
		}
	}
	return total
}

const deepseekContext = 128000

func getMaxContextTokens() int {
	if chatBody == nil || chatBody.Model == "" {
		return 0
	}
	modelName := chatBody.Model
	switch {
	case strings.Contains(cfg.CurrentAPI, "openrouter"):
		if orModelsData != nil {
			for i := range orModelsData.Data {
				m := &orModelsData.Data[i]
				if m.ID == modelName {
					return m.ContextLength
				}
			}
		}
	case strings.Contains(cfg.CurrentAPI, "deepseek"):
		return deepseekContext
	default:
		if localModelsData != nil {
			for i := range localModelsData.Data {
				m := &localModelsData.Data[i]
				if m.ID == modelName {
					for _, arg := range m.Status.Args {
						if strings.HasPrefix(arg, "--ctx-size") {
							if strings.Contains(arg, "=") {
								val := strings.Split(arg, "=")[1]
								if n, err := strconv.Atoi(val); err == nil {
									return n
								}
							} else {
								idx := -1
								for j, a := range m.Status.Args {
									if a == "--ctx-size" && j+1 < len(m.Status.Args) {
										idx = j + 1
										break
									}
								}
								if idx != -1 {
									if n, err := strconv.Atoi(m.Status.Args[idx]); err == nil {
										return n
									}
								}
							}
						}
					}
				}
			}
		}
	}
	return 0
}

// set of roles within card definition and mention in chat history
func listChatRoles() []string {
	currentChat, ok := chatMap[activeChatName]
	cbc := chatBody.ListRoles()
	if !ok {
		return cbc
	}
	currentCard := GetCardByRole(currentChat.Agent)
	if currentCard == nil {
		logger.Warn("failed to find current card", "agent", currentChat.Agent)
		return cbc
	}
	charset := []string{}
	for _, name := range currentCard.Characters {
		if !strInSlice(name, cbc) {
			charset = append(charset, name)
		}
	}
	charset = append(charset, cbc...)
	return charset
}

func deepseekModelValidator() error {
	if cfg.CurrentAPI == cfg.DeepSeekChatAPI || cfg.CurrentAPI == cfg.DeepSeekCompletionAPI {
		if chatBody.Model != "deepseek-chat" && chatBody.Model != "deepseek-reasoner" {
			showToast("bad request", "wrong deepseek model name")
			return nil
		}
	}
	return nil
}

// == shellmode ==

func toggleShellMode() {
	shellMode = !shellMode
	setShellMode(shellMode)
	if shellMode {
		shellInput.SetLabel(fmt.Sprintf("[%s]$ ", cfg.FilePickerDir))
	} else {
		textArea.SetPlaceholder("input is multiline; press <Enter> to start the next line;\npress <Esc> to send the message.")
	}
	updateStatusLine()
}

func updateFlexLayout() {
	if fullscreenMode {
		// flex already contains only focused widget; do nothing
		return
	}
	flex.Clear()
	flex.AddItem(textView, 0, 40, false)
	if shellMode {
		flex.AddItem(shellInput, 0, 10, false)
	} else {
		flex.AddItem(textArea, 0, 10, false)
	}
	if positionVisible {
		flex.AddItem(statusLineWidget, 0, 2, false)
	}
	// Keep focus on currently focused widget
	focused := app.GetFocus()
	switch {
	case focused == textView:
		app.SetFocus(textView)
	case shellMode:
		app.SetFocus(shellInput)
	default:
		app.SetFocus(textArea)
	}
}

func executeCommandAndDisplay(cmdText string) {
	cmdText = strings.TrimSpace(cmdText)
	if cmdText == "" {
		fmt.Fprintf(textView, "\n[red]Error: No command provided[-:-:-]\n")
		if scrollToEndEnabled {
			textView.ScrollToEnd()
		}
		colorText()
		return
	}
	workingDir := cfg.FilePickerDir
	// Handle cd command specially to update working directory
	if strings.HasPrefix(cmdText, "cd ") {
		newDir := strings.TrimPrefix(cmdText, "cd ")
		newDir = strings.TrimSpace(newDir)
		// Handle cd ~ or cdHOME
		if strings.HasPrefix(newDir, "~") {
			home := os.Getenv("HOME")
			newDir = strings.Replace(newDir, "~", home, 1)
		}
		// Check if directory exists
		if _, err := os.Stat(newDir); err == nil {
			workingDir = newDir
			cfg.FilePickerDir = workingDir
			// Update shell input label with new directory
			shellInput.SetLabel(fmt.Sprintf("[%s]$ ", cfg.FilePickerDir))
			outputContent := workingDir
			// Add the command being executed to the chat
			fmt.Fprintf(textView, "\n[-:-:b](%d) <%s>: [-:-:-]\n$ %s\n",
				len(chatBody.Messages), cfg.ToolRole, cmdText)
			fmt.Fprintf(textView, "%s\n", outputContent)
			combinedMsg := models.RoleMsg{
				Role:    cfg.ToolRole,
				Content: "$ " + cmdText + "\n\n" + outputContent,
			}
			chatBody.Messages = append(chatBody.Messages, combinedMsg)
			if scrollToEndEnabled {
				textView.ScrollToEnd()
			}
			colorText()
			return
		} else {
			outputContent := "cd: " + newDir + ": No such file or directory"
			fmt.Fprintf(textView, "\n[-:-:b](%d) <%s>: [-:-:-]\n$ %s\n",
				len(chatBody.Messages), cfg.ToolRole, cmdText)
			fmt.Fprintf(textView, "[red]%s[-:-:-]\n", outputContent)
			combinedMsg := models.RoleMsg{
				Role:    cfg.ToolRole,
				Content: "$ " + cmdText + "\n\n" + outputContent,
			}
			chatBody.Messages = append(chatBody.Messages, combinedMsg)
			if scrollToEndEnabled {
				textView.ScrollToEnd()
			}
			colorText()
			return
		}
	}

	// Use /bin/sh to support pipes, redirects, etc.
	cmd := exec.Command("/bin/sh", "-c", cmdText)
	cmd.Dir = workingDir
	// Execute the command and get output
	output, err := cmd.CombinedOutput()
	// Add the command being executed to the chat
	fmt.Fprintf(textView, "\n[-:-:b](%d) <%s>: [-:-:-]\n$ %s\n",
		len(chatBody.Messages), cfg.ToolRole, cmdText)
	var outputContent string
	if err != nil {
		// Include both output and error
		errorMsg := "Error: " + err.Error()
		fmt.Fprintf(textView, "[red]%s[-:-:-]\n", errorMsg)
		if len(output) > 0 {
			outputStr := string(output)
			fmt.Fprintf(textView, "[red]%s[-:-:-]\n", outputStr)
			outputContent = errorMsg + "\n" + outputStr
		} else {
			outputContent = errorMsg
		}
	} else {
		// Only output if successful
		if len(output) > 0 {
			outputStr := string(output)
			fmt.Fprintf(textView, "[green]%s[-:-:-]\n", outputStr)
			outputContent = outputStr
		} else {
			successMsg := "Command executed successfully (no output)"
			fmt.Fprintf(textView, "[green]%s[-:-:-]\n", successMsg)
			outputContent = successMsg
		}
	}
	// Combine command and output in a single message for chat history
	combinedContent := "$ " + cmdText + "\n\n" + outputContent
	combinedMsg := models.RoleMsg{
		Role:    cfg.ToolRole,
		Content: combinedContent,
	}
	chatBody.Messages = append(chatBody.Messages, combinedMsg)
	// Scroll to end and update colors
	if scrollToEndEnabled {
		textView.ScrollToEnd()
	}
	colorText()
	// Add command to history (avoid duplicates at the end)
	if len(shellHistory) == 0 || shellHistory[len(shellHistory)-1] != cmdText {
		shellHistory = append(shellHistory, cmdText)
	}
	shellHistoryPos = -1
}

// == search ==

// Global variables for search state
var searchResults []int
var searchResultLengths []int // To store the length of each match in the formatted string
var searchIndex int
var searchText string
var originalTextForSearch string

// performSearch searches for the given term in the textView content and highlights matches
func performSearch(term string) {
	searchText = term
	if searchText == "" {
		searchResults = nil
		searchResultLengths = nil
		originalTextForSearch = ""
		// Re-render text without highlights
		textView.SetText(chatToText(chatBody.Messages, cfg.ShowSys))
		colorText()
		return
	}
	// Get formatted text and search directly in it to avoid mapping issues
	formattedText := textView.GetText(true)
	originalTextForSearch = formattedText
	searchTermLower := strings.ToLower(searchText)
	formattedTextLower := strings.ToLower(formattedText)
	// Find all occurrences of the search term in the formatted text directly
	formattedSearchResults := []int{}
	searchStart := 0
	for {
		pos := strings.Index(formattedTextLower[searchStart:], searchTermLower)
		if pos == -1 {
			break
		}
		absolutePos := searchStart + pos
		formattedSearchResults = append(formattedSearchResults, absolutePos)
		searchStart = absolutePos + len(searchText)
	}
	if len(formattedSearchResults) == 0 {
		// No matches found
		searchResults = nil
		searchResultLengths = nil
		notification := "Pattern not found: " + term
		showToast("search", notification)
		return
	}
	// Store the formatted text positions and lengths for accurate highlighting
	searchResults = formattedSearchResults
	// Create lengths array - all matches have the same length as the search term
	searchResultLengths = make([]int, len(formattedSearchResults))
	for i := range searchResultLengths {
		searchResultLengths[i] = len(searchText)
	}
	searchIndex = 0
	highlightCurrentMatch()
}

// highlightCurrentMatch highlights the current search match and scrolls to it
func highlightCurrentMatch() {
	if len(searchResults) == 0 || searchIndex >= len(searchResults) {
		return
	}
	// Get the stored formatted text
	formattedText := originalTextForSearch
	// For tview to properly support highlighting and scrolling, we need to work with its region system
	// Instead of just applying highlights, we need to add region tags to the text
	highlightedText := addRegionTags(formattedText, searchResults, searchResultLengths, searchIndex, searchText)
	// Update the text view with the text that includes region tags
	textView.SetText(highlightedText)
	// Highlight the current region and scroll to it
	// Need to identify which position in the results array corresponds to the current match
	// The region ID will be search_<position>_<index>
	currentRegion := fmt.Sprintf("search_%d_%d", searchResults[searchIndex], searchIndex)
	textView.Highlight(currentRegion).ScrollToHighlight()
	// Send notification about which match we're at
	notification := fmt.Sprintf("Match %d of %d", searchIndex+1, len(searchResults))
	showToast("search", notification)
}

// showSearchBar shows the search input field as an overlay
func showSearchBar() {
	// Create a temporary flex to combine search and main content
	updatedFlex := tview.NewFlex().SetDirection(tview.FlexRow).
		AddItem(searchField, 3, 0, true). // Search field at top
		AddItem(flex, 0, 1, false)        // Main flex layout below
	// Add the search overlay as a page
	pages.AddPage(searchPageName, updatedFlex, true, true)
	app.SetFocus(searchField)
}

// hideSearchBar hides the search input field
func hideSearchBar() {
	pages.RemovePage(searchPageName)
	// Return focus to the text view
	app.SetFocus(textView)
	// Clear the search field
	searchField.SetText("")
}

// Global variables for index overlay functionality
var indexPageName = "indexOverlay"

// showIndexBar shows the index input field as an overlay at the top
func showIndexBar() {
	// Create a temporary flex to combine index input and main content
	updatedFlex := tview.NewFlex().SetDirection(tview.FlexRow).
		AddItem(indexPickWindow, 3, 0, true). // Index field at top
		AddItem(flex, 0, 1, false)            // Main flex layout below

	// Add the index overlay as a page
	pages.AddPage(indexPageName, updatedFlex, true, true)
	app.SetFocus(indexPickWindow)
}

// hideIndexBar hides the index input field
func hideIndexBar() {
	pages.RemovePage(indexPageName)
	// Return focus to the text view
	app.SetFocus(textView)
	// Clear the index field
	indexPickWindow.SetText("")
}

// addRegionTags adds region tags to search matches in the text for tview highlighting
func addRegionTags(text string, positions []int, lengths []int, currentIdx int, searchTerm string) string {
	if len(positions) == 0 {
		return text
	}
	var result strings.Builder
	lastEnd := 0
	for i, pos := range positions {
		endPos := pos + lengths[i]
		// Add text before this match
		if pos > lastEnd {
			result.WriteString(text[lastEnd:pos])
		}
		// The matched text, which may contain its own formatting tags
		actualText := text[pos:endPos]
		// Add region tag and highlighting for this match
		// Use a unique region id that includes the match index to avoid conflicts
		regionId := fmt.Sprintf("search_%d_%d", pos, i) // position + index to ensure uniqueness
		var highlightStart, highlightEnd string
		if i == currentIdx {
			// Current match - use different highlighting
			highlightStart = fmt.Sprintf(`["%s"][yellow:blue:b]`, regionId) // Current match with region and special highlight
			highlightEnd = `[-:-:-][""]`                                    // Reset formatting and close region
		} else {
			// Other matches - use regular highlighting
			highlightStart = fmt.Sprintf(`["%s"][gold:red:u]`, regionId) // Other matches with region and highlight
			highlightEnd = `[-:-:-][""]`                                 // Reset formatting and close region
		}
		result.WriteString(highlightStart)
		result.WriteString(actualText)
		result.WriteString(highlightEnd)
		lastEnd = endPos
	}
	// Add the rest of the text after the last processed match
	if lastEnd < len(text) {
		result.WriteString(text[lastEnd:])
	}
	return result.String()
}

// searchNext finds the next occurrence of the search term
func searchNext() {
	if len(searchResults) == 0 {
		showToast("search", "No search results to navigate")
		return
	}
	searchIndex = (searchIndex + 1) % len(searchResults)
	highlightCurrentMatch()
}

// searchPrev finds the previous occurrence of the search term
func searchPrev() {
	if len(searchResults) == 0 {
		showToast("search", "No search results to navigate")
		return
	}
	if searchIndex == 0 {
		searchIndex = len(searchResults) - 1
	} else {
		searchIndex--
	}
	highlightCurrentMatch()
}

// == tab completion ==

func scanFiles(dir, filter string) []string {
	const maxDepth = 3
	const maxFiles = 50
	var files []string
	var scanRecursive func(currentDir string, currentDepth int, relPath string)
	scanRecursive = func(currentDir string, currentDepth int, relPath string) {
		if len(files) >= maxFiles {
			return
		}
		if currentDepth > maxDepth {
			return
		}
		entries, err := os.ReadDir(currentDir)
		if err != nil {
			return
		}
		for _, entry := range entries {
			if len(files) >= maxFiles {
				return
			}
			name := entry.Name()
			if strings.HasPrefix(name, ".") {
				continue
			}
			fullPath := name
			if relPath != "" {
				fullPath = relPath + "/" + name
			}
			if entry.IsDir() {
				// Recursively scan subdirectories
				scanRecursive(filepath.Join(currentDir, name), currentDepth+1, fullPath)
				continue
			}
			// Check if file matches filter
			if filter == "" || strings.HasPrefix(strings.ToLower(fullPath), strings.ToLower(filter)) {
				files = append(files, fullPath)
			}
		}
	}
	scanRecursive(dir, 0, "")
	return files
}

// models logic that is too complex for models package
func MsgToText(i int, m *models.RoleMsg) string {
	var contentStr string
	var imageIndicators []string
	if !m.HasContentParts {
		contentStr = m.Content
	} else {
		var textParts []string
		for _, part := range m.ContentParts {
			switch p := part.(type) {
			case models.TextContentPart:
				if p.Type == "text" {
					textParts = append(textParts, p.Text)
				}
			case models.ImageContentPart:
				displayPath := p.Path
				if displayPath == "" {
					displayPath = "image"
				} else {
					displayPath = extractDisplayPath(displayPath, cfg.FilePickerDir)
				}
				imageIndicators = append(imageIndicators, fmt.Sprintf("[orange::i][image: %s][-:-:-]", displayPath))
			case map[string]any:
				if partType, exists := p["type"]; exists {
					switch partType {
					case "text":
						if textVal, textExists := p["text"]; textExists {
							if textStr, isStr := textVal.(string); isStr {
								textParts = append(textParts, textStr)
							}
						}
					case "image_url":
						var displayPath string
						if pathVal, pathExists := p["path"]; pathExists {
							if pathStr, isStr := pathVal.(string); isStr && pathStr != "" {
								displayPath = extractDisplayPath(pathStr, cfg.FilePickerDir)
							}
						}
						if displayPath == "" {
							displayPath = "image"
						}
						imageIndicators = append(imageIndicators, fmt.Sprintf("[orange::i][image: %s][-:-:-]", displayPath))
					}
				}
			}
		}
		contentStr = strings.Join(textParts, " ") + " "
	}
	contentStr, _ = strings.CutPrefix(contentStr, m.Role+":")
	icon := fmt.Sprintf("(%d) <%s>: ", i, m.Role)
	var finalContent strings.Builder
	if len(imageIndicators) > 0 {
		for _, indicator := range imageIndicators {
			finalContent.WriteString(indicator)
			finalContent.WriteString("\n")
		}
	}
	finalContent.WriteString(contentStr)
	if m.Stats != nil {
		fmt.Fprintf(&finalContent, "\n[gray::i][%d tok, %.1fs, %.1f t/s][-:-:-]", m.Stats.Tokens, m.Stats.Duration, m.Stats.TokensPerSec)
	}
	textMsg := fmt.Sprintf("[-:-:b]%s[-:-:-]\n%s\n", icon, finalContent.String())
	return strings.ReplaceAll(textMsg, "\n\n", "\n")
}

// extractDisplayPath returns a path suitable for display, potentially relative to imageBaseDir
func extractDisplayPath(p, bp string) string {
	if p == "" {
		return ""
	}
	// If base directory is set, try to make path relative to it
	if bp != "" {
		if rel, err := filepath.Rel(bp, p); err == nil {
			// Check if relative path doesn't start with ".." (meaning it's within base dir)
			// If it starts with "..", we might still want to show it as relative
			// but for now we show full path if it goes outside base dir
			if !strings.HasPrefix(rel, "..") {
				p = rel
			}
		}
	}
	// Truncate long paths to last 60 characters if needed
	if len(p) > 60 {
		return "..." + p[len(p)-60:]
	}
	return p
}