usbguard-tui/internal/ui/model.go

package ui

import (
	"fmt"
	"strings"
	"time"

	"charm.land/bubbles/v2/help"
	"charm.land/bubbles/v2/key"
	"charm.land/bubbles/v2/list"
	"charm.land/bubbles/v2/textinput"
	tea "charm.land/bubbletea/v2"
	"charm.land/lipgloss/v2"
	"github.com/anotherhadi/usbguard-tui/internal/guard"
)

type state int

const (
	stateList state = iota
	statePopup
)

type (
	tickMsg         time.Time
	devicesMsg      []guard.Device
	daemonStatusMsg string
	actionMsg       struct{ err error }
	nixRuleMsg      struct{ rule string }
)

type Model struct {
	state        state
	list         list.Model
	actionList   list.Model
	help         help.Model
	daemonStatus string
	width        int
	height       int
	notice       string
	selectedDev  *guard.Device
	rulesManaged bool
	pendingRules []string
}

func (m Model) PendingRules() []string { return m.pendingRules }

func New() Model {
	l := list.New(nil, deviceDelegate{}, 0, 0)
	l.SetShowHelp(false)
	l.SetFilteringEnabled(true)
	l.SetShowStatusBar(true)
	l.SetShowTitle(false)
	l.DisableQuitKeybindings()
	l.KeyMap.CursorUp = key.NewBinding(key.WithKeys("up", "k"), key.WithHelp("↑/k", "up"))
	l.KeyMap.CursorDown = key.NewBinding(key.WithKeys("down", "j"), key.WithHelp("↓/j", "down"))

	l.Styles = list.DefaultStyles(true)
	filterStyles := textinput.DefaultStyles(true)
	filterStyles.Focused.Prompt = filterStyles.Focused.Prompt.Foreground(colorAccent)
	filterStyles.Blurred.Prompt = filterStyles.Blurred.Prompt.Foreground(colorAccent)
	l.Styles.Filter = filterStyles

	h := help.New()
	h.Styles = help.DefaultStyles(true)

	rulesManaged := guard.IsRulesManaged()
	notice := ""
	if rulesManaged {
		notice = "Rules managed by NixOS config: permanent actions will print NixOS rules on exit."
		listKeys.AllowPerm.SetEnabled(false)
		listKeys.BlockPerm.SetEnabled(false)
		listKeys.RejectPerm.SetEnabled(false)
	}

	return Model{
		state:        stateList,
		list:         l,
		actionList:   makeActionList(rulesManaged),
		help:         h,
		rulesManaged: rulesManaged,
		notice:       notice,
	}
}

func makeActionList(rulesManaged bool) list.Model {
	var items []list.Item
	if rulesManaged {
		items = []list.Item{
			actionItem{"allow", guard.AllowDevice, false, guard.Allowed, false},
			actionItem{"allow (perm)", nil, true, guard.Allowed, true},
			actionItem{"block", guard.BlockDevice, false, guard.Blocked, false},
			actionItem{"block (perm)", nil, true, guard.Blocked, true},
			actionItem{"reject", guard.RejectDevice, false, guard.Rejected, false},
			actionItem{"reject (perm)", nil, true, guard.Rejected, true},
		}
	} else {
		items = []list.Item{
			actionItem{"allow", guard.AllowDevice, false, guard.Allowed, false},
			actionItem{"allow (permanent)", guard.AllowDevice, true, guard.Allowed, false},
			actionItem{"block", guard.BlockDevice, false, guard.Blocked, false},
			actionItem{"block (permanent)", guard.BlockDevice, true, guard.Blocked, false},
			actionItem{"reject", guard.RejectDevice, false, guard.Rejected, false},
			actionItem{"reject (permanent)", guard.RejectDevice, true, guard.Rejected, false},
		}
	}
	l := list.New(items, actionDelegate{}, 24, len(items))
	l.SetShowHelp(false)
	l.SetShowTitle(false)
	l.SetShowStatusBar(false)
	l.DisableQuitKeybindings()
	l.SetFilteringEnabled(false)
	return l
}

func (m Model) Init() tea.Cmd {
	return tea.Batch(fetchDevices, fetchDaemonStatus, tickCmd())
}

