cocommit/src/cmd/tui/tui_groups.go

package tui

import (
	"slices"
	"strings"

	"github.com/Slug-Boi/cocommit/src/cmd/utils"
	"github.com/charmbracelet/bubbles/paginator"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/charmbracelet/x/term"
)

// sessionState is used to track which model is focused

var (
	modelStyle = lipgloss.NewStyle().
			Width(20).
			Height(8).
			Align(lipgloss.Center, lipgloss.Center).
			BorderStyle(lipgloss.NormalBorder()).
			BorderForeground(lipgloss.Color("241"))
	focusedModelStyle = lipgloss.NewStyle().
				Width(20).
				Height(8).
				Align(lipgloss.Center, lipgloss.Center).
				BorderStyle(lipgloss.DoubleBorder()).
				BorderForeground(lipgloss.Color("170"))
)

type mainModel struct {
	content   []string
	index     int
	paginator paginator.Model
}

var cap, lines int

func newModel() mainModel {
	groups := utils.Groups

	content := []string{}

	for name, users := range groups {
		newUser := strings.Builder{}
		newUser.WriteString(name + ":\n")
		for _, user := range users {
			newUser.WriteString(user.Username + "\n")
		}
		content = append(content, newUser.String())
	}

	slices.Sort(content)

	// check if terminal 0 is a terminal
	var w,h int
	var err error
	if ok := term.IsTerminal(0); ok {
		// calculate term size
		w, h, err = term.GetSize(0)
		if err != nil {
			panic(err)
		}
	}

	// 25 is a magic number in terms of height but is roughly based on the element height 
	// of the group squares
	if h > 25 {
		lines = 2
	} else {
		lines = 1
	}
	
	// if the terminal size is 0, then skip
	if w > 0 {
		// 30 is a magic number don't question it
		cap = min(5, w/30)
	}
	cap = max(1, cap)

	if cap * 2 > len(content) {
		cap = len(content)
	}

	p := paginator.New()
	p.Type = paginator.Dots
	p.ActiveDot = lipgloss.NewStyle().Foreground(lipgloss.AdaptiveColor{Light: "170", Dark: "170"}).Render("•")
	p.InactiveDot = lipgloss.NewStyle().Foreground(lipgloss.AdaptiveColor{Light: "250", Dark: "238"}).Render("•")
	p.PerPage = cap * lines
	p.SetTotalPages(len(content))

	m := mainModel{content: content, paginator: p}
	return m
}

func (m mainModel) Init() tea.Cmd {
	// start the timer and spinner on program start
	return nil
}

func (m mainModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	var cmd tea.Cmd
	var cmds []tea.Cmd
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch msg.String() {
		case "ctrl+c", "q", "esc":
			m.content = nil
			return nil, nil
		case "enter":
			var group string
			if m.currentFocusedModel() != "" {
				group = strings.Split(m.currentFocusedModel(), ":")[0]
			}
			if group != "" {
				for _, sel := range selected {
					delete(selected, string(sel))
				}
				users := utils.Groups[group]
				//TODO: this may be able to be done in a more efficient way currently this would scale poorly
				for k, v := range dupProtect {
					if _, ok := selected[k]; !ok {
						for _, user := range users {
							split := strings.Split(user.Names, "/")
							if split[0] == v || split[1] == v {
								selectToggle(item(k))
							}
						}
					}
				}
			}
			return nil, nil
		case "tab", "right":
			m.next()
		case "left":
			m.previous()
		}
	}

	// Adrian is a fucking genius thanks for the idea :)
	m.paginator.Page = m.index / (cap * lines)
	cmds = append(cmds, cmd)

	return m, tea.Batch(cmds...)
}

func (m mainModel) View() string {
	var s string
	var squares []string
	for i, c := range m.content {
		// uses joinhorizontal to create a grid of squares
		if i == m.index {
			squares = append(squares, focusedModelStyle.Render(c))
		} else {
			squares = append(squares, modelStyle.Render(c))
		}
	}
	// Take the first 5 elements and join them horizontally
	// then take the next 5 and join them horizontally if there are more than 5
	// then join vertically

	start, end := m.paginator.GetSliceBounds(len(m.content))
	end = end - 1
	//println(start, end)
	for start <= end {
		s += lipgloss.JoinHorizontal(lipgloss.Top, squares[start:start+cap]...)
		s += "\n"
		start += cap
	}

	s += "\n" + m.paginator.View()

	s += helpStyle.Render("\ntab/right: focus next • left: focus previous • enter: select group • q/esq: exit\n")
	return s
}

func (m mainModel) currentFocusedModel() string {
	if m.index < len(m.content) {
		return m.content[m.index]
	}
	return ""
}

func (m *mainModel) next() {
	if m.index == len(m.content)-1 {
		m.index = 0
	} else {
		m.index++
	}
}

func (m *mainModel) previous() {
	if m.index == 0 {
		m.index = len(m.content) - 1
	} else {
		m.index--
	}
}