package autoassignment

import (
	"context"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"gorm.io/driver/sqlite"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"

	"github.com/gochat/gochat/internal/model"
)

func setupCapacityAssignmentDB(t *testing.T) *gorm.DB {
	t.Helper()
	db, err := gorm.Open(sqlite.Open("file::memory:"), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
	require.NoError(t, err)
	require.NoError(t, db.AutoMigrate(
		&model.Account{},
		&model.User{},
		&model.AccountUser{},
		&model.Inbox{},
		&model.InboxMember{},
		&model.Contact{},
		&model.Conversation{},
		&model.AgentCapacityPolicy{},
		&model.InboxCapacityLimit{},
	))
	return db
}

func TestAssignmentService_SkipsAgentsAtInboxCapacity(t *testing.T) {
	_, rdb := setupTestRedis(t)
	db := setupCapacityAssignmentDB(t)
	svc := NewAssignmentService(db, rdb)
	ctx := context.Background()

	account := &model.Account{Name: "Capacity Account"}
	require.NoError(t, db.Create(account).Error)
	inbox := &model.Inbox{AccountID: account.ID, Name: "Capacity Inbox", ChannelType: string(model.InboxChannelTypeWebWidget), EnableAutoAssignment: true}
	require.NoError(t, db.Create(inbox).Error)
	contact := &model.Contact{AccountID: account.ID, Name: "Capacity Contact"}
	require.NoError(t, db.Create(contact).Error)

	agentAtCapacity := createCapacityUser(t, db, account.ID, "full@test.com")
	agentWithCapacity := createCapacityUser(t, db, account.ID, "available@test.com")
	require.NoError(t, db.Create(&model.InboxMember{InboxID: inbox.ID, UserID: agentAtCapacity.ID}).Error)
	require.NoError(t, db.Create(&model.InboxMember{InboxID: inbox.ID, UserID: agentWithCapacity.ID}).Error)

	policy := &model.AgentCapacityPolicy{AccountID: account.ID, Name: "Capacity", AssignmentLogic: "round_robin", ExclusionRules: []byte(`{}`)}
	require.NoError(t, db.Create(policy).Error)
	require.NoError(t, db.Model(&model.AccountUser{}).
		Where("account_id = ? AND user_id = ?", account.ID, agentAtCapacity.ID).
		Update("agent_capacity_policy_id", policy.ID).Error)
	require.NoError(t, db.Create(&model.InboxCapacityLimit{AgentCapacityPolicyID: policy.ID, InboxID: inbox.ID, ConversationLimit: 1}).Error)

	now := time.Now().Unix()
	existing := &model.Conversation{AccountID: account.ID, InboxID: inbox.ID, ContactID: contact.ID, AssigneeID: &agentAtCapacity.ID, Status: string(model.ConversationStatusOpen), ChannelType: "web_widget", Channel: "web_widget", LastActivityAt: &now}
	unassigned := &model.Conversation{AccountID: account.ID, InboxID: inbox.ID, ContactID: contact.ID, Status: string(model.ConversationStatusOpen), ChannelType: "web_widget", Channel: "web_widget", LastActivityAt: &now}
	require.NoError(t, db.Create(existing).Error)
	require.NoError(t, db.Create(unassigned).Error)

	agentID, err := svc.AssignConversation(ctx, unassigned.ID, inbox.ID, account.ID)
	require.NoError(t, err)
	assert.Equal(t, agentWithCapacity.ID, agentID)
}

func createCapacityUser(t *testing.T, db *gorm.DB, accountID uint, email string) *model.User {
	t.Helper()
	user := &model.User{AccountID: accountID, Name: email, Email: email, Password: "hashed", Active: true, Available: true}
	require.NoError(t, db.Create(user).Error)
	require.NoError(t, db.Create(&model.AccountUser{AccountID: accountID, UserID: user.ID, Role: "agent", Availability: "online"}).Error)
	return user
}