wip: refactoring tui
This commit is contained in:
adamdottv
@@ -1,29 +1,23 @@
|
||||
package dialog
|
||||
|
||||
import (
|
||||
"github.com/charmbracelet/bubbles/v2/key"
|
||||
"context"
|
||||
|
||||
tea "github.com/charmbracelet/bubbletea/v2"
|
||||
"github.com/sst/opencode/internal/app"
|
||||
"github.com/sst/opencode/internal/components/modal"
|
||||
utilComponents "github.com/sst/opencode/internal/components/util"
|
||||
components "github.com/sst/opencode/internal/components/util"
|
||||
"github.com/sst/opencode/internal/layout"
|
||||
"github.com/sst/opencode/internal/state"
|
||||
"github.com/sst/opencode/internal/styles"
|
||||
"github.com/sst/opencode/internal/theme"
|
||||
"github.com/sst/opencode/internal/util"
|
||||
"github.com/sst/opencode/pkg/client"
|
||||
)
|
||||
|
||||
// CloseSessionDialogMsg is sent when the session dialog is closed
|
||||
type CloseSessionDialogMsg struct {
|
||||
Session *client.SessionInfo
|
||||
}
|
||||
|
||||
// SessionDialog interface for the session switching dialog
|
||||
type SessionDialog interface {
|
||||
tea.Model
|
||||
layout.Bindings
|
||||
SetSessions(sessions []client.SessionInfo)
|
||||
SetSelectedSession(sessionID string)
|
||||
Render(background string) string
|
||||
layout.Modal
|
||||
}
|
||||
|
||||
type sessionItem struct {
|
||||
@@ -49,163 +43,70 @@ func (s sessionItem) Render(selected bool, width int) string {
|
||||
return baseStyle.Padding(0, 1).Render(s.session.Title)
|
||||
}
|
||||
|
||||
// sessionDialogContent is the inner content of the session dialog
|
||||
type sessionDialogContent struct {
|
||||
sessions []client.SessionInfo
|
||||
type sessionDialog struct {
|
||||
width int
|
||||
height int
|
||||
modal *modal.Modal
|
||||
selectedSessionID string
|
||||
list utilComponents.SimpleList[sessionItem]
|
||||
list components.SimpleList[sessionItem]
|
||||
}
|
||||
|
||||
type sessionKeyMap struct {
|
||||
Enter key.Binding
|
||||
Escape key.Binding
|
||||
}
|
||||
|
||||
var sessionKeys = sessionKeyMap{
|
||||
Enter: key.NewBinding(
|
||||
key.WithKeys("enter"),
|
||||
key.WithHelp("enter", "select session"),
|
||||
),
|
||||
Escape: key.NewBinding(
|
||||
key.WithKeys("esc"),
|
||||
key.WithHelp("esc", "close"),
|
||||
),
|
||||
}
|
||||
|
||||
func (s *sessionDialogContent) Init() tea.Cmd {
|
||||
func (s *sessionDialog) Init() tea.Cmd {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *sessionDialogContent) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
func (s *sessionDialog) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
switch msg := msg.(type) {
|
||||
case tea.WindowSizeMsg:
|
||||
s.width = msg.Width
|
||||
s.height = msg.Height
|
||||
s.list.SetMaxWidth(layout.Current.Container.Width - 12)
|
||||
case tea.KeyMsg:
|
||||
switch {
|
||||
case key.Matches(msg, sessionKeys.Enter):
|
||||
switch msg.String() {
|
||||
case "enter":
|
||||
if item, idx := s.list.GetSelectedItem(); idx >= 0 {
|
||||
selectedSession := item.session
|
||||
s.selectedSessionID = selectedSession.Id
|
||||
|
||||
return s, util.CmdHandler(CloseSessionDialogMsg{
|
||||
Session: &selectedSession,
|
||||
})
|
||||
return s, tea.Batch(
|
||||
util.CmdHandler(state.SessionSelectedMsg(&selectedSession)),
|
||||
util.CmdHandler(modal.CloseModalMsg{}),
|
||||
)
|
||||
}
|
||||
case key.Matches(msg, sessionKeys.Escape):
|
||||
return s, util.CmdHandler(CloseSessionDialogMsg{})
|
||||
default:
|
||||
// Pass other key messages to the list component
|
||||
var cmd tea.Cmd
|
||||
listModel, cmd := s.list.Update(msg)
|
||||
s.list = listModel.(utilComponents.SimpleList[sessionItem])
|
||||
return s, cmd
|
||||
}
|
||||
}
|
||||
|
||||
// For non-key messages
|
||||
var cmd tea.Cmd
|
||||
listModel, cmd := s.list.Update(msg)
|
||||
s.list = listModel.(utilComponents.SimpleList[sessionItem])
|
||||
s.list = listModel.(components.SimpleList[sessionItem])
|
||||
return s, cmd
|
||||
}
|
||||
|
||||
func (s *sessionDialogContent) View() string {
|
||||
t := theme.CurrentTheme()
|
||||
baseStyle := styles.BaseStyle().Background(t.BackgroundElement())
|
||||
width := layout.Current.Container.Width - 12
|
||||
|
||||
if len(s.sessions) == 0 {
|
||||
return baseStyle.Padding(1, 2).
|
||||
Foreground(t.TextMuted()).
|
||||
Width(width).
|
||||
Render("No sessions available")
|
||||
}
|
||||
|
||||
// Set the max width for the list
|
||||
s.list.SetMaxWidth(width)
|
||||
|
||||
return s.list.View()
|
||||
func (s *sessionDialog) Render(background string) string {
|
||||
return s.modal.Render(s.list.View(), background)
|
||||
}
|
||||
|
||||
func (s *sessionDialogContent) BindingKeys() []key.Binding {
|
||||
// Combine session dialog keys with list keys
|
||||
dialogKeys := layout.KeyMapToSlice(sessionKeys)
|
||||
listKeys := s.list.BindingKeys()
|
||||
return append(dialogKeys, listKeys...)
|
||||
func (s *sessionDialog) Close() tea.Cmd {
|
||||
return nil
|
||||
}
|
||||
|
||||
// sessionDialogComponent wraps the content with a modal
|
||||
type sessionDialogComponent struct {
|
||||
content *sessionDialogContent
|
||||
modal *modal.Modal
|
||||
}
|
||||
// NewSessionDialog creates a new session switching dialog
|
||||
func NewSessionDialog(app *app.App) SessionDialog {
|
||||
sessions, _ := app.ListSessions(context.Background())
|
||||
|
||||
func (s *sessionDialogComponent) Init() tea.Cmd {
|
||||
return s.modal.Init()
|
||||
}
|
||||
|
||||
func (s *sessionDialogComponent) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
|
||||
m, cmd := s.modal.Update(msg)
|
||||
s.modal = m.(*modal.Modal)
|
||||
return s, cmd
|
||||
}
|
||||
|
||||
func (s *sessionDialogComponent) View() string {
|
||||
return s.modal.View()
|
||||
}
|
||||
|
||||
func (s *sessionDialogComponent) Render(background string) string {
|
||||
return s.modal.Render(background)
|
||||
}
|
||||
|
||||
func (s *sessionDialogComponent) BindingKeys() []key.Binding {
|
||||
return s.modal.BindingKeys()
|
||||
}
|
||||
|
||||
func (s *sessionDialogComponent) SetSessions(sessions []client.SessionInfo) {
|
||||
s.content.sessions = sessions
|
||||
|
||||
// Convert sessions to sessionItems
|
||||
var sessionItems []sessionItem
|
||||
|
||||
for _, sess := range sessions {
|
||||
sessionItems = append(sessionItems, sessionItem{session: sess})
|
||||
}
|
||||
|
||||
s.content.list.SetItems(sessionItems)
|
||||
}
|
||||
|
||||
func (s *sessionDialogComponent) SetSelectedSession(sessionID string) {
|
||||
s.content.selectedSessionID = sessionID
|
||||
|
||||
// Update the selected index if sessions are already loaded
|
||||
if len(s.content.sessions) > 0 {
|
||||
// Re-set the sessions to update the selection
|
||||
s.SetSessions(s.content.sessions)
|
||||
}
|
||||
}
|
||||
|
||||
// NewSessionDialogCmp creates a new session switching dialog
|
||||
func NewSessionDialogCmp() SessionDialog {
|
||||
list := utilComponents.NewSimpleList[sessionItem](
|
||||
[]sessionItem{},
|
||||
list := components.NewSimpleList(
|
||||
sessionItems,
|
||||
10, // maxVisibleSessions
|
||||
"No sessions available",
|
||||
true, // useAlphaNumericKeys
|
||||
)
|
||||
|
||||
content := &sessionDialogContent{
|
||||
sessions: []client.SessionInfo{},
|
||||
selectedSessionID: "",
|
||||
list: list,
|
||||
}
|
||||
|
||||
return &sessionDialogComponent{
|
||||
content: content,
|
||||
modal: modal.New(content, modal.WithTitle("Switch Session")),
|
||||
return &sessionDialog{
|
||||
list: list,
|
||||
modal: modal.New(modal.WithTitle("Switch Session"), modal.WithMaxWidth(80)),
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user