opencode-ide-plugin/internal/tui/components/dialog/models.go

package dialog

import (
	"github.com/charmbracelet/bubbles/key"
	tea "github.com/charmbracelet/bubbletea"
	"github.com/charmbracelet/lipgloss"
	"github.com/sst/opencode/internal/config"
	"github.com/sst/opencode/internal/tui/layout"
	"github.com/sst/opencode/internal/tui/styles"
	"github.com/sst/opencode/internal/tui/theme"
	"github.com/sst/opencode/internal/tui/util"
)

const (
	numVisibleModels = 10
	maxDialogWidth   = 40
)

// ModelSelectedMsg is sent when a model is selected
type ModelSelectedMsg struct {
	// Model models.Model
}

// CloseModelDialogMsg is sent when a model is selected
type CloseModelDialogMsg struct{}

// ModelDialog interface for the model selection dialog
type ModelDialog interface {
	tea.Model
	layout.Bindings
}

type modelDialogCmp struct {
	// models             []models.Model
	// provider           models.ModelProvider
	// availableProviders []models.ModelProvider

	selectedIdx     int
	width           int
	height          int
	scrollOffset    int
	hScrollOffset   int
	hScrollPossible bool
}

type modelKeyMap struct {
	Up     key.Binding
	Down   key.Binding
	Left   key.Binding
	Right  key.Binding
	Enter  key.Binding
	Escape key.Binding
	J      key.Binding
	K      key.Binding
	H      key.Binding
	L      key.Binding
}

var modelKeys = modelKeyMap{
	Up: key.NewBinding(
		key.WithKeys("up"),
		key.WithHelp("↑", "previous model"),
	),
	Down: key.NewBinding(
		key.WithKeys("down"),
		key.WithHelp("↓", "next model"),
	),
	Left: key.NewBinding(
		key.WithKeys("left"),
		key.WithHelp("←", "scroll left"),
	),
	Right: key.NewBinding(
		key.WithKeys("right"),
		key.WithHelp("→", "scroll right"),
	),
	Enter: key.NewBinding(
		key.WithKeys("enter"),
		key.WithHelp("enter", "select model"),
	),
	Escape: key.NewBinding(
		key.WithKeys("esc"),
		key.WithHelp("esc", "close"),
	),
	J: key.NewBinding(
		key.WithKeys("j"),
		key.WithHelp("j", "next model"),
	),
	K: key.NewBinding(
		key.WithKeys("k"),
		key.WithHelp("k", "previous model"),
	),
	H: key.NewBinding(
		key.WithKeys("h"),
		key.WithHelp("h", "scroll left"),
	),
	L: key.NewBinding(
		key.WithKeys("l"),
		key.WithHelp("l", "scroll right"),
	),
}

func (m *modelDialogCmp) Init() tea.Cmd {
	m.setupModels()
	return nil
}

func (m *modelDialogCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
	switch msg := msg.(type) {
	case tea.KeyMsg:
		switch {
		case key.Matches(msg, modelKeys.Up) || key.Matches(msg, modelKeys.K):
			m.moveSelectionUp()
		case key.Matches(msg, modelKeys.Down) || key.Matches(msg, modelKeys.J):
			m.moveSelectionDown()
		case key.Matches(msg, modelKeys.Left) || key.Matches(msg, modelKeys.H):
			if m.hScrollPossible {
				m.switchProvider(-1)
			}
		case key.Matches(msg, modelKeys.Right) || key.Matches(msg, modelKeys.L):
			if m.hScrollPossible {
				m.switchProvider(1)
			}
		case key.Matches(msg, modelKeys.Enter):
			// return m, util.CmdHandler(ModelSelectedMsg{Model: m.models[m.selectedIdx]})
		case key.Matches(msg, modelKeys.Escape):
			return m, util.CmdHandler(CloseModelDialogMsg{})
		}
	case tea.WindowSizeMsg:
		m.width = msg.Width
		m.height = msg.Height
	}

	return m, nil
}

// moveSelectionUp moves the selection up or wraps to bottom
func (m *modelDialogCmp) moveSelectionUp() {
	if m.selectedIdx > 0 {
		m.selectedIdx--
	} else {
		// m.selectedIdx = len(m.models) - 1
		// m.scrollOffset = max(0, len(m.models)-numVisibleModels)
	}

	// Keep selection visible
	if m.selectedIdx < m.scrollOffset {
		m.scrollOffset = m.selectedIdx
	}
}

// moveSelectionDown moves the selection down or wraps to top
func (m *modelDialogCmp) moveSelectionDown() {
	// if m.selectedIdx < len(m.models)-1 {
	// 	m.selectedIdx++
	// } else {
	// 	m.selectedIdx = 0
	// 	m.scrollOffset = 0
	// }

	// Keep selection visible
	if m.selectedIdx >= m.scrollOffset+numVisibleModels {
		m.scrollOffset = m.selectedIdx - (numVisibleModels - 1)
	}
}

func (m *modelDialogCmp) switchProvider(offset int) {
	newOffset := m.hScrollOffset + offset

	// Ensure we stay within bounds
	// if newOffset < 0 {
	// 	newOffset = len(m.availableProviders) - 1
	// }
	// if newOffset >= len(m.availableProviders) {
	// 	newOffset = 0
	// }

	m.hScrollOffset = newOffset
	// m.provider = m.availableProviders[m.hScrollOffset]
	// m.setupModelsForProvider(m.provider)
}

