package business

import (
	"context"
	"encoding/json"
	"sync"
	"sync/atomic"
	"testing"
	"time"

	"gorm.io/gorm"
	"kra/internal/biz/business"
)

func prepareAlertRecords(t *testing.T) (*gorm.DB, business.AlertRecordRepo) {
	t.Helper()
	db := prepareAccounts(t)
	for _, sql := range []string{
		"INSERT INTO sys_users(id,username,level,parent_id,id_path) VALUES(7,'owner',1,0,'0'),(8,'child',2,7,'0,7'),(9,'leaf',3,8,'0,7,8'),(10,'sibling',2,7,'0,7'),(99,'other',1,0,'0')",
		"INSERT INTO sys_user_bind_platform(user_id,source_db,platform_busier_id,status) VALUES(9,'lx',14,'0'),(10,'hw',14,'0'),(9,'hw',15,'3'),(99,'yq',14,'0')",
		"INSERT INTO tab_promote VALUES(100,'lx',14,1,0),(101,'lx',14,2,1),(100,'hw',14,1,1),(102,'hw',15,1,1),(100,'yq',14,1,1)",
		"CREATE TABLE media_file_upload_and_downloads(id INTEGER PRIMARY KEY,url TEXT,deleted_at DATETIME)",
		"INSERT INTO media_file_upload_and_downloads(id,url) VALUES(1,'/files/one.png'),(2,'/files/two.png')",
	} {
		if err := db.Exec(sql).Error; err != nil {
			t.Fatal(err)
		}
	}
	stamp := time.Date(2026, 11, 3, 10, 0, 0, 0, time.UTC)
	pos := []alertRecordPO{
		{ID: 1, PlatformCode: "lx", TargetPromoteID: 100, BusinessID: 7, RuleName: "注册", RuleType: "register_anomaly", Status: 1, CreatedAt: stamp, UpdatedAt: stamp},
		{ID: 2, PlatformCode: "lx", TargetPromoteID: 100, BusinessID: 7, RuleName: "充值", RuleType: "recharge_anomaly", Status: 3, CreatedAt: stamp, UpdatedAt: stamp},
		{ID: 3, PlatformCode: "hw", TargetPromoteID: 100, BusinessID: 7, Status: 2, CreatedAt: stamp},
		{ID: 4, PlatformCode: "yq", TargetPromoteID: 100, BusinessID: 99, Status: 5, CreatedAt: stamp},
		{ID: 5, PlatformCode: "lx", TargetPromoteID: 100, BusinessID: 99, Status: 1, CreatedAt: stamp},
		{ID: 6, PlatformCode: "lx", TargetPromoteID: 101, BusinessID: 7, Status: 1, CreatedAt: stamp},
		{ID: 7, PlatformCode: "hw", TargetPromoteID: 102, BusinessID: 7, Status: 1, CreatedAt: stamp},
	}
	if err := db.Create(&pos).Error; err != nil {
		t.Fatal(err)
	}
	return db, NewAlertRecordRepo(testData{db})
}

func alertTestScope(id uint) business.AlertRecordScope {
	return business.AlertRecordScope{Owner: business.LevelOneScope{UserID: 7}, Viewer: business.UserScope{SelfID: id, IncludeSelf: true, OnlySelf: id == 9, DescendantPath: map[uint]string{7: "0,7", 8: "0,7,8", 9: "0,7,8,9"}[id]}}
}

func TestAlertRecordsScopeAndSummaryUseDifferentFilters(t *testing.T) {
	_, repo := prepareAlertRecords(t)
	ctx := context.Background()
	for _, id := range []uint{8, 9} {
		page, err := repo.List(ctx, alertTestScope(id), business.AlertRecordFilter{})
		if err != nil || page.Total != 2 || len(page.Rows) != 2 || page.Rows[0].ID != 2 || page.Rows[1].ID != 1 {
			t.Fatalf("渠道状态、平台隔离、一级归属或排序错误: %+v %v", page, err)
		}
	}
	scope := alertTestScope(7)
	status := 1
	page, err := repo.List(ctx, scope, business.AlertRecordFilter{RuleName: "注册", Status: &status})
	if err != nil || page.Total != 1 || len(page.StatusCount) != 1 || page.StatusCount[1] != 1 {
		t.Fatalf("汇总应继承名称筛选并忽略状态自身: %+v %v", page, err)
	}
	selected := alertTestScope(8).Viewer
	scope.Selected = &selected
	page, err = repo.List(ctx, scope, business.AlertRecordFilter{})
	if err != nil || page.Total != 2 {
		t.Fatalf("二级商务筛选误用了 record.business_id: %+v %v", page, err)
	}
	outside := business.UserScope{SelfID: 99, OnlySelf: true}
	scope.Selected = &outside
	page, err = repo.List(ctx, scope, business.AlertRecordFilter{})
	if err != nil || page.Total != 0 || len(page.StatusCount) != 0 {
		t.Fatalf("指定其他商务绕过查看范围: %+v %v", page, err)
	}
	root := business.AlertRecordScope{Owner: business.LevelOneScope{All: true}, Viewer: business.UserScope{All: true}, Selected: &selected}
	page, err = repo.List(ctx, root, business.AlertRecordFilter{})
	if err != nil || page.Total != 3 {
		t.Fatalf("超管指定商务须收窄渠道但不增加一级归属约束: %+v %v", page, err)
	}
	if _, err := repo.Detail(ctx, alertTestScope(8), 3); err != business.ErrAlertRecordScope {
		t.Fatal("同号不同平台详情被放行")
	}
	if _, err := repo.Detail(ctx, alertTestScope(8), 5); err != business.ErrAlertRecordScope {
		t.Fatal("异常归属预警跨一级商务泄漏")
	}
}

