package printer

import (
	"strings"
	"testing"

	"olexsmir.xyz/clerk/internal/testutil/golden"
	"olexsmir.xyz/clerk/journal"
)

var tests = []string{"entries", "directives", "sample"}

func TestRoundTrip(t *testing.T) {
	for _, tname := range tests {
		t.Run(tname, func(t *testing.T) {
			a := golden.Read(t, tname)
			rj := journal.NewLoader().ResolveBytes(tname+".journal", a.Get("input"))
			pf := rj.Occurrences[0]
			var b strings.Builder
			if err := DefaultConfig.Fprint(&b, pf.Ast); err != nil {
				t.Fatal(err)
			}

			golden.Assert(t, a, b.String())
		})
	}
}

func BenchmarkPrinter(b *testing.B) {
	for _, tname := range tests {
		b.Run(tname, func(b *testing.B) {
			b.ReportAllocs()
			inp := golden.Read(b, tname).Get("input")
			rj := journal.NewLoader().ResolveBytes(tname+".journal", inp)
			pf := rj.Occurrences[0]
			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				var buf strings.Builder
				if err := DefaultConfig.Fprint(&buf, pf.Ast); err != nil {
					b.Fatal(err)
				}
			}
		})
	}
}

var testsWithConfig = map[string]*Config{
	"align_right":              {AlignStyle: AlignRight, AlignColumn: 50},
	"align_tab":                {AlignStyle: AlignTab, AlignColumn: 50},
	"commodity_pos_before":     {CommodityPos: CommodityPosBefore},
	"preserve_blanks":          {PreserveBlankLines: true},
	"commodity_space_preserve": {CommoditySpace: CommoditySpacePreserve},
	"commodity_space_always":   {CommoditySpace: CommoditySpaceAlways},
	"commodity_space_never":    {CommoditySpace: CommoditySpaceNever},
	"tab_indent":               {TabIndent: true},
	"indent_width":             {IndentWidth: 4},
}

func TestRoundTrip_WithConfig(t *testing.T) {
	for tname, tt := range testsWithConfig {
		t.Run(tname, func(t *testing.T) {
			a := golden.Read(t, tname)
			rj := journal.NewLoader().ResolveBytes(tname+".journal", a.Get("input"))
			pf := rj.Occurrences[0]
			var b strings.Builder
			if err := tt.Fprint(&b, pf.Ast); err != nil {
				t.Fatal(err)
			}

			golden.Assert(t, a, b.String())
		})
	}
}

func BenchmarkPrinter_Config(b *testing.B) {
	inp := golden.Read(b, "entries").Get("input")
	rj := journal.NewLoader().ResolveBytes("entries.journal", inp)
	pf := rj.Occurrences[0]
	for name, cfg := range testsWithConfig {
		b.Run(name, func(b *testing.B) {
			b.ReportAllocs()
			b.ResetTimer()
			for i := 0; i < b.N; i++ {
				var buf strings.Builder
				if err := cfg.Fprint(&buf, pf.Ast); err != nil {
					b.Fatal(err)
				}
			}
		})
	}
}