func (m *modelDialogCmp) View() string {
	t := theme.CurrentTheme()
	baseStyle := styles.BaseStyle()

	// Capitalize first letter of provider name
	// providerName := strings.ToUpper(string(m.provider)[:1]) + string(m.provider[1:])
	// title := baseStyle.
	// 	Foreground(t.Primary()).
	// 	Bold(true).
	// 	Width(maxDialogWidth).
	// 	Padding(0, 0, 1).
	// 	Render(fmt.Sprintf("Select %s Model", providerName))

	// Render visible models
	// endIdx := min(m.scrollOffset+numVisibleModels, len(m.models))
	// modelItems := make([]string, 0, endIdx-m.scrollOffset)
	//
	// for i := m.scrollOffset; i < endIdx; i++ {
	// 	itemStyle := baseStyle.Width(maxDialogWidth)
	// 	if i == m.selectedIdx {
	// 		itemStyle = itemStyle.Background(t.Primary()).
	// 			Foreground(t.Background()).Bold(true)
	// 	}
	// 	modelItems = append(modelItems, itemStyle.Render(m.models[i].Name))
	// }

	scrollIndicator := m.getScrollIndicators(maxDialogWidth)

	content := lipgloss.JoinVertical(
		lipgloss.Left,
		// title,
		// baseStyle.Width(maxDialogWidth).Render(lipgloss.JoinVertical(lipgloss.Left, modelItems...)),
		scrollIndicator,
	)

	return baseStyle.Padding(1, 2).
		Border(lipgloss.RoundedBorder()).
		BorderBackground(t.Background()).
		BorderForeground(t.TextMuted()).
		Width(lipgloss.Width(content) + 4).
		Render(content)
}

func (m *modelDialogCmp) getScrollIndicators(maxWidth int) string {
	var indicator string

	// if len(m.models) > numVisibleModels {
	// 	if m.scrollOffset > 0 {
	// 		indicator += "↑ "
	// 	}
	// 	if m.scrollOffset+numVisibleModels < len(m.models) {
	// 		indicator += "↓ "
	// 	}
	// }

	if m.hScrollPossible {
		if m.hScrollOffset > 0 {
			indicator = "← " + indicator
		}
		// if m.hScrollOffset < len(m.availableProviders)-1 {
		// 	indicator += "→"
		// }
	}

	if indicator == "" {
		return ""
	}

	t := theme.CurrentTheme()
	baseStyle := styles.BaseStyle()

	return baseStyle.
		Foreground(t.Primary()).
		Width(maxWidth).
		Align(lipgloss.Right).
		Bold(true).
		Render(indicator)
}

func (m *modelDialogCmp) BindingKeys() []key.Binding {
	return layout.KeyMapToSlice(modelKeys)
}

func (m *modelDialogCmp) setupModels() {
	// cfg := config.Get()
	// modelInfo := GetSelectedModel(cfg)
	// m.availableProviders = getEnabledProviders(cfg)
	// m.hScrollPossible = len(m.availableProviders) > 1
	//
	// m.provider = modelInfo.Provider
	// m.hScrollOffset = findProviderIndex(m.availableProviders, m.provider)
	//
	// m.setupModelsForProvider(m.provider)
}

func GetSelectedModel(cfg *config.Config) string {
	return "Claude Sonnet 4"
	// agentCfg := cfg.Agents[config.AgentPrimary]
	// selectedModelId := agentCfg.Model
	// return models.SupportedModels[selectedModelId]
}

func getEnabledProviders(cfg *config.Config) []string {
	return []string{"anthropic", "openai", "google"}
	// var providers []models.ModelProvider
	// for providerId, provider := range cfg.Providers {
	// 	if !provider.Disabled {
	// 		providers = append(providers, providerId)
	// 	}
	// }
	//
	// // Sort by provider popularity
	// slices.SortFunc(providers, func(a, b models.ModelProvider) int {
	// 	rA := models.ProviderPopularity[a]
	// 	rB := models.ProviderPopularity[b]
	//
	// 	// models not included in popularity ranking default to last
	// 	if rA == 0 {
	// 		rA = 999
	// 	}
	// 	if rB == 0 {
	// 		rB = 999
	// 	}
	// 	return rA - rB
	// })
	// return providers
}

// findProviderIndex returns the index of the provider in the list, or -1 if not found
func findProviderIndex(providers []string, provider string) int {
	for i, p := range providers {
		if p == provider {
			return i
		}
	}
	return -1
}

func (m *modelDialogCmp) setupModelsForProvider(provider string) {
	// cfg := config.Get()
	// agentCfg := cfg.Agents[config.AgentPrimary]
	// selectedModelId := agentCfg.Model

	// m.provider = provider
	// m.models = getModelsForProvider(provider)
	m.selectedIdx = 0
	m.scrollOffset = 0

	// Try to select the current model if it belongs to this provider
	// if provider == models.SupportedModels[selectedModelId].Provider {
	// 	for i, model := range m.models {
	// 		if model.ID == selectedModelId {
	// 			m.selectedIdx = i
	// 			// Adjust scroll position to keep selected model visible
	// 			if m.selectedIdx >= numVisibleModels {
	// 				m.scrollOffset = m.selectedIdx - (numVisibleModels - 1)
	// 			}
	// 			break
	// 		}
	// 	}
	// }
}

func getModelsForProvider(provider string) []string {
	return []string{"Claude Sonnet 4"}
	// var providerModels []models.Model
	// for _, model := range models.SupportedModels {
	// 	if model.Provider == provider {
	// 		providerModels = append(providerModels, model)
	// 	}
	// }

	// reverse alphabetical order (if llm naming was consistent latest would appear first)
	// slices.SortFunc(providerModels, func(a, b models.Model) int {
	// 	if a.Name > b.Name {
	// 		return -1
	// 	} else if a.Name < b.Name {
	// 		return 1
	// 	}
	// 	return 0
	// })

	// return providerModels
}

func NewModelDialogCmp() ModelDialog {
	return &modelDialogCmp{}
}