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 rich text.Rich
	ParseMessageRich(&rich, m, s)
	return rich
}

func ParseMessageRich(rich *text.Rich, m *discord.Message, s store.Cabinet) {
	var content = []byte(m.Content)
	var node = md.ParseWithMessage(content, s, m, true)

	r := renderer.New(content, node)
	r.Buffer.Grow(len(rich.Content))
	r.Buffer.WriteString(rich.Content)

	// 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)

	rich.Content = r.String()
	rich.Segments = append(rich.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)
}

func ParseWithMessageRich(b []byte, m *discord.Message, s store.Cabinet, msg bool) text.Rich {
	node := md.ParseWithMessage(b, s, m, msg)
	return renderer.RenderNode(b, node)
}