package noteButtons

import (
	"bytes"
	"fmt"
	"log"
	"strings"

	"github.com/yuin/goldmark/ast"
	"github.com/yuin/goldmark/parser"
	"github.com/yuin/goldmark/renderer"
	"github.com/yuin/goldmark/text"
	"github.com/yuin/goldmark/util"
)

type RollTableButton struct {
	ast.BaseBlock

	Uuid     string
	Campaign string
	Label    string
}

type RollTableButtonData struct {
	Uuid     string `json:"uuid"`
	Campaign string `json:"campaign"`
	Label    string `json:"label"`
}

var KindRollTableButton = ast.NewNodeKind("RollTableButton")

func (b *RollTableButton) Kind() ast.NodeKind {
	return KindRollTableButton
}

func (b *RollTableButton) Dump(source []byte, level int) {
	ast.DumpHelper(b, source, level, nil, nil)
}

type RollTableButtonParser struct{}

func (p *RollTableButtonParser) Trigger() []byte {
	return []byte{'{'}
}

func (p *RollTableButtonParser) Parse(parent ast.Node, reader text.Reader, pc parser.Context) ast.Node {
	line, _ := reader.PeekLine()

	if !bytes.HasPrefix(line, []byte("{{")) {
		return nil
	}

	// Find the closing }}
	end := bytes.Index(line[2:], []byte("}}"))
	if end < 0 {
		return nil
	}

	end += 2
	contents := strings.TrimSpace(string(line[2:end]))

	if !strings.HasPrefix(contents, "rollTableButton") {
		return nil
	}

	contents = strings.TrimSpace(strings.TrimPrefix(contents, "rollTableButton"))
	args, err := parseArgs(contents)

	if err != nil {
		log.Default().Println(err)
		return nil
	}

	campaign, _ := pc.Get(CampaignContextKey).(string)
	reader.Advance(end + 2)

	return &RollTableButton{
		Campaign: campaign,
		Uuid:     args["uuid"],
		Label:    args["label"],
	}
}

type RollTableButtonRenderer struct{}

func (r *RollTableButtonRenderer) renderRollTableButton(writer util.BufWriter, _ []byte, node ast.Node, entering bool) (ast.WalkStatus, error) {
	if !entering {
		return ast.WalkContinue, nil
	}

	button := node.(*RollTableButton)

	fmt.Fprintf(
		writer,
		`<button data-campaign="%s" data-uuid="%s" class="rollTableButton">%s</button>`,
		button.Campaign,
		button.Uuid,
		button.Label,
	)

	return ast.WalkContinue, nil
}

func (r *RollTableButtonRenderer) RegisterFuncs(reg renderer.NodeRendererFuncRegisterer) {
	reg.Register(KindRollTableButton, r.renderRollTableButton)
}