resultsview.go

package main

import (
	"errors"

	"github.com/nsf/termbox-go"
)

var ScrollOff = 5

type ResultsView struct {
	// Array of results to be filtered
	// initialset ResultArray
	results ResultArray

	// Current user input
	lastuserinput string

	// Visible result lines
	top_result    int
	bottom_result int

	// Total number of results
	resultCount int

	// Index of currently selected line
	result_selected int

	// View size
	x, y, h, w int
}

func (r *ResultsView) SelectFirst() {
	r.result_selected = 0
	r.top_result = 0

	if r.resultCount > r.h {
		r.bottom_result = r.h
	} else {
		r.bottom_result = r.resultCount
	}
}

func (r *ResultsView) SelectPrevious() {
	if r.result_selected > 0 {
		r.result_selected--
	}
	if r.top_result > 0 && r.result_selected < r.top_result+ScrollOff {
		r.top_result--
		r.bottom_result--
	}
}

func (r *ResultsView) SelectNext() {
	if r.result_selected < (r.resultCount - 1) {
		r.result_selected++

		if r.result_selected >= r.bottom_result-ScrollOff && r.bottom_result < r.resultCount {
			r.top_result++
			r.bottom_result++
		}
	}
}

func tbprint(x, y int, fg, bg termbox.Attribute, msg string) {
	for _, c := range msg {
		termbox.SetCell(x, y, c, fg, bg)
		x++
	}
}

func (r *ResultsView) Draw() {
	tclear(r.x, r.y, r.w, r.h)
	cy := r.y

	for cnt, res := range r.results[r.top_result:r.bottom_result] {
		is_selected := (cnt + r.top_result) == r.result_selected
		res.Draw(r.x, cy, r.w, is_selected)
		cy++
	}

}

func (r *ResultsView) ToggleMark() {
	if r.resultCount > 0 {
		r.results[r.result_selected].marked = !r.results[r.result_selected].marked
		r.SelectNext()
	}
}

func (r *ResultsView) ToggleMarkAll() error {
	if len(r.results) > 1000 {
		return errors.New("too many files to mark! I WONT DO IT!")
	}
	for _, res := range r.results {
		res.marked = !res.marked
	}
	return nil
}

func (r *ResultsView) SetSize(x, y, w, h int) {
	r.x, r.y, r.w, r.h = x, y, w, h

	r.top_result = 0
	if r.resultCount > r.h {
		r.bottom_result = r.h
	} else {
		r.bottom_result = r.resultCount
	}
}

func (r *ResultsView) Update(results ResultArray) {
	r.results = results
	r.resultCount = len(results)
	r.SetSize(r.x, r.y, r.w, r.h)

}

// If there isnt any marked, return the selection. Otherwise return the array of marked results.
func (rv *ResultsView) GetMarkedOrSelected() ResultArray {
	selected := make(ResultArray, 0, 1)

	for _, res := range rv.results {
		if res.marked {
			selected = append(selected, res)
		}
	}

	if len(selected) > 0 {
		return selected
	}

	if len(rv.results) > 0 && rv.result_selected < len(rv.results) {
		selected = append(selected, rv.results[rv.result_selected])
	}

	return selected
}