messages_list.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867
  1. package chat
  2. import (
  3. "context"
  4. "errors"
  5. "io"
  6. "log/slog"
  7. "net/http"
  8. "os"
  9. "path/filepath"
  10. "slices"
  11. "strings"
  12. "sync"
  13. "time"
  14. "github.com/ayn2op/tview/layers"
  15. "github.com/ayn2op/discordo/internal/clipboard"
  16. "github.com/ayn2op/discordo/internal/config"
  17. "github.com/ayn2op/discordo/internal/consts"
  18. "github.com/ayn2op/discordo/internal/markdown"
  19. "github.com/ayn2op/discordo/internal/ui"
  20. "github.com/ayn2op/tview"
  21. "github.com/diamondburned/arikawa/v3/api"
  22. "github.com/diamondburned/arikawa/v3/discord"
  23. "github.com/diamondburned/arikawa/v3/gateway"
  24. "github.com/diamondburned/arikawa/v3/state"
  25. "github.com/diamondburned/arikawa/v3/utils/json/option"
  26. "github.com/diamondburned/ningen/v3/discordmd"
  27. "github.com/gdamore/tcell/v3"
  28. "github.com/gdamore/tcell/v3/color"
  29. "github.com/skratchdot/open-golang/open"
  30. "github.com/yuin/goldmark/ast"
  31. "github.com/yuin/goldmark/parser"
  32. "github.com/yuin/goldmark/text"
  33. )
  34. type messagesList struct {
  35. *tview.List
  36. cfg *config.Config
  37. chatView *View
  38. messages []discord.Message
  39. renderer *markdown.Renderer
  40. // itemByID caches unselected message TextViews.
  41. itemByID map[discord.MessageID]*tview.TextView
  42. fetchingMembers struct {
  43. mu sync.Mutex
  44. value bool
  45. count uint
  46. done chan struct{}
  47. }
  48. }
  49. func newMessagesList(cfg *config.Config, chatView *View) *messagesList {
  50. ml := &messagesList{
  51. List: tview.NewList(),
  52. cfg: cfg,
  53. chatView: chatView,
  54. renderer: markdown.NewRenderer(cfg),
  55. itemByID: make(map[discord.MessageID]*tview.TextView),
  56. }
  57. ml.Box = ui.ConfigureBox(ml.Box, &cfg.Theme)
  58. ml.SetTitle("Messages")
  59. ml.SetBuilder(ml.buildItem)
  60. ml.SetTrackEnd(true)
  61. ml.SetInputCapture(ml.onInputCapture)
  62. return ml
  63. }
  64. func (ml *messagesList) reset() {
  65. ml.messages = nil
  66. clear(ml.itemByID)
  67. ml.
  68. Clear().
  69. SetBuilder(ml.buildItem).
  70. SetTitle("")
  71. }
  72. func (ml *messagesList) setTitle(channel discord.Channel) {
  73. title := ui.ChannelToString(channel, ml.cfg.Icons)
  74. if topic := channel.Topic; topic != "" {
  75. title += " - " + topic
  76. }
  77. ml.SetTitle(title)
  78. }
  79. func (ml *messagesList) setMessages(messages []discord.Message) {
  80. ml.messages = slices.Clone(messages)
  81. slices.Reverse(ml.messages)
  82. // New channel payload / refetch: replace the cache wholesale to keep it in
  83. // lockstep with the current message slice.
  84. clear(ml.itemByID)
  85. ml.MarkDirty()
  86. }
  87. func (ml *messagesList) addMessage(message discord.Message) {
  88. ml.messages = append(ml.messages, message)
  89. // Defensive invalidation for ID reuse/edits delivered out-of-order.
  90. delete(ml.itemByID, message.ID)
  91. ml.MarkDirty()
  92. }
  93. func (ml *messagesList) setMessage(index int, message discord.Message) {
  94. if index < 0 || index >= len(ml.messages) {
  95. return
  96. }
  97. ml.messages[index] = message
  98. delete(ml.itemByID, message.ID)
  99. ml.MarkDirty()
  100. }
  101. func (ml *messagesList) deleteMessage(index int) {
  102. if index < 0 || index >= len(ml.messages) {
  103. return
  104. }
  105. delete(ml.itemByID, ml.messages[index].ID)
  106. ml.messages = append(ml.messages[:index], ml.messages[index+1:]...)
  107. ml.MarkDirty()
  108. }
  109. func (ml *messagesList) clearSelection() {
  110. ml.SetCursor(-1)
  111. }
  112. func (ml *messagesList) buildItem(index int, cursor int) tview.ListItem {
  113. if index < 0 || index >= len(ml.messages) {
  114. return nil
  115. }
  116. message := ml.messages[index]
  117. if index == cursor {
  118. return tview.NewTextView().
  119. SetWrap(true).
  120. SetWordWrap(true).
  121. SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.SelectedMessageStyle.Style))
  122. }
  123. item, ok := ml.itemByID[message.ID]
  124. if !ok {
  125. item = tview.NewTextView().
  126. SetWrap(true).
  127. SetWordWrap(true).
  128. SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.MessageStyle.Style))
  129. ml.itemByID[message.ID] = item
  130. }
  131. return item
  132. }
  133. func (ml *messagesList) renderMessage(message discord.Message, baseStyle tcell.Style) []tview.Line {
  134. builder := tview.NewLineBuilder()
  135. ml.writeMessage(builder, message, baseStyle)
  136. return builder.Finish()
  137. }
  138. func (ml *messagesList) writeMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  139. if ml.cfg.HideBlockedUsers {
  140. isBlocked := ml.chatView.state.UserIsBlocked(message.Author.ID)
  141. if isBlocked {
  142. builder.Write("Blocked message", baseStyle.Foreground(color.Red).Bold(true))
  143. return
  144. }
  145. }
  146. switch message.Type {
  147. case discord.DefaultMessage:
  148. if message.Reference != nil && message.Reference.Type == discord.MessageReferenceTypeForward {
  149. ml.drawForwardedMessage(builder, message, baseStyle)
  150. } else {
  151. ml.drawDefaultMessage(builder, message, baseStyle)
  152. }
  153. case discord.GuildMemberJoinMessage:
  154. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  155. ml.drawAuthor(builder, message, baseStyle)
  156. builder.Write("joined the server.", baseStyle)
  157. case discord.InlinedReplyMessage:
  158. ml.drawReplyMessage(builder, message, baseStyle)
  159. case discord.ChannelPinnedMessage:
  160. ml.drawPinnedMessage(builder, message, baseStyle)
  161. default:
  162. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  163. ml.drawAuthor(builder, message, baseStyle)
  164. }
  165. }
  166. func (ml *messagesList) formatTimestamp(ts discord.Timestamp) string {
  167. return ts.Time().In(time.Local).Format(ml.cfg.Timestamps.Format)
  168. }
  169. func (ml *messagesList) drawTimestamps(builder *tview.LineBuilder, ts discord.Timestamp, baseStyle tcell.Style) {
  170. dimStyle := baseStyle.Dim(true)
  171. builder.Write(ml.formatTimestamp(ts)+" ", dimStyle)
  172. }
  173. func (ml *messagesList) drawAuthor(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  174. name := message.Author.DisplayOrUsername()
  175. foreground := tcell.ColorDefault
  176. // Webhooks do not have nicknames or roles.
  177. if message.GuildID.IsValid() && !message.WebhookID.IsValid() {
  178. member, err := ml.chatView.state.Cabinet.Member(message.GuildID, message.Author.ID)
  179. if err != nil {
  180. slog.Error("failed to get member from state", "guild_id", message.GuildID, "member_id", message.Author.ID, "err", err)
  181. } else {
  182. if member.Nick != "" {
  183. name = member.Nick
  184. }
  185. color, ok := state.MemberColor(member, func(id discord.RoleID) *discord.Role {
  186. r, _ := ml.chatView.state.Cabinet.Role(message.GuildID, id)
  187. return r
  188. })
  189. if ok {
  190. foreground = tcell.NewHexColor(int32(color))
  191. }
  192. }
  193. }
  194. style := baseStyle.Foreground(foreground).Bold(true)
  195. builder.Write(name+" ", style)
  196. }
  197. func (ml *messagesList) drawContent(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  198. c := []byte(message.Content)
  199. if ml.chatView.cfg.Markdown.Enabled {
  200. root := discordmd.ParseWithMessage(c, *ml.chatView.state.Cabinet, &message, false)
  201. lines := ml.renderer.RenderLines(c, root, baseStyle)
  202. if builder.HasCurrentLine() {
  203. startsWithCodeBlock := false
  204. if first := root.FirstChild(); first != nil {
  205. _, startsWithCodeBlock = first.(*ast.FencedCodeBlock)
  206. }
  207. if startsWithCodeBlock {
  208. // Keep code blocks visually separate from "timestamp + author".
  209. builder.NewLine()
  210. for len(lines) > 0 && len(lines[0]) == 0 {
  211. lines = lines[1:]
  212. }
  213. } else {
  214. for len(lines) > 1 && len(lines[0]) == 0 {
  215. lines = lines[1:]
  216. }
  217. }
  218. }
  219. builder.AppendLines(lines)
  220. } else {
  221. builder.Write(message.Content, baseStyle)
  222. }
  223. }
  224. func (ml *messagesList) drawSnapshotContent(builder *tview.LineBuilder, message discord.MessageSnapshotMessage, baseStyle tcell.Style) {
  225. c := []byte(message.Content)
  226. // discordmd doesn't support MessageSnapshotMessage, so we just use write it as is. todo?
  227. builder.Write(string(c), baseStyle)
  228. }
  229. func (ml *messagesList) drawDefaultMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  230. if ml.cfg.Timestamps.Enabled {
  231. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  232. }
  233. ml.drawAuthor(builder, message, baseStyle)
  234. ml.drawContent(builder, message, baseStyle)
  235. if message.EditedTimestamp.IsValid() {
  236. dimStyle := baseStyle.Dim(true)
  237. builder.Write(" (edited)", dimStyle)
  238. }
  239. attachmentStyle := ui.MergeStyle(baseStyle, ml.cfg.Theme.MessagesList.AttachmentStyle.Style)
  240. for _, a := range message.Attachments {
  241. builder.NewLine()
  242. if ml.cfg.ShowAttachmentLinks {
  243. builder.Write(a.Filename+":\n"+a.URL, attachmentStyle)
  244. } else {
  245. builder.Write(a.Filename, attachmentStyle)
  246. }
  247. }
  248. }
  249. func (ml *messagesList) drawForwardedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  250. dimStyle := baseStyle.Dim(true)
  251. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  252. ml.drawAuthor(builder, message, baseStyle)
  253. builder.Write(ml.cfg.Theme.MessagesList.ForwardedIndicator+" ", dimStyle)
  254. ml.drawSnapshotContent(builder, message.MessageSnapshots[0].Message, baseStyle)
  255. builder.Write(" ("+ml.formatTimestamp(message.MessageSnapshots[0].Message.Timestamp)+") ", dimStyle)
  256. }
  257. func (ml *messagesList) drawReplyMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  258. dimStyle := baseStyle.Dim(true)
  259. // indicator
  260. builder.Write(ml.cfg.Theme.MessagesList.ReplyIndicator+" ", dimStyle)
  261. if m := message.ReferencedMessage; m != nil {
  262. m.GuildID = message.GuildID
  263. ml.drawAuthor(builder, *m, dimStyle)
  264. ml.drawContent(builder, *m, dimStyle)
  265. } else {
  266. builder.Write("Original message was deleted", dimStyle)
  267. }
  268. builder.NewLine()
  269. // main
  270. ml.drawDefaultMessage(builder, message, baseStyle)
  271. }
  272. func (ml *messagesList) drawPinnedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  273. builder.Write(message.Author.DisplayOrUsername(), baseStyle)
  274. builder.Write(" pinned a message.", baseStyle)
  275. }
  276. func (ml *messagesList) selectedMessage() (*discord.Message, error) {
  277. if len(ml.messages) == 0 {
  278. return nil, errors.New("no messages available")
  279. }
  280. cursor := ml.Cursor()
  281. if cursor == -1 || cursor >= len(ml.messages) {
  282. return nil, errors.New("no message is currently selected")
  283. }
  284. return &ml.messages[cursor], nil
  285. }
  286. func (ml *messagesList) onInputCapture(event *tcell.EventKey) *tcell.EventKey {
  287. switch event.Name() {
  288. case ml.cfg.Keybinds.MessagesList.ScrollUp:
  289. ml.ScrollUp()
  290. return nil
  291. case ml.cfg.Keybinds.MessagesList.ScrollDown:
  292. ml.ScrollDown()
  293. return nil
  294. case ml.cfg.Keybinds.MessagesList.ScrollTop:
  295. ml.ScrollToStart()
  296. return nil
  297. case ml.cfg.Keybinds.MessagesList.ScrollBottom:
  298. ml.ScrollToEnd()
  299. return nil
  300. case ml.cfg.Keybinds.MessagesList.Cancel:
  301. ml.clearSelection()
  302. return nil
  303. 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:
  304. ml._select(event.Name())
  305. return nil
  306. case ml.cfg.Keybinds.MessagesList.YankID:
  307. ml.yankID()
  308. return nil
  309. case ml.cfg.Keybinds.MessagesList.YankContent:
  310. ml.yankContent()
  311. return nil
  312. case ml.cfg.Keybinds.MessagesList.YankURL:
  313. ml.yankURL()
  314. return nil
  315. case ml.cfg.Keybinds.MessagesList.Open:
  316. ml.open()
  317. return nil
  318. case ml.cfg.Keybinds.MessagesList.Reply:
  319. ml.reply(false)
  320. return nil
  321. case ml.cfg.Keybinds.MessagesList.ReplyMention:
  322. ml.reply(true)
  323. return nil
  324. case ml.cfg.Keybinds.MessagesList.Edit:
  325. ml.edit()
  326. return nil
  327. case ml.cfg.Keybinds.MessagesList.Delete:
  328. ml.delete()
  329. return nil
  330. case ml.cfg.Keybinds.MessagesList.DeleteConfirm:
  331. ml.confirmDelete()
  332. return nil
  333. }
  334. return event
  335. }
  336. func (ml *messagesList) _select(name string) {
  337. messages := ml.messages
  338. if len(messages) == 0 {
  339. return
  340. }
  341. cursor := ml.Cursor()
  342. switch name {
  343. case ml.cfg.Keybinds.MessagesList.SelectUp:
  344. switch {
  345. case cursor == -1:
  346. cursor = len(messages) - 1
  347. case cursor > 0:
  348. cursor--
  349. case cursor == 0:
  350. selectedChannel := ml.chatView.SelectedChannel()
  351. if selectedChannel == nil {
  352. return
  353. }
  354. channelID := selectedChannel.ID
  355. before := ml.messages[0].ID
  356. limit := uint(ml.cfg.MessagesLimit)
  357. messages, err := ml.chatView.state.MessagesBefore(channelID, before, limit)
  358. if err != nil {
  359. slog.Error("failed to fetch older messages", "err", err)
  360. return
  361. }
  362. if len(messages) == 0 {
  363. return
  364. }
  365. if guildID := selectedChannel.GuildID; guildID.IsValid() {
  366. ml.requestGuildMembers(guildID, messages)
  367. }
  368. older := slices.Clone(messages)
  369. slices.Reverse(older)
  370. // Defensive invalidation if Discord returns overlapping windows.
  371. for _, message := range older {
  372. delete(ml.itemByID, message.ID)
  373. }
  374. ml.messages = slices.Concat(older, ml.messages)
  375. cursor = len(messages) - 1
  376. }
  377. case ml.cfg.Keybinds.MessagesList.SelectDown:
  378. switch {
  379. case cursor == -1:
  380. cursor = len(messages) - 1
  381. case cursor < len(messages)-1:
  382. cursor++
  383. }
  384. case ml.cfg.Keybinds.MessagesList.SelectTop:
  385. cursor = 0
  386. case ml.cfg.Keybinds.MessagesList.SelectBottom:
  387. cursor = len(messages) - 1
  388. case ml.cfg.Keybinds.MessagesList.SelectReply:
  389. if cursor == -1 || cursor >= len(messages) {
  390. return
  391. }
  392. if ref := messages[cursor].ReferencedMessage; ref != nil {
  393. refIdx := slices.IndexFunc(messages, func(m discord.Message) bool {
  394. return m.ID == ref.ID
  395. })
  396. if refIdx != -1 {
  397. cursor = refIdx
  398. }
  399. }
  400. }
  401. ml.SetCursor(cursor)
  402. }
  403. func (ml *messagesList) yankID() {
  404. msg, err := ml.selectedMessage()
  405. if err != nil {
  406. slog.Error("failed to get selected message", "err", err)
  407. return
  408. }
  409. go clipboard.Write(clipboard.FmtText, []byte(msg.ID.String()))
  410. }
  411. func (ml *messagesList) yankContent() {
  412. msg, err := ml.selectedMessage()
  413. if err != nil {
  414. slog.Error("failed to get selected message", "err", err)
  415. return
  416. }
  417. go clipboard.Write(clipboard.FmtText, []byte(msg.Content))
  418. }
  419. func (ml *messagesList) yankURL() {
  420. msg, err := ml.selectedMessage()
  421. if err != nil {
  422. slog.Error("failed to get selected message", "err", err)
  423. return
  424. }
  425. go clipboard.Write(clipboard.FmtText, []byte(msg.URL()))
  426. }
  427. func (ml *messagesList) open() {
  428. msg, err := ml.selectedMessage()
  429. if err != nil {
  430. slog.Error("failed to get selected message", "err", err)
  431. return
  432. }
  433. var urls []string
  434. if msg.Content != "" {
  435. urls = extractURLs(msg.Content)
  436. }
  437. if len(urls) == 0 && len(msg.Attachments) == 0 {
  438. return
  439. }
  440. if len(urls)+len(msg.Attachments) == 1 {
  441. if len(urls) == 1 {
  442. go ml.openURL(urls[0])
  443. } else {
  444. attachment := msg.Attachments[0]
  445. if strings.HasPrefix(attachment.ContentType, "image/") {
  446. go ml.openAttachment(msg.Attachments[0])
  447. } else {
  448. go ml.openURL(attachment.URL)
  449. }
  450. }
  451. } else {
  452. ml.showAttachmentsList(urls, msg.Attachments)
  453. }
  454. }
  455. func extractURLs(content string) []string {
  456. src := []byte(content)
  457. node := parser.NewParser(
  458. parser.WithBlockParsers(discordmd.BlockParsers()...),
  459. parser.WithInlineParsers(discordmd.InlineParserWithLink()...),
  460. ).Parse(text.NewReader(src))
  461. var urls []string
  462. ast.Walk(node, func(n ast.Node, entering bool) (ast.WalkStatus, error) {
  463. if entering {
  464. switch n := n.(type) {
  465. case *ast.AutoLink:
  466. urls = append(urls, string(n.URL(src)))
  467. case *ast.Link:
  468. urls = append(urls, string(n.Destination))
  469. }
  470. }
  471. return ast.WalkContinue, nil
  472. })
  473. return urls
  474. }
  475. func (ml *messagesList) showAttachmentsList(urls []string, attachments []discord.Attachment) {
  476. type attachmentAction struct {
  477. label string
  478. shortcut rune
  479. open func()
  480. }
  481. closeList := func() {
  482. ml.chatView.RemoveLayer(attachmentsListLayerName)
  483. ml.chatView.app.SetFocus(ml)
  484. }
  485. var actions []attachmentAction
  486. for i, a := range attachments {
  487. attachment := a
  488. action := func() {
  489. if strings.HasPrefix(attachment.ContentType, "image/") {
  490. go ml.openAttachment(attachment)
  491. } else {
  492. go ml.openURL(attachment.URL)
  493. }
  494. }
  495. actions = append(actions, attachmentAction{
  496. label: attachment.Filename,
  497. shortcut: rune('a' + i),
  498. open: action,
  499. })
  500. }
  501. for i, u := range urls {
  502. url := u
  503. actions = append(actions, attachmentAction{
  504. label: url,
  505. shortcut: rune('1' + i),
  506. open: func() { go ml.openURL(url) },
  507. })
  508. }
  509. normalItems := make([]*tview.TextView, len(actions))
  510. selectedItems := make([]*tview.TextView, len(actions))
  511. for i, action := range actions {
  512. normalItems[i] = tview.NewTextView().
  513. SetScrollable(false).
  514. SetWrap(false).
  515. SetWordWrap(false).
  516. SetLines([]tview.Line{{{Text: action.label, Style: tcell.StyleDefault}}})
  517. selectedItems[i] = tview.NewTextView().
  518. SetScrollable(false).
  519. SetWrap(false).
  520. SetWordWrap(false).
  521. SetLines([]tview.Line{{{Text: action.label, Style: tcell.StyleDefault.Reverse(true)}}})
  522. }
  523. list := tview.NewList().
  524. SetSnapToItems(true).
  525. SetBuilder(func(index int, cursor int) tview.ListItem {
  526. if index < 0 || index >= len(actions) {
  527. return nil
  528. }
  529. if index == cursor {
  530. return selectedItems[index]
  531. }
  532. return normalItems[index]
  533. })
  534. list.Box = ui.ConfigureBox(list.Box, &ml.cfg.Theme)
  535. if len(actions) > 0 {
  536. list.SetCursor(0)
  537. }
  538. list.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
  539. switch event.Name() {
  540. case ml.cfg.Keybinds.MessagesList.SelectUp:
  541. return tcell.NewEventKey(tcell.KeyUp, "", tcell.ModNone)
  542. case ml.cfg.Keybinds.MessagesList.SelectDown:
  543. return tcell.NewEventKey(tcell.KeyDown, "", tcell.ModNone)
  544. case ml.cfg.Keybinds.MessagesList.SelectTop:
  545. return tcell.NewEventKey(tcell.KeyHome, "", tcell.ModNone)
  546. case ml.cfg.Keybinds.MessagesList.SelectBottom:
  547. return tcell.NewEventKey(tcell.KeyEnd, "", tcell.ModNone)
  548. case ml.cfg.Keybinds.MessagesList.Cancel:
  549. closeList()
  550. return nil
  551. }
  552. if event.Key() == tcell.KeyEnter || event.Key() == tcell.KeyRune && event.Str() == " " {
  553. index := list.Cursor()
  554. if index >= 0 && index < len(actions) {
  555. actions[index].open()
  556. closeList()
  557. }
  558. return nil
  559. }
  560. if event.Key() == tcell.KeyRune {
  561. key := event.Str()
  562. if key == "" {
  563. return event
  564. }
  565. ch := []rune(key)[0]
  566. for index, action := range actions {
  567. if action.shortcut == ch {
  568. list.SetCursor(index)
  569. actions[index].open()
  570. closeList()
  571. return nil
  572. }
  573. }
  574. }
  575. return event
  576. })
  577. ml.chatView.
  578. AddLayer(
  579. ui.Centered(list, 0, 0),
  580. layers.WithName(attachmentsListLayerName),
  581. layers.WithResize(true),
  582. layers.WithVisible(true),
  583. layers.WithOverlay(),
  584. ).
  585. SendToFront(attachmentsListLayerName)
  586. }
  587. func (ml *messagesList) openAttachment(attachment discord.Attachment) {
  588. resp, err := http.Get(attachment.URL)
  589. if err != nil {
  590. slog.Error("failed to fetch the attachment", "err", err, "url", attachment.URL)
  591. return
  592. }
  593. defer resp.Body.Close()
  594. path := filepath.Join(consts.CacheDir(), "attachments")
  595. if err := os.MkdirAll(path, os.ModePerm); err != nil {
  596. slog.Error("failed to create attachments dir", "err", err, "path", path)
  597. return
  598. }
  599. path = filepath.Join(path, attachment.Filename)
  600. file, err := os.Create(path)
  601. if err != nil {
  602. slog.Error("failed to create attachment file", "err", err, "path", path)
  603. return
  604. }
  605. defer file.Close()
  606. if _, err := io.Copy(file, resp.Body); err != nil {
  607. slog.Error("failed to copy attachment to file", "err", err)
  608. return
  609. }
  610. if err := open.Start(path); err != nil {
  611. slog.Error("failed to open attachment file", "err", err, "path", path)
  612. return
  613. }
  614. }
  615. func (ml *messagesList) openURL(url string) {
  616. if err := open.Start(url); err != nil {
  617. slog.Error("failed to open URL", "err", err, "url", url)
  618. }
  619. }
  620. func (ml *messagesList) reply(mention bool) {
  621. message, err := ml.selectedMessage()
  622. if err != nil {
  623. slog.Error("failed to get selected message", "err", err)
  624. return
  625. }
  626. name := message.Author.DisplayOrUsername()
  627. if message.GuildID.IsValid() {
  628. member, err := ml.chatView.state.Cabinet.Member(message.GuildID, message.Author.ID)
  629. if err != nil {
  630. slog.Error("failed to get member from state", "guild_id", message.GuildID, "member_id", message.Author.ID, "err", err)
  631. } else {
  632. if member.Nick != "" {
  633. name = member.Nick
  634. }
  635. }
  636. }
  637. data := ml.chatView.messageInput.sendMessageData
  638. data.Reference = &discord.MessageReference{MessageID: message.ID}
  639. data.AllowedMentions = &api.AllowedMentions{RepliedUser: option.False}
  640. title := "Replying to "
  641. if mention {
  642. data.AllowedMentions.RepliedUser = option.True
  643. title = "[@] " + title
  644. }
  645. ml.chatView.messageInput.sendMessageData = data
  646. ml.chatView.messageInput.SetTitle(title + name)
  647. ml.chatView.app.SetFocus(ml.chatView.messageInput)
  648. }
  649. func (ml *messagesList) edit() {
  650. message, err := ml.selectedMessage()
  651. if err != nil {
  652. slog.Error("failed to get selected message", "err", err)
  653. return
  654. }
  655. me, _ := ml.chatView.state.Cabinet.Me()
  656. if message.Author.ID != me.ID {
  657. slog.Error("failed to edit message; not the author", "channel_id", message.ChannelID, "message_id", message.ID)
  658. return
  659. }
  660. ml.chatView.messageInput.SetTitle("Editing")
  661. ml.chatView.messageInput.edit = true
  662. ml.chatView.messageInput.SetText(message.Content, true)
  663. ml.chatView.app.SetFocus(ml.chatView.messageInput)
  664. }
  665. func (ml *messagesList) confirmDelete() {
  666. onChoice := func(choice string) {
  667. if choice == "Yes" {
  668. ml.delete()
  669. }
  670. }
  671. ml.chatView.showConfirmModal(
  672. "Are you sure you want to delete this message?",
  673. []string{"Yes", "No"},
  674. onChoice,
  675. )
  676. }
  677. func (ml *messagesList) delete() {
  678. msg, err := ml.selectedMessage()
  679. if err != nil {
  680. slog.Error("failed to get selected message", "err", err)
  681. return
  682. }
  683. if msg.GuildID.IsValid() {
  684. me, _ := ml.chatView.state.Cabinet.Me()
  685. if msg.Author.ID != me.ID && !ml.chatView.state.HasPermissions(msg.ChannelID, discord.PermissionManageMessages) {
  686. slog.Error("failed to delete message; missing relevant permissions", "channel_id", msg.ChannelID, "message_id", msg.ID)
  687. return
  688. }
  689. }
  690. selected := ml.chatView.SelectedChannel()
  691. if selected == nil {
  692. return
  693. }
  694. if err := ml.chatView.state.DeleteMessage(selected.ID, msg.ID, ""); err != nil {
  695. slog.Error("failed to delete message", "channel_id", selected.ID, "message_id", msg.ID, "err", err)
  696. return
  697. }
  698. if err := ml.chatView.state.MessageRemove(selected.ID, msg.ID); err != nil {
  699. slog.Error("failed to delete message", "channel_id", selected.ID, "message_id", msg.ID, "err", err)
  700. return
  701. }
  702. }
  703. func (ml *messagesList) requestGuildMembers(guildID discord.GuildID, messages []discord.Message) {
  704. usersToFetch := make([]discord.UserID, 0, len(messages))
  705. seen := make(map[discord.UserID]struct{}, len(messages))
  706. for _, message := range messages {
  707. // Do not fetch member for a webhook message.
  708. if message.WebhookID.IsValid() {
  709. continue
  710. }
  711. if member, _ := ml.chatView.state.Cabinet.Member(guildID, message.Author.ID); member == nil {
  712. userID := message.Author.ID
  713. if _, ok := seen[userID]; !ok {
  714. seen[userID] = struct{}{}
  715. usersToFetch = append(usersToFetch, userID)
  716. }
  717. }
  718. }
  719. if len(usersToFetch) > 0 {
  720. err := ml.chatView.state.SendGateway(context.TODO(), &gateway.RequestGuildMembersCommand{
  721. GuildIDs: []discord.GuildID{guildID},
  722. UserIDs: usersToFetch,
  723. })
  724. if err != nil {
  725. slog.Error("failed to request guild members", "guild_id", guildID, "err", err)
  726. return
  727. }
  728. ml.setFetchingChunk(true, 0)
  729. ml.waitForChunkEvent()
  730. }
  731. }
  732. func (ml *messagesList) setFetchingChunk(value bool, count uint) {
  733. ml.fetchingMembers.mu.Lock()
  734. defer ml.fetchingMembers.mu.Unlock()
  735. if ml.fetchingMembers.value == value {
  736. return
  737. }
  738. ml.fetchingMembers.value = value
  739. if value {
  740. ml.fetchingMembers.done = make(chan struct{})
  741. } else {
  742. ml.fetchingMembers.count = count
  743. close(ml.fetchingMembers.done)
  744. }
  745. }
  746. func (ml *messagesList) waitForChunkEvent() uint {
  747. ml.fetchingMembers.mu.Lock()
  748. if !ml.fetchingMembers.value {
  749. ml.fetchingMembers.mu.Unlock()
  750. return 0
  751. }
  752. ml.fetchingMembers.mu.Unlock()
  753. <-ml.fetchingMembers.done
  754. return ml.fetchingMembers.count
  755. }