routing

type AffinityEntry struct {
	Service   *loadbalance.Service
	MessageID string
	LockedAt  time.Time
	ExpiresAt time.Time
}

type AffinityStage struct {
	// contains filtered or unexported fields
}

type AffinityStore interface {
	Get(ruleUUID, sessionID string) (*AffinityEntry, bool)
	Set(ruleUUID, sessionID string, entry *AffinityEntry)
	CountByService(serviceID string) int // count active sessions locked to this service
}

type HealthStage struct {
	// contains filtered or unexported fields
}

type LoadBalancer interface {
	SelectService(rule *typ.Rule) (*loadbalance.Service, error)
	UpdateServiceIndex(rule *typ.Rule, service *loadbalance.Service)
}

type LoadBalancerStage struct {
	// contains filtered or unexported fields
}

type ProviderResolver interface {
	GetProviderByUUID(uuid string) (*typ.Provider, error)
	SaveCurrentServiceID(ruleUUID string, serviceID string) error
}

type SelectionContext struct {
	// Rule is the routing rule being evaluated
	Rule *typ.Rule

	// Request is the parsed API request (OpenAI/Anthropic params)
	Request interface{}

	// SessionID is the resolved session identifier for affinity
	// Priority: Anthropic metadata.user_id > X-Tingly-Session-ID header > ClientIP
	SessionID typ.SessionID

	// GinContext provides access to HTTP headers and client info
	GinContext *gin.Context

	// Scenario identifies the request type (openai, anthropic, etc.)
	Scenario typ.RuleScenario

	// MatchedSmartRuleIndex tracks which smart routing rule matched (-1 if none)
	// This is set by SmartRoutingStage and used for smart_rule-scoped affinity
	MatchedSmartRuleIndex int
}

type SelectionResult struct {
	// Service is the selected load-balanced service
	Service *loadbalance.Service

	// FilteredServices contains a narrowed candidate set produced by filter stages.
	// When set, selector updates SelectionContext.CandidateServices with this value.
	FilteredServices []*loadbalance.Service

	// Provider is the resolved provider for the service
	Provider *typ.Provider

	// Source indicates which stage selected this service
	// Values: "affinity", "smart_routing", "load_balancer"
	Source string

	// EvaluatedStages tracks which stages were evaluated (for observability)
	EvaluatedStages []string

	// MatchedSmartRuleIndex is the index of the matched smart routing rule
	// -1 if no smart routing rule matched
	MatchedSmartRuleIndex int
}

type SelectionStage interface {
	// Name returns the stage identifier for logging and metrics
	Name() string

	// Evaluate attempts to select a service based on the context.
	// Returns:
	//   - (result, true) if this stage selected a service (stops pipeline)
	//   - (nil, false) if this stage cannot select (continue to next stage)
	Evaluate(ctx *SelectionContext, state *selectionState) (*SelectionResult, bool)
}

type ServiceSelector struct {
	// contains filtered or unexported fields
}

type SimpleSelector struct {
	// contains filtered or unexported fields
}

type SmartRoutingStage struct {
	// contains filtered or unexported fields
}