Struct ServerState 

pub struct ServerState {
    pub sessions: Arc<DashMap<String, UserSession>>,
    pub ws_tx: Sender<String>,
    pub auth: Option<Arc<AuthState>>,
    pub config_toml: Option<String>,
    pub secrets: Option<HashMap<String, String>>,
    pub build_info: Option<BuildInfo>,
    pub workflow_semaphore: Arc<Semaphore>,
}

Shared state accessible by all axum handlers.

Fields

sessions: Arc<DashMap<String, UserSession>>

Per-user session registry (TMU Phase 2).

ws_tx: Sender<String>

Broadcast sender for backward compat — delegates to the default user’s channel. New code should use user_ws_tx(user_key) instead.

auth: Option<Arc<AuthState>>

Auth state (None = auth disabled, all endpoints open).

config_toml: Option<String>

Shared config TOML (all users share the same agent config).

secrets: Option<HashMap<String, String>>

Shared secrets (injected into every per-user session).

build_info: Option<BuildInfo>

Shared build info (injected into every per-user session).

workflow_semaphore: Arc<Semaphore>

Global concurrency limiter — limits the number of simultaneous workflows across all users. Prevents resource exhaustion on shared infrastructure. Default: 8 concurrent workflows.

Implementations

impl ServerState

pub fn new( session: Session, ws_tx: Sender<String>, auth: Option<Arc<AuthState>>, ) -> Self

Create new shared state from a default session, broadcast sender, and optional auth.

The provided session becomes the "default" user’s session. When auth is enabled, authenticated users get their own sessions created on demand.

pub fn with_config_toml(self, config_toml: String) -> Self

Set the shared config TOML.

pub fn with_secrets(self, secrets: HashMap<String, String>) -> Self

Set the shared secrets.

pub fn with_build_info(self, build_info: BuildInfo) -> Self

Set the shared build info.

pub fn with_max_concurrent_workflows(self, max: usize) -> Self

Set the max concurrent workflows.

pub async fn ensure_user_session( &self, user_key: &str, ) -> (Arc<Mutex<Session>>, Sender<String>, Arc<MemoryTokenStore>)

Get or create a session for the given user key, returning the session + ws_tx.

This is the main entry point for route handlers. It ensures the user has a session, spawns a drain task if newly created, and returns references to the session mutex, broadcast sender, and token store.

pub fn spawn_drain_task(self, workflow_rx: UnboundedReceiver<TuiMessage>)

Spawn the default user’s drain task (backward compatibility).

Called during server startup for the initial session.

pub async fn resolve_user_key(&self, headers: &HeaderMap) -> String

Resolve the user key from request headers.

Returns "default" when auth is disabled. Returns the JWT sub claim (or dev:email for dev mode) when auth is enabled.

Trait Implementations

impl Clone for ServerState

fn clone(&self) -> ServerState

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.