discordo/messages_text.go
2023-01-01 19:30:30 +05:30

87 lines
1.9 KiB
Go

package main
import (
"bytes"
"time"
"github.com/diamondburned/arikawa/v3/discord"
"github.com/rivo/tview"
)
type MessagesText struct {
*tview.TextView
buffer bytes.Buffer
}
func newMessagesText() *MessagesText {
mt := &MessagesText{
TextView: tview.NewTextView(),
}
mt.SetDynamicColors(true)
mt.SetRegions(true)
mt.SetWordWrap(true)
mt.SetBorder(cfg.Theme.MessagesText.Border)
padding := cfg.Theme.MessagesText.BorderPadding
mt.SetBorderPadding(padding[0], padding[1], padding[2], padding[3])
return mt
}
func (mt *MessagesText) newMessage(m *discord.Message) error {
switch m.Type {
case discord.DefaultMessage, discord.InlinedReplyMessage:
// Region tags are square brackets that contain a region ID in double quotes
// https://pkg.go.dev/github.com/rivo/tview#hdr-Regions_and_Highlights
mt.buffer.WriteString(`["`)
mt.buffer.WriteString(m.ID.String())
mt.buffer.WriteString(`"]`)
if m.ReferencedMessage != nil {
mt.buffer.WriteString("> [::d]")
// Author
mt.newAuthor(m.ReferencedMessage)
// Content
mt.newContent(m.ReferencedMessage)
mt.buffer.WriteString("[::-]")
mt.buffer.WriteByte('\n')
}
// Author
mt.newAuthor(m)
// Timestamps
mt.newTimestamp(m)
// Content
mt.buffer.WriteByte('\n')
mt.newContent(m)
// Tags with no region ID ([""]) don't start new regions. They can therefore be used to mark the end of a region.
mt.buffer.WriteString(`[""]`)
}
mt.buffer.WriteString("\n\n")
_, err := mt.buffer.WriteTo(mt)
return err
}
func (mt *MessagesText) newAuthor(m *discord.Message) {
mt.buffer.WriteString("[blue]")
mt.buffer.WriteString(m.Author.Username)
mt.buffer.WriteString("[-] ")
}
func (mt *MessagesText) newTimestamp(m *discord.Message) {
mt.buffer.WriteString("[::d]")
mt.buffer.WriteString(m.Timestamp.Format(time.Kitchen))
mt.buffer.WriteString("[-:-:-] ")
}
func (mt *MessagesText) newContent(m *discord.Message) {
mt.buffer.WriteString(tview.Escape(m.Content))
}