hook

type AfterCallHook interface {
	Name() string
	AfterCall(ctx context.Context, call *CallContext, info AfterCallInfo) error
}

type AfterCallInfo struct {
	Status           string
	LatencyMs        int64
	ProviderType     string
	TargetModel      string
	PromptTokens     int64
	CompletionTokens int64
}

type AfterChatHook interface {
	Name() string
	AfterChat(ctx context.Context, info AfterChatInfo) error
}

type AfterChatInfo struct {
	Model        string
	Status       string
	LatencyMs    int64
	ProviderType string
	TargetModel  string
	// Dialect is the client wire format: "openai", "anthropic", or "gemini".
	Dialect string
	// Modality is the usage modality: "chat", "messages", or "generate_content".
	Modality string
}

type BeforeCallHook interface {
	Name() string
	BeforeCall(ctx context.Context, call *CallContext) (CallDecision, error)
}

type CallContext struct {
	Principal Principal
	Route     RouteContext
	Tags      map[string]string
	// Headers are sanitized inbound HTTP headers (lowercased keys) for policy/CEL.
	Headers  map[string]string
	Metadata map[string]any
	Body     []byte
	// RequestHeaders are applied to the upstream provider HTTP request after BeforeCall.
	// Keys are canonicalized with http.CanonicalHeaderKey when applied.
	RequestHeaders map[string]string
	// ResponseHeaders are merged onto the client HTTP response (allow and deny paths).
	ResponseHeaders map[string]string
}

type CallDecision struct {
	Allow   bool
	Status  int               // default 403 when deny and unset
	Reason  string            // machine code, e.g. policy_violation
	Message string            // human-readable; optional
	Headers map[string]string // optional response headers on deny
}

type ChatHook interface {
	Name() string
	BeforeChat(ctx context.Context, req chatir.Request) (chatir.Request, error)
}

type Principal struct {
	OrganizationID string
	ProjectID      string
	TeamID         string
	EnvironmentID  string
	APIKeyID       string
	SigningKeyID   string
	KeyID          string
	AuthMethod     string
	Kind           string
	OwnerUserID    string
	Name           string
}

type RouteContext struct {
	Model    string
	Path     string
	Stream   bool
	Modality string
}