package client

import (
	"bytes"
	"regexp"
	"testing"

	"github.com/stretchr/testify/require"

	renderer "github.com/bnema/vev-vt"
	"github.com/bnema/vev/internal/domain"
	"github.com/bnema/vev/internal/usecase/ui"
)

func TestDrawClientToastPlacement(t *testing.T) {
	size := domain.Size{Cols: 100, Rows: 30}
	tests := []struct {
		name   string
		anchor domain.Anchor
		check  func(t *testing.T, bounds domain.Rect)
	}{
		{name: "notice sits top-right", anchor: domain.AnchorTopRight, check: func(t *testing.T, bounds domain.Rect) {
			require.Less(t, bounds.Y, 3)
			require.Greater(t, bounds.X+bounds.Width, size.Cols-3)
		}},
		{name: "transition stays centered", anchor: domain.AnchorCenter, check: func(t *testing.T, bounds domain.Rect) {
			require.Equal(t, (size.Rows-bounds.Height)/2, bounds.Y)
			require.Equal(t, (size.Cols-bounds.Width)/2, bounds.X)
		}},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var out bytes.Buffer
			bounds, err := drawClientToast(&out, size, "SSH authentication failed", tt.anchor, domain.NoticeInfo)
			require.NoError(t, err)
			require.NotZero(t, out.Len())
			tt.check(t, bounds)
		})
	}
}

func TestClientToastLinesKeepExactWidth(t *testing.T) {
	sgr := regexp.MustCompile("\x1b\\[[0-9;]*m")
	tests := []struct {
		name    string
		message string
		border  string
	}{
		{name: "ascii", message: "host down"},
		{name: "wide rune", message: "hôte 東京 down"},
		{name: "colored border", message: "hôte 東京 down", border: toastBorderSGR(domain.NoticeError)},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			bounds := ui.ToastBounds(domain.Size{Cols: 100, Rows: 30}, ui.Toast{Message: tt.message, Anchor: domain.AnchorTopRight})
			for i, line := range clientToastLines(bounds, tt.message, tt.border) {
				width := 0
				for _, r := range sgr.ReplaceAllString(line, "") {
					width += renderer.RuneWidth(r)
				}
				require.Equal(t, bounds.Width, width, "row %d width", i)
			}
		})
	}
}

func TestDrawClientToastSeverityBorder(t *testing.T) {
	size := domain.Size{Cols: 100, Rows: 30}
	tests := []struct {
		name     string
		severity domain.NoticeSeverity
		want     string
	}{
		{name: "error border is red", severity: domain.NoticeError, want: "\x1b[38;5;167m"},
		{name: "warn border is yellow", severity: domain.NoticeWarn, want: "\x1b[38;5;179m"},
		{name: "info keeps the default color", severity: domain.NoticeInfo},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			var out bytes.Buffer
			_, err := drawClientToast(&out, size, "host down", domain.AnchorTopRight, tt.severity)
			require.NoError(t, err)
			if tt.want == "" {
				require.NotContains(t, out.String(), "\x1b[38;5;")
				return
			}
			require.Contains(t, out.String(), tt.want)
			require.Contains(t, out.String(), "host down")
		})
	}
}