package model

import (
	"encoding/json"
	"time"
)

// AgentCapacityPolicy defines agent capacity and assignment rules.
// Reference: Chatwoot AgentCapacityPolicy + InboxCapacityLimit + P2B M11 spec
type AgentCapacityPolicy struct {
	ID              uint            `gorm:"primaryKey" json:"id"`
	AccountID       uint            `gorm:"not null;index" json:"account_id"`
	Name            string          `gorm:"size:255;not null" json:"name"`
	Description     string          `gorm:"size:500" json:"description,omitempty"`
	AssignmentLogic string          `gorm:"size:50" json:"assignment_logic"` // round_robin/least_busy
	ExclusionRules  json.RawMessage `gorm:"type:jsonb" json:"exclusion_rules"`
	CreatedAt       time.Time       `gorm:"autoCreateTime" json:"created_at"`
	UpdatedAt       time.Time       `gorm:"autoUpdateTime" json:"updated_at"`

	Account             Account              `gorm:"foreignKey:AccountID" json:"account,omitempty"`
	AccountUsers        []AccountUser        `gorm:"foreignKey:AgentCapacityPolicyID" json:"account_users,omitempty"`
	InboxCapacityLimits []InboxCapacityLimit `gorm:"foreignKey:AgentCapacityPolicyID" json:"inbox_capacity_limits,omitempty"`
}

// InboxCapacityLimit associates a capacity policy with an inbox and its conversation limit.
// Reference: Chatwoot InboxCapacityLimit + P2B M11 spec
type InboxCapacityLimit struct {
	ID                    uint      `gorm:"primaryKey" json:"id"`
	AgentCapacityPolicyID uint      `gorm:"not null;index;uniqueIndex:idx_policy_inbox_capacity_limit" json:"agent_capacity_policy_id"`
	InboxID               uint      `gorm:"not null;index;uniqueIndex:idx_policy_inbox_capacity_limit" json:"inbox_id"`
	ConversationLimit     int       `gorm:"not null;check:conversation_limit >= 0" json:"conversation_limit"`
	CreatedAt             time.Time `gorm:"autoCreateTime" json:"created_at"`
	UpdatedAt             time.Time `gorm:"autoUpdateTime" json:"updated_at"`

	AgentCapacityPolicy AgentCapacityPolicy `gorm:"foreignKey:AgentCapacityPolicyID" json:"agent_capacity_policy,omitempty"`
	Inbox               Inbox               `gorm:"foreignKey:InboxID" json:"inbox,omitempty"`
}

func (InboxCapacityLimit) TableName() string { return "inbox_capacity_limits" }

func (AgentCapacityPolicy) TableName() string { return "agent_capacity_policies" }