package smtp_server

import (
	"bytes"
	"context"
	"encoding/json"
	"errors"
	"net"
	"strings"
	"testing"

	"git.pcux.cn/xuans/Smail/config"
	"git.pcux.cn/xuans/Smail/db/migration"
	"git.pcux.cn/xuans/Smail/services/logstore"
	appcontext "git.pcux.cn/xuans/Smail/utils/context"
)

type failedMailReader struct{}

func (failedMailReader) Read([]byte) (int, error) { return 0, errors.New("private-reader-error") }

func TestSMTPDataLogDistinguishesDiscardedMailAndReadFailure(t *testing.T) {
	engine, _ := newAuditTestEngine(t)
	config.Instance.SpamFilterLevel = 3
	migrations := migration.DefaultMigrations()
	for _, statement := range migrations[len(migrations)-1].StatementsFor(migration.DialectSQLite) {
		if _, err := engine.Exec(statement); err != nil {
			t.Fatal(err)
		}
	}
	store, err := logstore.New(engine.DB().DB, migration.DialectSQLite)
	if err != nil {
		t.Fatal(err)
	}
	runtime := logstore.Start(store)
	defer runtime.Stop()
	session := &Session{
		Ctx:  &appcontext.Context{Context: context.Background()},
		From: "sender@example.net", To: []string{"absent@example.com"},
		RemoteAddress: &net.TCPAddr{IP: net.IPv4(192, 168, 1, 2), Port: 25},
	}
	message := "From: sender@example.net\r\nTo: absent@example.com\r\nSubject: private-subject\r\n\r\nprivate-body"
	if err := session.Data(bytes.NewBufferString(message)); err != nil {
		t.Fatal(err)
	}
	assertAuditCounts(t, engine, 0, 0)
	for _, result := range []string{"discarded", "failure"} {
		if result == "failure" && session.Data(failedMailReader{}) == nil {
			t.Fatal("读取失败未返回错误")
		}
		entries, err := store.Query(context.Background(), logstore.Query{Kind: "mail", Limit: 1})
		if err != nil || len(entries) != 1 {
			t.Fatal("缺少收信处理记录", err)
		}
		entry := entries[0]
		var details logstore.MailDetails
		if err := json.Unmarshal(entry.Payload, &details); err != nil {
			t.Fatal(err)
		}
		if entry.Result != result || details.Trigger != "SMTP 收信" || details.Stage == "" || details.Outcome == "" {
			t.Fatalf("收信状态错误：%+v", entry)
		}
		if strings.Contains(string(entry.Payload), "private-") {
			t.Fatal("收信日志泄露内容或原始错误")
		}
	}
}