clean patch

terminal.go

@@ -879,6 +879,40 @@ func (t *Terminal) SetSize(width, height int) error {
 	return err
 }
 
+func (t *Terminal) GetHistory() []string {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	history := make([]string, t.history.size)
+	for i := 0; i < t.history.size; i++ {
+		history[i], _ = t.history.NthPreviousEntry(t.history.size - i - 1)
+	}
+	return history
+}
+
+func (t *Terminal) SetHistory(history []string) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	for _, entry := range history {
+		t.history.Add(entry)
+	}
+}
+
+func (t *Terminal) ClearHistory() {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	t.history.Clear()
+}
+
+func (t *Terminal) PopHistory() (string, bool) {
+	t.lock.Lock()
+	defer t.lock.Unlock()
+
+	return t.history.Pop()
+}
+
 type pasteIndicatorError struct{}
 
 func (pasteIndicatorError) Error() string {
@@ -944,6 +978,30 @@ func (s *stRingBuffer) NthPreviousEntry(n int) (value string, ok bool) {
 	return s.entries[index], true
 }
 
+// Clear removes all elements from the ring buffer.
+func (s *stRingBuffer) Clear() {
+	for i := range s.entries {
+		s.entries[i] = ""
+	}
+	s.size = 0
+}
+
+// Pop the last element from the history.
+func (s *stRingBuffer) Pop() (string, bool) {
+	if s.size == 0 {
+		return "", false
+	}
+	value := s.entries[s.head]
+	s.head--
+	if s.head < 0 {
+		s.head += s.max
+	}
+	if s.size > 0 {
+		s.size--
+	}
+	return value, true
+}
+
 // readPasswordLine reads from reader until it finds \n or io.EOF.
 // The slice returned does not include the \n.
 // readPasswordLine also ignores any \r it finds.

terminal_test.go

@@ -435,3 +435,50 @@ func TestOutputNewlines(t *testing.T) {
 		t.Errorf("incorrect output: was %q, expected %q", output, expected)
 	}
 }
+
+func TestGetSetHistory(t *testing.T) {
+	term := NewTerminal(nil, ">")
+	term.SetHistory([]string{"a", "b", "c"})
+	history := term.GetHistory()
+	if len(history) != 3 || history[0] != "a" || history[1] != "b" || history[2] != "c" {
+		t.Errorf("history not set correctly: %v", history)
+	}
+}
+
+func TestClearHistory(t *testing.T) {
+	term := NewTerminal(nil, ">")
+	term.SetHistory([]string{"a", "b", "c"})
+	term.ClearHistory()
+	history := term.GetHistory()
+	if len(history) != 0 {
+		t.Errorf("history not cleared: %v", history)
+	}
+}
+
+func TestPopHistory(t *testing.T) {
+	term := NewTerminal(nil, ">")
+	term.SetHistory([]string{"a", "b", "c"})
+	history := term.GetHistory()
+	if len(history) != 3 || history[0] != "a" || history[1] != "b" || history[2] != "c" {
+		t.Errorf("history not set correctly: %v", history)
+	}
+	h, ok := term.PopHistory()
+	if !ok {
+		t.Errorf("history not popped correctly: %v", h)
+	}
+	if h != "c" {
+		t.Errorf("history not popped correctly: %v", h)
+	}
+	history = term.GetHistory()
+	if len(history) != 2 || history[0] != "a" || history[1] != "b" {
+		t.Errorf("history not popped correctly: %v", history)
+	}
+	term.ClearHistory()
+	h, ok = term.PopHistory()
+	if ok {
+		t.Errorf("history not popped correctly: %v", h)
+	}
+	if h != "" {
+		t.Errorf("history not popped correctly: %v", h)
+	}
+}