package main

import (
	"fmt"
	"github.com/charmbracelet/bubbles/list"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"os"
	"strconv"
)

var (
	docStyle = lipgloss.NewStyle().Margin(1, 2)
)

type Person struct {
	Name string
	Age  int
}

func (p Person) Title() string       { return p.Name }
func (p Person) Description() string { return strconv.Itoa(p.Age) }

func (p Person) FilterValue() string { return p.Name }

type model struct {
	list list.Model
}

func (m model) Init() tea.Cmd {
	return nil
}

func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		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)
	}

	var cmd tea.Cmd
	m.list, cmd = m.list.Update(msg)
	return m, cmd
}

func (m model) View() string {
	return docStyle.Render(m.list.View())
}

func main() {
	people := []list.Item{
		Person{Name: "John Smith", Age: 24},
		Person{Name: "Jane Doe", Age: 23},
		Person{Name: "Emily Johnson", Age: 25},
		Person{Name: "Alex Davis", Age: 14},
		Person{Name: "Ethan Brown", Age: 10},
	}

	m := model{list: list.New(people, list.NewDefaultDelegate(), 0, 0)}
	m.list.Title = "People"

	p := tea.NewProgram(m, tea.WithAltScreen())
	if _, err := p.Run(); err != nil {
		fmt.Println("Error running program: ", err)
		os.Exit(1)
	}