Struct ServerState 

pub struct ServerState {
    pub sessions: Arc<DashMap<String, UserSessions>>,
    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>,
    pub max_sessions_per_user: usize,
}

Shared state accessible by all axum handlers.

Fields

sessions: Arc<DashMap<String, UserSessions>>

Per-user multi-session registry (MSU).

ws_tx: Sender<String>

Broadcast sender for backward compat — delegates to the default user’s channel.

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. Default: 8 concurrent workflows.

max_sessions_per_user: usize

Maximum number of concurrent sessions per user. Default: 4.

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.

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

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

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

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

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

Set the max sessions per user.

pub async fn create_session( &self, user_key: &str, session_name: Option<String>, ) -> Result<String, SessionError>

Create a new session for a user. Returns the session_id.

Creates a fresh Session::new(), copies shared config, sets per-user isolation, creates a broadcast channel, spawns a drain task, and inserts into the user’s session DashMap. The new session becomes the “active” one.

pub async fn get_session( &self, user_key: &str, session_id: &str, ) -> Option<(Arc<Mutex<Session>>, Sender<String>)>

Get a specific session by user_key + session_id. Updates last_active on the session entry.

pub async fn list_sessions(&self, user_key: &str) -> Vec<SessionListItem>

List all active sessions for a user, sorted by last_active desc.

pub async fn destroy_session( &self, user_key: &str, session_id: &str, ) -> Result<(), SessionError>

Destroy a session. The session must be Idle.

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

Get or create the user’s “active” session for legacy routes.

Behavior:

1. If user has no sessions → create one
2. If active session exists → return it
3. If active session was destroyed → create a new one

Returns: (session_id, session_arc, ws_tx, token_store)

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

DEPRECATED: Use ensure_active_session() instead. Kept for backward compatibility — same 3-tuple return type.

pub async fn set_active_session(&self, user_key: &str, session_id: &str)

Set a session as the user’s active session.

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.

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.