messages_list.go 44 KB

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