mirror of
https://github.com/sst/opencode.git
synced 2025-08-24 06:54:09 +00:00
132 lines
3.4 KiB
Go
132 lines
3.4 KiB
Go
package dialog
|
|
|
|
import (
|
|
tea "github.com/charmbracelet/bubbletea/v2"
|
|
list "github.com/sst/opencode/internal/components/list"
|
|
"github.com/sst/opencode/internal/components/modal"
|
|
"github.com/sst/opencode/internal/layout"
|
|
"github.com/sst/opencode/internal/styles"
|
|
"github.com/sst/opencode/internal/theme"
|
|
"github.com/sst/opencode/internal/util"
|
|
)
|
|
|
|
// ThemeSelectedMsg is sent when the theme is changed
|
|
type ThemeSelectedMsg struct {
|
|
ThemeName string
|
|
}
|
|
|
|
// ThemeDialog interface for the theme switching dialog
|
|
type ThemeDialog interface {
|
|
layout.Modal
|
|
}
|
|
|
|
type themeDialog struct {
|
|
width int
|
|
height int
|
|
|
|
modal *modal.Modal
|
|
list list.List[list.Item]
|
|
originalTheme string
|
|
themeApplied bool
|
|
}
|
|
|
|
func (t *themeDialog) Init() tea.Cmd {
|
|
return nil
|
|
}
|
|
|
|
func (t *themeDialog) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
|
switch msg := msg.(type) {
|
|
case tea.WindowSizeMsg:
|
|
t.width = msg.Width
|
|
t.height = msg.Height
|
|
case tea.KeyMsg:
|
|
switch msg.String() {
|
|
case "enter":
|
|
if item, idx := t.list.GetSelectedItem(); idx >= 0 {
|
|
if stringItem, ok := item.(list.StringItem); ok {
|
|
selectedTheme := string(stringItem)
|
|
if err := theme.SetTheme(selectedTheme); err != nil {
|
|
// status.Error(err.Error())
|
|
return t, nil
|
|
}
|
|
t.themeApplied = true
|
|
return t, tea.Sequence(
|
|
util.CmdHandler(modal.CloseModalMsg{}),
|
|
util.CmdHandler(ThemeSelectedMsg{ThemeName: selectedTheme}),
|
|
)
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
_, prevIdx := t.list.GetSelectedItem()
|
|
|
|
var cmd tea.Cmd
|
|
listModel, cmd := t.list.Update(msg)
|
|
t.list = listModel.(list.List[list.Item])
|
|
|
|
if item, newIdx := t.list.GetSelectedItem(); newIdx >= 0 && newIdx != prevIdx {
|
|
if stringItem, ok := item.(list.StringItem); ok {
|
|
theme.SetTheme(string(stringItem))
|
|
return t, util.CmdHandler(ThemeSelectedMsg{ThemeName: string(stringItem)})
|
|
}
|
|
}
|
|
return t, cmd
|
|
}
|
|
|
|
func (t *themeDialog) Render(background string) string {
|
|
return t.modal.Render(t.list.View(), background)
|
|
}
|
|
|
|
func (t *themeDialog) Close() tea.Cmd {
|
|
if !t.themeApplied {
|
|
theme.SetTheme(t.originalTheme)
|
|
return util.CmdHandler(ThemeSelectedMsg{ThemeName: t.originalTheme})
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// NewThemeDialog creates a new theme switching dialog
|
|
func NewThemeDialog() ThemeDialog {
|
|
themes := theme.AvailableThemes()
|
|
currentTheme := theme.CurrentThemeName()
|
|
|
|
var selectedIdx int
|
|
for i, name := range themes {
|
|
if name == currentTheme {
|
|
selectedIdx = i
|
|
}
|
|
}
|
|
|
|
// Convert themes to list items
|
|
items := make([]list.Item, len(themes))
|
|
for i, theme := range themes {
|
|
items[i] = list.StringItem(theme)
|
|
}
|
|
|
|
listComponent := list.NewListComponent(
|
|
list.WithItems(items),
|
|
list.WithMaxVisibleHeight[list.Item](10),
|
|
list.WithFallbackMessage[list.Item]("No themes available"),
|
|
list.WithAlphaNumericKeys[list.Item](true),
|
|
list.WithRenderFunc(func(item list.Item, selected bool, width int, baseStyle styles.Style) string {
|
|
return item.Render(selected, width, baseStyle)
|
|
}),
|
|
list.WithSelectableFunc(func(item list.Item) bool {
|
|
return item.Selectable()
|
|
}),
|
|
)
|
|
|
|
// Set the initial selection to the current theme
|
|
listComponent.SetSelectedIndex(selectedIdx)
|
|
|
|
// Set the max width for the list to match the modal width
|
|
listComponent.SetMaxWidth(36) // 40 (modal max width) - 4 (modal padding)
|
|
return &themeDialog{
|
|
list: listComponent,
|
|
modal: modal.New(modal.WithTitle("Select Theme"), modal.WithMaxWidth(40)),
|
|
originalTheme: currentTheme,
|
|
themeApplied: false,
|
|
}
|
|
}
|