func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {

	case tea.WindowSizeMsg:
		m.width = msg.Width
		m.height = msg.Height
		m.help.SetWidth(msg.Width)
		m.list.SetSize(msg.Width, m.listHeight())
		m.updateActionListSize()
		return m, nil

	case tickMsg:
		return m, tea.Batch(fetchDevices, fetchDaemonStatus, tickCmd())

	case devicesMsg:
		items := make([]list.Item, len(msg))
		for i, d := range msg {
			items[i] = d
		}
		cmd := m.list.SetItems(items)
		return m, cmd

	case daemonStatusMsg:
		m.daemonStatus = string(msg)
		return m, nil

	case nixRuleMsg:
		m.state = stateList
		m.selectedDev = nil
		m.pendingRules = append(m.pendingRules, msg.rule)
		count := len(m.pendingRules)
		if count == 1 {
			m.notice = "1 NixOS rule queued (printed on exit)"
		} else {
			m.notice = fmt.Sprintf("%d NixOS rules queued (printed on exit)", count)
		}
		return m, nil

	case actionMsg:
		m.state = stateList
		m.selectedDev = nil
		if msg.err != nil {
			switch msg.err {
			case guard.ErrReadOnly:
				m.notice = "Rules file is not writable: permanent changes are not supported."
			case guard.ErrPermission:
				m.notice = "Permission denied. Run with appropriate privileges."
			default:
				m.notice = msg.err.Error()
			}
		} else {
			m.notice = m.defaultNotice()
		}
		return m, fetchDevices

	case tea.KeyPressMsg:
		if m.state == statePopup {
			return m.updatePopup(msg)
		}
		return m.updateList(msg)

	case tea.MouseClickMsg:
		if m.state == statePopup {
			var cmd tea.Cmd
			m.actionList, cmd = m.actionList.Update(msg)
			return m, cmd
		}

	case tea.MouseWheelMsg:
		return m.updateMouseWheel(msg)
	}

	if m.state == stateList {
		var cmd tea.Cmd
		m.list, cmd = m.list.Update(msg)
		return m, cmd
	}

	return m, nil
}

func (m Model) updateList(msg tea.KeyPressMsg) (tea.Model, tea.Cmd) {
	if msg.String() == "ctrl+c" {
		return m, tea.Quit
	}
	if !m.list.SettingFilter() {
		id := m.selectedDevID()
		switch {
		case key.Matches(msg, listKeys.Quit):
			return m, tea.Quit
		case key.Matches(msg, listKeys.Refresh):
			m.notice = m.defaultNotice()
			return m, tea.Batch(fetchDevices, fetchDaemonStatus)
		case key.Matches(msg, listKeys.Help):
			m.help.ShowAll = !m.help.ShowAll
			m.list.SetSize(m.width, m.listHeight())
			return m, nil
		case key.Matches(msg, listKeys.Open):
			if item := m.list.SelectedItem(); item != nil {
				d := item.(guard.Device)
				m.selectedDev = &d
				m.updateActionListSize()
				m.actionList.Select(0)
				m.state = statePopup
				return m, nil
			}
		}
		if id >= 0 {
			if cmd := m.deviceActionCmd(msg, id); cmd != nil {
				return m, cmd
			}
		}
	}
	var cmd tea.Cmd
	m.list, cmd = m.list.Update(msg)
	return m, cmd
}

func (m Model) updatePopup(msg tea.KeyPressMsg) (tea.Model, tea.Cmd) {
	switch {
	case key.Matches(msg, cancelKey):
		m.state = stateList
		m.selectedDev = nil
		return m, nil
	case key.Matches(msg, listKeys.Open):
		if item := m.actionList.SelectedItem(); item != nil {
			a := item.(actionItem)
			if a.nixos && m.selectedDev != nil {
				rule := guard.NixOSRule(*m.selectedDev, a.status)
				return m, func() tea.Msg { return nixRuleMsg{rule: rule} }
			}
			return m, doAction(m.selectedDev.ID, a.fn, a.permanent)
		}
	}
	var cmd tea.Cmd
	m.actionList, cmd = m.actionList.Update(msg)
	return m, cmd
}

func (m Model) View() tea.View {
	return tea.View{
		Content:     m.renderContent(),
		AltScreen:   true,
		WindowTitle: "USBGuard TUI",
		MouseMode:   tea.MouseModeCellMotion,
	}
}

func (m Model) renderContent() string {
	header := m.renderHeader()
	notice := m.renderNotice()
	listView := strings.TrimRight(m.list.View(), "\n")
	helpView := strings.TrimRight(m.help.View(listKeys), "\n")
	bg := strings.Join([]string{header, listView, notice, helpView}, "\n")

	if m.state == statePopup && m.selectedDev != nil {
		return placeOverlay(bg, m.renderActionSelect(), m.width, m.height)
	}
	return bg
}

func (m Model) renderHeader() string {
	title := headerStyle.Render("USBGuard-tui")
	switch m.daemonStatus {
	case "active":
		return title + mutedStyle.Render(" - ") + daemonActiveStyle.Render("active")
	case "":
		return title
	default:
		return title + mutedStyle.Render(" - ") + daemonOtherStyle.Render(m.daemonStatus)
	}
}

func (m Model) renderNotice() string {
	if m.notice == "" {
		return ""
	}
	return warnStyle.Render(m.notice)
}

