loki5512344
742cf03a5f
Server: - PostgreSQL backend (Pool enum over Sqlite/PgPool, all storage methods dual-dialect) - E2EE for DM (4 new protobuf packets, handlers, storage of ciphertext only) - ChannelEdit packet + server-authoritative channel create/delete - Gateway → voice-node membership signaling via broadcast channel Client: - E2EE crypto (X25519 from Ed25519, HKDF, ChaCha20-Poly1305) + UI lock icons - E2EE net layer (PID 0x0071-0x0074, JSON payloads, dispatch, session handlers) - Slint migration: all .slint files, bridge.rs, update.rs, main.rs rewritten - Removed egui UI (~50 files) and all dead code (clean clippy -D warnings)
82 lines
2.7 KiB
Rust
82 lines
2.7 KiB
Rust
use std::collections::HashMap;
|
|
use std::sync::Arc;
|
|
use std::sync::atomic::AtomicUsize;
|
|
use tokio::sync::{RwLock, broadcast};
|
|
|
|
use crate::admin::metrics::Metrics;
|
|
use crate::bootstrap::server_identity::ServerIdentity;
|
|
use crate::domain::rate_limit::RateLimiter;
|
|
use crate::domain::{
|
|
channels::ChannelStore, config::Config, session::SessionStore, storage::Storage,
|
|
};
|
|
use crate::proto::PresenceInfo;
|
|
|
|
pub type VoiceMemberTx = tokio::sync::broadcast::Sender<String>;
|
|
|
|
#[derive(Clone)]
|
|
pub struct State {
|
|
pub sessions: SessionStore,
|
|
pub channels: ChannelStore,
|
|
pub storage: Arc<Storage>,
|
|
pub config: Arc<Config>,
|
|
pub server_identity: Arc<ServerIdentity>,
|
|
pub broadcast: broadcast::Sender<BroadcastMsg>,
|
|
/// user_id → presence info (Phase 1.2)
|
|
pub presences: Arc<RwLock<HashMap<String, PresenceInfo>>>,
|
|
/// Prometheus-style counters.
|
|
pub metrics: Arc<Metrics>,
|
|
/// Live count of authenticated sessions.
|
|
pub sessions_count: Arc<AtomicUsize>,
|
|
/// Live count of channels with at least one member.
|
|
#[allow(dead_code)]
|
|
pub channels_count: Arc<AtomicUsize>,
|
|
/// Per-session rate limiter (token bucket).
|
|
pub rate_limiter: Arc<RateLimiter>,
|
|
/// Gateway → voice-node membership bridge.
|
|
pub voice_member_tx: Option<VoiceMemberTx>,
|
|
/// E2EE key store: user_id -> X25519 public key bytes
|
|
pub e2ee_keys: Arc<RwLock<HashMap<String, Vec<u8>>>>,
|
|
}
|
|
|
|
impl State {
|
|
#[allow(clippy::too_many_arguments)]
|
|
pub fn new(
|
|
sessions: SessionStore,
|
|
channels: ChannelStore,
|
|
storage: Arc<Storage>,
|
|
config: Arc<Config>,
|
|
server_identity: Arc<ServerIdentity>,
|
|
broadcast: broadcast::Sender<BroadcastMsg>,
|
|
metrics: Arc<Metrics>,
|
|
sessions_count: Arc<AtomicUsize>,
|
|
channels_count: Arc<AtomicUsize>,
|
|
voice_member_tx: Option<VoiceMemberTx>,
|
|
) -> Self {
|
|
let rate_per_sec = config.gateway.message_rate_per_sec.unwrap_or(5.0);
|
|
let burst = config.gateway.message_rate_burst.unwrap_or(10);
|
|
let rate_limiter = Arc::new(RateLimiter::new(rate_per_sec, burst));
|
|
Self {
|
|
sessions,
|
|
channels,
|
|
storage,
|
|
config,
|
|
server_identity,
|
|
broadcast,
|
|
presences: Arc::new(RwLock::new(HashMap::new())),
|
|
metrics,
|
|
sessions_count,
|
|
channels_count,
|
|
rate_limiter,
|
|
voice_member_tx,
|
|
e2ee_keys: Arc::new(RwLock::new(HashMap::new())),
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Clone, Debug)]
|
|
pub struct BroadcastMsg {
|
|
pub channel_id: Option<String>,
|
|
pub exclude_session: Option<String>,
|
|
pub target_session_id: Option<String>,
|
|
pub data: Vec<u8>,
|
|
}
|