Merge pull request #1109 from rumpl/fix-history

Put user messages to the history

Author: dgageot

pkg/history/history.go

@@ -14,10 +14,31 @@ type History struct {
 	current int
 }
 
-func New() (*History, error) {
-	homeDir, err := os.UserHomeDir()
-	if err != nil {
-		return nil, err
+type options struct {
+	homeDir string
+}
+
+type Opt func(*options)
+
+func WithBaseDir(dir string) Opt {
+	return func(o *options) {
+		o.homeDir = dir
+	}
+}
+
+func New(opts ...Opt) (*History, error) {
+	o := &options{}
+	for _, opt := range opts {
+		opt(o)
+	}
+
+	homeDir := o.homeDir
+	if homeDir == "" {
+		var err error
+		homeDir, err = os.UserHomeDir()
+		if err != nil {
+			return nil, err
+		}
 	}
 
 	h := &History{

pkg/tui/page/chat/chat.go

@@ -646,6 +646,8 @@ func (p *chatPage) processMessage(msg editor.SendMsg) tea.Cmd {
 		p.msgCancel()
 	}
 
+	_ = p.history.Add(msg.Content)
+
 	var ctx context.Context
 	ctx, p.msgCancel = context.WithCancel(context.Background())
 

pkg/tui/page/chat/chat_test.go

@@ -0,0 +1,33 @@
+package chat
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+	"github.com/docker/cagent/pkg/history"
+	"github.com/docker/cagent/pkg/tui/components/editor"
+)
+
+func TestProcessMessage_AddsToHistory(t *testing.T) {
+	t.Parallel()
+
+	homeDir := t.TempDir()
+	hist, err := history.New(history.WithBaseDir(homeDir))
+	require.NoError(t, err)
+
+	p := &chatPage{
+		history: hist,
+	}
+
+	p.processMessage(editor.SendMsg{Content: "/new"})
+	p.processMessage(editor.SendMsg{Content: "/compact"})
+	p.processMessage(editor.SendMsg{Content: "/copy"})
+
+	assert.Equal(t, []string{"/new", "/compact", "/copy"}, hist.Messages)
+
+	hist2, err := history.New(history.WithBaseDir(homeDir))
+	require.NoError(t, err)
+	assert.Equal(t, []string{"/new", "/compact", "/copy"}, hist2.Messages)
+}