af6099665f
* fix(deps): update module github.com/charmbracelet/bubbletea to v0.25.0 * Remove deprecated and unused mouse events --------- Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com> Co-authored-by: Miccah Castorina <[email protected]>
114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package tabs
|
|
|
|
import (
|
|
"strings"
|
|
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/trufflesecurity/trufflehog/v3/pkg/tui/common"
|
|
)
|
|
|
|
// SelectTabMsg is a message that contains the index of the tab to select.
|
|
type SelectTabMsg int
|
|
|
|
// ActiveTabMsg is a message that contains the index of the current active tab.
|
|
type ActiveTabMsg int
|
|
|
|
// Tabs is bubbletea component that displays a list of tabs.
|
|
type Tabs struct {
|
|
common common.Common
|
|
tabs []string
|
|
activeTab int
|
|
TabSeparator lipgloss.Style
|
|
TabInactive lipgloss.Style
|
|
TabActive lipgloss.Style
|
|
TabDot lipgloss.Style
|
|
UseDot bool
|
|
}
|
|
|
|
// New creates a new Tabs component.
|
|
func New(c common.Common, tabs []string) *Tabs {
|
|
r := &Tabs{
|
|
common: c,
|
|
tabs: tabs,
|
|
activeTab: 0,
|
|
TabSeparator: c.Styles.TabSeparator,
|
|
TabInactive: c.Styles.TabInactive,
|
|
TabActive: c.Styles.TabActive,
|
|
}
|
|
return r
|
|
}
|
|
|
|
// SetSize implements common.Component.
|
|
func (t *Tabs) SetSize(width, height int) {
|
|
t.common.SetSize(width, height)
|
|
}
|
|
|
|
// Init implements tea.Model.
|
|
func (t *Tabs) Init() tea.Cmd {
|
|
t.activeTab = 0
|
|
return nil
|
|
}
|
|
|
|
// Update implements tea.Model.
|
|
func (t *Tabs) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
cmds := make([]tea.Cmd, 0)
|
|
switch msg := msg.(type) {
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "tab":
|
|
t.activeTab = (t.activeTab + 1) % len(t.tabs)
|
|
cmds = append(cmds, t.activeTabCmd)
|
|
case "shift+tab":
|
|
t.activeTab = (t.activeTab - 1 + len(t.tabs)) % len(t.tabs)
|
|
cmds = append(cmds, t.activeTabCmd)
|
|
}
|
|
case SelectTabMsg:
|
|
tab := int(msg)
|
|
if tab >= 0 && tab < len(t.tabs) {
|
|
t.activeTab = int(msg)
|
|
}
|
|
}
|
|
return t, tea.Batch(cmds...)
|
|
}
|
|
|
|
// View implements tea.Model.
|
|
func (t *Tabs) View() string {
|
|
s := strings.Builder{}
|
|
sep := t.TabSeparator
|
|
for i, tab := range t.tabs {
|
|
style := t.TabInactive.Copy()
|
|
prefix := " "
|
|
if i == t.activeTab {
|
|
style = t.TabActive.Copy()
|
|
prefix = t.TabDot.Render("• ")
|
|
}
|
|
if t.UseDot {
|
|
s.WriteString(prefix)
|
|
}
|
|
s.WriteString(
|
|
t.common.Zone.Mark(
|
|
tab,
|
|
style.Render(tab),
|
|
),
|
|
)
|
|
if i != len(t.tabs)-1 {
|
|
s.WriteString(sep.String())
|
|
}
|
|
}
|
|
return lipgloss.NewStyle().
|
|
MaxWidth(t.common.Width).
|
|
Render(s.String())
|
|
}
|
|
|
|
func (t *Tabs) activeTabCmd() tea.Msg {
|
|
return ActiveTabMsg(t.activeTab)
|
|
}
|
|
|
|
// SelectTabCmd is a bubbletea command that selects the tab at the given index.
|
|
func SelectTabCmd(tab int) tea.Cmd {
|
|
return func() tea.Msg {
|
|
return SelectTabMsg(tab)
|
|
}
|
|
}
|