Evan Hosinski
2b6c4eb4cd
Introduce Bubble Tea-based terminal UI to manage suspicious artifact findings, including file selection, type filtering, list view, and details.
110 lines
2.6 KiB
Plaintext
110 lines
2.6 KiB
Plaintext
package tui
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/charmbracelet/bubbles/list"
|
|
"github.com/charmbracelet/bubbles/spinner"
|
|
tea "github.com/charmbracelet/bubbletea"
|
|
"github.com/charmbracelet/lipgloss"
|
|
)
|
|
|
|
// FileSelectedMsg is emitted when a file is chosen
|
|
type FileSelectedMsg struct{ Path string }
|
|
|
|
// FilePicker is a simple list of .json files in the current directory
|
|
// Press Enter to pick, q/esc to quit
|
|
|
|
type fileItem struct {
|
|
title string
|
|
path string
|
|
}
|
|
|
|
func (i fileItem) Title() string { return i.title }
|
|
func (i fileItem) Description() string { return i.path }
|
|
func (i fileItem) FilterValue() string { return i.title }
|
|
|
|
type FilePickerModel struct {
|
|
list list.Model
|
|
spinner spinner.Model
|
|
error error
|
|
loading bool
|
|
}
|
|
|
|
func NewFilePicker() FilePickerModel {
|
|
delegate := list.NewDefaultDelegate()
|
|
delegate.ShowDescription = true
|
|
l := list.New([]list.Item{}, delegate, 0, 0)
|
|
l.Title = "Select JSON report"
|
|
l.Styles.Title = lipgloss.NewStyle().Bold(true)
|
|
l.SetShowHelp(false)
|
|
sp := spinner.New()
|
|
sp.Style = lipgloss.NewStyle().Foreground(lipgloss.Color("205"))
|
|
m := FilePickerModel{list: l, spinner: sp}
|
|
return m
|
|
}
|
|
|
|
func (m FilePickerModel) Init() tea.Cmd {
|
|
return tea.Batch(m.spinner.Tick, m.loadFilesCmd())
|
|
}
|
|
|
|
func (m FilePickerModel) loadFilesCmd() tea.Cmd {
|
|
return func() tea.Msg {
|
|
files, err := os.ReadDir(".")
|
|
if err != nil {
|
|
return errMsg{err}
|
|
}
|
|
var items []list.Item
|
|
for _, f := range files {
|
|
if f.IsDir() { continue }
|
|
name := f.Name()
|
|
if filepath.Ext(name) == ".json" {
|
|
items = append(items, fileItem{title: name, path: name})
|
|
}
|
|
}
|
|
return filesLoadedMsg{items}
|
|
}
|
|
}
|
|
|
|
type errMsg struct{ error }
|
|
type filesLoadedMsg struct{ items []list.Item }
|
|
|
|
func (m FilePickerModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.WindowSizeMsg:
|
|
m.list.SetSize(msg.Width, msg.Height-2)
|
|
case filesLoadedMsg:
|
|
m.loading = false
|
|
m.list.SetItems(msg.items)
|
|
case errMsg:
|
|
m.error = msg
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "q", "esc", "ctrl+c":
|
|
return m, tea.Quit
|
|
case "enter":
|
|
if it, ok := m.list.SelectedItem().(fileItem); ok {
|
|
return m, func() tea.Msg { return FileSelectedMsg{Path: it.path} }
|
|
}
|
|
}
|
|
}
|
|
var cmd tea.Cmd
|
|
m.list, cmd = m.list.Update(msg)
|
|
return m, cmd
|
|
}
|
|
|
|
func (m FilePickerModel) View() string {
|
|
if m.error != nil {
|
|
return lipgloss.NewStyle().Foreground(lipgloss.Color("203")).Render(fmt.Sprintf("Error: %v\n", m.error))
|
|
}
|
|
if m.loading {
|
|
return "Loading files...\n" + m.spinner.View()
|
|
}
|
|
if len(m.list.Items()) == 0 {
|
|
return "No .json files found in current directory. Press q to exit.\n"
|
|
}
|
|
return m.list.View()
|
|
}
|