messages_list.go 39 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444
  1. package chat
  2. import (
  3. "context"
  4. "errors"
  5. "io"
  6. "log/slog"
  7. "net/http"
  8. "net/url"
  9. "os"
  10. "path/filepath"
  11. "slices"
  12. "strings"
  13. "sync"
  14. "time"
  15. "unicode/utf8"
  16. "github.com/ayn2op/tview/layers"
  17. "github.com/ayn2op/discordo/internal/clipboard"
  18. "github.com/ayn2op/discordo/internal/config"
  19. "github.com/ayn2op/discordo/internal/consts"
  20. "github.com/ayn2op/discordo/internal/markdown"
  21. "github.com/ayn2op/discordo/internal/ui"
  22. "github.com/ayn2op/tview"
  23. "github.com/ayn2op/tview/help"
  24. "github.com/ayn2op/tview/keybind"
  25. "github.com/diamondburned/arikawa/v3/api"
  26. "github.com/diamondburned/arikawa/v3/discord"
  27. "github.com/diamondburned/arikawa/v3/gateway"
  28. "github.com/diamondburned/arikawa/v3/state"
  29. "github.com/diamondburned/arikawa/v3/utils/json/option"
  30. "github.com/diamondburned/ningen/v3/discordmd"
  31. "github.com/gdamore/tcell/v3"
  32. "github.com/gdamore/tcell/v3/color"
  33. "github.com/rivo/uniseg"
  34. "github.com/skratchdot/open-golang/open"
  35. "github.com/yuin/goldmark/ast"
  36. "github.com/yuin/goldmark/parser"
  37. "github.com/yuin/goldmark/text"
  38. )
  39. type messagesList struct {
  40. *tview.List
  41. cfg *config.Config
  42. chatView *Model
  43. messages []discord.Message
  44. // rows is the virtual list model rendered by tview (message rows +
  45. // date-separator rows). It is rebuilt lazily when rowsDirty is true.
  46. rows []messagesListRow
  47. rowsDirty bool
  48. renderer *markdown.Renderer
  49. // itemByID caches unselected message TextViews.
  50. itemByID map[discord.MessageID]*tview.TextView
  51. attachmentsPicker *attachmentsPicker
  52. fetchingMembers struct {
  53. mu sync.Mutex
  54. value bool
  55. count uint
  56. done chan struct{}
  57. }
  58. }
  59. var _ help.KeyMap = (*messagesList)(nil)
  60. type messagesListRowKind uint8
  61. const (
  62. messagesListRowMessage messagesListRowKind = iota
  63. messagesListRowSeparator
  64. )
  65. type messagesListRow struct {
  66. kind messagesListRowKind
  67. messageIndex int
  68. timestamp discord.Timestamp
  69. }
  70. func newMessagesList(cfg *config.Config, chatView *Model) *messagesList {
  71. ml := &messagesList{
  72. List: tview.NewList(),
  73. cfg: cfg,
  74. chatView: chatView,
  75. renderer: markdown.NewRenderer(cfg),
  76. itemByID: make(map[discord.MessageID]*tview.TextView),
  77. }
  78. ml.attachmentsPicker = newAttachmentsPicker(cfg, chatView)
  79. ml.Box = ui.ConfigureBox(ml.Box, &cfg.Theme)
  80. ml.SetTitle("Messages")
  81. ml.SetBuilder(ml.buildItem)
  82. ml.SetChangedFunc(ml.onRowCursorChanged)
  83. ml.SetTrackEnd(true)
  84. ml.SetScrollBarVisibility(cfg.Theme.ScrollBar.Visibility.ScrollBarVisibility)
  85. ml.SetScrollBar(tview.NewScrollBar().
  86. SetTrackStyle(cfg.Theme.ScrollBar.TrackStyle.Style).
  87. SetThumbStyle(cfg.Theme.ScrollBar.ThumbStyle.Style).
  88. SetGlyphSet(cfg.Theme.ScrollBar.GlyphSet.GlyphSet))
  89. return ml
  90. }
  91. func (ml *messagesList) reset() {
  92. ml.messages = nil
  93. ml.rows = nil
  94. ml.rowsDirty = false
  95. clear(ml.itemByID)
  96. ml.
  97. Clear().
  98. SetBuilder(ml.buildItem).
  99. SetTitle("")
  100. }
  101. func (ml *messagesList) setTitle(channel discord.Channel) {
  102. title := ui.ChannelToString(channel, ml.cfg.Icons, ml.chatView.state)
  103. if topic := channel.Topic; topic != "" {
  104. title += " - " + topic
  105. }
  106. ml.SetTitle(title)
  107. }
  108. func (ml *messagesList) setMessages(messages []discord.Message) {
  109. ml.messages = slices.Clone(messages)
  110. slices.Reverse(ml.messages)
  111. ml.invalidateRows()
  112. // New channel payload / refetch: replace the cache wholesale to keep it in
  113. // lockstep with the current message slice.
  114. clear(ml.itemByID)
  115. }
  116. func (ml *messagesList) addMessage(message discord.Message) {
  117. ml.messages = append(ml.messages, message)
  118. ml.invalidateRows()
  119. // Defensive invalidation for ID reuse/edits delivered out-of-order.
  120. delete(ml.itemByID, message.ID)
  121. }
  122. func (ml *messagesList) setMessage(index int, message discord.Message) {
  123. if index < 0 || index >= len(ml.messages) {
  124. return
  125. }
  126. ml.messages[index] = message
  127. delete(ml.itemByID, message.ID)
  128. ml.invalidateRows()
  129. }
  130. func (ml *messagesList) deleteMessage(index int) {
  131. if index < 0 || index >= len(ml.messages) {
  132. return
  133. }
  134. delete(ml.itemByID, ml.messages[index].ID)
  135. ml.messages = append(ml.messages[:index], ml.messages[index+1:]...)
  136. ml.invalidateRows()
  137. }
  138. func (ml *messagesList) clearSelection() {
  139. ml.SetCursor(-1)
  140. }
  141. func (ml *messagesList) buildItem(index int, cursor int) tview.ListItem {
  142. ml.ensureRows()
  143. if index < 0 || index >= len(ml.rows) {
  144. return nil
  145. }
  146. row := ml.rows[index]
  147. if row.kind == messagesListRowSeparator {
  148. return ml.buildSeparatorItem(row.timestamp)
  149. }
  150. message := ml.messages[row.messageIndex]
  151. if index == cursor {
  152. return tview.NewTextView().
  153. SetWrap(true).
  154. SetWordWrap(true).
  155. SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.SelectedMessageStyle.Style))
  156. }
  157. item, ok := ml.itemByID[message.ID]
  158. if !ok {
  159. item = tview.NewTextView().
  160. SetWrap(true).
  161. SetWordWrap(true).
  162. SetLines(ml.renderMessage(message, ml.cfg.Theme.MessagesList.MessageStyle.Style))
  163. ml.itemByID[message.ID] = item
  164. }
  165. return item
  166. }
  167. func (ml *messagesList) renderMessage(message discord.Message, baseStyle tcell.Style) []tview.Line {
  168. builder := tview.NewLineBuilder()
  169. ml.writeMessage(builder, message, baseStyle)
  170. return builder.Finish()
  171. }
  172. func (ml *messagesList) buildSeparatorItem(ts discord.Timestamp) *tview.TextView {
  173. builder := tview.NewLineBuilder()
  174. ml.drawDateSeparator(builder, ts, ml.cfg.Theme.MessagesList.MessageStyle.Style)
  175. return tview.NewTextView().
  176. SetScrollable(false).
  177. SetWrap(false).
  178. SetWordWrap(false).
  179. SetLines(builder.Finish())
  180. }
  181. func (ml *messagesList) drawDateSeparator(builder *tview.LineBuilder, ts discord.Timestamp, baseStyle tcell.Style) {
  182. date := ts.Time().In(time.Local).Format(ml.cfg.DateSeparator.Format)
  183. label := " " + date + " "
  184. fillChar := ml.cfg.DateSeparator.Character
  185. dimStyle := baseStyle.Dim(true)
  186. _, _, width, _ := ml.GetInnerRect()
  187. if width <= 0 {
  188. builder.Write(strings.Repeat(fillChar, 8)+label+strings.Repeat(fillChar, 8), dimStyle)
  189. return
  190. }
  191. labelWidth := utf8.RuneCountInString(label)
  192. if width <= labelWidth {
  193. builder.Write(date, dimStyle)
  194. return
  195. }
  196. fillWidth := width - labelWidth
  197. left := fillWidth / 2
  198. right := fillWidth - left
  199. builder.Write(strings.Repeat(fillChar, left)+label+strings.Repeat(fillChar, right), dimStyle)
  200. }
  201. func (ml *messagesList) rebuildRows() {
  202. rows := make([]messagesListRow, 0, len(ml.messages)*2)
  203. for i := range ml.messages {
  204. // Always show a date separator before the first message, and between messages on different days.
  205. if ml.cfg.DateSeparator.Enabled && (i == 0 || !sameLocalDate(ml.messages[i-1].Timestamp, ml.messages[i].Timestamp)) {
  206. rows = append(rows, messagesListRow{
  207. kind: messagesListRowSeparator,
  208. timestamp: ml.messages[i].Timestamp,
  209. })
  210. }
  211. rows = append(rows, messagesListRow{
  212. kind: messagesListRowMessage,
  213. messageIndex: i,
  214. })
  215. }
  216. ml.rows = rows
  217. ml.rowsDirty = false
  218. }
  219. func (ml *messagesList) invalidateRows() {
  220. ml.rowsDirty = true
  221. }
  222. // ensureRows lazily rebuilds list rows. This avoids repeated O(n) row rebuild
  223. // work when multiple message mutations happen close together.
  224. func (ml *messagesList) ensureRows() {
  225. if !ml.rowsDirty {
  226. return
  227. }
  228. ml.rebuildRows()
  229. }
  230. func sameLocalDate(a discord.Timestamp, b discord.Timestamp) bool {
  231. ta := a.Time().In(time.Local)
  232. tb := b.Time().In(time.Local)
  233. return ta.Year() == tb.Year() && ta.YearDay() == tb.YearDay()
  234. }
  235. // Cursor returns the selected message index, skipping separator rows.
  236. func (ml *messagesList) Cursor() int {
  237. ml.ensureRows()
  238. rowIndex := ml.List.Cursor()
  239. if rowIndex < 0 || rowIndex >= len(ml.rows) {
  240. return -1
  241. }
  242. row := ml.rows[rowIndex]
  243. if row.kind != messagesListRowMessage {
  244. return -1
  245. }
  246. return row.messageIndex
  247. }
  248. // SetCursor selects a message index and maps it to the corresponding row.
  249. func (ml *messagesList) SetCursor(index int) {
  250. ml.List.SetCursor(ml.messageToRowIndex(index))
  251. }
  252. func (ml *messagesList) messageToRowIndex(messageIndex int) int {
  253. ml.ensureRows()
  254. if messageIndex < 0 || messageIndex >= len(ml.messages) {
  255. return -1
  256. }
  257. for i, row := range ml.rows {
  258. if row.kind == messagesListRowMessage && row.messageIndex == messageIndex {
  259. return i
  260. }
  261. }
  262. return -1
  263. }
  264. func (ml *messagesList) onRowCursorChanged(rowIndex int) {
  265. ml.ensureRows()
  266. if rowIndex < 0 || rowIndex >= len(ml.rows) || ml.rows[rowIndex].kind == messagesListRowMessage {
  267. return
  268. }
  269. target := ml.nearestMessageRowIndex(rowIndex)
  270. ml.List.SetCursor(target)
  271. }
  272. // nearestMessageRowIndex expects rowIndex to be within bounds.
  273. func (ml *messagesList) nearestMessageRowIndex(rowIndex int) int {
  274. for i := rowIndex - 1; i >= 0; i-- {
  275. if ml.rows[i].kind == messagesListRowMessage {
  276. return i
  277. }
  278. }
  279. for i := rowIndex + 1; i < len(ml.rows); i++ {
  280. if ml.rows[i].kind == messagesListRowMessage {
  281. return i
  282. }
  283. }
  284. return -1
  285. }
  286. func (ml *messagesList) writeMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  287. if ml.cfg.HideBlockedUsers {
  288. isBlocked := ml.chatView.state.UserIsBlocked(message.Author.ID)
  289. if isBlocked {
  290. builder.Write("Blocked message", baseStyle.Foreground(color.Red).Bold(true))
  291. return
  292. }
  293. }
  294. switch message.Type {
  295. case discord.DefaultMessage:
  296. if message.Reference != nil && message.Reference.Type == discord.MessageReferenceTypeForward {
  297. ml.drawForwardedMessage(builder, message, baseStyle)
  298. } else {
  299. ml.drawDefaultMessage(builder, message, baseStyle)
  300. }
  301. case discord.GuildMemberJoinMessage:
  302. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  303. ml.drawAuthor(builder, message, baseStyle)
  304. builder.Write("joined the server.", baseStyle)
  305. case discord.InlinedReplyMessage:
  306. ml.drawReplyMessage(builder, message, baseStyle)
  307. case discord.ChannelPinnedMessage:
  308. ml.drawPinnedMessage(builder, message, baseStyle)
  309. default:
  310. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  311. ml.drawAuthor(builder, message, baseStyle)
  312. }
  313. }
  314. func (ml *messagesList) formatTimestamp(ts discord.Timestamp) string {
  315. return ts.Time().In(time.Local).Format(ml.cfg.Timestamps.Format)
  316. }
  317. func (ml *messagesList) drawTimestamps(builder *tview.LineBuilder, ts discord.Timestamp, baseStyle tcell.Style) {
  318. dimStyle := baseStyle.Dim(true)
  319. builder.Write(ml.formatTimestamp(ts)+" ", dimStyle)
  320. }
  321. func (ml *messagesList) drawAuthor(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  322. name := message.Author.DisplayOrUsername()
  323. foreground := tcell.ColorDefault
  324. if member := ml.memberForMessage(message); member != nil {
  325. if member.Nick != "" {
  326. name = member.Nick
  327. }
  328. color, ok := state.MemberColor(member, func(id discord.RoleID) *discord.Role {
  329. r, _ := ml.chatView.state.Cabinet.Role(message.GuildID, id)
  330. return r
  331. })
  332. if ok {
  333. foreground = tcell.NewHexColor(int32(color))
  334. }
  335. }
  336. style := baseStyle.Foreground(foreground).Bold(true)
  337. builder.Write(name+" ", style)
  338. }
  339. func (ml *messagesList) memberForMessage(message discord.Message) *discord.Member {
  340. // Webhooks do not have nicknames or roles.
  341. if !message.GuildID.IsValid() || message.WebhookID.IsValid() {
  342. return nil
  343. }
  344. member, err := ml.chatView.state.Cabinet.Member(message.GuildID, message.Author.ID)
  345. if err != nil {
  346. slog.Error("failed to get member from state", "guild_id", message.GuildID, "member_id", message.Author.ID, "err", err)
  347. return nil
  348. }
  349. return member
  350. }
  351. func (ml *messagesList) drawContent(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  352. lines, root := ml.renderContentLines(message, baseStyle)
  353. if ml.chatView.cfg.Markdown.Enabled && builder.HasCurrentLine() {
  354. startsWithCodeBlock := false
  355. if root != nil {
  356. if first := root.FirstChild(); first != nil {
  357. _, startsWithCodeBlock = first.(*ast.FencedCodeBlock)
  358. }
  359. }
  360. if startsWithCodeBlock {
  361. // Keep code blocks visually separate from "timestamp + author".
  362. builder.NewLine()
  363. for len(lines) > 0 && len(lines[0]) == 0 {
  364. lines = lines[1:]
  365. }
  366. } else {
  367. for len(lines) > 1 && len(lines[0]) == 0 {
  368. lines = lines[1:]
  369. }
  370. }
  371. }
  372. builder.AppendLines(lines)
  373. }
  374. func (ml *messagesList) renderContentLines(message discord.Message, baseStyle tcell.Style) ([]tview.Line, ast.Node) {
  375. return ml.renderContentLinesWithMarkdown(message, baseStyle, false)
  376. }
  377. func (ml *messagesList) renderContentLinesWithMarkdown(message discord.Message, baseStyle tcell.Style, forceMarkdown bool) ([]tview.Line, ast.Node) {
  378. // Keep one rendering path for both normal messages and embed fragments so we preserve mention/link parsing behavior consistently across both.
  379. if forceMarkdown || ml.chatView.cfg.Markdown.Enabled {
  380. c := []byte(message.Content)
  381. root := discordmd.ParseWithMessage(c, *ml.chatView.state.Cabinet, &message, false)
  382. return ml.renderer.RenderLines(c, root, baseStyle), root
  383. }
  384. b := tview.NewLineBuilder()
  385. b.Write(message.Content, baseStyle)
  386. return b.Finish(), nil
  387. }
  388. func (ml *messagesList) drawSnapshotContent(builder *tview.LineBuilder, parent discord.Message, snapshot discord.MessageSnapshotMessage, baseStyle tcell.Style) {
  389. // Convert discord.MessageSnapshotMessage to discord.Message with common fields.
  390. message := discord.Message{
  391. Type: snapshot.Type,
  392. Content: snapshot.Content,
  393. Embeds: snapshot.Embeds,
  394. Attachments: snapshot.Attachments,
  395. Timestamp: snapshot.Timestamp,
  396. EditedTimestamp: snapshot.EditedTimestamp,
  397. Flags: snapshot.Flags,
  398. Mentions: snapshot.Mentions,
  399. MentionRoleIDs: snapshot.MentionRoleIDs,
  400. Stickers: snapshot.Stickers,
  401. Components: snapshot.Components,
  402. ChannelID: parent.ChannelID,
  403. GuildID: parent.GuildID,
  404. }
  405. ml.drawContent(builder, message, baseStyle)
  406. }
  407. func (ml *messagesList) drawDefaultMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  408. if ml.cfg.Timestamps.Enabled {
  409. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  410. }
  411. ml.drawAuthor(builder, message, baseStyle)
  412. ml.drawContent(builder, message, baseStyle)
  413. if message.EditedTimestamp.IsValid() {
  414. dimStyle := baseStyle.Dim(true)
  415. builder.Write(" (edited)", dimStyle)
  416. }
  417. ml.drawEmbeds(builder, message, baseStyle)
  418. attachmentStyle := ui.MergeStyle(baseStyle, ml.cfg.Theme.MessagesList.AttachmentStyle.Style)
  419. for _, a := range message.Attachments {
  420. builder.NewLine()
  421. if ml.cfg.ShowAttachmentLinks {
  422. builder.Write(a.Filename+":\n"+a.URL, attachmentStyle)
  423. } else {
  424. builder.Write(a.Filename, attachmentStyle)
  425. }
  426. }
  427. }
  428. func (ml *messagesList) drawEmbeds(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  429. if len(message.Embeds) == 0 {
  430. return
  431. }
  432. contentListURLs := extractURLs(message.Content)
  433. contentURLs := make(map[string]struct{}, len(contentListURLs))
  434. for _, u := range contentListURLs {
  435. contentURLs[u] = struct{}{}
  436. }
  437. lineStyles := embedLineStyles(baseStyle, ml.cfg.Theme.MessagesList.Embeds)
  438. defaultBarStyle := baseStyle.Dim(true)
  439. prefixText := " ▎ "
  440. prefixWidth := tview.TaggedStringWidth(prefixText)
  441. _, _, innerWidth, _ := ml.GetInnerRect()
  442. // Wrap against the current list viewport. This keeps embed wrapping stable even when sidebars/panes are resized.
  443. wrapWidth := max(innerWidth-prefixWidth, 1)
  444. for _, embed := range message.Embeds {
  445. lines := embedLines(embed, contentURLs)
  446. if len(lines) == 0 {
  447. continue
  448. }
  449. embedContentLines := make([]tview.Line, 0, len(lines)*2)
  450. barStyle := defaultBarStyle
  451. if embed.Color != discord.NullColor && embed.Color != 0 {
  452. barStyle = barStyle.Foreground(tcell.NewHexColor(int32(embed.Color.Uint32())))
  453. }
  454. prefix := tview.NewSegment(prefixText, barStyle)
  455. builder.NewLine()
  456. for _, line := range lines {
  457. if strings.TrimSpace(line.Text) == "" {
  458. continue
  459. }
  460. msg := message
  461. msg.Content = line.Text
  462. lineStyle := lineStyles[line.Kind]
  463. // Embed descriptions are always markdown-rendered to match Discord's rich embed semantics, even when message markdown is globally disabled.
  464. rendered, _ := ml.renderContentLinesWithMarkdown(msg, lineStyle, line.Kind == embedLineDescription)
  465. for _, renderedLine := range rendered {
  466. if line.URL != "" {
  467. renderedLine = lineWithURL(renderedLine, line.URL)
  468. }
  469. // Prefix must be applied after wrapping so every visual line keeps the embed bar marker ("▎"), not only the first logical line.
  470. for _, wrapped := range wrapStyledLine(renderedLine, wrapWidth) {
  471. prefixed := make(tview.Line, 0, len(wrapped)+1)
  472. prefixed = append(prefixed, prefix)
  473. prefixed = append(prefixed, wrapped...)
  474. embedContentLines = append(embedContentLines, prefixed)
  475. }
  476. }
  477. }
  478. if len(embedContentLines) > 0 {
  479. builder.AppendLines(embedContentLines)
  480. }
  481. }
  482. }
  483. func wrapStyledLine(line tview.Line, width int) []tview.Line {
  484. if width <= 0 {
  485. return []tview.Line{line}
  486. }
  487. if len(line) == 0 {
  488. return []tview.Line{line}
  489. }
  490. lines := make([]tview.Line, 0, 2)
  491. current := make(tview.Line, 0, len(line))
  492. currentWidth := 0
  493. pushSegment := func(text string, style tcell.Style) {
  494. if text == "" {
  495. return
  496. }
  497. if n := len(current); n > 0 && current[n-1].Style == style {
  498. current[n-1].Text += text
  499. return
  500. }
  501. current = append(current, tview.Segment{Text: text, Style: style})
  502. }
  503. flush := func() {
  504. lineCopy := make(tview.Line, len(current))
  505. copy(lineCopy, current)
  506. lines = append(lines, lineCopy)
  507. current = current[:0]
  508. currentWidth = 0
  509. }
  510. for _, segment := range line {
  511. state := -1
  512. rest := segment.Text
  513. for len(rest) > 0 {
  514. cluster, nextRest, boundaries, nextState := uniseg.StepString(rest, state)
  515. state = nextState
  516. rest = nextRest
  517. if cluster == "" {
  518. continue
  519. }
  520. // Use grapheme width (not rune count) so wrapping stays correct with wide glyphs, emoji, and combining characters.
  521. clusterWidth := graphemeClusterWidth(boundaries)
  522. if currentWidth > 0 && currentWidth+clusterWidth > width {
  523. flush()
  524. }
  525. pushSegment(cluster, segment.Style)
  526. currentWidth += clusterWidth
  527. if currentWidth >= width {
  528. flush()
  529. }
  530. }
  531. }
  532. if len(current) > 0 {
  533. flush()
  534. }
  535. if len(lines) == 0 {
  536. return []tview.Line{{}}
  537. }
  538. return lines
  539. }
  540. func graphemeClusterWidth(boundaries int) int {
  541. return boundaries >> uniseg.ShiftWidth
  542. }
  543. func lineWithURL(line tview.Line, rawURL string) tview.Line {
  544. out := make(tview.Line, len(line))
  545. for i, segment := range line {
  546. out[i] = segment
  547. out[i].Style = out[i].Style.Url(rawURL)
  548. }
  549. return out
  550. }
  551. type embedLine struct {
  552. Text string
  553. Kind embedLineKind
  554. URL string
  555. }
  556. type embedLineKind uint8
  557. const (
  558. // Keep this ordering stable: drawEmbeds indexes precomputed style slots by this enum.
  559. embedLineProvider embedLineKind = iota
  560. embedLineAuthor
  561. embedLineTitle
  562. embedLineDescription
  563. embedLineFieldName
  564. embedLineFieldValue
  565. embedLineFooter
  566. embedLineURL
  567. )
  568. func embedLineStyles(baseStyle tcell.Style, theme config.MessagesListEmbedsTheme) [8]tcell.Style {
  569. styles := [8]tcell.Style{}
  570. styles[embedLineProvider] = ui.MergeStyle(baseStyle, theme.ProviderStyle.Style)
  571. styles[embedLineAuthor] = ui.MergeStyle(baseStyle, theme.AuthorStyle.Style)
  572. styles[embedLineTitle] = ui.MergeStyle(baseStyle, theme.TitleStyle.Style)
  573. styles[embedLineDescription] = ui.MergeStyle(baseStyle, theme.DescriptionStyle.Style)
  574. styles[embedLineFieldName] = ui.MergeStyle(baseStyle, theme.FieldNameStyle.Style)
  575. styles[embedLineFieldValue] = ui.MergeStyle(baseStyle, theme.FieldValueStyle.Style)
  576. styles[embedLineFooter] = ui.MergeStyle(baseStyle, theme.FooterStyle.Style)
  577. styles[embedLineURL] = ui.MergeStyle(baseStyle, theme.URLStyle.Style)
  578. return styles
  579. }
  580. type embedLineDedupKey struct {
  581. kind embedLineKind
  582. text string
  583. }
  584. func embedLines(embed discord.Embed, contentURLs map[string]struct{}) []embedLine {
  585. lines := make([]embedLine, 0, 8)
  586. seen := make(map[embedLineDedupKey]struct{}, 8)
  587. appendUnique := func(s string, kind embedLineKind, rawURL string) {
  588. s = strings.TrimSpace(s)
  589. if s == "" {
  590. return
  591. }
  592. // Deduplicate by kind+text so the same value can intentionally appear in multiple semantic slots with different styles (e.g. title vs. field).
  593. key := embedLineDedupKey{kind: kind, text: s}
  594. if _, ok := seen[key]; ok {
  595. return
  596. }
  597. seen[key] = struct{}{}
  598. lines = append(lines, embedLine{
  599. Text: s,
  600. Kind: kind,
  601. URL: rawURL,
  602. })
  603. }
  604. appendURL := func(url discord.URL) {
  605. u := strings.TrimSpace(string(url))
  606. if u == "" {
  607. return
  608. }
  609. // Avoid duplicating links that already appear in message body content.
  610. if _, ok := contentURLs[u]; ok {
  611. return
  612. }
  613. appendUnique(linkDisplayText(u), embedLineURL, u)
  614. }
  615. if embed.Provider != nil {
  616. appendUnique(embed.Provider.Name, embedLineProvider, "")
  617. }
  618. if embed.Author != nil {
  619. appendUnique(embed.Author.Name, embedLineAuthor, "")
  620. }
  621. appendUnique(embed.Title, embedLineTitle, string(embed.URL))
  622. // Some Discord embeds include markdown-escaped punctuation in raw payload text (e.g. "\."), so normalize for display.
  623. appendUnique(unescapeMarkdownEscapes(embed.Description), embedLineDescription, "")
  624. for _, field := range embed.Fields {
  625. switch {
  626. case field.Name != "" && field.Value != "":
  627. appendUnique(field.Name, embedLineFieldName, "")
  628. appendUnique(field.Value, embedLineFieldValue, "")
  629. case field.Name != "":
  630. appendUnique(field.Name, embedLineFieldName, "")
  631. default:
  632. appendUnique(field.Value, embedLineFieldValue, "")
  633. }
  634. }
  635. if embed.Footer != nil {
  636. appendUnique(embed.Footer.Text, embedLineFooter, "")
  637. }
  638. // Prefer media URLs after textual fields so previews read top-to-bottom before jumping to link targets.
  639. // When a title exists, embed.URL is represented by title Style.Url metadata instead of a separate URL row.
  640. if embed.Title == "" {
  641. appendURL(embed.URL)
  642. }
  643. if embed.Image != nil {
  644. appendURL(embed.Image.URL)
  645. }
  646. if embed.Video != nil {
  647. appendURL(embed.Video.URL)
  648. }
  649. return lines
  650. }
  651. func linkDisplayText(raw string) string {
  652. parsed, err := url.Parse(raw)
  653. if err != nil || parsed.Host == "" {
  654. return raw
  655. }
  656. path := strings.TrimSpace(parsed.EscapedPath())
  657. switch {
  658. case path == "", path == "/":
  659. return parsed.Host
  660. case len(path) > 48:
  661. return parsed.Host + path[:45] + "..."
  662. default:
  663. return parsed.Host + path
  664. }
  665. }
  666. func unescapeMarkdownEscapes(s string) string {
  667. if !strings.ContainsRune(s, '\\') {
  668. return s
  669. }
  670. var b strings.Builder
  671. b.Grow(len(s))
  672. for i := range len(s) {
  673. if s[i] == '\\' && i+1 < len(s) && isMarkdownEscapable(s[i+1]) {
  674. continue
  675. }
  676. b.WriteByte(s[i])
  677. }
  678. return b.String()
  679. }
  680. func isMarkdownEscapable(c byte) bool {
  681. switch c {
  682. case '\\', '`', '*', '_', '{', '}', '[', ']', '(', ')', '#', '+', '-', '.', '!', '|', '>', '~':
  683. return true
  684. default:
  685. return false
  686. }
  687. }
  688. func (ml *messagesList) drawForwardedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  689. dimStyle := baseStyle.Dim(true)
  690. ml.drawTimestamps(builder, message.Timestamp, baseStyle)
  691. ml.drawAuthor(builder, message, baseStyle)
  692. builder.Write(ml.cfg.Theme.MessagesList.ForwardedIndicator+" ", dimStyle)
  693. ml.drawSnapshotContent(builder, message, message.MessageSnapshots[0].Message, baseStyle)
  694. builder.Write(" ("+ml.formatTimestamp(message.MessageSnapshots[0].Message.Timestamp)+") ", dimStyle)
  695. }
  696. func (ml *messagesList) drawReplyMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  697. dimStyle := baseStyle.Dim(true)
  698. // indicator
  699. builder.Write(ml.cfg.Theme.MessagesList.ReplyIndicator+" ", dimStyle)
  700. if m := message.ReferencedMessage; m != nil {
  701. m.GuildID = message.GuildID
  702. ml.drawAuthor(builder, *m, dimStyle)
  703. ml.drawContent(builder, *m, dimStyle)
  704. } else {
  705. builder.Write("Original message was deleted", dimStyle)
  706. }
  707. builder.NewLine()
  708. // main
  709. ml.drawDefaultMessage(builder, message, baseStyle)
  710. }
  711. func (ml *messagesList) drawPinnedMessage(builder *tview.LineBuilder, message discord.Message, baseStyle tcell.Style) {
  712. builder.Write(message.Author.DisplayOrUsername(), baseStyle)
  713. builder.Write(" pinned a message.", baseStyle)
  714. }
  715. func (ml *messagesList) selectedMessage() (*discord.Message, error) {
  716. if len(ml.messages) == 0 {
  717. return nil, errors.New("no messages available")
  718. }
  719. cursor := ml.Cursor()
  720. if cursor == -1 || cursor >= len(ml.messages) {
  721. return nil, errors.New("no message is currently selected")
  722. }
  723. return &ml.messages[cursor], nil
  724. }
  725. func (ml *messagesList) HandleEvent(event tcell.Event) tview.Command {
  726. switch event := event.(type) {
  727. case *tview.KeyEvent:
  728. switch {
  729. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.ScrollUp.Keybind):
  730. ml.ScrollUp()
  731. return nil
  732. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.ScrollDown.Keybind):
  733. ml.ScrollDown()
  734. return nil
  735. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.ScrollTop.Keybind):
  736. ml.ScrollToStart()
  737. return nil
  738. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.ScrollBottom.Keybind):
  739. ml.ScrollToEnd()
  740. return nil
  741. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Cancel.Keybind):
  742. ml.clearSelection()
  743. return nil
  744. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectUp.Keybind):
  745. ml.selectUp()
  746. return nil
  747. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectDown.Keybind):
  748. ml.selectDown()
  749. return nil
  750. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectTop.Keybind):
  751. ml.selectTop()
  752. return nil
  753. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectBottom.Keybind):
  754. ml.selectBottom()
  755. return nil
  756. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.SelectReply.Keybind):
  757. ml.selectReply()
  758. return nil
  759. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.YankID.Keybind):
  760. return ml.yankMessageID()
  761. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.YankContent.Keybind):
  762. return ml.yankContent()
  763. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.YankURL.Keybind):
  764. return ml.yankURL()
  765. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Open.Keybind):
  766. ml.open()
  767. return nil
  768. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Reply.Keybind):
  769. ml.reply(false)
  770. return nil
  771. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.ReplyMention.Keybind):
  772. ml.reply(true)
  773. return nil
  774. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Edit.Keybind):
  775. ml.editSelectedMessage()
  776. return nil
  777. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.Delete.Keybind):
  778. return ml.deleteSelectedMessage()
  779. case keybind.Matches(event, ml.cfg.Keybinds.MessagesList.DeleteConfirm.Keybind):
  780. ml.confirmDelete()
  781. return nil
  782. }
  783. // Do not fall through to List defaults for unmatched keys.
  784. return nil
  785. }
  786. return ml.List.HandleEvent(event)
  787. }
  788. func (ml *messagesList) selectUp() {
  789. messages := ml.messages
  790. if len(messages) == 0 {
  791. return
  792. }
  793. cursor := ml.Cursor()
  794. switch {
  795. case cursor == -1:
  796. cursor = len(messages) - 1
  797. case cursor > 0:
  798. cursor--
  799. case cursor == 0:
  800. added := ml.prependOlderMessages()
  801. if added == 0 {
  802. return
  803. }
  804. cursor = added - 1
  805. }
  806. ml.SetCursor(cursor)
  807. }
  808. func (ml *messagesList) selectDown() {
  809. messages := ml.messages
  810. if len(messages) == 0 {
  811. return
  812. }
  813. cursor := ml.Cursor()
  814. switch {
  815. case cursor == -1:
  816. cursor = len(messages) - 1
  817. case cursor < len(messages)-1:
  818. cursor++
  819. }
  820. ml.SetCursor(cursor)
  821. }
  822. func (ml *messagesList) selectTop() {
  823. if len(ml.messages) == 0 {
  824. return
  825. }
  826. ml.SetCursor(0)
  827. }
  828. func (ml *messagesList) selectBottom() {
  829. if len(ml.messages) == 0 {
  830. return
  831. }
  832. ml.SetCursor(len(ml.messages) - 1)
  833. }
  834. func (ml *messagesList) selectReply() {
  835. messages := ml.messages
  836. if len(messages) == 0 {
  837. return
  838. }
  839. cursor := ml.Cursor()
  840. if cursor == -1 || cursor >= len(messages) {
  841. return
  842. }
  843. if ref := messages[cursor].ReferencedMessage; ref != nil {
  844. refIdx := slices.IndexFunc(messages, func(m discord.Message) bool {
  845. return m.ID == ref.ID
  846. })
  847. if refIdx != -1 {
  848. ml.SetCursor(refIdx)
  849. }
  850. }
  851. }
  852. func (ml *messagesList) prependOlderMessages() int {
  853. selectedChannel := ml.chatView.SelectedChannel()
  854. if selectedChannel == nil {
  855. return 0
  856. }
  857. channelID := selectedChannel.ID
  858. before := ml.messages[0].ID
  859. limit := uint(ml.cfg.MessagesLimit)
  860. messages, err := ml.chatView.state.MessagesBefore(channelID, before, limit)
  861. if err != nil {
  862. slog.Error("failed to fetch older messages", "err", err)
  863. return 0
  864. }
  865. if len(messages) == 0 {
  866. return 0
  867. }
  868. if guildID := selectedChannel.GuildID; guildID.IsValid() {
  869. ml.requestGuildMembers(guildID, messages)
  870. }
  871. older := slices.Clone(messages)
  872. slices.Reverse(older)
  873. // Defensive invalidation if Discord returns overlapping windows.
  874. for _, message := range older {
  875. delete(ml.itemByID, message.ID)
  876. }
  877. ml.messages = slices.Concat(older, ml.messages)
  878. ml.invalidateRows()
  879. return len(messages)
  880. }
  881. func (ml *messagesList) yankMessageID() tview.Command {
  882. msg, err := ml.selectedMessage()
  883. if err != nil {
  884. slog.Error("failed to get selected message", "err", err)
  885. return nil
  886. }
  887. return func() tcell.Event {
  888. if err := clipboard.Write(clipboard.FmtText, []byte(msg.ID.String())); err != nil {
  889. slog.Error("failed to copy message id", "err", err)
  890. }
  891. return nil
  892. }
  893. }
  894. func (ml *messagesList) yankContent() tview.Command {
  895. msg, err := ml.selectedMessage()
  896. if err != nil {
  897. slog.Error("failed to get selected message", "err", err)
  898. return nil
  899. }
  900. return func() tcell.Event {
  901. if err := clipboard.Write(clipboard.FmtText, []byte(msg.Content)); err != nil {
  902. slog.Error("failed to copy message content", "err", err)
  903. }
  904. return nil
  905. }
  906. }
  907. func (ml *messagesList) yankURL() tview.Command {
  908. msg, err := ml.selectedMessage()
  909. if err != nil {
  910. slog.Error("failed to get selected message", "err", err)
  911. return nil
  912. }
  913. return func() tcell.Event {
  914. if err := clipboard.Write(clipboard.FmtText, []byte(msg.URL())); err != nil {
  915. slog.Error("failed to copy message url", "err", err)
  916. }
  917. return nil
  918. }
  919. }
  920. func (ml *messagesList) open() {
  921. msg, err := ml.selectedMessage()
  922. if err != nil {
  923. slog.Error("failed to get selected message", "err", err)
  924. return
  925. }
  926. urls := messageURLs(*msg)
  927. if len(urls) == 0 && len(msg.Attachments) == 0 {
  928. return
  929. }
  930. if len(urls)+len(msg.Attachments) == 1 {
  931. if len(urls) == 1 {
  932. go ml.openURL(urls[0])
  933. } else {
  934. attachment := msg.Attachments[0]
  935. if strings.HasPrefix(attachment.ContentType, "image/") {
  936. go ml.openAttachment(msg.Attachments[0])
  937. } else {
  938. go ml.openURL(attachment.URL)
  939. }
  940. }
  941. } else {
  942. ml.showAttachmentsList(urls, msg.Attachments)
  943. }
  944. }
  945. func extractURLs(content string) []string {
  946. src := []byte(content)
  947. node := parser.NewParser(
  948. parser.WithBlockParsers(discordmd.BlockParsers()...),
  949. parser.WithInlineParsers(discordmd.InlineParserWithLink()...),
  950. ).Parse(text.NewReader(src))
  951. var urls []string
  952. ast.Walk(node, func(n ast.Node, entering bool) (ast.WalkStatus, error) {
  953. if entering {
  954. switch n := n.(type) {
  955. case *ast.AutoLink:
  956. urls = append(urls, string(n.URL(src)))
  957. case *ast.Link:
  958. urls = append(urls, string(n.Destination))
  959. }
  960. }
  961. return ast.WalkContinue, nil
  962. })
  963. return urls
  964. }
  965. func extractEmbedURLs(embeds []discord.Embed) []string {
  966. urls := make([]string, 0, len(embeds)*3)
  967. for _, embed := range embeds {
  968. if embed.URL != "" {
  969. urls = append(urls, string(embed.URL))
  970. }
  971. if embed.Image != nil && embed.Image.URL != "" {
  972. urls = append(urls, string(embed.Image.URL))
  973. }
  974. if embed.Video != nil && embed.Video.URL != "" {
  975. urls = append(urls, string(embed.Video.URL))
  976. }
  977. }
  978. return urls
  979. }
  980. func messageURLs(msg discord.Message) []string {
  981. combined := extractURLs(msg.Content)
  982. combined = append(combined, extractEmbedURLs(msg.Embeds)...)
  983. urls := make([]string, 0, len(combined))
  984. seen := make(map[string]struct{}, len(combined))
  985. for _, u := range combined {
  986. u = strings.TrimSpace(u)
  987. if u == "" {
  988. continue
  989. }
  990. if _, ok := seen[u]; ok {
  991. continue
  992. }
  993. seen[u] = struct{}{}
  994. urls = append(urls, u)
  995. }
  996. return urls
  997. }
  998. func (ml *messagesList) showAttachmentsList(urls []string, attachments []discord.Attachment) {
  999. var items []attachmentItem
  1000. for _, a := range attachments {
  1001. attachment := a
  1002. action := func() {
  1003. if strings.HasPrefix(attachment.ContentType, "image/") {
  1004. go ml.openAttachment(attachment)
  1005. } else {
  1006. go ml.openURL(attachment.URL)
  1007. }
  1008. }
  1009. items = append(items, attachmentItem{
  1010. label: attachment.Filename,
  1011. open: action,
  1012. })
  1013. }
  1014. for _, u := range urls {
  1015. url := u
  1016. items = append(items, attachmentItem{
  1017. label: url,
  1018. open: func() { go ml.openURL(url) },
  1019. })
  1020. }
  1021. ml.attachmentsPicker.SetItems(items)
  1022. ml.chatView.
  1023. AddLayer(
  1024. ui.Centered(ml.attachmentsPicker, 0, 0),
  1025. layers.WithName(attachmentsListLayerName),
  1026. layers.WithResize(true),
  1027. layers.WithVisible(true),
  1028. layers.WithOverlay(),
  1029. ).
  1030. SendToFront(attachmentsListLayerName)
  1031. ml.chatView.app.SetFocus(ml.attachmentsPicker)
  1032. }
  1033. func (ml *messagesList) openAttachment(attachment discord.Attachment) {
  1034. resp, err := http.Get(attachment.URL)
  1035. if err != nil {
  1036. slog.Error("failed to fetch the attachment", "err", err, "url", attachment.URL)
  1037. return
  1038. }
  1039. defer resp.Body.Close()
  1040. path := filepath.Join(consts.CacheDir(), "attachments")
  1041. if err := os.MkdirAll(path, os.ModePerm); err != nil {
  1042. slog.Error("failed to create attachments dir", "err", err, "path", path)
  1043. return
  1044. }
  1045. path = filepath.Join(path, attachment.Filename)
  1046. file, err := os.Create(path)
  1047. if err != nil {
  1048. slog.Error("failed to create attachment file", "err", err, "path", path)
  1049. return
  1050. }
  1051. defer file.Close()
  1052. if _, err := io.Copy(file, resp.Body); err != nil {
  1053. slog.Error("failed to copy attachment to file", "err", err)
  1054. return
  1055. }
  1056. if err := open.Start(path); err != nil {
  1057. slog.Error("failed to open attachment file", "err", err, "path", path)
  1058. return
  1059. }
  1060. }
  1061. func (ml *messagesList) openURL(url string) {
  1062. if err := open.Start(url); err != nil {
  1063. slog.Error("failed to open URL", "err", err, "url", url)
  1064. }
  1065. }
  1066. func (ml *messagesList) reply(mention bool) {
  1067. message, err := ml.selectedMessage()
  1068. if err != nil {
  1069. slog.Error("failed to get selected message", "err", err)
  1070. return
  1071. }
  1072. name := message.Author.DisplayOrUsername()
  1073. if member := ml.memberForMessage(*message); member != nil && member.Nick != "" {
  1074. name = member.Nick
  1075. }
  1076. data := ml.chatView.messageInput.sendMessageData
  1077. data.Reference = &discord.MessageReference{MessageID: message.ID}
  1078. data.AllowedMentions = &api.AllowedMentions{RepliedUser: option.False}
  1079. title := "Replying to "
  1080. if mention {
  1081. data.AllowedMentions.RepliedUser = option.True
  1082. title = "[@] " + title
  1083. }
  1084. ml.chatView.messageInput.sendMessageData = data
  1085. ml.chatView.messageInput.SetTitle(title + name)
  1086. ml.chatView.app.SetFocus(ml.chatView.messageInput)
  1087. }
  1088. func (ml *messagesList) editSelectedMessage() {
  1089. message, err := ml.selectedMessage()
  1090. if err != nil {
  1091. slog.Error("failed to get selected message", "err", err)
  1092. return
  1093. }
  1094. me, _ := ml.chatView.state.Cabinet.Me()
  1095. if message.Author.ID != me.ID {
  1096. slog.Error("failed to edit message; not the author", "channel_id", message.ChannelID, "message_id", message.ID)
  1097. return
  1098. }
  1099. ml.chatView.messageInput.SetTitle("Editing")
  1100. ml.chatView.messageInput.edit = true
  1101. ml.chatView.messageInput.SetText(message.Content, true)
  1102. ml.chatView.app.SetFocus(ml.chatView.messageInput)
  1103. }
  1104. func (ml *messagesList) confirmDelete() {
  1105. onChoice := func(choice string) {
  1106. if choice == "Yes" {
  1107. if command := ml.deleteSelectedMessage(); command != nil {
  1108. command()
  1109. }
  1110. }
  1111. }
  1112. ml.chatView.showConfirmModal(
  1113. "Are you sure you want to delete this message?",
  1114. []string{"Yes", "No"},
  1115. onChoice,
  1116. )
  1117. }
  1118. func (ml *messagesList) deleteSelectedMessage() tview.Command {
  1119. selectedMessage, err := ml.selectedMessage()
  1120. if err != nil {
  1121. slog.Error("failed to get selected message", "err", err)
  1122. return nil
  1123. }
  1124. return func() tcell.Event {
  1125. if selectedMessage.GuildID.IsValid() {
  1126. me, _ := ml.chatView.state.Cabinet.Me()
  1127. if selectedMessage.Author.ID != me.ID && !ml.chatView.state.HasPermissions(selectedMessage.ChannelID, discord.PermissionManageMessages) {
  1128. slog.Error("failed to delete message; missing relevant permissions", "channel_id", selectedMessage.ChannelID, "message_id", selectedMessage.ID)
  1129. return nil
  1130. }
  1131. }
  1132. if err := ml.chatView.state.DeleteMessage(selectedMessage.ChannelID, selectedMessage.ID, ""); err != nil {
  1133. slog.Error("failed to delete message", "channel_id", selectedMessage.ChannelID, "message_id", selectedMessage.ID, "err", err)
  1134. return nil
  1135. }
  1136. if err := ml.chatView.state.MessageRemove(selectedMessage.ChannelID, selectedMessage.ID); err != nil {
  1137. slog.Error("failed to delete message", "channel_id", selectedMessage.ChannelID, "message_id", selectedMessage.ID, "err", err)
  1138. return nil
  1139. }
  1140. return nil
  1141. }
  1142. }
  1143. func (ml *messagesList) requestGuildMembers(guildID discord.GuildID, messages []discord.Message) {
  1144. usersToFetch := make([]discord.UserID, 0, len(messages))
  1145. seen := make(map[discord.UserID]struct{}, len(messages))
  1146. for _, message := range messages {
  1147. // Do not fetch member for a webhook message.
  1148. if message.WebhookID.IsValid() {
  1149. continue
  1150. }
  1151. if member, _ := ml.chatView.state.Cabinet.Member(guildID, message.Author.ID); member == nil {
  1152. userID := message.Author.ID
  1153. if _, ok := seen[userID]; !ok {
  1154. seen[userID] = struct{}{}
  1155. usersToFetch = append(usersToFetch, userID)
  1156. }
  1157. }
  1158. }
  1159. if len(usersToFetch) > 0 {
  1160. err := ml.chatView.state.SendGateway(context.Background(), &gateway.RequestGuildMembersCommand{
  1161. GuildIDs: []discord.GuildID{guildID},
  1162. UserIDs: usersToFetch,
  1163. })
  1164. if err != nil {
  1165. slog.Error("failed to request guild members", "guild_id", guildID, "err", err)
  1166. return
  1167. }
  1168. ml.setFetchingChunk(true, 0)
  1169. ml.waitForChunkEvent()
  1170. }
  1171. }
  1172. func (ml *messagesList) setFetchingChunk(value bool, count uint) {
  1173. ml.fetchingMembers.mu.Lock()
  1174. defer ml.fetchingMembers.mu.Unlock()
  1175. if ml.fetchingMembers.value == value {
  1176. return
  1177. }
  1178. ml.fetchingMembers.value = value
  1179. if value {
  1180. ml.fetchingMembers.done = make(chan struct{})
  1181. } else {
  1182. ml.fetchingMembers.count = count
  1183. close(ml.fetchingMembers.done)
  1184. }
  1185. }
  1186. func (ml *messagesList) waitForChunkEvent() uint {
  1187. ml.fetchingMembers.mu.Lock()
  1188. if !ml.fetchingMembers.value {
  1189. ml.fetchingMembers.mu.Unlock()
  1190. return 0
  1191. }
  1192. ml.fetchingMembers.mu.Unlock()
  1193. <-ml.fetchingMembers.done
  1194. return ml.fetchingMembers.count
  1195. }
  1196. func (ml *messagesList) ShortHelp() []keybind.Keybind {
  1197. cfg := ml.cfg.Keybinds.MessagesList
  1198. help := []keybind.Keybind{
  1199. cfg.SelectUp.Keybind,
  1200. cfg.SelectDown.Keybind,
  1201. cfg.Cancel.Keybind,
  1202. }
  1203. if msg, err := ml.selectedMessage(); err == nil {
  1204. me, _ := ml.chatView.state.Cabinet.Me()
  1205. if msg.Author.ID != me.ID {
  1206. help = append(help, cfg.Reply.Keybind)
  1207. }
  1208. }
  1209. return help
  1210. }
  1211. func (ml *messagesList) FullHelp() [][]keybind.Keybind {
  1212. cfg := ml.cfg.Keybinds.MessagesList
  1213. canSelectReply := false
  1214. canReply := false
  1215. canEdit := false
  1216. canDelete := false
  1217. canOpen := false
  1218. if msg, err := ml.selectedMessage(); err == nil {
  1219. canSelectReply = msg.ReferencedMessage != nil
  1220. canOpen = len(messageURLs(*msg)) != 0 || len(msg.Attachments) != 0
  1221. me, _ := ml.chatView.state.Cabinet.Me()
  1222. canReply = msg.Author.ID != me.ID
  1223. canEdit = msg.Author.ID == me.ID
  1224. canDelete = canEdit
  1225. if !canDelete {
  1226. selected := ml.chatView.SelectedChannel()
  1227. canDelete = selected != nil && ml.chatView.state.HasPermissions(selected.ID, discord.PermissionManageMessages)
  1228. }
  1229. }
  1230. actions := make([]keybind.Keybind, 0, 4)
  1231. if canReply {
  1232. actions = append(actions, cfg.Reply.Keybind, cfg.ReplyMention.Keybind)
  1233. }
  1234. if canSelectReply {
  1235. actions = append(actions, cfg.SelectReply.Keybind)
  1236. }
  1237. actions = append(actions, cfg.Cancel.Keybind)
  1238. manage := make([]keybind.Keybind, 0, 4)
  1239. if canEdit {
  1240. manage = append(manage, cfg.Edit.Keybind)
  1241. }
  1242. if canDelete {
  1243. manage = append(manage, cfg.DeleteConfirm.Keybind, cfg.Delete.Keybind)
  1244. }
  1245. if canOpen {
  1246. manage = append(manage, cfg.Open.Keybind)
  1247. }
  1248. return [][]keybind.Keybind{
  1249. {cfg.SelectUp.Keybind, cfg.SelectDown.Keybind, cfg.SelectTop.Keybind, cfg.SelectBottom.Keybind},
  1250. {cfg.ScrollUp.Keybind, cfg.ScrollDown.Keybind, cfg.ScrollTop.Keybind, cfg.ScrollBottom.Keybind},
  1251. actions,
  1252. manage,
  1253. {cfg.YankContent.Keybind, cfg.YankURL.Keybind, cfg.YankID.Keybind},
  1254. }
  1255. }