package mail

import (
	"bytes"
	"html/template"
	"testing"

	"github.com/tamakyi/TamaBox/templates"
)

// TestSecurityAlertTemplateRenders 保证 security-alert.html 模板可解析，
// 且新增的「网络归属 / 代理网络提示」字段在参数齐全时正确渲染、
// proxy_hint 为空时不渲染提示块。
func TestSecurityAlertTemplateRenders(t *testing.T) {
	tmpl, err := template.ParseFS(templates.FS, "mail/security-alert.html")
	if err != nil {
		t.Fatalf("parse template: %v", err)
	}

	base := map[string]string{
		"ip":       "203.0.113.7",
		"provider": "Cloudflare（CDN）",
		"paths":    "/admin、/.env",
		"duration": "15",
		"time":     "2026-09-08 12:00:00",
		"year":     "2026",
	}

	// 1. 代理网络命中：应有归属行 + 代理提示
	var withHint bytes.Buffer
	params := cloneParams(base)
	params["proxy_hint"] = "1"
	if err := tmpl.Execute(&withHint, params); err != nil {
		t.Fatalf("execute with proxy_hint: %v", err)
	}
	out := withHint.String()
	for _, want := range []string{"203.0.113.7", "网络归属", "Cloudflare（CDN）", "代理网络"} {
		if !bytes.Contains([]byte(out), []byte(want)) {
			t.Errorf("rendered mail missing %q", want)
		}
	}

	// 2. 普通网络：不应出现代理提示块
	var noHint bytes.Buffer
	if err := tmpl.Execute(&noHint, cloneParams(base)); err != nil {
		t.Fatalf("execute without proxy_hint: %v", err)
	}
	if bytes.Contains(noHint.Bytes(), []byte("代理网络")) {
		t.Errorf("proxy hint rendered without proxy_hint param")
	}
}

func cloneParams(src map[string]string) map[string]string {
	dst := make(map[string]string, len(src))
	for k, v := range src {
		dst[k] = v
	}
	return dst
}