func TestAlertRecordsMissingChannelsFailClosed(t *testing.T) {
	db, repo := prepareAlertRecords(t)
	if err := db.Exec("DROP TABLE tab_promote").Error; err != nil {
		t.Fatal(err)
	}
	ctx := context.Background()
	page, err := repo.List(ctx, alertTestScope(8), business.AlertRecordFilter{})
	if err != nil || page.Available || page.Total != 0 {
		t.Fatalf("缺渠道表未明确不可用: %+v %v", page, err)
	}
	if _, err := repo.Detail(ctx, alertTestScope(8), 1); err != business.ErrAlertChannelsUnavailable {
		t.Fatalf("详情应拒绝未知范围: %v", err)
	}
	root := business.AlertRecordScope{Owner: business.LevelOneScope{All: true}, Viewer: business.UserScope{All: true}}
	page, err = repo.List(ctx, root, business.AlertRecordFilter{})
	if err != nil || !page.Available || page.Total != 7 {
		t.Fatal("不限范围的超管查询不应依赖渠道表")
	}
	root.Selected = &business.UserScope{SelfID: 500, OnlySelf: true}
	page, err = repo.List(ctx, root, business.AlertRecordFilter{})
	if err != nil || !page.Available || page.Total != 0 {
		t.Fatal("无绑定应是确定空集")
	}
}

func TestBindingPortraitAlertsStayWithinSelectedSDKAccount(t *testing.T) {
	db, repo := prepareAlertRecords(t)
	if err := db.Exec("INSERT INTO sys_user_bind_platform(user_id,source_db,platform_busier_id,status) VALUES(9,'lx',16,'0')").Error; err != nil {
		t.Fatal(err)
	}
	if err := db.Exec("INSERT INTO tab_promote VALUES(103,'lx',16,1,1)").Error; err != nil {
		t.Fatal(err)
	}
	if err := db.Create(&alertRecordPO{ID: 8, PlatformCode: "lx", TargetPromoteID: 103, BusinessID: 7, Status: 1, CreatedAt: time.Now()}).Error; err != nil {
		t.Fatal(err)
	}
	ctx := context.Background()
	for _, sample := range []struct {
		busier uint64
		want   int64
	}{{14, 2}, {16, 1}, {999, 0}} {
		page, err := repo.List(ctx, alertTestScope(9), business.AlertRecordFilter{PlatformCodes: []string{"lx"}, PlatformBusierID: &sample.busier})
		if err != nil || page.Total != sample.want {
			t.Fatalf("binding %d: %+v %v", sample.busier, page, err)
		}
		var total int64
		for _, count := range page.StatusCount {
			total += count
		}
		if total != sample.want {
			t.Fatal("binding filter must also constrain status counts")
		}
	}
	busier := uint64(16)
	page, err := repo.List(ctx, alertTestScope(10), business.AlertRecordFilter{PlatformCodes: []string{"lx"}, PlatformBusierID: &busier})
	if err != nil || page.Total != 0 {
		t.Fatal("binding filter must not expand the viewer's permissions")
	}
}

