Somewhat functional concept

internal/tuimode/tuimode.go

@@ -4,37 +4,100 @@ package tuimode
 // from the Bubbles component library.
 
 import (
+	"bufio"
+	"bytes"
+	"io"
 	"log"
+	"os/exec"
+	"strings"
 
+	"github.com/HikariKnight/ls-iommu/pkg/errorcheck"
+	"github.com/charmbracelet/bubbles/list"
 	tea "github.com/charmbracelet/bubbletea"
+	"github.com/charmbracelet/lipgloss"
 )
 
-type model struct{}
+var docStyle = lipgloss.NewStyle().Margin(1, 2)
+
+type item struct {
+	title, desc string
+}
+
+func (i item) Title() string       { return i.title }
+func (i item) Description() string { return i.desc }
+func (i item) FilterValue() string { return i.title }
+
+type model struct {
+	lists map[string]list.Model
+	list  list.Model
+}
+
+func initLists() model {
+	m := model{lists: make(map[string]list.Model)}
+	return m
+}
+
+func (m model) addItems(obj string, items []list.Item) {
+	m.lists[obj] = list.New(items, list.NewDefaultDelegate(), 0, 0)
+}
 
 func (m model) Init() tea.Cmd {
 	return nil
 }
 
-// This is where we handle all the events
 func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	switch msg := msg.(type) {
 	case tea.KeyMsg:
-		switch msg.String() {
-		case "ctrl+c":
+		if msg.String() == "ctrl+c" {
 			return m, tea.Quit
 		}
+	case tea.WindowSizeMsg:
+		h, v := docStyle.GetFrameSize()
+		m.list.SetSize(msg.Width-h, msg.Height-v)
 	}
-	return m, nil
+
+	var cmd tea.Cmd
+	m.list, cmd = m.list.Update(msg)
+	return m, cmd
 }
 
-// This is where we change what is shown on the screen
 func (m model) View() string {
-	return "test\n"
+	return docStyle.Render(m.list.View())
 }
 
 // This is where we build everything
 func App() {
-	p := tea.NewProgram(model{}, tea.WithAltScreen())
+	var stdout, stderr bytes.Buffer
+
+	cmd := exec.Command("utils/ls-iommu", "-g", "-F", "name,device_id,optional_revision")
+	cmd.Stderr = &stderr
+	cmd.Stdout = &stdout
+
+	err := cmd.Run()
+	errorcheck.ErrorCheck(err)
+
+	items := []list.Item{}
+	output, _ := io.ReadAll(&stdout)
+
+	// If we get an error from ls-iommu then print the error and exit
+	// As the system might not have IOMMU enabled
+	if stderr.String() != "" {
+		log.Fatal(stderr.String())
+	}
+
+	scanner := bufio.NewScanner(strings.NewReader(string(output)))
+	for scanner.Scan() {
+		objects := strings.Split(scanner.Text(), ": ")
+		items = append(items, item{title: objects[1], desc: objects[0]})
+	}
+
+	l := list.New(items, list.NewDefaultDelegate(), 0, 0)
+	l.Title = "Select a GPU to check the IOMMU groups of"
+
+	m := model{list: l}
+
+	// Start the program with the model
+	p := tea.NewProgram(m, tea.WithAltScreen())
 	if _, err := p.Run(); err != nil {
 		log.Fatal(err)
 	}