func (m Model) renderActionSelect() string {
	dev := m.selectedDev
	color := statusColors[dev.Status]
	innerW := m.actionListInnerWidth()

	title := popupTitleStyle.Foreground(color).Width(innerW).Render(dev.Name)
	hint := lipgloss.NewStyle().Foreground(colorMuted).Width(innerW).Render("↑↓ navigate  enter confirm  esc cancel")

	parts := []string{title, m.actionList.View(), ""}
	if m.rulesManaged {
		nixosHint := lipgloss.NewStyle().Foreground(colorMuted).Width(innerW).Render("[NixOS: perm rules printed on exit]")
		parts = append(parts, nixosHint)
	}
	parts = append(parts, hint)
	return popupStyle.Width(innerW).Render(strings.Join(parts, "\n"))
}

func (m Model) popupOuterWidth() int {
	w := m.width - 6
	if w > 60 {
		w = 60
	}
	if w < 32 {
		w = 32
	}
	return w
}

func (m Model) actionListInnerWidth() int {
	return m.popupOuterWidth() - 8 // border(2) + padding_h(6)
}

func (m Model) defaultNotice() string {
	if m.rulesManaged {
		return "Rules managed by NixOS config: permanent actions will print NixOS rules on exit."
	}
	return ""
}

func (m Model) actionItemCount() int {
	return 6
}

// updateActionListSize sizes the action list and toggles pagination based on available space.
// When there is enough room for all items: pagination is hidden and height is set exactly,
// avoiding the phantom line that bubbles/list reserves when showPagination=true.
// When space is limited: pagination is shown naturally by bubbles/list.
func (m *Model) updateActionListSize() {
	items := m.actionItemCount()
	innerW := m.actionListInnerWidth()
	// popup overhead: border(2) + padding_v(2) + title(1) + blank(1) + hint(1) = 7; +1 for NixOS footer
	overhead := 7
	if m.rulesManaged {
		overhead = 8
	}
	available := m.height - overhead - 2 // 2 lines margin
	if available >= items {
		m.actionList.SetShowPagination(false)
		m.actionList.SetSize(innerW, items)
	} else {
		m.actionList.SetShowPagination(true)
		h := available
		if h < 2 {
			h = 2
		}
		m.actionList.SetSize(innerW, h)
	}
}

func (m Model) listHeight() int {
	helpH := lipgloss.Height(strings.TrimRight(m.help.View(listKeys), "\n"))
	return m.height - 1 - helpH - 1 // header - help - notice
}

func (m Model) selectedDevID() int {
	if item := m.list.SelectedItem(); item != nil {
		return item.(guard.Device).ID
	}
	return -1
}

func tickCmd() tea.Cmd {
	return tea.Tick(2*time.Second, func(t time.Time) tea.Msg {
		return tickMsg(t)
	})
}

func fetchDevices() tea.Msg {
	devices, err := guard.ListDevices()
	if err != nil {
		return actionMsg{err: err}
	}
	return devicesMsg(devices)
}

func fetchDaemonStatus() tea.Msg {
	return daemonStatusMsg(guard.DaemonStatus())
}

func doAction(id int, fn func(int, bool) error, permanent bool) tea.Cmd {
	return func() tea.Msg {
		return actionMsg{err: fn(id, permanent)}
	}
}

type actionBinding struct {
	binding       key.Binding
	fn            func(int, bool) error
	perm          bool
	needsWritable bool
}

var deviceActionBindings = []actionBinding{
	{listKeys.Allow, guard.AllowDevice, false, false},
	{listKeys.AllowPerm, guard.AllowDevice, true, true},
	{listKeys.Block, guard.BlockDevice, false, false},
	{listKeys.BlockPerm, guard.BlockDevice, true, true},
	{listKeys.Reject, guard.RejectDevice, false, false},
	{listKeys.RejectPerm, guard.RejectDevice, true, true},
}

func (m Model) deviceActionCmd(msg tea.KeyPressMsg, id int) tea.Cmd {
	for _, b := range deviceActionBindings {
		if (!b.needsWritable || !m.rulesManaged) && key.Matches(msg, b.binding) {
			return doAction(id, b.fn, b.perm)
		}
	}
	return nil
}

func (m Model) updateMouseWheel(msg tea.MouseWheelMsg) (tea.Model, tea.Cmd) {
	switch msg.Button {
	case tea.MouseWheelUp:
		if m.state == statePopup {
			m.actionList.CursorUp()
		} else {
			m.list.CursorUp()
		}
	case tea.MouseWheelDown:
		if m.state == statePopup {
			m.actionList.CursorDown()
		} else {
			m.list.CursorDown()
		}
	}
	return m, nil
}