messages_list.go 39 KB

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