| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854 |
- package chat
- import (
- "context"
- "errors"
- "io"
- "log/slog"
- "net/http"
- "os"
- "path/filepath"
- "slices"
- "strings"
- "sync"
- "time"
- "github.com/ayn2op/tview/layers"
- "github.com/ayn2op/discordo/internal/clipboard"
- "github.com/ayn2op/discordo/internal/config"
- "github.com/ayn2op/discordo/internal/consts"
- "github.com/ayn2op/discordo/internal/markdown"
- "github.com/ayn2op/discordo/internal/ui"
- "github.com/ayn2op/tview"
- "github.com/diamondburned/arikawa/v3/api"
- "github.com/diamondburned/arikawa/v3/discord"
- "github.com/diamondburned/arikawa/v3/gateway"
- "github.com/diamondburned/arikawa/v3/state"
- "github.com/diamondburned/arikawa/v3/utils/json/option"
- "github.com/diamondburned/ningen/v3/discordmd"
- "github.com/gdamore/tcell/v3"
- "github.com/gdamore/tcell/v3/color"
- "github.com/skratchdot/open-golang/open"
- "github.com/yuin/goldmark/ast"
- "github.com/yuin/goldmark/parser"
- "github.com/yuin/goldmark/text"
- )
- type messagesList struct {
- *tview.List
- cfg *config.Config
- chatView *View
- messages []discord.Message
- renderer *markdown.Renderer
- // itemByID caches unselected message TextViews.
- itemByID map[discord.MessageID]*tview.TextView
- fetchingMembers struct {
- mu sync.Mutex
- value bool
- count uint
- done chan struct{}
- }
- }
- func newMessagesList(cfg *config.Config, chatView *View) *messagesList {
- ml := &messagesList{
- List: tview.NewList(),
- cfg: cfg,
- chatView: chatView,
- renderer: markdown.NewRenderer(cfg.Theme.MessagesList),
- itemByID: make(map[discord.MessageID]*tview.TextView),
- }
- ml.Box = ui.ConfigureBox(ml.Box, &cfg.Theme)
- ml.SetTitle("Messages")
- ml.SetBuilder(ml.buildItem)
- ml.SetTrackEnd(true)
- ml.SetInputCapture(ml.onInputCapture)
- return ml
- }
- func (ml *messagesList) reset() {
- ml.messages = nil
- clear(ml.itemByID)
- ml.
- Clear().
- SetBuilder(ml.buildItem).
- SetTitle("")
- }
- func (ml *messagesList) setTitle(channel discord.Channel) {
- title := ui.ChannelToString(channel, ml.cfg.Icons)
- if topic := channel.Topic; topic != "" {
- title += " - " + topic
- }
- ml.SetTitle(title)
- }
- func (ml *messagesList) setMessages(messages []discord.Message) {
- ml.messages = slices.Clone(messages)
- slices.Reverse(ml.messages)
- // New channel payload / refetch: replace the cache wholesale to keep it in
- // lockstep with the current message slice.
- clear(ml.itemByID)
- ml.MarkDirty()
- }
- func (ml *messagesList) addMessage(message discord.Message) {
- ml.messages = append(ml.messages, message)
- // Defensive invalidation for ID reuse/edits delivered out-of-order.
- delete(ml.itemByID, message.ID)
- ml.MarkDirty()
- }
- func (ml *messagesList) setMessage(index int, message discord.Message) {
- if index < 0 || index >= len(ml.messages) {
- return
- }
- ml.messages[index] = message
- delete(ml.itemByID, message.ID)
- ml.MarkDirty()
- }
- func (ml *messagesList) deleteMessage(index int) {
- if index < 0 || index >= len(ml.messages) {
- return
- }
- delete(ml.itemByID, ml.messages[index].ID)
- ml.messages = append(ml.messages[:index], ml.messages[index+1:]...)
- ml.MarkDirty()
- }
- func (ml *messagesList) clearSelection() {
- ml.SetCursor(-1)
- }
- func (ml *messagesList) buildItem(index int, cursor int) tview.ListItem {
- if index < 0 || index >= len(ml.messages) {
- return nil
- }
- message := ml.messages[index]
- if index == cursor {
- return tview.NewTextView().
- SetWrap(true).
- SetWordWrap(true).
- SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.SelectedMessageStyle.Style))
- }
- item, ok := ml.itemByID[message.ID]
- if !ok {
- item = tview.NewTextView().
- SetWrap(true).
- SetWordWrap(true).
- SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.MessageStyle.Style))
- ml.itemByID[message.ID] = item
- }
- return item
- }
- func (ml *messagesList) renderMessage(message discord.Message, baseStyle tcell.Style) []tview.Line {
- builder := tview.NewLineBuilder()
- ml.writeMessage(builder, message, baseStyle)
- return builder.Finish()
- }
- func (ml *messagesList) writeMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- if ml.cfg.HideBlockedUsers {
- isBlocked := ml.chatView.state.UserIsBlocked(message.Author.ID)
- if isBlocked {
- builder.Write("Blocked message", baseStyle.Foreground(color.Red).Bold(true))
- return
- }
- }
- switch message.Type {
- case discord.DefaultMessage:
- if message.Reference != nil && message.Reference.Type == discord.MessageReferenceTypeForward {
- ml.drawForwardedMessage(builder, message, baseStyle)
- } else {
- ml.drawDefaultMessage(builder, message, baseStyle)
- }
- case discord.GuildMemberJoinMessage:
- ml.drawTimestamps(builder, message.Timestamp, baseStyle)
- ml.drawAuthor(builder, message, baseStyle)
- builder.Write("joined the server.", baseStyle)
- case discord.InlinedReplyMessage:
- ml.drawReplyMessage(builder, message, baseStyle)
- case discord.ChannelPinnedMessage:
- ml.drawPinnedMessage(builder, message, baseStyle)
- default:
- ml.drawTimestamps(builder, message.Timestamp, baseStyle)
- ml.drawAuthor(builder, message, baseStyle)
- }
- }
- func (ml *messagesList) formatTimestamp(ts discord.Timestamp) string {
- return ts.Time().In(time.Local).Format(ml.cfg.Timestamps.Format)
- }
- func (ml *messagesList) drawTimestamps(builder *tview.LineBuilder, ts discord.Timestamp, baseStyle tcell.Style) {
- dimStyle := baseStyle.Dim(true)
- builder.Write(ml.formatTimestamp(ts)+" ", dimStyle)
- }
- func (ml *messagesList) drawAuthor(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- name := message.Author.DisplayOrUsername()
- foreground := tcell.ColorDefault
- // Webhooks do not have nicknames or roles.
- if message.GuildID.IsValid() && !message.WebhookID.IsValid() {
- member, err := ml.chatView.state.Cabinet.Member(message.GuildID, message.Author.ID)
- if err != nil {
- slog.Error("failed to get member from state", "guild_id", message.GuildID, "member_id", message.Author.ID, "err", err)
- } else {
- if member.Nick != "" {
- name = member.Nick
- }
- color, ok := state.MemberColor(member, func(id discord.RoleID) *discord.Role {
- r, _ := ml.chatView.state.Cabinet.Role(message.GuildID, id)
- return r
- })
- if ok {
- foreground = tcell.NewHexColor(int32(color))
- }
- }
- }
- style := baseStyle.Foreground(foreground).Bold(true)
- builder.Write(name+" ", style)
- }
- func (ml *messagesList) drawContent(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- c := []byte(message.Content)
- if ml.chatView.cfg.Markdown {
- root := discordmd.ParseWithMessage(c, *ml.chatView.state.Cabinet, &message, false)
- lines := ml.renderer.RenderLines(c, root, baseStyle)
- if builder.HasCurrentLine() {
- for len(lines) > 1 && len(lines[0]) == 0 {
- lines = lines[1:]
- }
- }
- builder.AppendLines(lines)
- } else {
- builder.Write(message.Content, baseStyle)
- }
- }
- func (ml *messagesList) drawSnapshotContent(builder *tview.LineBuilder, message discord.MessageSnapshotMessage, baseStyle tcell.Style) {
- c := []byte(message.Content)
- // discordmd doesn't support MessageSnapshotMessage, so we just use write it as is. todo?
- builder.Write(string(c), baseStyle)
- }
- func (ml *messagesList) drawDefaultMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- if ml.cfg.Timestamps.Enabled {
- ml.drawTimestamps(builder, message.Timestamp, baseStyle)
- }
- ml.drawAuthor(builder, message, baseStyle)
- ml.drawContent(builder, message, baseStyle)
- if message.EditedTimestamp.IsValid() {
- dimStyle := baseStyle.Dim(true)
- builder.Write(" (edited)", dimStyle)
- }
- attachmentStyle := ui.MergeStyle(baseStyle, ml.cfg.Theme.MessagesList.AttachmentStyle.Style)
- for _, a := range message.Attachments {
- builder.NewLine()
- if ml.cfg.ShowAttachmentLinks {
- builder.Write(a.Filename+":\n"+a.URL, attachmentStyle)
- } else {
- builder.Write(a.Filename, attachmentStyle)
- }
- }
- }
- func (ml *messagesList) drawForwardedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- dimStyle := baseStyle.Dim(true)
- ml.drawTimestamps(builder, message.Timestamp, baseStyle)
- ml.drawAuthor(builder, message, baseStyle)
- builder.Write(ml.cfg.Theme.MessagesList.ForwardedIndicator+" ", dimStyle)
- ml.drawSnapshotContent(builder, message.MessageSnapshots[0].Message, baseStyle)
- builder.Write(" ("+ml.formatTimestamp(message.MessageSnapshots[0].Message.Timestamp)+") ", dimStyle)
- }
- func (ml *messagesList) drawReplyMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- dimStyle := baseStyle.Dim(true)
- // indicator
- builder.Write(ml.cfg.Theme.MessagesList.ReplyIndicator+" ", dimStyle)
- if m := message.ReferencedMessage; m != nil {
- m.GuildID = message.GuildID
- ml.drawAuthor(builder, *m, dimStyle)
- ml.drawContent(builder, *m, dimStyle)
- } else {
- builder.Write("Original message was deleted", dimStyle)
- }
- builder.NewLine()
- // main
- ml.drawDefaultMessage(builder, message, baseStyle)
- }
- func (ml *messagesList) drawPinnedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
- builder.Write(message.Author.DisplayOrUsername(), baseStyle)
- builder.Write(" pinned a message.", baseStyle)
- }
- func (ml *messagesList) selectedMessage() (*discord.Message, error) {
- if len(ml.messages) == 0 {
- return nil, errors.New("no messages available")
- }
- cursor := ml.Cursor()
- if cursor == -1 || cursor >= len(ml.messages) {
- return nil, errors.New("no message is currently selected")
- }
- return &ml.messages[cursor], nil
- }
- func (ml *messagesList) onInputCapture(event *tcell.EventKey) *tcell.EventKey {
- switch event.Name() {
- case ml.cfg.Keybinds.MessagesList.ScrollUp:
- ml.ScrollUp()
- return nil
- case ml.cfg.Keybinds.MessagesList.ScrollDown:
- ml.ScrollDown()
- return nil
- case ml.cfg.Keybinds.MessagesList.ScrollTop:
- ml.ScrollToStart()
- return nil
- case ml.cfg.Keybinds.MessagesList.ScrollBottom:
- ml.ScrollToEnd()
- return nil
- case ml.cfg.Keybinds.MessagesList.Cancel:
- ml.clearSelection()
- return nil
- case ml.cfg.Keybinds.MessagesList.SelectUp, ml.cfg.Keybinds.MessagesList.SelectDown, ml.cfg.Keybinds.MessagesList.SelectTop, ml.cfg.Keybinds.MessagesList.SelectBottom, ml.cfg.Keybinds.MessagesList.SelectReply:
- ml._select(event.Name())
- return nil
- case ml.cfg.Keybinds.MessagesList.YankID:
- ml.yankID()
- return nil
- case ml.cfg.Keybinds.MessagesList.YankContent:
- ml.yankContent()
- return nil
- case ml.cfg.Keybinds.MessagesList.YankURL:
- ml.yankURL()
- return nil
- case ml.cfg.Keybinds.MessagesList.Open:
- ml.open()
- return nil
- case ml.cfg.Keybinds.MessagesList.Reply:
- ml.reply(false)
- return nil
- case ml.cfg.Keybinds.MessagesList.ReplyMention:
- ml.reply(true)
- return nil
- case ml.cfg.Keybinds.MessagesList.Edit:
- ml.edit()
- return nil
- case ml.cfg.Keybinds.MessagesList.Delete:
- ml.delete()
- return nil
- case ml.cfg.Keybinds.MessagesList.DeleteConfirm:
- ml.confirmDelete()
- return nil
- }
- return event
- }
- func (ml *messagesList) _select(name string) {
- messages := ml.messages
- if len(messages) == 0 {
- return
- }
- cursor := ml.Cursor()
- switch name {
- case ml.cfg.Keybinds.MessagesList.SelectUp:
- switch {
- case cursor == -1:
- cursor = len(messages) - 1
- case cursor > 0:
- cursor--
- case cursor == 0:
- selectedChannel := ml.chatView.SelectedChannel()
- if selectedChannel == nil {
- return
- }
- channelID := selectedChannel.ID
- before := ml.messages[0].ID
- limit := uint(ml.cfg.MessagesLimit)
- messages, err := ml.chatView.state.MessagesBefore(channelID, before, limit)
- if err != nil {
- slog.Error("failed to fetch older messages", "err", err)
- return
- }
- if len(messages) == 0 {
- return
- }
- if guildID := selectedChannel.GuildID; guildID.IsValid() {
- ml.requestGuildMembers(guildID, messages)
- }
- older := slices.Clone(messages)
- slices.Reverse(older)
- // Defensive invalidation if Discord returns overlapping windows.
- for _, message := range older {
- delete(ml.itemByID, message.ID)
- }
- ml.messages = slices.Concat(older, ml.messages)
- cursor = len(messages) - 1
- }
- case ml.cfg.Keybinds.MessagesList.SelectDown:
- switch {
- case cursor == -1:
- cursor = len(messages) - 1
- case cursor < len(messages)-1:
- cursor++
- }
- case ml.cfg.Keybinds.MessagesList.SelectTop:
- cursor = 0
- case ml.cfg.Keybinds.MessagesList.SelectBottom:
- cursor = len(messages) - 1
- case ml.cfg.Keybinds.MessagesList.SelectReply:
- if cursor == -1 || cursor >= len(messages) {
- return
- }
- if ref := messages[cursor].ReferencedMessage; ref != nil {
- refIdx := slices.IndexFunc(messages, func(m discord.Message) bool {
- return m.ID == ref.ID
- })
- if refIdx != -1 {
- cursor = refIdx
- }
- }
- }
- ml.SetCursor(cursor)
- }
- func (ml *messagesList) yankID() {
- msg, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- go clipboard.Write(clipboard.FmtText, []byte(msg.ID.String()))
- }
- func (ml *messagesList) yankContent() {
- msg, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- go clipboard.Write(clipboard.FmtText, []byte(msg.Content))
- }
- func (ml *messagesList) yankURL() {
- msg, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- go clipboard.Write(clipboard.FmtText, []byte(msg.URL()))
- }
- func (ml *messagesList) open() {
- msg, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- var urls []string
- if msg.Content != "" {
- urls = extractURLs(msg.Content)
- }
- if len(urls) == 0 && len(msg.Attachments) == 0 {
- return
- }
- if len(urls)+len(msg.Attachments) == 1 {
- if len(urls) == 1 {
- go ml.openURL(urls[0])
- } else {
- attachment := msg.Attachments[0]
- if strings.HasPrefix(attachment.ContentType, "image/") {
- go ml.openAttachment(msg.Attachments[0])
- } else {
- go ml.openURL(attachment.URL)
- }
- }
- } else {
- ml.showAttachmentsList(urls, msg.Attachments)
- }
- }
- func extractURLs(content string) []string {
- src := []byte(content)
- node := parser.NewParser(
- parser.WithBlockParsers(discordmd.BlockParsers()...),
- parser.WithInlineParsers(discordmd.InlineParserWithLink()...),
- ).Parse(text.NewReader(src))
- var urls []string
- ast.Walk(node, func(n ast.Node, entering bool) (ast.WalkStatus, error) {
- if entering {
- switch n := n.(type) {
- case *ast.AutoLink:
- urls = append(urls, string(n.URL(src)))
- case *ast.Link:
- urls = append(urls, string(n.Destination))
- }
- }
- return ast.WalkContinue, nil
- })
- return urls
- }
- func (ml *messagesList) showAttachmentsList(urls []string, attachments []discord.Attachment) {
- type attachmentAction struct {
- label string
- shortcut rune
- open func()
- }
- closeList := func() {
- ml.chatView.RemoveLayer(attachmentsListLayerName)
- ml.chatView.app.SetFocus(ml)
- }
- var actions []attachmentAction
- for i, a := range attachments {
- attachment := a
- action := func() {
- if strings.HasPrefix(attachment.ContentType, "image/") {
- go ml.openAttachment(attachment)
- } else {
- go ml.openURL(attachment.URL)
- }
- }
- actions = append(actions, attachmentAction{
- label: attachment.Filename,
- shortcut: rune('a' + i),
- open: action,
- })
- }
- for i, u := range urls {
- url := u
- actions = append(actions, attachmentAction{
- label: url,
- shortcut: rune('1' + i),
- open: func() { go ml.openURL(url) },
- })
- }
- normalItems := make([]*tview.TextView, len(actions))
- selectedItems := make([]*tview.TextView, len(actions))
- for i, action := range actions {
- normalItems[i] = tview.NewTextView().
- SetScrollable(false).
- SetWrap(false).
- SetWordWrap(false).
- SetLines([]tview.Line{{{Text: action.label, Style: tcell.StyleDefault}}})
- selectedItems[i] = tview.NewTextView().
- SetScrollable(false).
- SetWrap(false).
- SetWordWrap(false).
- SetLines([]tview.Line{{{Text: action.label, Style: tcell.StyleDefault.Reverse(true)}}})
- }
- list := tview.NewList().
- SetSnapToItems(true).
- SetBuilder(func(index int, cursor int) tview.ListItem {
- if index < 0 || index >= len(actions) {
- return nil
- }
- if index == cursor {
- return selectedItems[index]
- }
- return normalItems[index]
- })
- list.Box = ui.ConfigureBox(list.Box, &ml.cfg.Theme)
- if len(actions) > 0 {
- list.SetCursor(0)
- }
- list.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
- switch event.Name() {
- case ml.cfg.Keybinds.MessagesList.SelectUp:
- return tcell.NewEventKey(tcell.KeyUp, "", tcell.ModNone)
- case ml.cfg.Keybinds.MessagesList.SelectDown:
- return tcell.NewEventKey(tcell.KeyDown, "", tcell.ModNone)
- case ml.cfg.Keybinds.MessagesList.SelectTop:
- return tcell.NewEventKey(tcell.KeyHome, "", tcell.ModNone)
- case ml.cfg.Keybinds.MessagesList.SelectBottom:
- return tcell.NewEventKey(tcell.KeyEnd, "", tcell.ModNone)
- case ml.cfg.Keybinds.MessagesList.Cancel:
- closeList()
- return nil
- }
- if event.Key() == tcell.KeyEnter || event.Key() == tcell.KeyRune && event.Str() == " " {
- index := list.Cursor()
- if index >= 0 && index < len(actions) {
- actions[index].open()
- closeList()
- }
- return nil
- }
- if event.Key() == tcell.KeyRune {
- key := event.Str()
- if key == "" {
- return event
- }
- ch := []rune(key)[0]
- for index, action := range actions {
- if action.shortcut == ch {
- list.SetCursor(index)
- actions[index].open()
- closeList()
- return nil
- }
- }
- }
- return event
- })
- ml.chatView.
- AddLayer(
- ui.Centered(list, 0, 0),
- layers.WithName(attachmentsListLayerName),
- layers.WithResize(true),
- layers.WithVisible(true),
- layers.WithOverlay(),
- ).
- SendToFront(attachmentsListLayerName)
- }
- func (ml *messagesList) openAttachment(attachment discord.Attachment) {
- resp, err := http.Get(attachment.URL)
- if err != nil {
- slog.Error("failed to fetch the attachment", "err", err, "url", attachment.URL)
- return
- }
- defer resp.Body.Close()
- path := filepath.Join(consts.CacheDir(), "attachments")
- if err := os.MkdirAll(path, os.ModePerm); err != nil {
- slog.Error("failed to create attachments dir", "err", err, "path", path)
- return
- }
- path = filepath.Join(path, attachment.Filename)
- file, err := os.Create(path)
- if err != nil {
- slog.Error("failed to create attachment file", "err", err, "path", path)
- return
- }
- defer file.Close()
- if _, err := io.Copy(file, resp.Body); err != nil {
- slog.Error("failed to copy attachment to file", "err", err)
- return
- }
- if err := open.Start(path); err != nil {
- slog.Error("failed to open attachment file", "err", err, "path", path)
- return
- }
- }
- func (ml *messagesList) openURL(url string) {
- if err := open.Start(url); err != nil {
- slog.Error("failed to open URL", "err", err, "url", url)
- }
- }
- func (ml *messagesList) reply(mention bool) {
- message, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- name := message.Author.DisplayOrUsername()
- if message.GuildID.IsValid() {
- member, err := ml.chatView.state.Cabinet.Member(message.GuildID, message.Author.ID)
- if err != nil {
- slog.Error("failed to get member from state", "guild_id", message.GuildID, "member_id", message.Author.ID, "err", err)
- } else {
- if member.Nick != "" {
- name = member.Nick
- }
- }
- }
- data := ml.chatView.messageInput.sendMessageData
- data.Reference = &discord.MessageReference{MessageID: message.ID}
- data.AllowedMentions = &api.AllowedMentions{RepliedUser: option.False}
- title := "Replying to "
- if mention {
- data.AllowedMentions.RepliedUser = option.True
- title = "[@] " + title
- }
- ml.chatView.messageInput.sendMessageData = data
- ml.chatView.messageInput.SetTitle(title + name)
- ml.chatView.app.SetFocus(ml.chatView.messageInput)
- }
- func (ml *messagesList) edit() {
- message, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- me, _ := ml.chatView.state.Cabinet.Me()
- if message.Author.ID != me.ID {
- slog.Error("failed to edit message; not the author", "channel_id", message.ChannelID, "message_id", message.ID)
- return
- }
- ml.chatView.messageInput.SetTitle("Editing")
- ml.chatView.messageInput.edit = true
- ml.chatView.messageInput.SetText(message.Content, true)
- ml.chatView.app.SetFocus(ml.chatView.messageInput)
- }
- func (ml *messagesList) confirmDelete() {
- onChoice := func(choice string) {
- if choice == "Yes" {
- ml.delete()
- }
- }
- ml.chatView.showConfirmModal(
- "Are you sure you want to delete this message?",
- []string{"Yes", "No"},
- onChoice,
- )
- }
- func (ml *messagesList) delete() {
- msg, err := ml.selectedMessage()
- if err != nil {
- slog.Error("failed to get selected message", "err", err)
- return
- }
- if msg.GuildID.IsValid() {
- me, _ := ml.chatView.state.Cabinet.Me()
- if msg.Author.ID != me.ID && !ml.chatView.state.HasPermissions(msg.ChannelID, discord.PermissionManageMessages) {
- slog.Error("failed to delete message; missing relevant permissions", "channel_id", msg.ChannelID, "message_id", msg.ID)
- return
- }
- }
- selected := ml.chatView.SelectedChannel()
- if selected == nil {
- return
- }
- if err := ml.chatView.state.DeleteMessage(selected.ID, msg.ID, ""); err != nil {
- slog.Error("failed to delete message", "channel_id", selected.ID, "message_id", msg.ID, "err", err)
- return
- }
- if err := ml.chatView.state.MessageRemove(selected.ID, msg.ID); err != nil {
- slog.Error("failed to delete message", "channel_id", selected.ID, "message_id", msg.ID, "err", err)
- return
- }
- }
- func (ml *messagesList) requestGuildMembers(guildID discord.GuildID, messages []discord.Message) {
- usersToFetch := make([]discord.UserID, 0, len(messages))
- seen := make(map[discord.UserID]struct{}, len(messages))
- for _, message := range messages {
- // Do not fetch member for a webhook message.
- if message.WebhookID.IsValid() {
- continue
- }
- if member, _ := ml.chatView.state.Cabinet.Member(guildID, message.Author.ID); member == nil {
- userID := message.Author.ID
- if _, ok := seen[userID]; !ok {
- seen[userID] = struct{}{}
- usersToFetch = append(usersToFetch, userID)
- }
- }
- }
- if len(usersToFetch) > 0 {
- err := ml.chatView.state.SendGateway(context.TODO(), &gateway.RequestGuildMembersCommand{
- GuildIDs: []discord.GuildID{guildID},
- UserIDs: usersToFetch,
- })
- if err != nil {
- slog.Error("failed to request guild members", "guild_id", guildID, "err", err)
- return
- }
- ml.setFetchingChunk(true, 0)
- ml.waitForChunkEvent()
- }
- }
- func (ml *messagesList) setFetchingChunk(value bool, count uint) {
- ml.fetchingMembers.mu.Lock()
- defer ml.fetchingMembers.mu.Unlock()
- if ml.fetchingMembers.value == value {
- return
- }
- ml.fetchingMembers.value = value
- if value {
- ml.fetchingMembers.done = make(chan struct{})
- } else {
- ml.fetchingMembers.count = count
- close(ml.fetchingMembers.done)
- }
- }
- func (ml *messagesList) waitForChunkEvent() uint {
- ml.fetchingMembers.mu.Lock()
- if !ml.fetchingMembers.value {
- ml.fetchingMembers.mu.Unlock()
- return 0
- }
- ml.fetchingMembers.mu.Unlock()
- <-ml.fetchingMembers.done
- return ml.fetchingMembers.count
- }
|