import Spec.Prelude

/-!
# Capacity —— SaaS capacity admission and abuse controls (ADR-0022)

初始生产服务共享有限的单机资源,但不能让一个 Organization 垄断容量或让无界输入拖垮
其他租户。ADR-0022 定义分层限制、持久 admission、显式背压和紧急制动;具体数值由
生产式容量测试校准,保持 `OPEN`。
-/

namespace Spec.System

/-- 首个生产版本必须有硬边界的容量维度(`PINNED`, ADR-0022)。金额与 token 用量是
统计和软告警,不在本硬限制集合中;各维度的具体数值为 `OPEN`,须由容量测试决定。 -/
inductive CapacityDimension where
  | requestRate
  | requestBodySize
  | agentConcurrency
  | admissionQueueLength
  | admissionQueueWait
  | fileSize
  | attachmentCount
  | archiveExpansion
  | projectStorage
  | organizationStorage
  | memberCount
  | projectCount
  | teamCount
  | folderCount
  | sessionCount
  | runWallTime
  | runTurns
  | runToolCalls
  | toolWallTime
  | runOutputSize
  | processMemory
  | processCpu
  | processCount

/-- 一个容量维度的分层限制(`PINNED`, ADR-0022):platform ceiling 永远存在且不可被
Organization 突破;Organization policy 可以缺省,也只能配置更低的限制。 -/
structure LayeredLimit where
  /-- 由版本化生产部署配置提供的不可突破上限(`PINNED`, ADR-0022)。 -/
  platformCeiling : Nat
  /-- Organization 管理员可选的更低策略限制(`PINNED`, ADR-0022)。 -/
  organizationLimit : Option Nat

/-- 分层限制是良构的 iff Organization policy 未设置或不高于 platform ceiling
(`PINNED`, ADR-0022)。不允许用 `none` 表示 platform unlimited。 -/
def LayeredLimit.Valid (limit : LayeredLimit) : Prop :=
  match limit.organizationLimit with
  | none => True
  | some organizationLimit => organizationLimit ≤ limit.platformCeiling

/-- 有效限制(`PINNED`, ADR-0022)取 platform ceiling 与 Organization policy 中较低者;
Organization policy 缺省时直接采用 platform ceiling,永不退化为 unlimited。 -/
def LayeredLimit.effective (limit : LayeredLimit) : Nat :=
  match limit.organizationLimit with
  | none => limit.platformCeiling
  | some organizationLimit => min limit.platformCeiling organizationLimit

/-- 已被服务接受的 agent run request 状态(`PINNED`, ADR-0022)。`expired` 和
`canceled` 都是可审计终态且不得自动执行;`started` 表示已从 admission queue 移交给
一个 AgentRun。被容量规则拒绝的输入从未被接受,因此不伪装成 queued request。 -/
inductive RunRequestState where
  | queued
  | started
  | expired
  | canceled

/-- 平台紧急工作负载制动模式(`PINNED`, ADR-0022)。`drain` 停止新 admission 与队列
启动但允许当前 run 完成;`stopNow` 还会取消当前 run。它们不等同于删除或封禁 org。 -/
inductive WorkloadBrakeMode where
  | open
  | drain
  | stopNow

/-- 紧急制动是否允许接收新的 agent 工作(`PINNED`, ADR-0022):只有 `open` 允许。 -/
def WorkloadBrakeMode.AcceptsNew : WorkloadBrakeMode → Prop
  | .open => True
  | .drain | .stopNow => False

/-- 紧急制动是否允许当前 agent run 继续(`PINNED`, ADR-0022):`drain` 允许收尾,
`stopNow` 要求取消。 -/
def WorkloadBrakeMode.AllowsActive : WorkloadBrakeMode → Prop
  | .open | .drain => True
  | .stopNow => False

end Spec.System
