history.go

// TODO: Split this out into its own module, it's kinda neat.
package main

import "sync"

// History contains the history entries
type History struct {
	entries []string
	head    int
	size    int
	lock    sync.Mutex
}

// NewHistory constructs a new history of the given size
func NewHistory(size int) *History {
	return &History{
		entries: make([]string, size),
	}
}

// Add adds the given entry to the entries in the history
func (h *History) Add(entry string) {
	h.lock.Lock()
	defer h.lock.Unlock()

	max := cap(h.entries)
	h.head = (h.head + 1) % max
	h.entries[h.head] = entry
	if h.size < max {
		h.size++
	}
}

// Len returns the number of entries in the history
func (h *History) Len() int {
	return h.size
}

// Get the entry with the given number
func (h *History) Get(num int) []string {
	h.lock.Lock()
	defer h.lock.Unlock()

	max := cap(h.entries)
	if num > h.size {
		num = h.size
	}

	r := make([]string, num)
	for i := 0; i < num; i++ {
		idx := (h.head - i) % max
		if idx < 0 {
			idx += max
		}
		r[num-i-1] = h.entries[idx]
	}

	return r
}