discordo-plus/internal/ui/chat/attachments_picker.go

package chat

import (
	"github.com/ayn2op/discordo/internal/config"
	"github.com/ayn2op/discordo/internal/ui"
	"github.com/ayn2op/tview"
	"github.com/ayn2op/tview/help"
	"github.com/ayn2op/tview/keybind"
	"github.com/ayn2op/tview/list"
	"github.com/ayn2op/tview/picker"
)

type attachmentItem struct {
	label string
	open  func()
}

type attachmentsPicker struct {
	*picker.Model
	cfg      *config.Config
	chatView *Model
	items    []attachmentItem
}

var _ help.KeyMap = (*attachmentsPicker)(nil)

func newAttachmentsPicker(cfg *config.Config, chatView *Model) *attachmentsPicker {
	ap := &attachmentsPicker{
		Model:    picker.NewModel(),
		cfg:      cfg,
		chatView: chatView,
	}
	ap.Box = ui.ConfigureBox(tview.NewBox(), &cfg.Theme)
	ap.
		SetBlurFunc(nil).
		SetFocusFunc(nil).
		SetBorderSet(cfg.Theme.Border.ActiveSet.BorderSet).
		SetBorderStyle(cfg.Theme.Border.ActiveStyle.Style).
		SetTitleStyle(cfg.Theme.Title.ActiveStyle.Style).
		SetFooterStyle(cfg.Theme.Footer.ActiveStyle.Style)

	ap.SetTitle("Attachments")
	ap.SetKeybinds(picker.Keybinds{
		Cancel: cfg.Keybinds.Picker.Cancel.Keybind,
		Keybinds: list.Keybinds{
			SelectUp:     cfg.Keybinds.Picker.Up.Keybind,
			SelectDown:   cfg.Keybinds.Picker.Down.Keybind,
			SelectTop:    cfg.Keybinds.Picker.Top.Keybind,
			SelectBottom: cfg.Keybinds.Picker.Bottom.Keybind,
		},
		Select: cfg.Keybinds.Picker.Select.Keybind,
	})
	ap.SetScrollBarVisibility(cfg.Theme.ScrollBar.Visibility.ScrollBarVisibility)
	ap.SetScrollBar(tview.NewScrollBar().
		SetTrackStyle(cfg.Theme.ScrollBar.TrackStyle.Style).
		SetThumbStyle(cfg.Theme.ScrollBar.ThumbStyle.Style).
		SetGlyphSet(cfg.Theme.ScrollBar.GlyphSet.GlyphSet))
	return ap
}

func (ap *attachmentsPicker) SetItems(items []attachmentItem) {
	ap.items = items
	ap.ClearItems()
	for i, item := range items {
		ap.AddItem(picker.Item{
			Text:       item.label,
			FilterText: item.label,
			Reference:  i,
		})
	}
	ap.Update()
}

func (ap *attachmentsPicker) close() {
	ap.chatView.RemoveLayer(attachmentsListLayerName)
	ap.chatView.app.SetFocus(ap.chatView.messagesList)
}

func (ap *attachmentsPicker) HandleEvent(event tview.Event) tview.Command {
	switch event := event.(type) {
	case *picker.SelectedEvent:
		index, ok := event.Reference.(int)
		if !ok {
			return nil
		}
		if index < 0 || index >= len(ap.items) {
			return nil
		}
		ap.items[index].open()
		ap.close()
		return nil
	case *picker.CancelEvent:
		ap.close()
		return nil
	}

	return ap.Model.HandleEvent(event)
}

func (ap *attachmentsPicker) ShortHelp() []keybind.Keybind {
	cfg := ap.cfg.Keybinds.Picker
	return []keybind.Keybind{cfg.Up.Keybind, cfg.Down.Keybind, cfg.Select.Keybind, cfg.Cancel.Keybind}
}

func (ap *attachmentsPicker) FullHelp() [][]keybind.Keybind {
	cfg := ap.cfg.Keybinds.Picker
	return [][]keybind.Keybind{
		{cfg.Up.Keybind, cfg.Down.Keybind, cfg.Top.Keybind, cfg.Bottom.Keybind},
		{cfg.Select.Keybind, cfg.Cancel.Keybind},
	}
}