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

package chat

import (
	"fmt"
	"log/slog"

	"github.com/ayn2op/discordo/internal/config"
	"github.com/ayn2op/tview"
	"github.com/ayn2op/tview/help"
	"github.com/ayn2op/tview/keybind"
	"github.com/ayn2op/tview/picker"
	"github.com/diamondburned/arikawa/v3/discord"
	"github.com/gdamore/tcell/v3"
)

const emojiPickerLayerName = "emojiPicker"

type emojiPicker struct {
	*picker.Model
	chatView   *Model
	browseMode bool
	favorites  *emojiFavorites

	targetMessageID discord.MessageID
	targetChannelID discord.ChannelID
}

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

func newEmojiPicker(cfg *config.Config, chatView *Model) *emojiPicker {
	ep := &emojiPicker{
		Model:     picker.NewModel(),
		chatView:  chatView,
		favorites: loadEmojiFavorites(),
	}
	ConfigurePicker(ep.Model, cfg, "Emoji")
	return ep
}

// resetBrowse starts the emoji picker in browse (scroll) mode by default.
func (ep *emojiPicker) resetBrowse() { ep.browseMode = true }

var commonEmoji = []struct {
	emoji string
	names string
}{
	{"👍", "thumbsup thumbs_up +1 like"},
	{"👎", "thumbsdown thumbs_down -1 dislike"},
	{"❤️", "heart love red_heart"},
	{"😂", "joy laughing tears"},
	{"😭", "sob crying"},
	{"😊", "blush smile happy"},
	{"😍", "heart_eyes love"},
	{"🤣", "rofl rolling"},
	{"🙂", "slightly_smiling_face"},
	{"😔", "pensive sad"},
	{"🔥", "fire lit hot"},
	{"✅", "white_check_mark check yes"},
	{"❌", "x cross no"},
	{"👀", "eyes look"},
	{"🎉", "tada party celebration"},
	{"💀", "skull dead"},
	{"🤔", "thinking hmm"},
	{"😐", "neutral_face"},
	{"🤷", "shrug"},
	{"👋", "wave hello hi"},
	{"🙏", "pray please thanks"},
	{"💯", "100 hundred perfect"},
	{"😎", "sunglasses cool"},
	{"🥳", "partying_face party"},
	{"😢", "cry sad"},
	{"😅", "sweat_smile nervous"},
	{"🤗", "hugs hugging"},
	{"😤", "triumph angry"},
	{"🥺", "pleading_face"},
	{"💪", "muscle strong flex"},
	{"✨", "sparkles stars"},
	{"💜", "purple_heart"},
	{"💙", "blue_heart"},
	{"💚", "green_heart"},
	{"🧡", "orange_heart"},
	{"💛", "yellow_heart"},
	{"🖤", "black_heart"},
	{"🤍", "white_heart"},
	{"⭐", "star"},
	{"🌟", "star2 glowing_star"},
	{"😮", "open_mouth surprised"},
	{"😱", "scream shocked"},
	{"😈", "smiling_imp devil"},
	{"🤡", "clown clown_face"},
	{"🫡", "salute saluting_face"},
	{"🫠", "melting_face"},
	{"💤", "zzz sleep"},
	{"🤝", "handshake"},
	{"🏳️", "white_flag surrender"},
}

func (ep *emojiPicker) update() {
	items := make(picker.Items, 0, maxFavoriteEmoji+len(commonEmoji)+50)

	allItems := make(picker.Items, 0, len(commonEmoji)+50)
	for _, e := range commonEmoji {
		allItems = append(allItems, picker.Item{
			Text:       e.emoji + " " + e.names[:min(len(e.names), indexOf(e.names, ' '))],
			FilterText: e.names,
			Reference:  discord.APIEmoji(e.emoji),
		})
	}

	// Guild custom emoji
	selectedChannel := ep.chatView.SelectedChannel()
	if selectedChannel != nil && selectedChannel.GuildID.IsValid() {
		emojis, err := ep.chatView.state.Cabinet.Emojis(selectedChannel.GuildID)
		if err == nil {
			for _, emoji := range emojis {
				if !emoji.Available {
					continue
				}
				allItems = append(allItems, picker.Item{
					Text:       ":" + emoji.Name + ":",
					FilterText: emoji.Name,
					Reference:  emoji.APIString(),
				})
			}
		}
	}

	// Index for favorite lookup.
	byKey := make(map[string]picker.Item, len(allItems))
	for _, item := range allItems {
		byKey[emojiItemKey(item)] = item
	}

	// Prepend favorites.
	favSet := make(map[string]struct{})
	for _, key := range ep.favorites.list() {
		if item, ok := byKey[key]; ok {
			items = append(items, picker.Item{
				Text:       "★ " + item.Text,
				FilterText: "favorite " + item.FilterText,
				Reference:  item.Reference,
			})
			favSet[key] = struct{}{}
		}
	}

	for _, item := range allItems {
		if _, isFav := favSet[emojiItemKey(item)]; !isFav {
			items = append(items, item)
		}
	}

	ep.Model.SetItems(items)
}

func emojiItemKey(item picker.Item) string {
	if ref, ok := item.Reference.(discord.APIEmoji); ok {
		return string(ref)
	}
	return ""
}

func indexOf(s string, c byte) int {
	for i := range len(s) {
		if s[i] == c {
			return i
		}
	}
	return len(s)
}

func (ep *emojiPicker) HandleEvent(event tview.Event) tview.Command {
	switch event := event.(type) {
	case *tview.KeyEvent:
		if cmd, handled := pickerBrowseHandleKey(
			event, &ep.browseMode, ep.Model,
			func() { ep.chatView.closeEmojiPicker() },
			ep.handleFavoriteKey,
		); handled {
			return cmd
		}
	case *picker.SelectedEvent:
		apiEmoji, ok := event.Reference.(discord.APIEmoji)
		if !ok {
			return nil
		}

		channelID := ep.targetChannelID
		messageID := ep.targetMessageID
		ep.chatView.closeEmojiPicker()

		return func() tview.Event {
			if err := ep.chatView.state.React(channelID, messageID, apiEmoji); err != nil {
				slog.Error("failed to add reaction", "err", err, "emoji", fmt.Sprint(apiEmoji))
			}
			return nil
		}
	case *picker.CancelEvent:
		ep.chatView.closeEmojiPicker()
		return nil
	}
	return ep.Model.HandleEvent(event)
}

func (ep *emojiPicker) handleFavoriteKey(event *tview.KeyEvent) (tview.Command, bool) {
	if event.Key() != tcell.KeyRune || event.Str() != "f" {
		return nil, false
	}

	// Trigger a select to get the highlighted item's reference without
	// dispatching the event through the normal loop.
	cmd := ep.Model.HandleEvent(tcell.NewEventKey(tcell.KeyEnter, "", tcell.ModNone))
	if cmd == nil {
		return nil, true
	}
	if sel, ok := cmd().(*picker.SelectedEvent); ok {
		key := emojiItemKey(sel.Item)
		if key != "" {
			ep.favorites.toggle(key)
			ep.update()
		}
	}
	return nil, true
}

func (ep *emojiPicker) ShortHelp() []keybind.Keybind {
	return pickerShortHelp(ep.chatView.cfg.Keybinds.Picker)
}

func (ep *emojiPicker) FullHelp() [][]keybind.Keybind {
	return pickerFullHelp(ep.chatView.cfg.Keybinds.Picker)
}