func TestAlertRecordTransactionRollsBackWhenLogFails(t *testing.T) {
	db, repo := prepareAlertRecords(t)
	if err := db.Exec("CREATE TRIGGER reject_note BEFORE INSERT ON sys_alert_operation_log WHEN NEW.log_type = 2 BEGIN SELECT RAISE(ABORT, 'test failure'); END").Error; err != nil {
		t.Fatal(err)
	}
	err := repo.WithRecord(context.Background(), alertTestScope(8), 1, func(r *business.AlertRecord) ([]*business.AlertLog, error) {
		r.Status = 2
		return []*business.AlertLog{{LogType: 1, ActionDetail: "start"}, {LogType: 2, NoteContent: "备注"}}, nil
	})
	if err == nil {
		t.Fatal("日志失败应回滚")
	}
	var record alertRecordPO
	if err := db.First(&record, 1).Error; err != nil {
		t.Fatal(err)
	}
	var count int64
	db.Model(&alertLogPO{}).Count(&count)
	if record.Status != 1 || count != 0 {
		t.Fatal("留下半次状态流转或日志")
	}
}

func TestAlertRecordRepeatedConcurrentActionCreatesOneLog(t *testing.T) {
	db, repo := prepareAlertRecords(t)
	pool, _ := db.DB()
	// SQLite 用单连接串行事务；生产 MySQL 使用同一行锁回调，状态必须在锁内校验。
	pool.SetMaxOpenConns(1)
	var success atomic.Int32
	var wg sync.WaitGroup
	for i := 0; i < 12; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			err := repo.WithRecord(context.Background(), alertTestScope(8), 1, func(r *business.AlertRecord) ([]*business.AlertLog, error) {
				if r.Status != 1 {
					return nil, business.ErrAlertRecordChanged
				}
				r.Status = 2
				return []*business.AlertLog{{LogType: 1, ActionDetail: "start"}}, nil
			})
			if err == nil {
				success.Add(1)
			}
		}()
	}
	wg.Wait()
	var count int64
	db.Model(&alertLogPO{}).Count(&count)
	if success.Load() != 1 || count != 1 {
		t.Fatalf("重复请求写入多次: successes=%d logs=%d", success.Load(), count)
	}
}

func TestAlertRecordLogAttachmentsAndStableOrder(t *testing.T) {
	db, repo := prepareAlertRecords(t)
	stamp := time.Now()
	for i, raw := range []string{`[2,"1",999,2]`, `["/legacy/a.png","https://files.invalid/b.png"]`, `{"bad":true}`} {
		value := raw
		if err := db.Create(&alertLogPO{ID: uint64(i + 1), AlertID: 1, LogType: 2, AttachmentIDs: &value, CreatedAt: stamp}).Error; err != nil {
			t.Fatal(err)
		}
	}
	detail, err := repo.Detail(context.Background(), alertTestScope(8), 1)
	if err != nil {
		t.Fatal(err)
	}
	if len(detail.Logs) != 3 || detail.Logs[0].ID != 3 || len(detail.Logs[0].Attachments) != 0 || len(detail.Logs[1].Attachments) != 2 || len(detail.Logs[2].Attachments) != 2 || detail.Logs[2].Attachments[0] != "/files/two.png" {
		t.Fatalf("附件或顺序错误: %+v", detail.Logs)
	}
}

func TestAlertAttachmentMigrationUsesLegacyBackupAndIsIdempotent(t *testing.T) {
	db, _ := prepareAlertRecords(t)
	for _, sql := range []string{
		"CREATE TABLE bak_sw2kra_20261103_sys_alert_operation_log(id INTEGER,attachment_ids TEXT)",
		"INSERT INTO sw_kra_id_map(entity,old_id,new_id) VALUES('file','9007199254740993',2),('file','1',1)",
		`INSERT INTO sys_alert_operation_log(id,alert_id,attachment_ids) VALUES(10,1,'[9007199254740993,999]'),(11,1,'["/legacy.png"]'),(12,1,'[1]'),(13,1,'[2]')`,
		`INSERT INTO bak_sw2kra_20261103_sys_alert_operation_log VALUES(10,'[9007199254740993,999]'),(11,'["/legacy.png"]'),(13,'[9007199254740993]')`,
	} {
		if err := db.Exec(sql).Error; err != nil {
			t.Fatal(err)
		}
	}
	for i := 0; i < 2; i++ {
		if err := migrateAlertAttachments(db); err != nil {
			t.Fatal(err)
		}
	}
	var rows []alertLogPO
	db.Order("id").Find(&rows)
	if len(rows) != 4 {
		t.Fatal("日志丢失")
	}
	var ids []int64
	if json.Unmarshal([]byte(*rows[0].AttachmentIDs), &ids) != nil || len(ids) != 1 || ids[0] != 2 {
		t.Fatal("大 ID 映射或失效引用处理错误")
	}
	if *rows[1].AttachmentIDs != `["/legacy.png"]` || *rows[2].AttachmentIDs != `[1]` || *rows[3].AttachmentIDs != `[2]` {
		t.Fatal("改动了 URL、新日志或已经转换的附件")
	}
}