messages_list.go 30 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081
  1. package chat
  2. import (
  3. "context"
  4. "errors"
  5. "log/slog"
  6. "slices"
  7. "strings"
  8. "sync"
  9. "time"
  10. "unicode/utf8"
  11. "github.com/ayn2op/discordo/internal/clipboard"
  12. "github.com/ayn2op/discordo/internal/config"
  13. "github.com/ayn2op/discordo/internal/markdown"
  14. "github.com/ayn2op/discordo/internal/ui"
  15. "github.com/ayn2op/tview"
  16. "github.com/ayn2op/tview/help"
  17. "github.com/ayn2op/tview/keybind"
  18. "github.com/ayn2op/tview/list"
  19. "github.com/diamondburned/arikawa/v3/api"
  20. "github.com/diamondburned/arikawa/v3/discord"
  21. "github.com/diamondburned/arikawa/v3/gateway"
  22. "github.com/diamondburned/arikawa/v3/state"
  23. "github.com/diamondburned/arikawa/v3/utils/json/option"
  24. "github.com/diamondburned/ningen/v3/discordmd"
  25. "github.com/gdamore/tcell/v3"
  26. "github.com/gdamore/tcell/v3/color"
  27. "github.com/yuin/goldmark/ast"
  28. )
  29. type messagesList struct {
  30. *list.Model
  31. cfg *config.Config
  32. chatView *Model
  33. messages []discord.Message
  34. // rows is the virtual list model rendered by tview (message rows +
  35. // date-separator rows). It is rebuilt lazily when rowsDirty is true.
  36. rows []messagesListRow
  37. rowsDirty bool
  38. renderer *markdown.Renderer
  39. // itemByID caches unselected message TextViews.
  40. itemByID map[discord.MessageID]*tview.TextView
  41. attachmentsPicker *attachmentsPicker
  42. fetchingMembers struct {
  43. mu sync.Mutex
  44. value bool
  45. count uint
  46. done chan struct{}
  47. }
  48. }
  49. var _ help.KeyMap = (*messagesList)(nil)
  50. type messagesListRowKind uint8
  51. const (
  52. messagesListRowMessage messagesListRowKind = iota
  53. messagesListRowSeparator
  54. )
  55. type messagesListRow struct {
  56. kind messagesListRowKind
  57. messageIndex int
  58. timestamp discord.Timestamp
  59. }
  60. func newMessagesList(cfg *config.Config, chatView *Model) *messagesList {
  61. ml := &messagesList{
  62. Model: list.NewModel(),
  63. cfg: cfg,
  64. chatView: chatView,
  65. renderer: markdown.NewRenderer(cfg),
  66. itemByID: make(map[discord.MessageID]*tview.TextView),
  67. }
  68. ml.attachmentsPicker = newAttachmentsPicker(cfg, chatView)
  69. ml.Box = ui.ConfigureBox(ml.Box, &cfg.Theme)
  70. ml.SetTitle("Messages")
  71. ml.SetBuilder(ml.buildItem)
  72. ml.SetChangedFunc(ml.onRowCursorChanged)
  73. ml.SetTrackEnd(true)
  74. ml.SetKeybinds(list.Keybinds{
  75. ScrollUp: cfg.Keybinds.MessagesList.ScrollUp.Keybind,
  76. ScrollDown: cfg.Keybinds.MessagesList.ScrollDown.Keybind,
  77. ScrollTop: cfg.Keybinds.MessagesList.ScrollTop.Keybind,
  78. ScrollBottom: cfg.Keybinds.MessagesList.ScrollBottom.Keybind,
  79. })
  80. ml.SetScrollBarVisibility(cfg.Theme.ScrollBar.Visibility.ScrollBarVisibility)
  81. ml.SetScrollBar(tview.NewScrollBar().
  82. SetTrackStyle(cfg.Theme.ScrollBar.TrackStyle.Style).
  83. SetThumbStyle(cfg.Theme.ScrollBar.ThumbStyle.Style).
  84. SetGlyphSet(cfg.Theme.ScrollBar.GlyphSet.GlyphSet))
  85. return ml
  86. }
  87. func (ml *messagesList) reset() {
  88. ml.messages = nil
  89. ml.rows = nil
  90. ml.rowsDirty = false
  91. clear(ml.itemByID)
  92. ml.
  93. Clear().
  94. SetBuilder(ml.buildItem).
  95. SetTitle("")
  96. }
  97. func (ml *messagesList) setTitle(channel discord.Channel) {
  98. title := ui.ChannelToString(channel, ml.cfg.Icons, ml.chatView.state)
  99. if topic := channel.Topic; topic != "" {
  100. title += " - " + topic
  101. }
  102. ml.SetTitle(title)
  103. }
  104. func (ml *messagesList) setMessages(messages []discord.Message) {
  105. ml.messages = slices.Clone(messages)
  106. slices.Reverse(ml.messages)
  107. ml.invalidateRows()
  108. // New channel payload / refetch: replace the cache wholesale to keep it in
  109. // lockstep with the current message slice.
  110. clear(ml.itemByID)
  111. }
  112. func (ml *messagesList) addMessage(message discord.Message) {
  113. ml.messages = append(ml.messages, message)
  114. ml.invalidateRows()
  115. // Defensive invalidation for ID reuse/edits delivered out-of-order.
  116. delete(ml.itemByID, message.ID)
  117. }
  118. func (ml *messagesList) setMessage(index int, message discord.Message) {
  119. if index < 0 || index >= len(ml.messages) {
  120. return
  121. }
  122. ml.messages[index] = message
  123. delete(ml.itemByID, message.ID)
  124. ml.invalidateRows()
  125. }
  126. func (ml *messagesList) deleteMessage(index int) {
  127. if index < 0 || index >= len(ml.messages) {
  128. return
  129. }
  130. delete(ml.itemByID, ml.messages[index].ID)
  131. ml.messages = append(ml.messages[:index], ml.messages[index+1:]...)
  132. ml.invalidateRows()
  133. }
  134. func (ml *messagesList) clearSelection() {
  135. ml.SetCursor(-1)
  136. }
  137. func (ml *messagesList) buildItem(index int, cursor int) list.Item {
  138. ml.ensureRows()
  139. if index < 0 || index >= len(ml.rows) {
  140. return nil
  141. }
  142. row := ml.rows[index]
  143. if row.kind == messagesListRowSeparator {
  144. return ml.buildSeparatorItem(row.timestamp)
  145. }
  146. message := ml.messages[row.messageIndex]
  147. if index == cursor {
  148. return tview.NewTextView().
  149. SetWrap(true).
  150. SetWordWrap(true).
  151. SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.SelectedMessageStyle.Style))
  152. }
  153. item, ok := ml.itemByID[message.ID]
  154. if !ok {
  155. item = tview.NewTextView().
  156. SetWrap(true).
  157. SetWordWrap(true).
  158. SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.MessageStyle.Style))
  159. ml.itemByID[message.ID] = item
  160. }
  161. return item
  162. }
  163. func (ml *messagesList) renderMessage(message discord.Message, baseStyle tcell.Style) []tview.Line {
  164. builder := tview.NewLineBuilder()
  165. ml.writeMessage(builder, message, baseStyle)
  166. return builder.Finish()
  167. }
  168. func (ml *messagesList) buildSeparatorItem(ts discord.Timestamp) *tview.TextView {
  169. builder := tview.NewLineBuilder()
  170. ml.drawDateSeparator(builder, ts, ml.cfg.Theme.MessagesList.MessageStyle.Style)
  171. return tview.NewTextView().
  172. SetScrollable(false).
  173. SetWrap(false).
  174. SetWordWrap(false).
  175. SetLines(builder.Finish())
  176. }
  177. func (ml *messagesList) drawDateSeparator(builder *tview.LineBuilder, ts discord.Timestamp, baseStyle tcell.Style) {
  178. date := ts.Time().In(time.Local).Format(ml.cfg.DateSeparator.Format)
  179. label := " " + date + " "
  180. fillChar := ml.cfg.DateSeparator.Character
  181. dimStyle := baseStyle.Dim(true)
  182. _, _, width, _ := ml.InnerRect()
  183. if width <= 0 {
  184. builder.Write(strings.Repeat(fillChar, 8)+label+strings.Repeat(fillChar, 8), dimStyle)
  185. return
  186. }
  187. labelWidth := utf8.RuneCountInString(label)
  188. if width <= labelWidth {
  189. builder.Write(date, dimStyle)
  190. return
  191. }
  192. fillWidth := width - labelWidth
  193. left := fillWidth / 2
  194. right := fillWidth - left
  195. builder.Write(strings.Repeat(fillChar, left)+label+strings.Repeat(fillChar, right), dimStyle)
  196. }
  197. func (ml *messagesList) rebuildRows() {
  198. rows := make([]messagesListRow, 0, len(ml.messages)*2)
  199. for i := range ml.messages {
  200. // Always show a date separator before the first message, and between messages on different days.
  201. if ml.cfg.DateSeparator.Enabled && (i == 0 || !sameLocalDate(ml.messages[i-1].Timestamp, ml.messages[i].Timestamp)) {
  202. rows = append(rows, messagesListRow{
  203. kind: messagesListRowSeparator,
  204. timestamp: ml.messages[i].Timestamp,
  205. })
  206. }
  207. rows = append(rows, messagesListRow{
  208. kind: messagesListRowMessage,
  209. messageIndex: i,
  210. })
  211. }
  212. ml.rows = rows
  213. ml.rowsDirty = false
  214. }
  215. func (ml *messagesList) invalidateRows() {
  216. ml.rowsDirty = true
  217. }
  218. // ensureRows lazily rebuilds list rows. This avoids repeated O(n) row rebuild
  219. // work when multiple message mutations happen close together.
  220. func (ml *messagesList) ensureRows() {
  221. if !ml.rowsDirty {
  222. return
  223. }
  224. ml.rebuildRows()
  225. }
  226. func sameLocalDate(a discord.Timestamp, b discord.Timestamp) bool {
  227. ta := a.Time().In(time.Local)
  228. tb := b.Time().In(time.Local)
  229. return ta.Year() == tb.Year() && ta.YearDay() == tb.YearDay()
  230. }
  231. // Cursor returns the selected message index, skipping separator rows.
  232. func (ml *messagesList) Cursor() int {
  233. ml.ensureRows()
  234. rowIndex := ml.Model.Cursor()
  235. if rowIndex < 0 || rowIndex >= len(ml.rows) {
  236. return -1
  237. }
  238. row := ml.rows[rowIndex]
  239. if row.kind != messagesListRowMessage {
  240. return -1
  241. }
  242. return row.messageIndex
  243. }
  244. // SetCursor selects a message index and maps it to the corresponding row.
  245. func (ml *messagesList) SetCursor(index int) {
  246. ml.Model.SetCursor(ml.messageToRowIndex(index))
  247. }
  248. func (ml *messagesList) messageToRowIndex(messageIndex int) int {
  249. ml.ensureRows()
  250. if messageIndex < 0 || messageIndex >= len(ml.messages) {
  251. return -1
  252. }
  253. for i, row := range ml.rows {
  254. if row.kind == messagesListRowMessage && row.messageIndex == messageIndex {
  255. return i
  256. }
  257. }
  258. return -1
  259. }
  260. func (ml *messagesList) onRowCursorChanged(rowIndex int) {
  261. ml.ensureRows()
  262. if rowIndex < 0 || rowIndex >= len(ml.rows) || ml.rows[rowIndex].kind == messagesListRowMessage {
  263. return
  264. }
  265. target := ml.nearestMessageRowIndex(rowIndex)
  266. ml.Model.SetCursor(target)
  267. }
  268. // nearestMessageRowIndex expects rowIndex to be within bounds.
  269. func (ml *messagesList) nearestMessageRowIndex(rowIndex int) int {
  270. for i := rowIndex - 1; i >= 0; i-- {
  271. if ml.rows[i].kind == messagesListRowMessage {
  272. return i
  273. }
  274. }
  275. for i := rowIndex + 1; i < len(ml.rows); i++ {
  276. if ml.rows[i].kind == messagesListRowMessage {
  277. return i
  278. }
  279. }
  280. return -1
  281. }
  282. func (ml *messagesList) writeMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  283. if ml.cfg.HideBlockedUsers {
  284. isBlocked := ml.chatView.state.UserIsBlocked(message.Author.ID)
  285. if isBlocked {
  286. builder.Write("Blocked message", baseStyle.Foreground(color.Red).Bold(true))
  287. return
  288. }
  289. }
  290. switch message.Type {
  291. case discord.DefaultMessage:
  292. if message.Reference != nil && message.Reference.Type == discord.MessageReferenceTypeForward {
  293. ml.drawForwardedMessage(builder, message, baseStyle)
  294. } else {
  295. ml.drawDefaultMessage(builder, message, baseStyle)
  296. }
  297. case discord.GuildMemberJoinMessage:
  298. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  299. ml.drawAuthor(builder, message, baseStyle)
  300. builder.Write("joined the server.", baseStyle)
  301. case discord.InlinedReplyMessage:
  302. ml.drawReplyMessage(builder, message, baseStyle)
  303. case discord.ChannelPinnedMessage:
  304. ml.drawPinnedMessage(builder, message, baseStyle)
  305. default:
  306. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  307. ml.drawAuthor(builder, message, baseStyle)
  308. }
  309. }
  310. func (ml *messagesList) formatTimestamp(ts discord.Timestamp) string {
  311. return ts.Time().In(time.Local).Format(ml.cfg.Timestamps.Format)
  312. }
  313. func (ml *messagesList) drawTimestamps(builder *tview.LineBuilder, ts discord.Timestamp, baseStyle tcell.Style) {
  314. dimStyle := baseStyle.Dim(true)
  315. builder.Write(ml.formatTimestamp(ts)+" ", dimStyle)
  316. }
  317. func (ml *messagesList) drawAuthor(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  318. name := message.Author.DisplayOrUsername()
  319. foreground := tcell.ColorDefault
  320. if member := ml.memberForMessage(message); member != nil {
  321. if member.Nick != "" {
  322. name = member.Nick
  323. }
  324. color, ok := state.MemberColor(member, func(id discord.RoleID) *discord.Role {
  325. r, _ := ml.chatView.state.Cabinet.Role(message.GuildID, id)
  326. return r
  327. })
  328. if ok {
  329. foreground = tcell.NewHexColor(int32(color))
  330. }
  331. }
  332. style := baseStyle.Foreground(foreground).Bold(true)
  333. builder.Write(name+" ", style)
  334. }
  335. func (ml *messagesList) memberForMessage(message discord.Message) *discord.Member {
  336. // Webhooks do not have nicknames or roles.
  337. if !message.GuildID.IsValid() || message.WebhookID.IsValid() {
  338. return nil
  339. }
  340. member, err := ml.chatView.state.Cabinet.Member(message.GuildID, message.Author.ID)
  341. if err != nil {
  342. slog.Error("failed to get member from state", "guild_id", message.GuildID, "member_id", message.Author.ID, "err", err)
  343. return nil
  344. }
  345. return member
  346. }
  347. func (ml *messagesList) drawContent(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  348. lines, root := ml.renderContentLines(message, baseStyle)
  349. if ml.chatView.cfg.Markdown.Enabled && builder.HasCurrentLine() {
  350. startsWithCodeBlock := false
  351. if root != nil {
  352. if first := root.FirstChild(); first != nil {
  353. _, startsWithCodeBlock = first.(*ast.FencedCodeBlock)
  354. }
  355. }
  356. if startsWithCodeBlock {
  357. // Keep code blocks visually separate from "timestamp + author".
  358. builder.NewLine()
  359. for len(lines) > 0 && len(lines[0]) == 0 {
  360. lines = lines[1:]
  361. }
  362. } else {
  363. for len(lines) > 1 && len(lines[0]) == 0 {
  364. lines = lines[1:]
  365. }
  366. }
  367. }
  368. builder.AppendLines(lines)
  369. }
  370. func (ml *messagesList) renderContentLines(message discord.Message, baseStyle tcell.Style) ([]tview.Line, ast.Node) {
  371. return ml.renderContentLinesWithMarkdown(message, baseStyle, false)
  372. }
  373. func (ml *messagesList) renderContentLinesWithMarkdown(message discord.Message, baseStyle tcell.Style, forceMarkdown bool) ([]tview.Line, ast.Node) {
  374. // Keep one rendering path for both normal messages and embed fragments so we preserve mention/link parsing behavior consistently across both.
  375. if forceMarkdown || ml.chatView.cfg.Markdown.Enabled {
  376. c := []byte(message.Content)
  377. root := discordmd.ParseWithMessage(c, *ml.chatView.state.Cabinet, &message, false)
  378. return ml.renderer.RenderLines(c, root, baseStyle), root
  379. }
  380. b := tview.NewLineBuilder()
  381. b.Write(message.Content, baseStyle)
  382. return b.Finish(), nil
  383. }
  384. func (ml *messagesList) drawSnapshotContent(builder *tview.LineBuilder, parent discord.Message, snapshot discord.MessageSnapshotMessage, baseStyle tcell.Style) {
  385. // Convert discord.MessageSnapshotMessage to discord.Message with common fields.
  386. message := discord.Message{
  387. Type: snapshot.Type,
  388. Content: snapshot.Content,
  389. Embeds: snapshot.Embeds,
  390. Attachments: snapshot.Attachments,
  391. Timestamp: snapshot.Timestamp,
  392. EditedTimestamp: snapshot.EditedTimestamp,
  393. Flags: snapshot.Flags,
  394. Mentions: snapshot.Mentions,
  395. MentionRoleIDs: snapshot.MentionRoleIDs,
  396. Stickers: snapshot.Stickers,
  397. Components: snapshot.Components,
  398. ChannelID: parent.ChannelID,
  399. GuildID: parent.GuildID,
  400. }
  401. ml.drawContent(builder, message, baseStyle)
  402. }
  403. func (ml *messagesList) drawDefaultMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  404. if ml.cfg.Timestamps.Enabled {
  405. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  406. }
  407. ml.drawAuthor(builder, message, baseStyle)
  408. ml.drawContent(builder, message, baseStyle)
  409. if message.EditedTimestamp.IsValid() {
  410. dimStyle := baseStyle.Dim(true)
  411. builder.Write(" (edited)", dimStyle)
  412. }
  413. ml.drawEmbeds(builder, message, baseStyle)
  414. attachmentStyle := ui.MergeStyle(baseStyle, ml.cfg.Theme.MessagesList.AttachmentStyle.Style)
  415. for _, a := range message.Attachments {
  416. builder.NewLine()
  417. if ml.cfg.ShowAttachmentLinks {
  418. builder.Write(a.Filename+":", attachmentStyle)
  419. builder.NewLine()
  420. builder.Write(a.URL, attachmentStyle.Url(a.URL))
  421. } else {
  422. builder.Write(a.Filename, attachmentStyle)
  423. }
  424. }
  425. }
  426. func (ml *messagesList) drawEmbeds(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  427. if len(message.Embeds) == 0 {
  428. return
  429. }
  430. contentListURLs := extractURLs(message.Content)
  431. contentURLs := make(map[string]struct{}, len(contentListURLs))
  432. for _, u := range contentListURLs {
  433. contentURLs[u] = struct{}{}
  434. }
  435. lineStyles := embedLineStyles(baseStyle, ml.cfg.Theme.MessagesList.Embeds)
  436. defaultBarStyle := baseStyle.Dim(true)
  437. prefixText := " ▎ "
  438. prefixWidth := tview.TaggedStringWidth(prefixText)
  439. _, _, innerWidth, _ := ml.InnerRect()
  440. // Wrap against the current list viewport. This keeps embed wrapping stable even when sidebars/panes are resized.
  441. wrapWidth := max(innerWidth-prefixWidth, 1)
  442. for _, embed := range message.Embeds {
  443. lines := embedLines(embed, contentURLs)
  444. if len(lines) == 0 {
  445. continue
  446. }
  447. embedContentLines := make([]tview.Line, 0, len(lines)*2)
  448. barStyle := defaultBarStyle
  449. if embed.Color != discord.NullColor && embed.Color != 0 {
  450. barStyle = barStyle.Foreground(tcell.NewHexColor(int32(embed.Color.Uint32())))
  451. }
  452. prefix := tview.NewSegment(prefixText, barStyle)
  453. builder.NewLine()
  454. for _, line := range lines {
  455. if strings.TrimSpace(line.Text) == "" {
  456. continue
  457. }
  458. msg := message
  459. msg.Content = line.Text
  460. lineStyle := lineStyles[line.Kind]
  461. // Embed descriptions are always markdown-rendered to match Discord's rich embed semantics, even when message markdown is globally disabled.
  462. rendered, _ := ml.renderContentLinesWithMarkdown(msg, lineStyle, line.Kind == embedLineDescription)
  463. for _, renderedLine := range rendered {
  464. if line.URL != "" {
  465. renderedLine = lineWithURL(renderedLine, line.URL)
  466. }
  467. // Prefix must be applied after wrapping so every visual line keeps the embed bar marker ("▎"), not only the first logical line.
  468. for _, wrapped := range wrapStyledLine(renderedLine, wrapWidth) {
  469. prefixed := make(tview.Line, 0, len(wrapped)+1)
  470. prefixed = append(prefixed, prefix)
  471. prefixed = append(prefixed, wrapped...)
  472. embedContentLines = append(embedContentLines, prefixed)
  473. }
  474. }
  475. }
  476. if len(embedContentLines) > 0 {
  477. builder.AppendLines(embedContentLines)
  478. }
  479. }
  480. }
  481. func (ml *messagesList) drawForwardedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  482. dimStyle := baseStyle.Dim(true)
  483. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  484. ml.drawAuthor(builder, message, baseStyle)
  485. builder.Write(ml.cfg.Theme.MessagesList.ForwardedIndicator+" ", dimStyle)
  486. ml.drawSnapshotContent(builder, message, message.MessageSnapshots[0].Message, baseStyle)
  487. builder.Write(" ("+ml.formatTimestamp(message.MessageSnapshots[0].Message.Timestamp)+") ", dimStyle)
  488. }
  489. func (ml *messagesList) drawReplyMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  490. dimStyle := baseStyle.Dim(true)
  491. // indicator
  492. builder.Write(ml.cfg.Theme.MessagesList.ReplyIndicator+" ", dimStyle)
  493. if m := message.ReferencedMessage; m != nil {
  494. m.GuildID = message.GuildID
  495. ml.drawAuthor(builder, *m, dimStyle)
  496. ml.drawContent(builder, *m, dimStyle)
  497. } else {
  498. builder.Write("Original message was deleted", dimStyle)
  499. }
  500. builder.NewLine()
  501. // main
  502. ml.drawDefaultMessage(builder, message, baseStyle)
  503. }
  504. func (ml *messagesList) drawPinnedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  505. builder.Write(message.Author.DisplayOrUsername(), baseStyle)
  506. builder.Write(" pinned a message.", baseStyle)
  507. }
  508. func (ml *messagesList) selectedMessage() (*discord.Message, error) {
  509. if len(ml.messages) == 0 {
  510. return nil, errors.New("no messages available")
  511. }
  512. cursor := ml.Cursor()
  513. if cursor == -1 || cursor >= len(ml.messages) {
  514. return nil, errors.New("no message is currently selected")
  515. }
  516. return &ml.messages[cursor], nil
  517. }
  518. func (ml *messagesList) HandleEvent(event tview.Event) tview.Command {
  519. switch event := event.(type) {
  520. case *tview.KeyEvent:
  521. switch {
  522. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Cancel.Keybind):
  523. ml.clearSelection()
  524. return nil
  525. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectUp.Keybind):
  526. return ml.selectUp()
  527. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectDown.Keybind):
  528. ml.selectDown()
  529. return nil
  530. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectTop.Keybind):
  531. ml.selectTop()
  532. return nil
  533. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectBottom.Keybind):
  534. ml.selectBottom()
  535. return nil
  536. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectReply.Keybind):
  537. ml.selectReply()
  538. return nil
  539. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.YankID.Keybind):
  540. return ml.yankMessageID()
  541. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.YankContent.Keybind):
  542. return ml.yankContent()
  543. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.YankURL.Keybind):
  544. return ml.yankURL()
  545. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Open.Keybind):
  546. ml.open()
  547. return nil
  548. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SaveImage.Keybind):
  549. ml.saveImage()
  550. return nil
  551. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Reply.Keybind):
  552. ml.reply(false)
  553. return nil
  554. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.ReplyMention.Keybind):
  555. ml.reply(true)
  556. return nil
  557. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Edit.Keybind):
  558. ml.editSelectedMessage()
  559. return nil
  560. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Delete.Keybind):
  561. return ml.deleteSelectedMessage()
  562. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.DeleteConfirm.Keybind):
  563. ml.confirmDelete()
  564. return nil
  565. }
  566. return ml.Model.HandleEvent(event)
  567. case *olderMessagesLoadedEvent:
  568. selectedChannel := ml.chatView.SelectedChannel()
  569. if selectedChannel == nil || selectedChannel.ID != event.ChannelID {
  570. return nil
  571. }
  572. // Defensive invalidation if Discord returns overlapping windows.
  573. for _, message := range event.Older {
  574. delete(ml.itemByID, message.ID)
  575. }
  576. ml.messages = slices.Concat(event.Older, ml.messages)
  577. ml.invalidateRows()
  578. ml.SetCursor(len(event.Older) - 1)
  579. return nil
  580. }
  581. return ml.Model.HandleEvent(event)
  582. }
  583. func (ml *messagesList) selectUp() tview.Command {
  584. messages := ml.messages
  585. if len(messages) == 0 {
  586. return nil
  587. }
  588. cursor := ml.Cursor()
  589. switch {
  590. case cursor == -1:
  591. cursor = len(messages) - 1
  592. case cursor > 0:
  593. cursor--
  594. case cursor == 0:
  595. return ml.fetchOlderMessages()
  596. }
  597. ml.SetCursor(cursor)
  598. return nil
  599. }
  600. func (ml *messagesList) selectDown() {
  601. messages := ml.messages
  602. if len(messages) == 0 {
  603. return
  604. }
  605. cursor := ml.Cursor()
  606. switch {
  607. case cursor == -1:
  608. cursor = len(messages) - 1
  609. case cursor < len(messages)-1:
  610. cursor++
  611. }
  612. ml.SetCursor(cursor)
  613. }
  614. func (ml *messagesList) selectTop() {
  615. if len(ml.messages) == 0 {
  616. return
  617. }
  618. ml.SetCursor(0)
  619. }
  620. func (ml *messagesList) selectBottom() {
  621. if len(ml.messages) == 0 {
  622. return
  623. }
  624. ml.SetCursor(len(ml.messages) - 1)
  625. }
  626. func (ml *messagesList) selectReply() {
  627. messages := ml.messages
  628. if len(messages) == 0 {
  629. return
  630. }
  631. cursor := ml.Cursor()
  632. if cursor == -1 || cursor >= len(messages) {
  633. return
  634. }
  635. if ref := messages[cursor].ReferencedMessage; ref != nil {
  636. refIdx := slices.IndexFunc(messages, func(m discord.Message) bool {
  637. return m.ID == ref.ID
  638. })
  639. if refIdx != -1 {
  640. ml.SetCursor(refIdx)
  641. }
  642. }
  643. }
  644. func (ml *messagesList) fetchOlderMessages() tview.Command {
  645. selectedChannel := ml.chatView.SelectedChannel()
  646. if selectedChannel == nil {
  647. return nil
  648. }
  649. channelID := selectedChannel.ID
  650. before := ml.messages[0].ID
  651. limit := uint(ml.cfg.MessagesLimit)
  652. return func() tview.Event {
  653. messages, err := ml.chatView.state.MessagesBefore(channelID, before, limit)
  654. if err != nil {
  655. slog.Error("failed to fetch older messages", "err", err)
  656. return nil
  657. }
  658. if len(messages) == 0 {
  659. return nil
  660. }
  661. if guildID := selectedChannel.GuildID; guildID.IsValid() {
  662. ml.requestGuildMembers(guildID, messages)
  663. }
  664. older := slices.Clone(messages)
  665. slices.Reverse(older)
  666. return newOlderMessagesLoadedEvent(channelID, older)
  667. }
  668. }
  669. func (ml *messagesList) yankMessageID() tview.Command {
  670. msg, err := ml.selectedMessage()
  671. if err != nil {
  672. slog.Error("failed to get selected message", "err", err)
  673. return nil
  674. }
  675. return func() tview.Event {
  676. if err := clipboard.Write(clipboard.FmtText, []byte(msg.ID.String())); err != nil {
  677. slog.Error("failed to copy message id", "err", err)
  678. }
  679. return nil
  680. }
  681. }
  682. func (ml *messagesList) yankContent() tview.Command {
  683. msg, err := ml.selectedMessage()
  684. if err != nil {
  685. slog.Error("failed to get selected message", "err", err)
  686. return nil
  687. }
  688. return func() tview.Event {
  689. if err := clipboard.Write(clipboard.FmtText, []byte(msg.Content)); err != nil {
  690. slog.Error("failed to copy message content", "err", err)
  691. }
  692. return nil
  693. }
  694. }
  695. func (ml *messagesList) yankURL() tview.Command {
  696. msg, err := ml.selectedMessage()
  697. if err != nil {
  698. slog.Error("failed to get selected message", "err", err)
  699. return nil
  700. }
  701. return func() tview.Event {
  702. if err := clipboard.Write(clipboard.FmtText, []byte(msg.URL())); err != nil {
  703. slog.Error("failed to copy message url", "err", err)
  704. }
  705. return nil
  706. }
  707. }
  708. func (ml *messagesList) open() {
  709. msg, err := ml.selectedMessage()
  710. if err != nil {
  711. slog.Error("failed to get selected message", "err", err)
  712. return
  713. }
  714. urls := messageURLs(*msg)
  715. if len(urls) == 0 && len(msg.Attachments) == 0 {
  716. return
  717. }
  718. if len(urls)+len(msg.Attachments) == 1 {
  719. if len(urls) == 1 {
  720. go ml.openURL(urls[0])
  721. } else {
  722. attachment := msg.Attachments[0]
  723. if strings.HasPrefix(attachment.ContentType, "image/") {
  724. go ml.openAttachment(msg.Attachments[0])
  725. } else {
  726. go ml.openURL(attachment.URL)
  727. }
  728. }
  729. } else {
  730. ml.showAttachmentsList(urls, msg.Attachments)
  731. }
  732. }
  733. func (ml *messagesList) reply(mention bool) {
  734. message, err := ml.selectedMessage()
  735. if err != nil {
  736. slog.Error("failed to get selected message", "err", err)
  737. return
  738. }
  739. name := message.Author.DisplayOrUsername()
  740. if member := ml.memberForMessage(*message); member != nil && member.Nick != "" {
  741. name = member.Nick
  742. }
  743. data := ml.chatView.messageInput.sendMessageData
  744. data.Reference = &discord.MessageReference{MessageID: message.ID}
  745. data.AllowedMentions = &api.AllowedMentions{RepliedUser: option.False}
  746. title := "Replying to "
  747. if mention {
  748. data.AllowedMentions.RepliedUser = option.True
  749. title = "[@] " + title
  750. }
  751. ml.chatView.messageInput.sendMessageData = data
  752. ml.chatView.messageInput.SetTitle(title + name)
  753. ml.chatView.app.SetFocus(ml.chatView.messageInput)
  754. }
  755. func (ml *messagesList) editSelectedMessage() {
  756. message, err := ml.selectedMessage()
  757. if err != nil {
  758. slog.Error("failed to get selected message", "err", err)
  759. return
  760. }
  761. me, err := ml.chatView.state.Cabinet.Me()
  762. if err != nil {
  763. slog.Error("failed to get current user", "err", err)
  764. return
  765. }
  766. if message.Author.ID != me.ID {
  767. slog.Error("failed to edit message; not the author", "channel_id", message.ChannelID, "message_id", message.ID)
  768. return
  769. }
  770. ml.chatView.messageInput.SetTitle("Editing")
  771. ml.chatView.messageInput.edit = true
  772. ml.chatView.messageInput.SetText(message.Content, true)
  773. ml.chatView.app.SetFocus(ml.chatView.messageInput)
  774. }
  775. func (ml *messagesList) confirmDelete() {
  776. onChoice := func(choice string) {
  777. if choice == "Yes" {
  778. if command := ml.deleteSelectedMessage(); command != nil {
  779. command()
  780. }
  781. }
  782. }
  783. ml.chatView.showConfirmModal(
  784. "Are you sure you want to delete this message?",
  785. []string{"Yes", "No"},
  786. onChoice,
  787. )
  788. }
  789. func (ml *messagesList) deleteSelectedMessage() tview.Command {
  790. selectedMessage, err := ml.selectedMessage()
  791. if err != nil {
  792. slog.Error("failed to get selected message", "err", err)
  793. return nil
  794. }
  795. return func() tview.Event {
  796. if selectedMessage.GuildID.IsValid() {
  797. me, err := ml.chatView.state.Cabinet.Me()
  798. if err != nil {
  799. slog.Error("failed to get current user", "err", err)
  800. return nil
  801. }
  802. if selectedMessage.Author.ID != me.ID && !ml.chatView.state.HasPermissions(selectedMessage.ChannelID, discord.PermissionManageMessages) {
  803. slog.Error("failed to delete message; missing relevant permissions", "channel_id", selectedMessage.ChannelID, "message_id", selectedMessage.ID)
  804. return nil
  805. }
  806. }
  807. if err := ml.chatView.state.DeleteMessage(selectedMessage.ChannelID, selectedMessage.ID, ""); err != nil {
  808. slog.Error("failed to delete message", "channel_id", selectedMessage.ChannelID, "message_id", selectedMessage.ID, "err", err)
  809. return nil
  810. }
  811. if err := ml.chatView.state.MessageRemove(selectedMessage.ChannelID, selectedMessage.ID); err != nil {
  812. slog.Error("failed to delete message", "channel_id", selectedMessage.ChannelID, "message_id", selectedMessage.ID, "err", err)
  813. return nil
  814. }
  815. return nil
  816. }
  817. }
  818. func (ml *messagesList) requestGuildMembers(guildID discord.GuildID, messages []discord.Message) {
  819. usersToFetch := make([]discord.UserID, 0, len(messages))
  820. seen := make(map[discord.UserID]struct{}, len(messages))
  821. for _, message := range messages {
  822. // Do not fetch member for a webhook message.
  823. if message.WebhookID.IsValid() {
  824. continue
  825. }
  826. if member, _ := ml.chatView.state.Cabinet.Member(guildID, message.Author.ID); member == nil {
  827. userID := message.Author.ID
  828. if _, ok := seen[userID]; !ok {
  829. seen[userID] = struct{}{}
  830. usersToFetch = append(usersToFetch, userID)
  831. }
  832. }
  833. }
  834. if len(usersToFetch) > 0 {
  835. err := ml.chatView.state.SendGateway(context.Background(), &gateway.RequestGuildMembersCommand{
  836. GuildIDs: []discord.GuildID{guildID},
  837. UserIDs: usersToFetch,
  838. })
  839. if err != nil {
  840. slog.Error("failed to request guild members", "guild_id", guildID, "err", err)
  841. return
  842. }
  843. ml.setFetchingChunk(true, 0)
  844. ml.waitForChunkEvent()
  845. }
  846. }
  847. func (ml *messagesList) setFetchingChunk(value bool, count uint) {
  848. ml.fetchingMembers.mu.Lock()
  849. defer ml.fetchingMembers.mu.Unlock()
  850. if ml.fetchingMembers.value == value {
  851. return
  852. }
  853. ml.fetchingMembers.value = value
  854. if value {
  855. ml.fetchingMembers.done = make(chan struct{})
  856. } else {
  857. ml.fetchingMembers.count = count
  858. close(ml.fetchingMembers.done)
  859. }
  860. }
  861. func (ml *messagesList) waitForChunkEvent() uint {
  862. ml.fetchingMembers.mu.Lock()
  863. if !ml.fetchingMembers.value {
  864. ml.fetchingMembers.mu.Unlock()
  865. return 0
  866. }
  867. ml.fetchingMembers.mu.Unlock()
  868. <-ml.fetchingMembers.done
  869. return ml.fetchingMembers.count
  870. }
  871. func (ml *messagesList) ShortHelp() []keybind.Keybind {
  872. cfg := ml.cfg.Keybinds.MessagesList
  873. help := []keybind.Keybind{
  874. cfg.SelectUp.Keybind,
  875. cfg.SelectDown.Keybind,
  876. cfg.Cancel.Keybind,
  877. }
  878. if msg, err := ml.selectedMessage(); err == nil {
  879. if me, err := ml.chatView.state.Cabinet.Me(); err == nil {
  880. if msg.Author.ID != me.ID {
  881. help = append(help, cfg.Reply.Keybind)
  882. }
  883. }
  884. if len(messageURLs(*msg)) != 0 || len(msg.Attachments) != 0 {
  885. help = append(help, cfg.Open.Keybind)
  886. }
  887. }
  888. return help
  889. }
  890. func (ml *messagesList) FullHelp() [][]keybind.Keybind {
  891. cfg := ml.cfg.Keybinds.MessagesList
  892. canSelectReply := false
  893. canReply := false
  894. canEdit := false
  895. canDelete := false
  896. canOpen := false
  897. if msg, err := ml.selectedMessage(); err == nil {
  898. canSelectReply = msg.ReferencedMessage != nil
  899. canOpen = len(messageURLs(*msg)) != 0 || len(msg.Attachments) != 0
  900. if me, err := ml.chatView.state.Cabinet.Me(); err == nil {
  901. canReply = msg.Author.ID != me.ID
  902. canEdit = msg.Author.ID == me.ID
  903. canDelete = canEdit
  904. }
  905. if !canDelete {
  906. selected := ml.chatView.SelectedChannel()
  907. canDelete = selected != nil && ml.chatView.state.HasPermissions(selected.ID, discord.PermissionManageMessages)
  908. }
  909. }
  910. actions := make([]keybind.Keybind, 0, 4)
  911. if canReply {
  912. actions = append(actions, cfg.Reply.Keybind, cfg.ReplyMention.Keybind)
  913. }
  914. if canSelectReply {
  915. actions = append(actions, cfg.SelectReply.Keybind)
  916. }
  917. actions = append(actions, cfg.Cancel.Keybind)
  918. manage := make([]keybind.Keybind, 0, 4)
  919. if canEdit {
  920. manage = append(manage, cfg.Edit.Keybind)
  921. }
  922. if canDelete {
  923. manage = append(manage, cfg.DeleteConfirm.Keybind, cfg.Delete.Keybind)
  924. }
  925. if canOpen {
  926. manage = append(manage, cfg.Open.Keybind, cfg.SaveImage.Keybind)
  927. }
  928. return [][]keybind.Keybind{
  929. {cfg.SelectUp.Keybind, cfg.SelectDown.Keybind, cfg.SelectTop.Keybind, cfg.SelectBottom.Keybind},
  930. {cfg.ScrollUp.Keybind, cfg.ScrollDown.Keybind, cfg.ScrollTop.Keybind, cfg.ScrollBottom.Keybind},
  931. actions,
  932. manage,
  933. {cfg.YankContent.Keybind, cfg.YankURL.Keybind, cfg.YankID.Keybind},
  934. }
  935. }