43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package segments
|
|
|
|
import (
|
|
"github.com/diamondburned/arikawa/v2/discord"
|
|
"github.com/diamondburned/arikawa/v2/state/store"
|
|
"github.com/diamondburned/cchat-discord/internal/segments/embed"
|
|
"github.com/diamondburned/cchat-discord/internal/segments/renderer"
|
|
"github.com/diamondburned/cchat/text"
|
|
"github.com/diamondburned/ningen/v2/md"
|
|
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/blockquote"
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/codeblock"
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/colored"
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/emoji"
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/inline"
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/link"
|
|
_ "github.com/diamondburned/cchat-discord/internal/segments/mention"
|
|
)
|
|
|
|
func ParseMessage(m *discord.Message, s store.Cabinet) text.Rich {
|
|
var content = []byte(m.Content)
|
|
var node = md.ParseWithMessage(content, s, m, true)
|
|
|
|
r := renderer.New(content, node)
|
|
// Register the needed states for some renderers.
|
|
r.WithState(m, s)
|
|
// Render the main message body.
|
|
r.Walk(node)
|
|
// Render the extra bits.
|
|
embed.RenderAttachments(r, m.Attachments)
|
|
embed.RenderEmbeds(r, m.Embeds, m, s)
|
|
|
|
return text.Rich{
|
|
Content: r.String(),
|
|
Segments: r.Segments,
|
|
}
|
|
}
|
|
|
|
func ParseWithMessage(b []byte, m *discord.Message, s store.Cabinet, msg bool) text.Rich {
|
|
node := md.ParseWithMessage(b, s, m, msg)
|
|
return renderer.RenderNode(b, node)
|
|
}
|