| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830 |
- package chat
- import (
- "encoding/base64"
- "fmt"
- "log/slog"
- "os"
- "path/filepath"
- "strconv"
- "strings"
- "unicode/utf8"
- "github.com/charmbracelet/bubbles/v2/spinner"
- tea "github.com/charmbracelet/bubbletea/v2"
- "github.com/charmbracelet/lipgloss/v2"
- "github.com/google/uuid"
- "github.com/sst/opencode-sdk-go"
- "github.com/sst/opencode/internal/app"
- "github.com/sst/opencode/internal/attachment"
- "github.com/sst/opencode/internal/clipboard"
- "github.com/sst/opencode/internal/commands"
- "github.com/sst/opencode/internal/components/dialog"
- "github.com/sst/opencode/internal/components/textarea"
- "github.com/sst/opencode/internal/components/toast"
- "github.com/sst/opencode/internal/styles"
- "github.com/sst/opencode/internal/theme"
- "github.com/sst/opencode/internal/util"
- )
- type EditorComponent interface {
- tea.Model
- tea.ViewModel
- Content() string
- Lines() int
- Value() string
- Length() int
- Focused() bool
- Focus() (tea.Model, tea.Cmd)
- Blur()
- Submit() (tea.Model, tea.Cmd)
- Clear() (tea.Model, tea.Cmd)
- Paste() (tea.Model, tea.Cmd)
- Newline() (tea.Model, tea.Cmd)
- SetValue(value string)
- SetValueWithAttachments(value string)
- SetInterruptKeyInDebounce(inDebounce bool)
- SetExitKeyInDebounce(inDebounce bool)
- RestoreFromHistory(index int)
- }
- type editorComponent struct {
- app *app.App
- width int
- textarea textarea.Model
- spinner spinner.Model
- interruptKeyInDebounce bool
- exitKeyInDebounce bool
- historyIndex int // -1 means current (not in history)
- currentText string // Store current text when navigating history
- pasteCounter int
- reverted bool
- }
- func (m *editorComponent) Init() tea.Cmd {
- return tea.Batch(m.textarea.Focus(), m.spinner.Tick, tea.EnableReportFocus)
- }
- func (m *editorComponent) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
- var cmds []tea.Cmd
- var cmd tea.Cmd
- switch msg := msg.(type) {
- case tea.WindowSizeMsg:
- m.width = msg.Width - 4
- return m, nil
- case spinner.TickMsg:
- m.spinner, cmd = m.spinner.Update(msg)
- return m, cmd
- case tea.KeyPressMsg:
- // Handle up/down arrows and ctrl+p/ctrl+n for history navigation
- switch msg.String() {
- case "up", "ctrl+p":
- // Only navigate history if cursor is at the first line and column (for arrow keys)
- // or allow ctrl+p from anywhere
- if (msg.String() == "ctrl+p" || (m.textarea.Line() == 0 && m.textarea.CursorColumn() == 0)) && len(m.app.State.MessageHistory) > 0 {
- if m.historyIndex == -1 {
- // Save current text before entering history
- m.currentText = m.textarea.Value()
- m.textarea.MoveToBegin()
- }
- // Move up in history (older messages)
- if m.historyIndex < len(m.app.State.MessageHistory)-1 {
- m.historyIndex++
- m.RestoreFromHistory(m.historyIndex)
- m.textarea.MoveToBegin()
- }
- return m, nil
- }
- case "down", "ctrl+n":
- // Only navigate history if cursor is at the last line and we're in history navigation (for arrow keys)
- // or allow ctrl+n from anywhere if we're in history navigation
- if (msg.String() == "ctrl+n" || m.textarea.IsCursorAtEnd()) && m.historyIndex > -1 {
- // Move down in history (newer messages)
- m.historyIndex--
- if m.historyIndex == -1 {
- // Restore current text
- m.textarea.Reset()
- m.textarea.SetValue(m.currentText)
- m.currentText = ""
- } else {
- m.RestoreFromHistory(m.historyIndex)
- m.textarea.MoveToEnd()
- }
- return m, nil
- } else if m.historyIndex > -1 && msg.String() == "down" {
- m.textarea.MoveToEnd()
- return m, nil
- }
- }
- // Reset history navigation on any other input
- if m.historyIndex != -1 {
- m.historyIndex = -1
- m.currentText = ""
- }
- // Maximize editor responsiveness for printable characters
- if msg.Text != "" {
- m.reverted = false
- m.textarea, cmd = m.textarea.Update(msg)
- cmds = append(cmds, cmd)
- return m, tea.Batch(cmds...)
- }
- case app.MessageRevertedMsg:
- if msg.Session.ID == m.app.Session.ID {
- switch msg.Message.Info.(type) {
- case opencode.UserMessage:
- prompt, err := msg.Message.ToPrompt()
- if err != nil {
- return m, toast.NewErrorToast("Failed to revert message")
- }
- m.RestoreFromPrompt(*prompt)
- m.textarea.MoveToEnd()
- m.reverted = true
- return m, nil
- }
- }
- case app.SessionUnrevertedMsg:
- if msg.Session.ID == m.app.Session.ID {
- if m.reverted {
- updated, cmd := m.Clear()
- m = updated.(*editorComponent)
- return m, cmd
- }
- return m, nil
- }
- case tea.PasteMsg:
- text := string(msg)
- if filePath := strings.TrimSpace(strings.TrimPrefix(text, "@")); strings.HasPrefix(text, "@") && filePath != "" {
- statPath := filePath
- if !filepath.IsAbs(filePath) {
- statPath = filepath.Join(m.app.Info.Path.Cwd, filePath)
- }
- if _, err := os.Stat(statPath); err == nil {
- attachment := m.createAttachmentFromPath(filePath)
- if attachment != nil {
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- return m, nil
- }
- }
- }
- text = strings.ReplaceAll(text, "\\", "")
- text, err := strconv.Unquote(`"` + text + `"`)
- if err != nil {
- slog.Error("Failed to unquote text", "error", err)
- text := string(msg)
- if m.shouldSummarizePastedText(text) {
- m.handleLongPaste(text)
- } else {
- m.textarea.InsertRunesFromUserInput([]rune(msg))
- }
- return m, nil
- }
- if _, err := os.Stat(text); err != nil {
- slog.Error("Failed to paste file", "error", err)
- text := string(msg)
- if m.shouldSummarizePastedText(text) {
- m.handleLongPaste(text)
- } else {
- m.textarea.InsertRunesFromUserInput([]rune(msg))
- }
- return m, nil
- }
- filePath := text
- attachment := m.createAttachmentFromFile(filePath)
- if attachment == nil {
- if m.shouldSummarizePastedText(text) {
- m.handleLongPaste(text)
- } else {
- m.textarea.InsertRunesFromUserInput([]rune(msg))
- }
- return m, nil
- }
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- case tea.ClipboardMsg:
- text := string(msg)
- // Check if the pasted text is long and should be summarized
- if m.shouldSummarizePastedText(text) {
- m.handleLongPaste(text)
- } else {
- m.textarea.InsertRunesFromUserInput([]rune(text))
- }
- case dialog.ThemeSelectedMsg:
- m.textarea = updateTextareaStyles(m.textarea)
- m.spinner = createSpinner()
- return m, tea.Batch(m.textarea.Focus(), m.spinner.Tick)
- case dialog.CompletionSelectedMsg:
- switch msg.Item.ProviderID {
- case "commands":
- commandName := strings.TrimPrefix(msg.Item.Value, "/")
- updated, cmd := m.Clear()
- m = updated.(*editorComponent)
- cmds = append(cmds, cmd)
- cmds = append(cmds, util.CmdHandler(commands.ExecuteCommandMsg(m.app.Commands[commands.CommandName(commandName)])))
- return m, tea.Batch(cmds...)
- case "files":
- atIndex := m.textarea.LastRuneIndex('@')
- if atIndex == -1 {
- // Should not happen, but as a fallback, just insert.
- m.textarea.InsertString(msg.Item.Value + " ")
- return m, nil
- }
- // The range to replace is from the '@' up to the current cursor position.
- // Replace the search term (e.g., "@search") with an empty string first.
- cursorCol := m.textarea.CursorColumn()
- m.textarea.ReplaceRange(atIndex, cursorCol, "")
- // Now, insert the attachment at the position where the '@' was.
- // The cursor is now at `atIndex` after the replacement.
- filePath := msg.Item.Value
- attachment := m.createAttachmentFromPath(filePath)
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- return m, nil
- case "symbols":
- atIndex := m.textarea.LastRuneIndex('@')
- if atIndex == -1 {
- // Should not happen, but as a fallback, just insert.
- m.textarea.InsertString(msg.Item.Value + " ")
- return m, nil
- }
- cursorCol := m.textarea.CursorColumn()
- m.textarea.ReplaceRange(atIndex, cursorCol, "")
- symbol := msg.Item.RawData.(opencode.Symbol)
- parts := strings.Split(symbol.Name, ".")
- lastPart := parts[len(parts)-1]
- attachment := &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "symbol",
- Display: "@" + lastPart,
- URL: msg.Item.Value,
- Filename: lastPart,
- MediaType: "text/plain",
- Source: &attachment.SymbolSource{
- Path: symbol.Location.Uri,
- Name: symbol.Name,
- Kind: int(symbol.Kind),
- Range: attachment.SymbolRange{
- Start: attachment.Position{
- Line: int(symbol.Location.Range.Start.Line),
- Char: int(symbol.Location.Range.Start.Character),
- },
- End: attachment.Position{
- Line: int(symbol.Location.Range.End.Line),
- Char: int(symbol.Location.Range.End.Character),
- },
- },
- },
- }
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- return m, nil
- case "agents":
- atIndex := m.textarea.LastRuneIndex('@')
- if atIndex == -1 {
- // Should not happen, but as a fallback, just insert.
- m.textarea.InsertString(msg.Item.Value + " ")
- return m, nil
- }
- cursorCol := m.textarea.CursorColumn()
- m.textarea.ReplaceRange(atIndex, cursorCol, "")
- name := msg.Item.Value
- attachment := &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "agent",
- Display: "@" + name,
- Source: &attachment.AgentSource{
- Name: name,
- },
- }
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- return m, nil
- default:
- slog.Debug("Unknown provider", "provider", msg.Item.ProviderID)
- return m, nil
- }
- }
- m.spinner, cmd = m.spinner.Update(msg)
- cmds = append(cmds, cmd)
- m.textarea, cmd = m.textarea.Update(msg)
- cmds = append(cmds, cmd)
- return m, tea.Batch(cmds...)
- }
- func (m *editorComponent) Content() string {
- width := m.width
- if m.app.Session.ID == "" {
- width = min(width, 80)
- }
- t := theme.CurrentTheme()
- base := styles.NewStyle().Foreground(t.Text()).Background(t.Background()).Render
- muted := styles.NewStyle().Foreground(t.TextMuted()).Background(t.Background()).Render
- promptStyle := styles.NewStyle().Foreground(t.Primary()).
- Padding(0, 0, 0, 1).
- Bold(true)
- prompt := promptStyle.Render(">")
- m.textarea.SetWidth(width - 6)
- textarea := lipgloss.JoinHorizontal(
- lipgloss.Top,
- prompt,
- m.textarea.View(),
- )
- borderForeground := t.Border()
- if m.app.IsLeaderSequence {
- borderForeground = t.Accent()
- }
- textarea = styles.NewStyle().
- Background(t.BackgroundElement()).
- Width(width).
- PaddingTop(1).
- PaddingBottom(1).
- BorderStyle(lipgloss.ThickBorder()).
- BorderForeground(borderForeground).
- BorderBackground(t.Background()).
- BorderLeft(true).
- BorderRight(true).
- Render(textarea)
- hint := base(m.getSubmitKeyText()) + muted(" send ")
- if m.exitKeyInDebounce {
- keyText := m.getExitKeyText()
- hint = base(keyText+" again") + muted(" to exit")
- } else if m.app.IsBusy() {
- keyText := m.getInterruptKeyText()
- status := "working"
- if m.app.CurrentPermission.ID != "" {
- status = "waiting for permission"
- }
- if m.interruptKeyInDebounce && m.app.CurrentPermission.ID == "" {
- hint = muted(
- status,
- ) + m.spinner.View() + muted(
- " ",
- ) + base(
- keyText+" again",
- ) + muted(
- " interrupt",
- )
- } else {
- hint = muted(status) + m.spinner.View()
- if m.app.CurrentPermission.ID == "" {
- hint += muted(" ") + base(keyText) + muted(" interrupt")
- }
- }
- }
- model := ""
- if m.app.Model != nil {
- model = muted(m.app.Provider.Name) + base(" "+m.app.Model.Name)
- }
- space := width - 2 - lipgloss.Width(model) - lipgloss.Width(hint)
- spacer := styles.NewStyle().Background(t.Background()).Width(space).Render("")
- info := hint + spacer + model
- info = styles.NewStyle().Background(t.Background()).Padding(0, 1).Render(info)
- content := strings.Join([]string{"", textarea, info}, "\n")
- return content
- }
- func (m *editorComponent) View() string {
- width := m.width
- if m.app.Session.ID == "" {
- width = min(width, 80)
- }
- if m.Lines() > 1 {
- return lipgloss.Place(
- width,
- 5,
- lipgloss.Center,
- lipgloss.Center,
- "",
- styles.WhitespaceStyle(theme.CurrentTheme().Background()),
- )
- }
- return m.Content()
- }
- func (m *editorComponent) Focused() bool {
- return m.textarea.Focused()
- }
- func (m *editorComponent) Focus() (tea.Model, tea.Cmd) {
- return m, m.textarea.Focus()
- }
- func (m *editorComponent) Blur() {
- m.textarea.Blur()
- }
- func (m *editorComponent) Lines() int {
- return m.textarea.LineCount()
- }
- func (m *editorComponent) Value() string {
- return m.textarea.Value()
- }
- func (m *editorComponent) Length() int {
- return m.textarea.Length()
- }
- func (m *editorComponent) Submit() (tea.Model, tea.Cmd) {
- value := strings.TrimSpace(m.Value())
- if value == "" {
- return m, nil
- }
- switch value {
- case "exit", "quit", "q", ":q":
- return m, tea.Quit
- }
- if len(value) > 0 && value[len(value)-1] == '\\' {
- // If the last character is a backslash, remove it and add a newline
- backslashCol := m.textarea.CurrentRowLength() - 1
- m.textarea.ReplaceRange(backslashCol, backslashCol+1, "")
- m.textarea.InsertString("\n")
- return m, nil
- }
- var cmds []tea.Cmd
- attachments := m.textarea.GetAttachments()
- prompt := app.Prompt{Text: value, Attachments: attachments}
- m.app.State.AddPromptToHistory(prompt)
- cmds = append(cmds, m.app.SaveState())
- updated, cmd := m.Clear()
- m = updated.(*editorComponent)
- cmds = append(cmds, cmd)
- cmds = append(cmds, util.CmdHandler(app.SendPrompt(prompt)))
- return m, tea.Batch(cmds...)
- }
- func (m *editorComponent) Clear() (tea.Model, tea.Cmd) {
- m.textarea.Reset()
- m.historyIndex = -1
- m.currentText = ""
- m.pasteCounter = 0
- return m, nil
- }
- func (m *editorComponent) Paste() (tea.Model, tea.Cmd) {
- imageBytes := clipboard.Read(clipboard.FmtImage)
- if imageBytes != nil {
- attachmentCount := len(m.textarea.GetAttachments())
- attachmentIndex := attachmentCount + 1
- base64EncodedFile := base64.StdEncoding.EncodeToString(imageBytes)
- attachment := &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "file",
- MediaType: "image/png",
- Display: fmt.Sprintf("[Image #%d]", attachmentIndex),
- Filename: fmt.Sprintf("image-%d.png", attachmentIndex),
- URL: fmt.Sprintf("data:image/png;base64,%s", base64EncodedFile),
- Source: &attachment.FileSource{
- Path: fmt.Sprintf("image-%d.png", attachmentIndex),
- Mime: "image/png",
- Data: imageBytes,
- },
- }
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- return m, nil
- }
- textBytes := clipboard.Read(clipboard.FmtText)
- if textBytes != nil {
- text := string(textBytes)
- // Check if the pasted text is long and should be summarized
- if m.shouldSummarizePastedText(text) {
- m.handleLongPaste(text)
- } else {
- m.textarea.InsertRunesFromUserInput([]rune(text))
- }
- return m, nil
- }
- // fallback to reading the clipboard using OSC52
- return m, tea.ReadClipboard
- }
- func (m *editorComponent) Newline() (tea.Model, tea.Cmd) {
- m.textarea.Newline()
- return m, nil
- }
- func (m *editorComponent) SetInterruptKeyInDebounce(inDebounce bool) {
- m.interruptKeyInDebounce = inDebounce
- }
- func (m *editorComponent) SetValue(value string) {
- m.textarea.SetValue(value)
- }
- func (m *editorComponent) SetValueWithAttachments(value string) {
- m.textarea.Reset()
- i := 0
- for i < len(value) {
- r, size := utf8.DecodeRuneInString(value[i:])
- // Check if filepath and add attachment
- if r == '@' {
- start := i + size
- end := start
- for end < len(value) {
- nextR, nextSize := utf8.DecodeRuneInString(value[end:])
- if nextR == ' ' || nextR == '\t' || nextR == '\n' || nextR == '\r' {
- break
- }
- end += nextSize
- }
- if end > start {
- filePath := value[start:end]
- slog.Debug("test", "filePath", filePath)
- if _, err := os.Stat(filepath.Join(m.app.Info.Path.Cwd, filePath)); err == nil {
- slog.Debug("test", "found", true)
- attachment := m.createAttachmentFromFile(filePath)
- if attachment != nil {
- m.textarea.InsertAttachment(attachment)
- i = end
- continue
- }
- }
- }
- }
- // Not a valid file path, insert the character normally
- m.textarea.InsertRune(r)
- i += size
- }
- }
- func (m *editorComponent) SetExitKeyInDebounce(inDebounce bool) {
- m.exitKeyInDebounce = inDebounce
- }
- func (m *editorComponent) getInterruptKeyText() string {
- return m.app.Commands[commands.SessionInterruptCommand].Keys()[0]
- }
- func (m *editorComponent) getSubmitKeyText() string {
- return m.app.Commands[commands.InputSubmitCommand].Keys()[0]
- }
- func (m *editorComponent) getExitKeyText() string {
- return m.app.Commands[commands.AppExitCommand].Keys()[0]
- }
- // shouldSummarizePastedText determines if pasted text should be summarized
- func (m *editorComponent) shouldSummarizePastedText(text string) bool {
- lines := strings.Split(text, "\n")
- lineCount := len(lines)
- charCount := len(text)
- // Consider text long if it has more than 3 lines or more than 150 characters
- return lineCount > 3 || charCount > 150
- }
- // handleLongPaste handles long pasted text by creating a summary attachment
- func (m *editorComponent) handleLongPaste(text string) {
- lines := strings.Split(text, "\n")
- lineCount := len(lines)
- // Increment paste counter
- m.pasteCounter++
- // Create attachment with full text as base64 encoded data
- fileBytes := []byte(text)
- base64EncodedText := base64.StdEncoding.EncodeToString(fileBytes)
- url := fmt.Sprintf("data:text/plain;base64,%s", base64EncodedText)
- fileName := fmt.Sprintf("pasted-text-%d.txt", m.pasteCounter)
- displayText := fmt.Sprintf("[pasted #%d %d+ lines]", m.pasteCounter, lineCount)
- attachment := &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "text",
- MediaType: "text/plain",
- Display: displayText,
- URL: url,
- Filename: fileName,
- Source: &attachment.TextSource{
- Value: text,
- },
- }
- m.textarea.InsertAttachment(attachment)
- m.textarea.InsertString(" ")
- }
- func updateTextareaStyles(ta textarea.Model) textarea.Model {
- t := theme.CurrentTheme()
- bgColor := t.BackgroundElement()
- textColor := t.Text()
- textMutedColor := t.TextMuted()
- ta.Styles.Blurred.Base = styles.NewStyle().Foreground(textColor).Background(bgColor).Lipgloss()
- ta.Styles.Blurred.CursorLine = styles.NewStyle().Background(bgColor).Lipgloss()
- ta.Styles.Blurred.Placeholder = styles.NewStyle().
- Foreground(textMutedColor).
- Background(bgColor).
- Lipgloss()
- ta.Styles.Blurred.Text = styles.NewStyle().Foreground(textColor).Background(bgColor).Lipgloss()
- ta.Styles.Focused.Base = styles.NewStyle().Foreground(textColor).Background(bgColor).Lipgloss()
- ta.Styles.Focused.CursorLine = styles.NewStyle().Background(bgColor).Lipgloss()
- ta.Styles.Focused.Placeholder = styles.NewStyle().
- Foreground(textMutedColor).
- Background(bgColor).
- Lipgloss()
- ta.Styles.Focused.Text = styles.NewStyle().Foreground(textColor).Background(bgColor).Lipgloss()
- ta.Styles.Attachment = styles.NewStyle().
- Foreground(t.Secondary()).
- Background(bgColor).
- Lipgloss()
- ta.Styles.SelectedAttachment = styles.NewStyle().
- Foreground(t.Text()).
- Background(t.Secondary()).
- Lipgloss()
- ta.Styles.Cursor.Color = t.Primary()
- return ta
- }
- func createSpinner() spinner.Model {
- t := theme.CurrentTheme()
- return spinner.New(
- spinner.WithSpinner(spinner.Ellipsis),
- spinner.WithStyle(
- styles.NewStyle().
- Background(t.Background()).
- Foreground(t.TextMuted()).
- Width(3).
- Lipgloss(),
- ),
- )
- }
- func NewEditorComponent(app *app.App) EditorComponent {
- s := createSpinner()
- ta := textarea.New()
- ta.Prompt = " "
- ta.ShowLineNumbers = false
- ta.CharLimit = -1
- ta = updateTextareaStyles(ta)
- m := &editorComponent{
- app: app,
- textarea: ta,
- spinner: s,
- interruptKeyInDebounce: false,
- historyIndex: -1,
- pasteCounter: 0,
- }
- return m
- }
- func (m *editorComponent) RestoreFromPrompt(prompt app.Prompt) {
- m.textarea.Reset()
- m.textarea.SetValue(prompt.Text)
- // Sort attachments by start index in reverse order (process from end to beginning)
- // This prevents index shifting issues
- attachmentsCopy := make([]*attachment.Attachment, len(prompt.Attachments))
- copy(attachmentsCopy, prompt.Attachments)
- for i := 0; i < len(attachmentsCopy)-1; i++ {
- for j := i + 1; j < len(attachmentsCopy); j++ {
- if attachmentsCopy[i].StartIndex < attachmentsCopy[j].StartIndex {
- attachmentsCopy[i], attachmentsCopy[j] = attachmentsCopy[j], attachmentsCopy[i]
- }
- }
- }
- for _, att := range attachmentsCopy {
- m.textarea.SetCursorColumn(att.StartIndex)
- m.textarea.ReplaceRange(att.StartIndex, att.EndIndex, "")
- m.textarea.InsertAttachment(att)
- }
- }
- // RestoreFromHistory restores a message from history at the given index
- func (m *editorComponent) RestoreFromHistory(index int) {
- if index < 0 || index >= len(m.app.State.MessageHistory) {
- return
- }
- entry := m.app.State.MessageHistory[index]
- m.RestoreFromPrompt(entry)
- }
- func getMediaTypeFromExtension(ext string) string {
- switch strings.ToLower(ext) {
- case ".jpg":
- return "image/jpeg"
- case ".png", ".jpeg", ".gif", ".webp":
- return "image/" + ext[1:]
- case ".pdf":
- return "application/pdf"
- default:
- return "text/plain"
- }
- }
- func (m *editorComponent) createAttachmentFromFile(filePath string) *attachment.Attachment {
- ext := strings.ToLower(filepath.Ext(filePath))
- mediaType := getMediaTypeFromExtension(ext)
- absolutePath := filePath
- if !filepath.IsAbs(filePath) {
- absolutePath = filepath.Join(m.app.Info.Path.Cwd, filePath)
- }
- // For text files, create a simple file reference
- if mediaType == "text/plain" {
- return &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "file",
- Display: "@" + filePath,
- URL: fmt.Sprintf("file://%s", absolutePath),
- Filename: filePath,
- MediaType: mediaType,
- Source: &attachment.FileSource{
- Path: absolutePath,
- Mime: mediaType,
- },
- }
- }
- // For binary files (images, PDFs), read and encode
- fileBytes, err := os.ReadFile(filePath)
- if err != nil {
- slog.Error("Failed to read file", "error", err)
- return nil
- }
- base64EncodedFile := base64.StdEncoding.EncodeToString(fileBytes)
- url := fmt.Sprintf("data:%s;base64,%s", mediaType, base64EncodedFile)
- attachmentCount := len(m.textarea.GetAttachments())
- attachmentIndex := attachmentCount + 1
- label := "File"
- if strings.HasPrefix(mediaType, "image/") {
- label = "Image"
- }
- return &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "file",
- MediaType: mediaType,
- Display: fmt.Sprintf("[%s #%d]", label, attachmentIndex),
- URL: url,
- Filename: filePath,
- Source: &attachment.FileSource{
- Path: absolutePath,
- Mime: mediaType,
- Data: fileBytes,
- },
- }
- }
- func (m *editorComponent) createAttachmentFromPath(filePath string) *attachment.Attachment {
- extension := filepath.Ext(filePath)
- mediaType := getMediaTypeFromExtension(extension)
- absolutePath := filePath
- if !filepath.IsAbs(filePath) {
- absolutePath = filepath.Join(m.app.Info.Path.Cwd, filePath)
- }
- return &attachment.Attachment{
- ID: uuid.NewString(),
- Type: "file",
- Display: "@" + filePath,
- URL: fmt.Sprintf("file://%s", absolutePath),
- Filename: filePath,
- MediaType: mediaType,
- Source: &attachment.FileSource{
- Path: absolutePath,
- Mime: mediaType,
- },
- }
- }
|