discordo/internal/ui/chat/message_input.go

752 lines
20 KiB
Go

package chat
import (
"bytes"
"io"
"log/slog"
"os"
"path/filepath"
"regexp"
"slices"
"strings"
"sync"
"time"
"unicode"
"github.com/ayn2op/discordo/internal/cache"
"github.com/ayn2op/discordo/internal/clipboard"
"github.com/ayn2op/discordo/internal/config"
"github.com/ayn2op/discordo/internal/consts"
"github.com/ayn2op/discordo/internal/ui"
"github.com/ayn2op/tview"
"github.com/ayn2op/tview/help"
"github.com/ayn2op/tview/keybind"
"github.com/diamondburned/arikawa/v3/api"
"github.com/diamondburned/arikawa/v3/discord"
"github.com/diamondburned/arikawa/v3/state"
"github.com/diamondburned/arikawa/v3/utils/json/option"
"github.com/diamondburned/arikawa/v3/utils/sendpart"
"github.com/diamondburned/ningen/v3"
"github.com/diamondburned/ningen/v3/discordmd"
"github.com/gdamore/tcell/v3"
"github.com/ncruces/zenity"
"github.com/sahilm/fuzzy"
"github.com/yuin/goldmark/ast"
)
const tmpFilePattern = consts.Name + "_*.md"
var mentionRegex = regexp.MustCompile("@[a-zA-Z0-9._]+")
type messageInput struct {
*tview.TextArea
chat *Model
cfg *config.Config
edit bool
sendMessageData *api.SendMessageData
cache *cache.Cache
mentionsList *mentionsList
lastSearch time.Time
typingTimerMu sync.Mutex
typingTimer *time.Timer
}
var _ help.KeyMap = (*messageInput)(nil)
func newMessageInput(cfg *config.Config, chatView *Model) *messageInput {
mi := &messageInput{
TextArea: tview.NewTextArea(),
cfg: cfg,
chat: chatView,
sendMessageData: &api.SendMessageData{},
cache: cache.NewCache(),
mentionsList: newMentionsList(cfg),
}
mi.Box = ui.ConfigureBox(mi.Box, &cfg.Theme)
mi.
SetPlaceholder(tview.NewLine(tview.NewSegment("Select a channel to start chatting", tcell.StyleDefault.Dim(true)))).
SetClipboard(
func(s string) {
if err := clipboard.Write(clipboard.FmtText, []byte(s)); err != nil {
slog.Error("failed to write clipboard text", "err", err)
}
},
func() string {
data, err := clipboard.Read(clipboard.FmtText)
if err != nil {
slog.Error("failed to read clipboard text", "err", err)
return ""
}
return string(data)
},
).
SetDisabled(true)
return mi
}
func (mi *messageInput) reset() {
mi.edit = false
mi.sendMessageData = &api.SendMessageData{}
mi.SetTitle("")
mi.SetFooter("")
mi.SetText("", true)
}
func (mi *messageInput) stopTypingTimer() {
if mi.typingTimer != nil {
mi.typingTimer.Stop()
mi.typingTimer = nil
}
}
func (mi *messageInput) HandleEvent(event tcell.Event) tview.Command {
switch event := event.(type) {
case *tview.KeyEvent:
handler := mi.TextArea.HandleEvent
switch {
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.Paste.Keybind):
mi.paste()
return handler(tcell.NewEventKey(tcell.KeyCtrlV, "", tcell.ModNone))
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.Send.Keybind):
if mi.chat.GetVisible(mentionsListLayerName) {
mi.tabComplete()
} else {
mi.send()
}
return nil
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.OpenEditor.Keybind):
var cmds []tview.Command
mi.stopTabCompletion(func(next tview.Command) {
if next != nil {
cmds = append(cmds, next)
}
})
mi.editor()
return tview.Batch(cmds...)
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.OpenFilePicker.Keybind):
var cmds []tview.Command
mi.stopTabCompletion(func(next tview.Command) {
if next != nil {
cmds = append(cmds, next)
}
})
mi.openFilePicker()
return tview.Batch(cmds...)
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.Cancel.Keybind):
var cmds []tview.Command
if mi.chat.GetVisible(mentionsListLayerName) {
mi.stopTabCompletion(func(next tview.Command) {
if next != nil {
cmds = append(cmds, next)
}
})
} else {
mi.reset()
}
return tview.Batch(cmds...)
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.TabComplete.Keybind):
go mi.chat.app.QueueUpdateDraw(func() { mi.tabComplete() })
return nil
case keybind.Matches(event, mi.cfg.Keybinds.MessageInput.Undo.Keybind):
return handler(tcell.NewEventKey(tcell.KeyCtrlZ, "", tcell.ModNone))
}
if mi.cfg.TypingIndicator.Send && mi.typingTimer == nil {
mi.typingTimer = time.AfterFunc(typingDuration, func() {
mi.typingTimerMu.Lock()
mi.typingTimer = nil
mi.typingTimerMu.Unlock()
})
if selectedChannel := mi.chat.SelectedChannel(); selectedChannel != nil {
go mi.chat.state.Typing(selectedChannel.ID)
}
}
if mi.cfg.AutocompleteLimit > 0 {
if mi.chat.GetVisible(mentionsListLayerName) {
switch {
case keybind.Matches(event, mi.cfg.Keybinds.MentionsList.Up.Keybind):
mi.mentionsList.HandleEvent(tcell.NewEventKey(tcell.KeyUp, "", tcell.ModNone))
return nil
case keybind.Matches(event, mi.cfg.Keybinds.MentionsList.Down.Keybind):
mi.mentionsList.HandleEvent(tcell.NewEventKey(tcell.KeyDown, "", tcell.ModNone))
return nil
case keybind.Matches(event, mi.cfg.Keybinds.MentionsList.Top.Keybind):
mi.mentionsList.HandleEvent(tcell.NewEventKey(tcell.KeyHome, "", tcell.ModNone))
return nil
case keybind.Matches(event, mi.cfg.Keybinds.MentionsList.Bottom.Keybind):
mi.mentionsList.HandleEvent(tcell.NewEventKey(tcell.KeyEnd, "", tcell.ModNone))
return nil
}
}
go mi.chat.app.QueueUpdateDraw(func() { mi.tabSuggestion() })
}
return handler(event)
}
return mi.TextArea.HandleEvent(event)
}
func (mi *messageInput) paste() {
data, err := clipboard.Read(clipboard.FmtImage)
if err != nil {
slog.Error("failed to read clipboard image", "err", err)
return
}
if data != nil {
name := "clipboard.png"
mi.attach(name, bytes.NewReader(data))
}
}
func (mi *messageInput) send() {
selected := mi.chat.SelectedChannel()
if selected == nil {
return
}
text := strings.TrimSpace(mi.GetText())
if text == "" && len(mi.sendMessageData.Files) == 0 {
return
}
// Close attached files on return
defer func() {
for _, file := range mi.sendMessageData.Files {
if closer, ok := file.Reader.(io.Closer); ok {
closer.Close()
}
}
}()
text = mi.processText(selected, []byte(text))
if mi.edit {
m, err := mi.chat.messagesList.selectedMessage()
if err != nil {
slog.Error("failed to get selected message", "err", err)
return
}
data := api.EditMessageData{Content: option.NewNullableString(text)}
if _, err := mi.chat.state.EditMessageComplex(m.ChannelID, m.ID, data); err != nil {
slog.Error("failed to edit message", "err", err)
}
mi.edit = false
} else {
data := mi.sendMessageData
data.Content = text
if _, err := mi.chat.state.SendMessageComplex(selected.ID, *data); err != nil {
slog.Error("failed to send message in channel", "channel_id", selected.ID, "err", err)
}
}
if mi.typingTimer != nil {
mi.typingTimer.Stop()
mi.typingTimer = nil
}
mi.reset()
mi.chat.messagesList.clearSelection()
mi.chat.messagesList.ScrollToEnd()
}
func (mi *messageInput) processText(channel *discord.Channel, src []byte) string {
// Fast path: no mentions to expand.
if bytes.IndexByte(src, '@') == -1 {
return string(src)
}
// Fast path: no back ticks (code blocks), so expand mentions directly.
if bytes.IndexByte(src, '`') == -1 {
return string(mi.expandMentions(channel, src))
}
var (
ranges [][2]int
canMention = true
)
ast.Walk(discordmd.Parse(src), func(node ast.Node, enter bool) (ast.WalkStatus, error) {
switch node := node.(type) {
case *ast.CodeBlock, *ast.FencedCodeBlock:
canMention = !enter
case *discordmd.Inline:
if (node.Attr & discordmd.AttrMonospace) != 0 {
canMention = !enter
}
case *ast.Text:
if canMention {
ranges = append(ranges, [2]int{node.Segment.Start,
node.Segment.Stop})
}
}
return ast.WalkContinue, nil
})
for _, rng := range ranges {
src = slices.Replace(src, rng[0], rng[1], mi.expandMentions(channel, src[rng[0]:rng[1]])...)
}
return string(src)
}
func (mi *messageInput) expandMentions(c *discord.Channel, src []byte) []byte {
state := mi.chat.state
return mentionRegex.ReplaceAllFunc(src, func(input []byte) []byte {
output := input
name := string(input[1:])
if c.Type == discord.DirectMessage || c.Type == discord.GroupDM {
for _, user := range c.DMRecipients {
if strings.EqualFold(user.Username, name) {
return []byte(user.ID.Mention())
}
}
// self ping
me, _ := state.Cabinet.Me()
if strings.EqualFold(me.Username, name) {
return []byte(me.ID.Mention())
}
return output
}
state.MemberStore.Each(c.GuildID, func(m *discord.Member) bool {
if strings.EqualFold(m.User.Username, name) {
if channelHasUser(state, c.ID, m.User.ID) {
output = []byte(m.User.ID.Mention())
}
return true
}
return false
})
return output
})
}
func (mi *messageInput) tabComplete() {
posEnd, name, r := mi.GetWordUnderCursor(func(r rune) bool {
return unicode.IsLetter(r) || unicode.IsDigit(r) || r == '_' || r == '.'
})
if r != '@' {
mi.stopTabCompletion(nil)
return
}
pos := posEnd - (len(name) + 1)
selected := mi.chat.SelectedChannel()
if selected == nil {
return
}
gID := selected.GuildID
if mi.cfg.AutocompleteLimit == 0 {
if !gID.IsValid() {
users := selected.DMRecipients
res := fuzzy.FindFrom(name, userList(users))
if len(res) > 0 {
mi.Replace(pos, posEnd, "@"+users[res[0].Index].Username+" ")
}
} else {
mi.searchMember(gID, name)
members, err := mi.chat.state.Cabinet.Members(gID)
if err != nil {
slog.Error("failed to get members from state", "guild_id", gID, "err", err)
return
}
res := fuzzy.FindFrom(name, memberList(members))
for _, r := range res {
if channelHasUser(mi.chat.state, selected.ID, members[r.Index].User.ID) {
mi.Replace(pos, posEnd, "@"+members[r.Index].User.Username+" ")
return
}
}
}
return
}
if mi.mentionsList.itemCount() == 0 {
return
}
name, ok := mi.mentionsList.selectedInsertText()
if !ok {
return
}
mi.Replace(pos, posEnd, "@"+name+" ")
mi.stopTabCompletion(nil)
}
func (mi *messageInput) tabSuggestion() {
_, name, r := mi.GetWordUnderCursor(func(r rune) bool {
return unicode.IsLetter(r) || unicode.IsDigit(r) || r == '_' || r == '.'
})
if r != '@' {
mi.stopTabCompletion(nil)
return
}
selected := mi.chat.SelectedChannel()
if selected == nil {
return
}
gID := selected.GuildID
cID := selected.ID
mi.mentionsList.clear()
var shown map[string]struct{}
var userDone struct{}
if name == "" {
shown = make(map[string]struct{})
// Don't show @me in the list of recent authors
me, _ := mi.chat.state.Cabinet.Me()
shown[me.Username] = userDone
}
// DMs have recipients, not members
if !gID.IsValid() {
if name == "" { // show recent messages' authors
msgs, err := mi.chat.state.Cabinet.Messages(cID)
if err != nil {
return
}
for _, m := range msgs {
if _, ok := shown[m.Author.Username]; ok {
continue
}
shown[m.Author.Username] = userDone
mi.addMentionUser(&m.Author)
}
} else {
users := selected.DMRecipients
me, _ := mi.chat.state.Cabinet.Me()
users = append(users, *me)
res := fuzzy.FindFrom(name, userList(users))
for _, r := range res {
mi.addMentionUser(&users[r.Index])
}
}
} else if name == "" { // show recent messages' authors
msgs, err := mi.chat.state.Cabinet.Messages(cID)
if err != nil {
return
}
for _, m := range msgs {
if _, ok := shown[m.Author.Username]; ok {
continue
}
shown[m.Author.Username] = userDone
mi.chat.state.MemberState.RequestMember(gID, m.Author.ID)
if mem, err := mi.chat.state.Cabinet.Member(gID, m.Author.ID); err == nil {
if mi.addMentionMember(gID, mem) {
break
}
}
}
} else {
mi.searchMember(gID, name)
mems, err := mi.chat.state.Cabinet.Members(gID)
if err != nil {
slog.Error("fetching members failed", "err", err)
return
}
res := fuzzy.FindFrom(name, memberList(mems))
if len(res) > int(mi.cfg.AutocompleteLimit) {
res = res[:int(mi.cfg.AutocompleteLimit)]
}
for _, r := range res {
if channelHasUser(mi.chat.state, cID, mems[r.Index].User.ID) &&
mi.addMentionMember(gID, &mems[r.Index]) {
break
}
}
}
if mi.mentionsList.itemCount() == 0 {
mi.stopTabCompletion(nil)
return
}
mi.mentionsList.rebuild()
mi.showMentionList()
}
type memberList []discord.Member
type userList []discord.User
func (ml memberList) String(i int) string {
return ml[i].Nick + ml[i].User.DisplayName + ml[i].User.Tag()
}
func (ml memberList) Len() int {
return len(ml)
}
func (ul userList) String(i int) string {
return ul[i].DisplayName + ul[i].Tag()
}
func (ul userList) Len() int {
return len(ul)
}
// channelHasUser checks if a user has permission to view the specified channel.
func channelHasUser(state *ningen.State, channelID discord.ChannelID, userID discord.UserID) bool {
perms, err := state.Permissions(channelID, userID)
if err != nil {
slog.Error("failed to get permissions", "err", err, "channel", channelID, "user", userID)
return false
}
return perms.Has(discord.PermissionViewChannel)
}
func (mi *messageInput) searchMember(gID discord.GuildID, name string) {
key := gID.String() + " " + name
if mi.cache.Exists(key) {
return
}
// If searching for "ab" returns less than SearchLimit,
// then "abc" would not return anything new because we already searched
// everything starting with "ab". This will still be true even if a new
// member joins because arikawa loads new members into the state.
if k := key[:len(key)-1]; mi.cache.Exists(k) {
if c := mi.cache.Get(k); c < mi.chat.state.MemberState.SearchLimit {
mi.cache.Create(key, c)
return
}
}
// Rate limit on our side because we can't distinguish between a successful search and SearchMember not doing anything because of its internal rate limit that we can't detect
if mi.lastSearch.Add(mi.chat.state.MemberState.SearchFrequency).After(time.Now()) {
return
}
mi.lastSearch = time.Now()
mi.chat.messagesList.waitForChunkEvent()
mi.chat.messagesList.setFetchingChunk(true, 0)
mi.chat.state.MemberState.SearchMember(gID, name)
mi.cache.Create(key, mi.chat.messagesList.waitForChunkEvent())
}
func (mi *messageInput) showMentionList() {
borders := 0
if mi.cfg.Theme.Border.Enabled {
borders = 1
}
l := mi.mentionsList
x, _, _, _ := mi.GetInnerRect()
_, y, _, _ := mi.GetRect()
_, _, maxW, maxH := mi.chat.messagesList.GetInnerRect()
if t := int(mi.cfg.Theme.MentionsList.MaxHeight); t != 0 {
maxH = min(maxH, t)
}
count := mi.mentionsList.itemCount() + borders
h := min(count, maxH) + borders + mi.cfg.Theme.Border.Padding[1]
y -= h
w := int(mi.cfg.Theme.MentionsList.MinWidth)
if w == 0 {
w = maxW
} else {
w = max(w, mi.mentionsList.maxDisplayWidth())
w = min(w+borders*2, maxW)
_, col, _, _ := mi.GetCursor()
x += min(col, maxW-w)
}
l.SetRect(x, y, w, h)
mi.chat.ShowLayer(mentionsListLayerName).SendToFront(mentionsListLayerName)
mi.chat.app.SetFocus(mi)
}
func (mi *messageInput) addMentionMember(gID discord.GuildID, m *discord.Member) bool {
if m == nil {
return false
}
name := m.User.DisplayOrUsername()
if m.Nick != "" {
name = m.Nick
}
style := tcell.StyleDefault
// This avoids a slower member color lookup path.
color, ok := state.MemberColor(m, func(id discord.RoleID) *discord.Role {
r, _ := mi.chat.state.Cabinet.Role(gID, id)
return r
})
if ok {
style = style.Foreground(tcell.NewHexColor(int32(color)))
}
presence, err := mi.chat.state.Cabinet.Presence(gID, m.User.ID)
if err != nil {
slog.Info("failed to get presence from state", "guild_id", gID, "user_id", m.User.ID, "err", err)
} else if presence.Status == discord.OfflineStatus {
style = style.Dim(true)
}
mi.mentionsList.append(mentionsListItem{
insertText: m.User.Username,
displayText: name,
style: style,
})
return mi.mentionsList.itemCount() > int(mi.cfg.AutocompleteLimit)
}
func (mi *messageInput) addMentionUser(user *discord.User) {
if user == nil {
return
}
name := user.DisplayOrUsername()
style := tcell.StyleDefault
presence, err := mi.chat.state.Cabinet.Presence(discord.NullGuildID, user.ID)
if err != nil {
slog.Info("failed to get presence from state", "user_id", user.ID, "err", err)
} else if presence.Status == discord.OfflineStatus {
style = style.Dim(true)
}
mi.mentionsList.append(mentionsListItem{
insertText: user.Username,
displayText: name,
style: style,
})
}
func (mi *messageInput) removeMentionsList() {
mi.chat.HideLayer(mentionsListLayerName)
}
func (mi *messageInput) stopTabCompletion(emit func(tview.Command)) {
if mi.cfg.AutocompleteLimit > 0 {
mi.mentionsList.clear()
if emit != nil {
emit(closeLayer(mentionsListLayerName))
emit(tview.SetFocus(mi))
} else {
mi.removeMentionsList()
mi.chat.app.SetFocus(mi)
}
}
}
func (mi *messageInput) editor() {
file, err := os.CreateTemp("", tmpFilePattern)
if err != nil {
slog.Error("failed to create tmp file", "err", err)
return
}
defer file.Close()
defer os.Remove(file.Name())
file.WriteString(mi.GetText())
if mi.cfg.Editor == "" {
slog.Warn("Attempt to open file with editor, but no editor is set")
return
}
cmd := mi.cfg.CreateEditorCommand(file.Name())
if cmd == nil {
return
}
cmd.Stdin = os.Stdin
cmd.Stdout = os.Stdout
cmd.Stderr = os.Stderr
mi.chat.app.Suspend(func() {
err := cmd.Run()
if err != nil {
slog.Error("failed to run command", "args", cmd.Args, "err", err)
return
}
})
msg, err := os.ReadFile(file.Name())
if err != nil {
slog.Error("failed to read tmp file", "name", file.Name(), "err", err)
return
}
mi.SetText(strings.TrimSpace(string(msg)), true)
}
func (mi *messageInput) openFilePicker() {
if mi.chat.SelectedChannel() == nil {
return
}
paths, err := zenity.SelectFileMultiple()
if err != nil {
slog.Error("failed to open file dialog", "err", err)
return
}
for _, path := range paths {
file, err := os.Open(path)
if err != nil {
slog.Error("failed to open file", "path", path, "err", err)
continue
}
name := filepath.Base(path)
mi.attach(name, file)
}
}
func (mi *messageInput) attach(name string, reader io.Reader) {
mi.sendMessageData.Files = append(mi.sendMessageData.Files, sendpart.File{Name: name, Reader: reader})
var names []string
for _, file := range mi.sendMessageData.Files {
names = append(names, file.Name)
}
mi.SetFooter("Attached " + humanJoin(names))
}
func (mi *messageInput) ShortHelp() []keybind.Keybind {
if mi.chat.GetVisible(mentionsListLayerName) {
cfg := mi.cfg.Keybinds.MentionsList
icfg := mi.cfg.Keybinds.MessageInput
short := []keybind.Keybind{cfg.Up.Keybind, cfg.Down.Keybind, icfg.Cancel.Keybind}
if selected := mi.chat.SelectedChannel(); selected != nil && mi.chat.state.HasPermissions(selected.ID, discord.PermissionAttachFiles) {
short = append(short, icfg.OpenFilePicker.Keybind)
}
return short
}
cfg := mi.cfg.Keybinds.MessageInput
short := []keybind.Keybind{cfg.Send.Keybind, cfg.Cancel.Keybind, cfg.Paste.Keybind, cfg.OpenEditor.Keybind}
if selected := mi.chat.SelectedChannel(); selected != nil && mi.chat.state.HasPermissions(selected.ID, discord.PermissionAttachFiles) {
short = append(short, cfg.OpenFilePicker.Keybind)
}
return short
}
func (mi *messageInput) FullHelp() [][]keybind.Keybind {
if mi.chat.GetVisible(mentionsListLayerName) {
mcfg := mi.cfg.Keybinds.MentionsList
icfg := mi.cfg.Keybinds.MessageInput
return [][]keybind.Keybind{
{mcfg.Up.Keybind, mcfg.Down.Keybind, mcfg.Top.Keybind, mcfg.Bottom.Keybind},
{icfg.TabComplete.Keybind, icfg.Cancel.Keybind},
}
}
cfg := mi.cfg.Keybinds.MessageInput
openEditor := []keybind.Keybind{cfg.Paste.Keybind, cfg.OpenEditor.Keybind}
if selected := mi.chat.SelectedChannel(); selected != nil && mi.chat.state.HasPermissions(selected.ID, discord.PermissionAttachFiles) {
openEditor = append(openEditor, cfg.OpenFilePicker.Keybind)
}
return [][]keybind.Keybind{
{cfg.Send.Keybind, cfg.Cancel.Keybind, cfg.TabComplete.Keybind, cfg.Undo.Keybind},
openEditor,
}
}