messages_list.go 29 KB

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