Skip to main content

sockudo_core/
options.rs

1use crate::app::App;
2use crate::utils::{parse_bool_env, parse_env, parse_env_optional};
3use ahash::AHashMap;
4use serde::{Deserialize, Serialize};
5use std::str::FromStr;
6use tracing::{info, warn};
7use url::Url;
8
9// Custom deserializer for octal permission mode (string format only, like chmod)
10fn deserialize_octal_permission<'de, D>(deserializer: D) -> Result<u32, D::Error>
11where
12    D: serde::Deserializer<'de>,
13{
14    use serde::de::{self};
15
16    let s = String::deserialize(deserializer)?;
17
18    // Validate that the string contains only octal digits
19    if !s.chars().all(|c| c.is_digit(8)) {
20        return Err(de::Error::custom(format!(
21            "invalid octal permission mode '{}': must contain only digits 0-7",
22            s
23        )));
24    }
25
26    // Parse as octal
27    let mode = u32::from_str_radix(&s, 8)
28        .map_err(|_| de::Error::custom(format!("invalid octal permission mode: {}", s)))?;
29
30    // Validate it's within valid Unix permission range
31    if mode > 0o777 {
32        return Err(de::Error::custom(format!(
33            "permission mode '{}' exceeds maximum value 777",
34            s
35        )));
36    }
37
38    Ok(mode)
39}
40
41// Helper function to parse driver enums with fallback behavior (matches main.rs)
42fn parse_driver_enum<T: FromStr + Clone + std::fmt::Debug>(
43    driver_str: String,
44    default_driver: T,
45    driver_name: &str,
46) -> T
47where
48    <T as FromStr>::Err: std::fmt::Debug,
49{
50    match T::from_str(&driver_str.to_lowercase()) {
51        Ok(driver_enum) => driver_enum,
52        Err(e) => {
53            warn!(
54                "Failed to parse {} driver from string '{}': {:?}. Using default: {:?}.",
55                driver_name, driver_str, e, default_driver
56            );
57            default_driver
58        }
59    }
60}
61
62fn override_db_pool_settings(db_conn: &mut DatabaseConnection, prefix: &str) {
63    if let Some(min) = parse_env_optional::<u32>(&format!("{}_POOL_MIN", prefix)) {
64        db_conn.pool_min = Some(min);
65    }
66    if let Some(max) = parse_env_optional::<u32>(&format!("{}_POOL_MAX", prefix)) {
67        db_conn.pool_max = Some(max);
68    }
69}
70
71// --- Enums for Driver Types ---
72
73#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
74#[serde(rename_all = "lowercase")]
75pub enum AdapterDriver {
76    #[default]
77    Local,
78    Redis,
79    #[serde(rename = "redis-cluster")]
80    RedisCluster,
81    Nats,
82    Pulsar,
83    RabbitMq,
84    #[serde(rename = "google-pubsub")]
85    GooglePubSub,
86    Kafka,
87    Iggy,
88}
89
90#[derive(Debug, Clone, Serialize, Deserialize)]
91#[serde(default)]
92pub struct DynamoDbSettings {
93    pub region: String,
94    pub table_name: String,
95    pub endpoint_url: Option<String>,
96    pub aws_access_key_id: Option<String>,
97    pub aws_secret_access_key: Option<String>,
98    pub aws_profile_name: Option<String>,
99}
100
101impl Default for DynamoDbSettings {
102    fn default() -> Self {
103        Self {
104            region: "us-east-1".to_string(),
105            table_name: "sockudo-applications".to_string(),
106            endpoint_url: None,
107            aws_access_key_id: None,
108            aws_secret_access_key: None,
109            aws_profile_name: None,
110        }
111    }
112}
113
114#[derive(Debug, Clone, Serialize, Deserialize)]
115#[serde(default)]
116pub struct ScyllaDbSettings {
117    pub nodes: Vec<String>,
118    pub keyspace: String,
119    pub table_name: String,
120    pub username: Option<String>,
121    pub password: Option<String>,
122    pub replication_class: String,
123    pub replication_factor: u32,
124}
125
126impl Default for ScyllaDbSettings {
127    fn default() -> Self {
128        Self {
129            nodes: vec!["127.0.0.1:9042".to_string()],
130            keyspace: "sockudo".to_string(),
131            table_name: "applications".to_string(),
132            username: None,
133            password: None,
134            replication_class: "SimpleStrategy".to_string(),
135            replication_factor: 3,
136        }
137    }
138}
139
140#[derive(Debug, Clone, Serialize, Deserialize)]
141#[serde(default)]
142pub struct SurrealDbSettings {
143    pub url: String,
144    pub namespace: String,
145    pub database: String,
146    pub username: String,
147    pub password: String,
148    pub table_name: String,
149    pub cache_ttl: u64,
150    pub cache_max_capacity: u64,
151}
152
153impl Default for SurrealDbSettings {
154    fn default() -> Self {
155        Self {
156            url: "ws://127.0.0.1:8000".to_string(),
157            namespace: "sockudo".to_string(),
158            database: "sockudo".to_string(),
159            username: "root".to_string(),
160            password: "root".to_string(),
161            table_name: "applications".to_string(),
162            cache_ttl: 300,
163            cache_max_capacity: 100,
164        }
165    }
166}
167
168impl FromStr for AdapterDriver {
169    type Err = String;
170    fn from_str(s: &str) -> Result<Self, Self::Err> {
171        match s.to_lowercase().as_str() {
172            "local" => Ok(AdapterDriver::Local),
173            "redis" => Ok(AdapterDriver::Redis),
174            "redis-cluster" => Ok(AdapterDriver::RedisCluster),
175            "nats" => Ok(AdapterDriver::Nats),
176            "pulsar" => Ok(AdapterDriver::Pulsar),
177            "rabbitmq" | "rabbit-mq" => Ok(AdapterDriver::RabbitMq),
178            "google-pubsub" | "gcp-pubsub" | "pubsub" => Ok(AdapterDriver::GooglePubSub),
179            "kafka" => Ok(AdapterDriver::Kafka),
180            "iggy" | "apache-iggy" | "apache_iggy" => Ok(AdapterDriver::Iggy),
181            _ => Err(format!("Unknown adapter driver: {s}")),
182        }
183    }
184}
185
186#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
187#[serde(rename_all = "lowercase")]
188pub enum AppManagerDriver {
189    #[default]
190    Memory,
191    Mysql,
192    Dynamodb,
193    PgSql,
194    SurrealDb,
195    ScyllaDb,
196}
197impl FromStr for AppManagerDriver {
198    type Err = String;
199    fn from_str(s: &str) -> Result<Self, Self::Err> {
200        match s.to_lowercase().as_str() {
201            "memory" => Ok(AppManagerDriver::Memory),
202            "mysql" => Ok(AppManagerDriver::Mysql),
203            "dynamodb" => Ok(AppManagerDriver::Dynamodb),
204            "pgsql" | "postgres" | "postgresql" => Ok(AppManagerDriver::PgSql),
205            "surreal" | "surrealdb" => Ok(AppManagerDriver::SurrealDb),
206            "scylladb" | "scylla" => Ok(AppManagerDriver::ScyllaDb),
207            _ => Err(format!("Unknown app manager driver: {s}")),
208        }
209    }
210}
211
212#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
213#[serde(rename_all = "lowercase")]
214pub enum CacheDriver {
215    #[default]
216    Memory,
217    Redis,
218    #[serde(rename = "redis-cluster")]
219    RedisCluster,
220    None,
221}
222
223impl FromStr for CacheDriver {
224    type Err = String;
225    fn from_str(s: &str) -> Result<Self, Self::Err> {
226        match s.to_lowercase().as_str() {
227            "memory" => Ok(CacheDriver::Memory),
228            "redis" => Ok(CacheDriver::Redis),
229            "redis-cluster" => Ok(CacheDriver::RedisCluster),
230            "none" => Ok(CacheDriver::None),
231            _ => Err(format!("Unknown cache driver: {s}")),
232        }
233    }
234}
235
236#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
237#[serde(rename_all = "lowercase")]
238pub enum QueueDriver {
239    Memory,
240    #[default]
241    Redis,
242    #[serde(rename = "redis-cluster")]
243    RedisCluster,
244    Nats,
245    Pulsar,
246    RabbitMq,
247    #[serde(rename = "google-pubsub")]
248    GooglePubSub,
249    Kafka,
250    Iggy,
251    Sqs,
252    Sns,
253    None,
254}
255
256impl FromStr for QueueDriver {
257    type Err = String;
258    fn from_str(s: &str) -> Result<Self, Self::Err> {
259        match s.to_lowercase().as_str() {
260            "memory" => Ok(QueueDriver::Memory),
261            "redis" => Ok(QueueDriver::Redis),
262            "redis-cluster" => Ok(QueueDriver::RedisCluster),
263            "nats" => Ok(QueueDriver::Nats),
264            "pulsar" => Ok(QueueDriver::Pulsar),
265            "rabbitmq" | "rabbit-mq" => Ok(QueueDriver::RabbitMq),
266            "google-pubsub" | "gcp-pubsub" | "pubsub" => Ok(QueueDriver::GooglePubSub),
267            "kafka" => Ok(QueueDriver::Kafka),
268            "iggy" | "apache-iggy" | "apache_iggy" => Ok(QueueDriver::Iggy),
269            "sqs" => Ok(QueueDriver::Sqs),
270            "sns" => Ok(QueueDriver::Sns),
271            "none" => Ok(QueueDriver::None),
272            _ => Err(format!("Unknown queue driver: {s}")),
273        }
274    }
275}
276
277#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
278#[serde(rename_all = "snake_case")]
279pub enum DeltaCoordinationBackend {
280    #[default]
281    Auto,
282    None,
283    Redis,
284    RedisCluster,
285    Nats,
286}
287
288impl FromStr for DeltaCoordinationBackend {
289    type Err = String;
290
291    fn from_str(s: &str) -> Result<Self, Self::Err> {
292        match s.trim().to_ascii_lowercase().as_str() {
293            "auto" => Ok(Self::Auto),
294            "none" => Ok(Self::None),
295            "redis" => Ok(Self::Redis),
296            "redis-cluster" | "redis_cluster" => Ok(Self::RedisCluster),
297            "nats" => Ok(Self::Nats),
298            _ => Err(format!("Unknown delta coordination backend: {s}")),
299        }
300    }
301}
302
303impl AsRef<str> for QueueDriver {
304    fn as_ref(&self) -> &str {
305        match self {
306            QueueDriver::Memory => "memory",
307            QueueDriver::Redis => "redis",
308            QueueDriver::RedisCluster => "redis-cluster",
309            QueueDriver::Nats => "nats",
310            QueueDriver::Pulsar => "pulsar",
311            QueueDriver::RabbitMq => "rabbitmq",
312            QueueDriver::GooglePubSub => "google-pubsub",
313            QueueDriver::Kafka => "kafka",
314            QueueDriver::Iggy => "iggy",
315            QueueDriver::Sqs => "sqs",
316            QueueDriver::Sns => "sns",
317            QueueDriver::None => "none",
318        }
319    }
320}
321
322#[derive(Debug, Clone, Serialize, Deserialize)]
323#[serde(default)]
324pub struct RedisClusterQueueConfig {
325    pub concurrency: u32,
326    pub prefix: Option<String>,
327    pub nodes: Vec<String>,
328    pub request_timeout_ms: u64,
329}
330
331impl Default for RedisClusterQueueConfig {
332    fn default() -> Self {
333        Self {
334            concurrency: 5,
335            prefix: Some("sockudo_queue:".to_string()),
336            nodes: vec!["redis://127.0.0.1:6379".to_string()],
337            request_timeout_ms: 5000,
338        }
339    }
340}
341
342#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
343#[serde(rename_all = "lowercase")]
344pub enum MetricsDriver {
345    #[default]
346    Prometheus,
347}
348
349#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
350#[serde(rename_all = "lowercase")]
351pub enum LogOutputFormat {
352    #[default]
353    Human,
354    Json,
355}
356
357impl FromStr for LogOutputFormat {
358    type Err = String;
359    fn from_str(s: &str) -> Result<Self, Self::Err> {
360        match s.to_lowercase().as_str() {
361            "human" => Ok(LogOutputFormat::Human),
362            "json" => Ok(LogOutputFormat::Json),
363            _ => Err(format!("Unknown log output format: {s}")),
364        }
365    }
366}
367
368impl FromStr for MetricsDriver {
369    type Err = String;
370    fn from_str(s: &str) -> Result<Self, Self::Err> {
371        match s.to_lowercase().as_str() {
372            "prometheus" => Ok(MetricsDriver::Prometheus),
373            _ => Err(format!("Unknown metrics driver: {s}")),
374        }
375    }
376}
377
378impl AsRef<str> for MetricsDriver {
379    fn as_ref(&self) -> &str {
380        match self {
381            MetricsDriver::Prometheus => "prometheus",
382        }
383    }
384}
385
386// --- Main Configuration Struct ---
387#[derive(Debug, Clone, Serialize, Deserialize)]
388#[serde(default)]
389pub struct ServerOptions {
390    pub adapter: AdapterConfig,
391    pub app_manager: AppManagerConfig,
392    pub cache: CacheConfig,
393    pub channel_limits: ChannelLimits,
394    pub cors: CorsConfig,
395    pub database: DatabaseConfig,
396    pub database_pooling: DatabasePooling,
397    pub debug: bool,
398    pub event_limits: EventLimits,
399    pub host: String,
400    pub http_api: HttpApiConfig,
401    pub instance: InstanceConfig,
402    pub logging: Option<LoggingConfig>,
403    pub metrics: MetricsConfig,
404    pub mode: String,
405    pub port: u16,
406    pub path_prefix: String,
407    pub presence: PresenceConfig,
408    pub queue: QueueConfig,
409    pub rate_limiter: RateLimiterConfig,
410    pub shutdown_grace_period: u64,
411    pub ssl: SslConfig,
412    pub user_authentication_timeout: u64,
413    pub webhooks: WebhooksConfig,
414    pub websocket_max_payload_kb: u32,
415    pub cleanup: CleanupConfig,
416    pub activity_timeout: u64,
417    pub cluster_health: ClusterHealthConfig,
418    pub unix_socket: UnixSocketConfig,
419    pub delta_compression: DeltaCompressionOptionsConfig,
420    pub tag_filtering: TagFilteringConfig,
421    pub websocket: WebSocketConfig,
422    pub connection_recovery: ConnectionRecoveryConfig,
423    pub history: HistoryConfig,
424    pub presence_history: PresenceHistoryConfig,
425    pub idempotency: IdempotencyConfig,
426    pub ephemeral: EphemeralConfig,
427    pub echo_control: EchoControlConfig,
428    pub event_name_filtering: EventNameFilteringConfig,
429    pub versioned_messages: VersionedMessagesConfig,
430    pub annotations: AnnotationsConfig,
431    pub push: PushConfig,
432    /// Timeout in milliseconds for each subsystem check in the `/up` health endpoint.
433    /// Applies to adapter, cache, queue, and app manager checks independently.
434    pub health_check_timeout_ms: u64,
435}
436
437// --- Configuration Sub-Structs ---
438
439#[derive(Debug, Clone, Serialize, Deserialize)]
440#[serde(default)]
441pub struct SqsQueueConfig {
442    pub region: String,
443    pub queue_url_prefix: Option<String>,
444    pub visibility_timeout: i32,
445    pub endpoint_url: Option<String>,
446    pub max_messages: i32,
447    pub wait_time_seconds: i32,
448    pub concurrency: u32,
449    pub fifo: bool,
450    pub message_group_id: Option<String>,
451}
452
453#[derive(Debug, Clone, Serialize, Deserialize)]
454#[serde(default)]
455pub struct SnsQueueConfig {
456    pub region: String,
457    pub topic_arn: String,
458    pub endpoint_url: Option<String>,
459}
460
461#[derive(Debug, Clone, Serialize, Deserialize)]
462#[serde(default)]
463pub struct AdapterConfig {
464    pub driver: AdapterDriver,
465    pub redis: RedisAdapterConfig,
466    pub cluster: RedisClusterAdapterConfig,
467    pub nats: NatsAdapterConfig,
468    pub pulsar: PulsarAdapterConfig,
469    pub rabbitmq: RabbitMqAdapterConfig,
470    pub google_pubsub: GooglePubSubAdapterConfig,
471    pub kafka: KafkaAdapterConfig,
472    pub iggy: IggyConfig,
473    #[serde(default = "default_buffer_multiplier_per_cpu")]
474    pub buffer_multiplier_per_cpu: usize,
475    pub cluster_health: ClusterHealthConfig,
476    #[serde(default = "default_enable_socket_counting")]
477    pub enable_socket_counting: bool,
478    #[serde(default = "default_fallback_to_local")]
479    pub fallback_to_local: bool,
480}
481
482fn default_enable_socket_counting() -> bool {
483    true
484}
485
486fn default_fallback_to_local() -> bool {
487    true
488}
489
490fn default_buffer_multiplier_per_cpu() -> usize {
491    64
492}
493
494impl Default for AdapterConfig {
495    fn default() -> Self {
496        Self {
497            driver: AdapterDriver::default(),
498            redis: RedisAdapterConfig::default(),
499            cluster: RedisClusterAdapterConfig::default(),
500            nats: NatsAdapterConfig::default(),
501            pulsar: PulsarAdapterConfig::default(),
502            rabbitmq: RabbitMqAdapterConfig::default(),
503            google_pubsub: GooglePubSubAdapterConfig::default(),
504            kafka: KafkaAdapterConfig::default(),
505            iggy: IggyConfig::default(),
506            buffer_multiplier_per_cpu: default_buffer_multiplier_per_cpu(),
507            cluster_health: ClusterHealthConfig::default(),
508            enable_socket_counting: default_enable_socket_counting(),
509            fallback_to_local: default_fallback_to_local(),
510        }
511    }
512}
513
514#[derive(Debug, Clone, Serialize, Deserialize)]
515#[serde(default)]
516pub struct RedisAdapterConfig {
517    pub requests_timeout: u64,
518    pub prefix: String,
519    pub redis_pub_options: AHashMap<String, sonic_rs::Value>,
520    pub redis_sub_options: AHashMap<String, sonic_rs::Value>,
521    pub cluster_mode: bool,
522}
523
524#[derive(Debug, Clone, Serialize, Deserialize)]
525#[serde(default)]
526pub struct RedisClusterAdapterConfig {
527    pub nodes: Vec<String>,
528    pub prefix: String,
529    pub request_timeout_ms: u64,
530    pub use_connection_manager: bool,
531    #[serde(default)]
532    pub use_sharded_pubsub: bool,
533}
534
535#[derive(Debug, Clone, Serialize, Deserialize)]
536#[serde(default)]
537pub struct NatsAdapterConfig {
538    pub servers: Vec<String>,
539    pub prefix: String,
540    pub request_timeout_ms: u64,
541    pub username: Option<String>,
542    pub password: Option<String>,
543    pub token: Option<String>,
544    pub connection_timeout_ms: u64,
545    pub nodes_number: Option<u32>,
546    pub discovery_max_wait_ms: u64,
547    pub discovery_idle_wait_ms: u64,
548    pub subscription_capacity: Option<usize>,
549    pub client_capacity: Option<usize>,
550    pub max_reconnects: Option<usize>,
551    pub presence_sync_chunk_size: Option<usize>,
552}
553
554#[derive(Debug, Clone, Serialize, Deserialize)]
555#[serde(default)]
556pub struct PulsarAdapterConfig {
557    pub url: String,
558    pub prefix: String,
559    pub request_timeout_ms: u64,
560    pub token: Option<String>,
561    pub nodes_number: Option<u32>,
562}
563
564#[derive(Debug, Clone, Serialize, Deserialize)]
565#[serde(default)]
566pub struct RabbitMqAdapterConfig {
567    pub url: String,
568    pub prefix: String,
569    pub request_timeout_ms: u64,
570    pub connection_timeout_ms: u64,
571    pub nodes_number: Option<u32>,
572}
573
574#[derive(Debug, Clone, Serialize, Deserialize)]
575#[serde(default)]
576pub struct GooglePubSubAdapterConfig {
577    pub project_id: String,
578    pub prefix: String,
579    pub request_timeout_ms: u64,
580    pub emulator_host: Option<String>,
581    pub nodes_number: Option<u32>,
582}
583
584#[derive(Debug, Clone, Serialize, Deserialize)]
585#[serde(default)]
586pub struct KafkaAdapterConfig {
587    pub brokers: Vec<String>,
588    pub prefix: String,
589    pub request_timeout_ms: u64,
590    pub security_protocol: Option<String>,
591    pub sasl_mechanism: Option<String>,
592    pub sasl_username: Option<String>,
593    pub sasl_password: Option<String>,
594    pub nodes_number: Option<u32>,
595}
596
597#[derive(Debug, Clone, Serialize, Deserialize)]
598#[serde(default)]
599pub struct IggyConfig {
600    pub connection_string: String,
601    pub username: Option<String>,
602    pub password: Option<String>,
603    pub consumer_name: Option<String>,
604    pub stream: String,
605    pub topic_prefix: String,
606    pub queue_topic_prefix: String,
607    pub consumer_group_prefix: String,
608    pub request_timeout_ms: u64,
609    pub poll_interval_ms: u64,
610    pub poll_batch_size: u32,
611    pub partitions_count: u32,
612    pub partition_id: u32,
613    pub auto_create: bool,
614    pub start_from_latest: bool,
615    pub nodes_number: Option<u32>,
616}
617
618#[derive(Debug, Clone, Serialize, Deserialize, Default)]
619#[serde(default)]
620pub struct AppManagerConfig {
621    pub driver: AppManagerDriver,
622    pub array: ArrayConfig,
623    pub cache: CacheSettings,
624    pub scylladb: ScyllaDbSettings,
625}
626
627#[derive(Debug, Clone, Serialize, Deserialize, Default)]
628#[serde(default)]
629pub struct ArrayConfig {
630    pub apps: Vec<App>,
631}
632
633#[derive(Debug, Clone, Serialize, Deserialize)]
634#[serde(default)]
635pub struct CacheSettings {
636    pub enabled: bool,
637    pub ttl: u64,
638}
639
640#[derive(Debug, Clone, Serialize, Deserialize)]
641#[serde(default)]
642pub struct MemoryCacheOptions {
643    pub ttl: u64,
644    pub cleanup_interval: u64,
645    pub max_capacity: u64,
646}
647
648#[derive(Debug, Clone, Serialize, Deserialize)]
649#[serde(default)]
650pub struct CacheConfig {
651    pub driver: CacheDriver,
652    pub redis: RedisConfig,
653    pub memory: MemoryCacheOptions,
654}
655
656#[derive(Debug, Clone, Serialize, Deserialize, Default)]
657#[serde(default)]
658pub struct RedisConfig {
659    pub prefix: Option<String>,
660    pub url_override: Option<String>,
661    pub cluster_mode: bool,
662}
663
664#[derive(Debug, Clone, Serialize, Deserialize)]
665#[serde(default)]
666pub struct ChannelLimits {
667    pub max_name_length: u32,
668    pub cache_ttl: u64,
669}
670
671#[derive(Debug, Clone, Serialize, Deserialize)]
672#[serde(default)]
673pub struct CorsConfig {
674    pub credentials: bool,
675    #[serde(deserialize_with = "deserialize_and_validate_cors_origins")]
676    pub origin: Vec<String>,
677    pub methods: Vec<String>,
678    pub allowed_headers: Vec<String>,
679}
680
681fn deserialize_and_validate_cors_origins<'de, D>(deserializer: D) -> Result<Vec<String>, D::Error>
682where
683    D: serde::Deserializer<'de>,
684{
685    use serde::de::Error;
686    let origins = Vec::<String>::deserialize(deserializer)?;
687
688    if let Err(e) = crate::origin_validation::OriginValidator::validate_patterns(&origins) {
689        return Err(D::Error::custom(format!(
690            "CORS origin pattern validation failed: {}",
691            e
692        )));
693    }
694
695    Ok(origins)
696}
697
698#[derive(Debug, Clone, Serialize, Deserialize, Default)]
699#[serde(default)]
700pub struct DatabaseConfig {
701    pub mysql: DatabaseConnection,
702    pub postgres: DatabaseConnection,
703    pub redis: RedisConnection,
704    pub dynamodb: DynamoDbSettings,
705    pub surrealdb: SurrealDbSettings,
706    pub scylladb: ScyllaDbSettings,
707}
708
709#[derive(Debug, Clone, Serialize, Deserialize)]
710#[serde(default)]
711pub struct DatabaseConnection {
712    pub host: String,
713    pub port: u16,
714    pub username: String,
715    pub password: String,
716    pub database: String,
717    pub table_name: String,
718    pub connection_pool_size: u32,
719    pub pool_min: Option<u32>,
720    pub pool_max: Option<u32>,
721    pub cache_ttl: u64,
722    pub cache_cleanup_interval: u64,
723    pub cache_max_capacity: u64,
724}
725
726#[derive(Debug, Clone, Serialize, Deserialize)]
727#[serde(default)]
728pub struct RedisConnection {
729    pub host: String,
730    pub port: u16,
731    pub db: u32,
732    pub username: Option<String>,
733    pub password: Option<String>,
734    pub key_prefix: String,
735    pub sentinels: Vec<RedisSentinel>,
736    pub sentinel_password: Option<String>,
737    pub name: String,
738    pub cluster: RedisClusterConnection,
739    /// Legacy field kept for backward compatibility. Prefer `database.redis.cluster.nodes`.
740    pub cluster_nodes: Vec<ClusterNode>,
741}
742
743#[derive(Debug, Clone, Serialize, Deserialize)]
744#[serde(default)]
745pub struct RedisSentinel {
746    pub host: String,
747    pub port: u16,
748}
749
750#[derive(Debug, Clone, Serialize, Deserialize, Default)]
751#[serde(default)]
752pub struct RedisClusterConnection {
753    pub nodes: Vec<ClusterNode>,
754    pub username: Option<String>,
755    pub password: Option<String>,
756    #[serde(alias = "useTLS")]
757    pub use_tls: bool,
758}
759
760impl RedisConnection {
761    /// Returns true if Redis Sentinel is configured.
762    pub fn is_sentinel_configured(&self) -> bool {
763        !self.sentinels.is_empty()
764    }
765
766    /// Builds a Redis connection URL based on the configuration.
767    pub fn to_url(&self) -> String {
768        if self.is_sentinel_configured() {
769            self.build_sentinel_url()
770        } else {
771            self.build_standard_url()
772        }
773    }
774
775    fn build_standard_url(&self) -> String {
776        // Extract scheme from host if present, otherwise default to redis://
777        let (scheme, host) = if self.host.starts_with("rediss://") {
778            ("rediss://", self.host.trim_start_matches("rediss://"))
779        } else if self.host.starts_with("redis://") {
780            ("redis://", self.host.trim_start_matches("redis://"))
781        } else {
782            ("redis://", self.host.as_str())
783        };
784
785        let mut url = String::from(scheme);
786
787        if let Some(ref username) = self.username {
788            url.push_str(username);
789            if let Some(ref password) = self.password {
790                url.push(':');
791                url.push_str(&urlencoding::encode(password));
792            }
793            url.push('@');
794        } else if let Some(ref password) = self.password {
795            url.push(':');
796            url.push_str(&urlencoding::encode(password));
797            url.push('@');
798        }
799
800        url.push_str(host);
801        url.push(':');
802        url.push_str(&self.port.to_string());
803        url.push('/');
804        url.push_str(&self.db.to_string());
805
806        url
807    }
808
809    fn build_sentinel_url(&self) -> String {
810        let mut url = String::from("redis+sentinel://");
811
812        if let Some(ref sentinel_password) = self.sentinel_password {
813            url.push(':');
814            url.push_str(&urlencoding::encode(sentinel_password));
815            url.push('@');
816        }
817
818        let sentinel_hosts: Vec<String> = self
819            .sentinels
820            .iter()
821            .map(|s| format!("{}:{}", s.host, s.port))
822            .collect();
823        url.push_str(&sentinel_hosts.join(","));
824
825        url.push('/');
826        url.push_str(&self.name);
827        url.push('/');
828        url.push_str(&self.db.to_string());
829
830        let mut params = Vec::new();
831        if let Some(ref password) = self.password {
832            params.push(format!("password={}", urlencoding::encode(password)));
833        }
834        if let Some(ref username) = self.username {
835            params.push(format!("username={}", urlencoding::encode(username)));
836        }
837
838        if !params.is_empty() {
839            url.push('?');
840            url.push_str(&params.join("&"));
841        }
842
843        url
844    }
845
846    /// Returns true when cluster nodes are configured via either the new (`cluster.nodes`)
847    /// or legacy (`cluster_nodes`) field.
848    pub fn has_cluster_nodes(&self) -> bool {
849        !self.cluster.nodes.is_empty() || !self.cluster_nodes.is_empty()
850    }
851
852    /// Returns normalized Redis Cluster seed URLs from the canonical cluster configuration.
853    /// Falls back to legacy `cluster_nodes` for backward compatibility.
854    pub fn cluster_node_urls(&self) -> Vec<String> {
855        if !self.cluster.nodes.is_empty() {
856            return self.build_cluster_urls(&self.cluster.nodes);
857        }
858        self.build_cluster_urls(&self.cluster_nodes)
859    }
860
861    /// Normalizes any list of seed strings (`host:port`, `redis://...`, `rediss://...`) using
862    /// shared cluster auth/TLS options.
863    pub fn normalize_cluster_seed_urls(&self, seeds: &[String]) -> Vec<String> {
864        self.build_cluster_urls(
865            &seeds
866                .iter()
867                .filter_map(|seed| ClusterNode::from_seed(seed))
868                .collect::<Vec<ClusterNode>>(),
869        )
870    }
871
872    fn build_cluster_urls(&self, nodes: &[ClusterNode]) -> Vec<String> {
873        let username = self
874            .cluster
875            .username
876            .as_deref()
877            .or(self.username.as_deref());
878        let password = self
879            .cluster
880            .password
881            .as_deref()
882            .or(self.password.as_deref());
883        let use_tls = self.cluster.use_tls;
884
885        nodes
886            .iter()
887            .map(|node| node.to_url_with_options(use_tls, username, password))
888            .collect()
889    }
890}
891
892impl RedisSentinel {
893    pub fn to_host_port(&self) -> String {
894        format!("{}:{}", self.host, self.port)
895    }
896}
897
898#[derive(Debug, Clone, Serialize, Deserialize)]
899#[serde(default)]
900pub struct ClusterNode {
901    pub host: String,
902    pub port: u16,
903}
904
905impl ClusterNode {
906    pub fn to_url(&self) -> String {
907        self.to_url_with_options(false, None, None)
908    }
909
910    pub fn to_url_with_options(
911        &self,
912        use_tls: bool,
913        username: Option<&str>,
914        password: Option<&str>,
915    ) -> String {
916        let host = self.host.trim();
917
918        if host.starts_with("redis://") || host.starts_with("rediss://") {
919            if let Ok(parsed) = Url::parse(host)
920                && let Some(host_str) = parsed.host_str()
921            {
922                let scheme = parsed.scheme();
923                let port = parsed.port_or_known_default().unwrap_or(self.port);
924                let parsed_username = (!parsed.username().is_empty()).then_some(parsed.username());
925                let parsed_password = parsed.password();
926                let has_embedded_auth = parsed_username.is_some() || parsed_password.is_some();
927                let (effective_username, effective_password) = if has_embedded_auth {
928                    (parsed_username, parsed_password)
929                } else {
930                    (username, password)
931                };
932
933                return build_redis_url(
934                    scheme,
935                    host_str,
936                    port,
937                    effective_username,
938                    effective_password,
939                );
940            }
941
942            // Fallback for malformed URLs
943            let has_port = if let Some(bracket_pos) = host.rfind(']') {
944                host[bracket_pos..].contains(':')
945            } else {
946                host.split(':').count() >= 3
947            };
948            let base = if has_port {
949                host.to_string()
950            } else {
951                format!("{}:{}", host, self.port)
952            };
953
954            if let Ok(parsed) = Url::parse(&base) {
955                let parsed_username = (!parsed.username().is_empty()).then_some(parsed.username());
956                let parsed_password = parsed.password();
957                if let Some(host_str) = parsed.host_str() {
958                    let port = parsed.port_or_known_default().unwrap_or(self.port);
959                    let has_embedded_auth = parsed_username.is_some() || parsed_password.is_some();
960                    let (effective_username, effective_password) = if has_embedded_auth {
961                        (parsed_username, parsed_password)
962                    } else {
963                        (username, password)
964                    };
965                    return build_redis_url(
966                        parsed.scheme(),
967                        host_str,
968                        port,
969                        effective_username,
970                        effective_password,
971                    );
972                }
973            }
974            return base;
975        }
976
977        let (normalized_host, normalized_port) = split_plain_host_and_port(host, self.port);
978        let scheme = if use_tls { "rediss" } else { "redis" };
979        build_redis_url(
980            scheme,
981            &normalized_host,
982            normalized_port,
983            username,
984            password,
985        )
986    }
987
988    pub fn from_seed(seed: &str) -> Option<Self> {
989        let trimmed = seed.trim();
990        if trimmed.is_empty() {
991            return None;
992        }
993
994        if trimmed.starts_with("redis://") || trimmed.starts_with("rediss://") {
995            let port = Url::parse(trimmed)
996                .ok()
997                .and_then(|parsed| parsed.port_or_known_default())
998                .unwrap_or(6379);
999            return Some(Self {
1000                host: trimmed.to_string(),
1001                port,
1002            });
1003        }
1004
1005        let (host, port) = split_plain_host_and_port(trimmed, 6379);
1006        Some(Self { host, port })
1007    }
1008}
1009
1010fn split_plain_host_and_port(raw_host: &str, default_port: u16) -> (String, u16) {
1011    let host = raw_host.trim();
1012
1013    // Handle bracketed IPv6: [::1]:6379
1014    if host.starts_with('[') {
1015        if let Some(end_bracket) = host.find(']') {
1016            let host_part = host[1..end_bracket].to_string();
1017            let remainder = &host[end_bracket + 1..];
1018            if let Some(port_str) = remainder.strip_prefix(':')
1019                && let Ok(port) = port_str.parse::<u16>()
1020            {
1021                return (host_part, port);
1022            }
1023            return (host_part, default_port);
1024        }
1025        return (host.to_string(), default_port);
1026    }
1027
1028    // Handle hostname/IP with port: host:6379
1029    if host.matches(':').count() == 1
1030        && let Some((host_part, port_part)) = host.rsplit_once(':')
1031        && let Ok(port) = port_part.parse::<u16>()
1032    {
1033        return (host_part.to_string(), port);
1034    }
1035
1036    (host.to_string(), default_port)
1037}
1038
1039fn build_redis_url(
1040    scheme: &str,
1041    host: &str,
1042    port: u16,
1043    username: Option<&str>,
1044    password: Option<&str>,
1045) -> String {
1046    let mut url = format!("{scheme}://");
1047
1048    if let Some(user) = username {
1049        url.push_str(&urlencoding::encode(user));
1050        if let Some(pass) = password {
1051            url.push(':');
1052            url.push_str(&urlencoding::encode(pass));
1053        }
1054        url.push('@');
1055    } else if let Some(pass) = password {
1056        url.push(':');
1057        url.push_str(&urlencoding::encode(pass));
1058        url.push('@');
1059    }
1060
1061    if host.contains(':') && !host.starts_with('[') {
1062        url.push('[');
1063        url.push_str(host);
1064        url.push(']');
1065    } else {
1066        url.push_str(host);
1067    }
1068    url.push(':');
1069    url.push_str(&port.to_string());
1070    url
1071}
1072
1073#[derive(Debug, Clone, Serialize, Deserialize)]
1074#[serde(default)]
1075pub struct DatabasePooling {
1076    pub enabled: bool,
1077    pub min: u32,
1078    pub max: u32,
1079}
1080
1081#[derive(Debug, Clone, Serialize, Deserialize)]
1082#[serde(default)]
1083pub struct EventLimits {
1084    pub max_channels_at_once: u32,
1085    pub max_name_length: u32,
1086    pub max_payload_in_kb: u32,
1087    pub max_batch_size: u32,
1088}
1089
1090#[derive(Debug, Clone, Serialize, Deserialize)]
1091#[serde(default)]
1092pub struct IdempotencyConfig {
1093    /// Whether idempotency key support is enabled
1094    pub enabled: bool,
1095    /// TTL in seconds for idempotency keys (default: 120s, like Ably's 2-minute window)
1096    pub ttl_seconds: u64,
1097    /// Maximum length of an idempotency key
1098    pub max_key_length: usize,
1099}
1100
1101#[derive(Debug, Clone, Serialize, Deserialize)]
1102#[serde(default)]
1103pub struct EphemeralConfig {
1104    /// Whether ephemeral message handling is enabled.
1105    pub enabled: bool,
1106}
1107
1108#[derive(Debug, Clone, Serialize, Deserialize)]
1109#[serde(default)]
1110pub struct EchoControlConfig {
1111    /// Whether connection-level and per-message echo control is enabled.
1112    pub enabled: bool,
1113    /// Default echo behavior for new V2 connections when the query param is omitted.
1114    pub default_echo_messages: bool,
1115}
1116
1117#[derive(Debug, Clone, Serialize, Deserialize)]
1118#[serde(default)]
1119pub struct EventNameFilteringConfig {
1120    /// Whether per-subscription event name filtering is enabled.
1121    pub enabled: bool,
1122    /// Maximum event names allowed in a single filter.
1123    pub max_events_per_filter: usize,
1124    /// Maximum length for each event name in a filter.
1125    pub max_event_name_length: usize,
1126}
1127
1128#[derive(Debug, Clone, Serialize, Deserialize)]
1129#[serde(default)]
1130pub struct ConnectionRecoveryConfig {
1131    /// Whether connection recovery (resume) is enabled.
1132    /// When enabled, the server keeps a bounded replay buffer per channel so that
1133    /// reconnecting clients can receive missed messages. Disabled by default for
1134    /// Pusher protocol compatibility.
1135    pub enabled: bool,
1136    /// How long messages stay in the replay buffer (seconds). Default: 120 (2 min).
1137    pub buffer_ttl_seconds: u64,
1138    /// Maximum number of messages kept per channel. Default: 100.
1139    pub max_buffer_size: usize,
1140}
1141
1142#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1143#[serde(rename_all = "lowercase")]
1144pub enum VersionStoreDriver {
1145    #[default]
1146    Memory,
1147    Postgres,
1148    Mysql,
1149    DynamoDb,
1150    ScyllaDb,
1151    SurrealDb,
1152}
1153
1154impl FromStr for VersionStoreDriver {
1155    type Err = String;
1156    fn from_str(s: &str) -> Result<Self, Self::Err> {
1157        match s.to_lowercase().as_str() {
1158            "postgres" | "postgresql" | "pgsql" => Ok(Self::Postgres),
1159            "mysql" => Ok(Self::Mysql),
1160            "dynamodb" => Ok(Self::DynamoDb),
1161            "scylladb" | "scylla" => Ok(Self::ScyllaDb),
1162            "surrealdb" | "surreal" => Ok(Self::SurrealDb),
1163            "memory" => Ok(Self::Memory),
1164            _ => Err(format!("Unknown version store driver: {s}")),
1165        }
1166    }
1167}
1168
1169#[derive(Debug, Clone, Serialize, Deserialize)]
1170#[serde(default)]
1171pub struct VersionedMessagesConfig {
1172    /// Whether V2 mutable message HTTP/retrieval surfaces are enabled.
1173    pub enabled: bool,
1174    /// Storage driver for versioned messages. Defaults to memory.
1175    pub driver: VersionStoreDriver,
1176    /// Maximum page size for version-history retrieval.
1177    pub max_page_size: usize,
1178    /// Retention window for version entries, in seconds. `0` disables expiry
1179    /// (entries are kept forever).
1180    ///
1181    /// Backends with native row TTL (ScyllaDB `USING TTL`, DynamoDB TTL
1182    /// attribute) apply this at the storage layer. Other backends (MySQL,
1183    /// PostgreSQL, SurrealDB, Memory) enforce it via a periodic background
1184    /// purge worker controlled by `purge_interval_seconds`.
1185    pub retention_window_seconds: u64,
1186    /// Interval between purge worker runs, in seconds. Only consulted for
1187    /// backends without native TTL. Clamped to a minimum of 10 seconds.
1188    pub purge_interval_seconds: u64,
1189    /// Maximum rows deleted per purge query. Bounds lock/transaction sizes
1190    /// for SQL backends. The worker loops until no more expired rows remain
1191    /// or `max_purge_per_tick` is reached.
1192    pub purge_batch_size: usize,
1193    /// Hard cap on rows deleted per purge tick across all loop iterations.
1194    /// Prevents a backlog of expired rows from monopolising a worker run.
1195    pub max_purge_per_tick: usize,
1196}
1197
1198#[derive(Debug, Clone, Serialize, Deserialize, Default)]
1199#[serde(default)]
1200pub struct AnnotationsConfig {
1201    /// Whether Sockudo-native annotation APIs and realtime annotation protocol
1202    /// surfaces are enabled. Disabled by default while the feature is opt-in.
1203    pub enabled: bool,
1204}
1205
1206#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1207#[serde(rename_all = "lowercase")]
1208pub enum HistoryBackend {
1209    #[default]
1210    Postgres,
1211    Mysql,
1212    DynamoDb,
1213    SurrealDb,
1214    ScyllaDb,
1215    Memory,
1216}
1217
1218impl FromStr for HistoryBackend {
1219    type Err = String;
1220    fn from_str(s: &str) -> Result<Self, Self::Err> {
1221        match s.to_lowercase().as_str() {
1222            "postgres" | "postgresql" | "pgsql" => Ok(Self::Postgres),
1223            "mysql" => Ok(Self::Mysql),
1224            "dynamodb" => Ok(Self::DynamoDb),
1225            "surrealdb" | "surreal" => Ok(Self::SurrealDb),
1226            "scylladb" | "scylla" => Ok(Self::ScyllaDb),
1227            "memory" => Ok(Self::Memory),
1228            _ => Err(format!("Unknown history backend: {s}")),
1229        }
1230    }
1231}
1232
1233#[derive(Debug, Clone, Serialize, Deserialize)]
1234#[serde(default)]
1235pub struct PostgresHistoryConfig {
1236    pub table_prefix: String,
1237    pub write_timeout_ms: u64,
1238}
1239
1240impl Default for PostgresHistoryConfig {
1241    fn default() -> Self {
1242        Self {
1243            table_prefix: "sockudo_history".to_string(),
1244            write_timeout_ms: 5000,
1245        }
1246    }
1247}
1248
1249#[derive(Debug, Clone, Serialize, Deserialize)]
1250#[serde(default)]
1251pub struct MySqlHistoryConfig {
1252    pub table_prefix: String,
1253    pub write_timeout_ms: u64,
1254}
1255
1256impl Default for MySqlHistoryConfig {
1257    fn default() -> Self {
1258        Self {
1259            table_prefix: "sockudo_history".to_string(),
1260            write_timeout_ms: 5000,
1261        }
1262    }
1263}
1264
1265#[derive(Debug, Clone, Serialize, Deserialize)]
1266#[serde(default)]
1267pub struct DynamoDbHistoryConfig {
1268    pub table_prefix: String,
1269    pub write_timeout_ms: u64,
1270}
1271
1272impl Default for DynamoDbHistoryConfig {
1273    fn default() -> Self {
1274        Self {
1275            table_prefix: "sockudo_history".to_string(),
1276            write_timeout_ms: 5000,
1277        }
1278    }
1279}
1280
1281#[derive(Debug, Clone, Serialize, Deserialize)]
1282#[serde(default)]
1283pub struct SurrealDbHistoryConfig {
1284    pub table_prefix: String,
1285    pub write_timeout_ms: u64,
1286}
1287
1288impl Default for SurrealDbHistoryConfig {
1289    fn default() -> Self {
1290        Self {
1291            table_prefix: "sockudo_history".to_string(),
1292            write_timeout_ms: 5000,
1293        }
1294    }
1295}
1296
1297#[derive(Debug, Clone, Serialize, Deserialize)]
1298#[serde(default)]
1299pub struct ScyllaDbHistoryConfig {
1300    pub table_prefix: String,
1301    pub write_timeout_ms: u64,
1302}
1303
1304impl Default for ScyllaDbHistoryConfig {
1305    fn default() -> Self {
1306        Self {
1307            table_prefix: "sockudo_history".to_string(),
1308            write_timeout_ms: 5000,
1309        }
1310    }
1311}
1312
1313#[derive(Debug, Clone, Serialize, Deserialize)]
1314#[serde(default)]
1315pub struct HistoryConfig {
1316    pub enabled: bool,
1317    pub rewind_enabled: bool,
1318    pub backend: HistoryBackend,
1319    pub retention_window_seconds: u64,
1320    pub max_page_size: usize,
1321    pub max_messages_per_channel: Option<usize>,
1322    pub max_bytes_per_channel: Option<u64>,
1323    pub writer_shards: usize,
1324    pub writer_queue_capacity: usize,
1325    pub purge_interval_seconds: u64,
1326    pub purge_batch_size: usize,
1327    pub max_purge_per_tick: usize,
1328    pub postgres: PostgresHistoryConfig,
1329    pub mysql: MySqlHistoryConfig,
1330    pub dynamodb: DynamoDbHistoryConfig,
1331    pub surrealdb: SurrealDbHistoryConfig,
1332    pub scylladb: ScyllaDbHistoryConfig,
1333}
1334
1335impl Default for HistoryConfig {
1336    fn default() -> Self {
1337        Self {
1338            enabled: false,
1339            rewind_enabled: true,
1340            backend: HistoryBackend::Postgres,
1341            retention_window_seconds: 86400,
1342            max_page_size: 100,
1343            max_messages_per_channel: None,
1344            max_bytes_per_channel: None,
1345            writer_shards: 16,
1346            writer_queue_capacity: 4096,
1347            purge_interval_seconds: 300,
1348            purge_batch_size: 1000,
1349            max_purge_per_tick: 100_000,
1350            postgres: PostgresHistoryConfig::default(),
1351            mysql: MySqlHistoryConfig::default(),
1352            dynamodb: DynamoDbHistoryConfig::default(),
1353            surrealdb: SurrealDbHistoryConfig::default(),
1354            scylladb: ScyllaDbHistoryConfig::default(),
1355        }
1356    }
1357}
1358
1359impl Default for VersionedMessagesConfig {
1360    fn default() -> Self {
1361        Self {
1362            enabled: false,
1363            driver: VersionStoreDriver::Memory,
1364            max_page_size: 100,
1365            retention_window_seconds: 0,
1366            purge_interval_seconds: 300,
1367            purge_batch_size: 1000,
1368            max_purge_per_tick: 100_000,
1369        }
1370    }
1371}
1372
1373#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1374#[serde(rename_all = "lowercase")]
1375pub enum PushStorageDriver {
1376    #[default]
1377    Memory,
1378    Postgres,
1379    Mysql,
1380    DynamoDb,
1381    SurrealDb,
1382    ScyllaDb,
1383}
1384
1385impl FromStr for PushStorageDriver {
1386    type Err = String;
1387    fn from_str(s: &str) -> Result<Self, Self::Err> {
1388        match s.to_lowercase().as_str() {
1389            "memory" => Ok(Self::Memory),
1390            "postgres" | "postgresql" | "pgsql" => Ok(Self::Postgres),
1391            "mysql" => Ok(Self::Mysql),
1392            "dynamodb" => Ok(Self::DynamoDb),
1393            "surrealdb" | "surreal" => Ok(Self::SurrealDb),
1394            "scylladb" | "scylla" => Ok(Self::ScyllaDb),
1395            _ => Err(format!("Unknown push storage driver: {s}")),
1396        }
1397    }
1398}
1399
1400#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1401#[serde(rename_all = "lowercase")]
1402pub enum PushQueueDriver {
1403    #[default]
1404    Memory,
1405    Redis,
1406    #[serde(rename = "redis-cluster")]
1407    RedisCluster,
1408    Nats,
1409    Pulsar,
1410    RabbitMq,
1411    #[serde(rename = "google-pubsub")]
1412    GooglePubsub,
1413    Kafka,
1414    Iggy,
1415    Sqs,
1416    Sns,
1417}
1418
1419impl FromStr for PushQueueDriver {
1420    type Err = String;
1421    fn from_str(s: &str) -> Result<Self, Self::Err> {
1422        match s.to_lowercase().as_str() {
1423            "memory" => Ok(Self::Memory),
1424            "redis" => Ok(Self::Redis),
1425            "redis-cluster" | "redis_cluster" => Ok(Self::RedisCluster),
1426            "nats" => Ok(Self::Nats),
1427            "pulsar" => Ok(Self::Pulsar),
1428            "rabbitmq" | "rabbit-mq" => Ok(Self::RabbitMq),
1429            "google-pubsub" | "google_pubsub" | "gcp-pubsub" | "pubsub" => Ok(Self::GooglePubsub),
1430            "kafka" => Ok(Self::Kafka),
1431            "iggy" | "apache-iggy" | "apache_iggy" => Ok(Self::Iggy),
1432            "sqs" => Ok(Self::Sqs),
1433            "sns" => Ok(Self::Sns),
1434            _ => Err(format!("Unknown push queue driver: {s}")),
1435        }
1436    }
1437}
1438
1439#[derive(Debug, Clone, Serialize, Deserialize)]
1440#[serde(default)]
1441pub struct PushConfig {
1442    pub storage_driver: PushStorageDriver,
1443    pub queue_driver: PushQueueDriver,
1444    pub fcm_enabled: bool,
1445    pub apns_enabled: bool,
1446    pub webpush_enabled: bool,
1447    pub hms_enabled: bool,
1448    pub wns_enabled: bool,
1449    pub fcm_credential_ref: Option<String>,
1450    pub apns_credential_ref: Option<String>,
1451    pub webpush_credential_ref: Option<String>,
1452    pub hms_credential_ref: Option<String>,
1453    pub wns_credential_ref: Option<String>,
1454    pub accept_worker_count: u32,
1455    pub planner_worker_count: u32,
1456    pub shard_worker_count: u32,
1457    pub dispatch_worker_count: u32,
1458    pub feedback_worker_count: u32,
1459    pub queue_partition_count: u32,
1460    pub channel_shard_count: u32,
1461    pub fanout_fast_threshold: u64,
1462    pub fanout_shard_size: u64,
1463    pub fanout_sync_threshold: u64,
1464    pub backpressure_lag_threshold_secs: u64,
1465    pub publish_status_ttl_days: u64,
1466    pub stale_device_max_age_days: u64,
1467    pub retry: PushRetryConfig,
1468    pub circuit_breaker: PushCircuitBreakerConfig,
1469    pub default_quotas: PushDefaultQuotas,
1470    pub credential_encryption_key: Option<String>,
1471    pub kms_key_ref: Option<String>,
1472    pub vault_secret_ref: Option<String>,
1473    pub dry_run: bool,
1474    pub analytics_enabled: bool,
1475    pub analytics_retention_days: u64,
1476    pub payload_redaction: PushPayloadRedactionConfig,
1477    pub scheduler_interval_secs: u64,
1478}
1479
1480impl Default for PushConfig {
1481    fn default() -> Self {
1482        Self {
1483            storage_driver: PushStorageDriver::Memory,
1484            queue_driver: PushQueueDriver::Memory,
1485            fcm_enabled: false,
1486            apns_enabled: false,
1487            webpush_enabled: false,
1488            hms_enabled: false,
1489            wns_enabled: false,
1490            fcm_credential_ref: None,
1491            apns_credential_ref: None,
1492            webpush_credential_ref: None,
1493            hms_credential_ref: None,
1494            wns_credential_ref: None,
1495            accept_worker_count: 1,
1496            planner_worker_count: 1,
1497            shard_worker_count: 1,
1498            dispatch_worker_count: 1,
1499            feedback_worker_count: 1,
1500            queue_partition_count: 1,
1501            channel_shard_count: 1,
1502            fanout_fast_threshold: 10_000,
1503            fanout_shard_size: 100_000,
1504            fanout_sync_threshold: 0,
1505            backpressure_lag_threshold_secs: 60,
1506            publish_status_ttl_days: 30,
1507            stale_device_max_age_days: 90,
1508            retry: PushRetryConfig::default(),
1509            circuit_breaker: PushCircuitBreakerConfig::default(),
1510            default_quotas: PushDefaultQuotas::default(),
1511            credential_encryption_key: None,
1512            kms_key_ref: None,
1513            vault_secret_ref: None,
1514            dry_run: false,
1515            analytics_enabled: false,
1516            analytics_retention_days: 30,
1517            payload_redaction: PushPayloadRedactionConfig::default(),
1518            scheduler_interval_secs: 5,
1519        }
1520    }
1521}
1522
1523#[derive(Debug, Clone, Serialize, Deserialize)]
1524#[serde(default)]
1525pub struct PushRetryConfig {
1526    pub max_attempts: u32,
1527    pub initial_backoff_ms: u64,
1528    pub max_backoff_ms: u64,
1529    pub max_elapsed_secs: u64,
1530    pub jitter: bool,
1531    pub respect_retry_after: bool,
1532}
1533
1534impl Default for PushRetryConfig {
1535    fn default() -> Self {
1536        Self {
1537            max_attempts: 5,
1538            initial_backoff_ms: 1_000,
1539            max_backoff_ms: 60_000,
1540            max_elapsed_secs: 86_400,
1541            jitter: true,
1542            respect_retry_after: true,
1543        }
1544    }
1545}
1546
1547#[derive(Debug, Clone, Serialize, Deserialize)]
1548#[serde(default)]
1549pub struct PushCircuitBreakerConfig {
1550    pub failure_threshold: u32,
1551    pub cooldown_secs: u64,
1552    pub half_open_max_inflight: u32,
1553}
1554
1555impl Default for PushCircuitBreakerConfig {
1556    fn default() -> Self {
1557        Self {
1558            failure_threshold: 5,
1559            cooldown_secs: 60,
1560            half_open_max_inflight: 10,
1561        }
1562    }
1563}
1564
1565#[derive(Debug, Clone, Serialize, Deserialize)]
1566#[serde(default)]
1567pub struct PushDefaultQuotas {
1568    pub acceptance_rps: u64,
1569    pub delivery_quota_daily: u64,
1570    pub fanout_max: u64,
1571    pub inflight_max: u64,
1572}
1573
1574impl Default for PushDefaultQuotas {
1575    fn default() -> Self {
1576        Self {
1577            acceptance_rps: 100,
1578            delivery_quota_daily: 0,
1579            fanout_max: 0,
1580            inflight_max: 1_000,
1581        }
1582    }
1583}
1584
1585#[derive(Debug, Clone, Serialize, Deserialize)]
1586#[serde(default)]
1587pub struct PushPayloadRedactionConfig {
1588    pub redact_payload: bool,
1589    pub redact_template_data: bool,
1590    pub redact_provider_overrides: bool,
1591    pub allow_debug_payload_logging: bool,
1592}
1593
1594impl Default for PushPayloadRedactionConfig {
1595    fn default() -> Self {
1596        Self {
1597            redact_payload: true,
1598            redact_template_data: true,
1599            redact_provider_overrides: true,
1600            allow_debug_payload_logging: false,
1601        }
1602    }
1603}
1604
1605#[derive(Debug, Clone, Serialize, Deserialize)]
1606#[serde(default)]
1607pub struct PresenceHistoryConfig {
1608    pub enabled: bool,
1609    pub retention_window_seconds: u64,
1610    pub max_page_size: usize,
1611    pub max_events_per_channel: Option<usize>,
1612    pub max_bytes_per_channel: Option<u64>,
1613}
1614
1615impl Default for PresenceHistoryConfig {
1616    fn default() -> Self {
1617        Self {
1618            enabled: false,
1619            retention_window_seconds: 86400,
1620            max_page_size: 100,
1621            max_events_per_channel: None,
1622            max_bytes_per_channel: None,
1623        }
1624    }
1625}
1626
1627#[derive(Debug, Clone, Serialize, Deserialize)]
1628#[serde(default)]
1629pub struct HttpApiConfig {
1630    pub request_limit_in_mb: u32,
1631    pub accept_traffic: AcceptTraffic,
1632    pub usage_enabled: bool,
1633}
1634
1635#[derive(Debug, Clone, Serialize, Deserialize)]
1636#[serde(default)]
1637pub struct AcceptTraffic {
1638    pub memory_threshold: f64,
1639}
1640
1641#[derive(Debug, Clone, Serialize, Deserialize)]
1642#[serde(default)]
1643pub struct InstanceConfig {
1644    pub process_id: String,
1645}
1646
1647#[derive(Debug, Clone, Serialize, Deserialize)]
1648#[serde(default)]
1649pub struct MetricsConfig {
1650    pub enabled: bool,
1651    pub driver: MetricsDriver,
1652    pub host: String,
1653    pub prometheus: PrometheusConfig,
1654    pub port: u16,
1655}
1656
1657#[derive(Debug, Clone, Serialize, Deserialize)]
1658#[serde(default)]
1659pub struct PrometheusConfig {
1660    pub prefix: String,
1661}
1662
1663#[derive(Debug, Clone, Serialize, Deserialize)]
1664#[serde(default)]
1665pub struct LoggingConfig {
1666    pub colors_enabled: bool,
1667    pub include_target: bool,
1668}
1669
1670#[derive(Debug, Clone, Serialize, Deserialize)]
1671#[serde(default)]
1672pub struct PresenceConfig {
1673    pub max_members_per_channel: u32,
1674    pub max_member_size_in_kb: u32,
1675}
1676
1677/// WebSocket connection buffer configuration
1678/// Controls backpressure handling for slow consumers
1679#[derive(Debug, Clone, Serialize, Deserialize)]
1680#[serde(default)]
1681pub struct WebSocketConfig {
1682    pub max_messages: Option<usize>,
1683    pub max_bytes: Option<usize>,
1684    pub disconnect_on_buffer_full: bool,
1685    pub max_message_size: usize,
1686    pub max_frame_size: usize,
1687    pub write_buffer_size: usize,
1688    pub max_backpressure: usize,
1689    pub auto_ping: bool,
1690    pub ping_interval: u32,
1691    pub idle_timeout: u32,
1692    pub compression: String,
1693}
1694
1695impl Default for WebSocketConfig {
1696    fn default() -> Self {
1697        Self {
1698            max_messages: Some(1000),
1699            max_bytes: None,
1700            disconnect_on_buffer_full: true,
1701            max_message_size: 64 * 1024 * 1024,
1702            max_frame_size: 16 * 1024 * 1024,
1703            write_buffer_size: 16 * 1024,
1704            max_backpressure: 1024 * 1024,
1705            auto_ping: true,
1706            ping_interval: 30,
1707            idle_timeout: 120,
1708            compression: "disabled".to_string(),
1709        }
1710    }
1711}
1712
1713impl WebSocketConfig {
1714    /// Convert to WebSocketBufferConfig for runtime use
1715    pub fn to_buffer_config(&self) -> crate::websocket::WebSocketBufferConfig {
1716        use crate::websocket::{BufferLimit, WebSocketBufferConfig};
1717
1718        let limit = match (self.max_messages, self.max_bytes) {
1719            (Some(messages), Some(bytes)) => BufferLimit::Both { messages, bytes },
1720            (Some(messages), None) => BufferLimit::Messages(messages),
1721            (None, Some(bytes)) => BufferLimit::Bytes(bytes),
1722            (None, None) => BufferLimit::Messages(1000),
1723        };
1724
1725        WebSocketBufferConfig {
1726            limit,
1727            disconnect_on_full: self.disconnect_on_buffer_full,
1728        }
1729    }
1730
1731    /// Convert to native sockudo-ws runtime configuration.
1732    pub fn to_sockudo_ws_config(
1733        &self,
1734        websocket_max_payload_kb: u32,
1735        activity_timeout: u64,
1736    ) -> sockudo_ws::Config {
1737        use sockudo_ws::Compression;
1738
1739        let compression = match self.compression.to_lowercase().as_str() {
1740            "dedicated" => Compression::Dedicated,
1741            "shared" => Compression::Shared,
1742            "window256b" => Compression::Window256B,
1743            "window1kb" => Compression::Window1KB,
1744            "window2kb" => Compression::Window2KB,
1745            "window4kb" => Compression::Window4KB,
1746            "window8kb" => Compression::Window8KB,
1747            "window16kb" => Compression::Window16KB,
1748            "window32kb" => Compression::Window32KB,
1749            _ => Compression::Disabled,
1750        };
1751
1752        sockudo_ws::Config::builder()
1753            .max_payload_length(
1754                self.max_bytes
1755                    .unwrap_or(websocket_max_payload_kb as usize * 1024),
1756            )
1757            .max_message_size(self.max_message_size)
1758            .max_frame_size(self.max_frame_size)
1759            .write_buffer_size(self.write_buffer_size)
1760            .max_backpressure(self.max_backpressure)
1761            .idle_timeout(self.idle_timeout)
1762            .auto_ping(self.auto_ping)
1763            .ping_interval(self.ping_interval.max((activity_timeout / 2).max(5) as u32))
1764            .compression(compression)
1765            .build()
1766    }
1767}
1768
1769#[derive(Debug, Clone, Serialize, Deserialize, Default)]
1770#[serde(default)]
1771pub struct QueueConfig {
1772    pub driver: QueueDriver,
1773    pub redis: RedisQueueConfig,
1774    pub redis_cluster: RedisClusterQueueConfig,
1775    pub nats: NatsAdapterConfig,
1776    pub pulsar: PulsarAdapterConfig,
1777    pub rabbitmq: RabbitMqAdapterConfig,
1778    pub google_pubsub: GooglePubSubAdapterConfig,
1779    pub kafka: KafkaAdapterConfig,
1780    pub iggy: IggyConfig,
1781    pub sqs: SqsQueueConfig,
1782    pub sns: SnsQueueConfig,
1783}
1784
1785#[derive(Debug, Clone, Serialize, Deserialize)]
1786#[serde(default)]
1787pub struct RedisQueueConfig {
1788    pub concurrency: u32,
1789    pub prefix: Option<String>,
1790    pub url_override: Option<String>,
1791    pub cluster_mode: bool,
1792}
1793
1794#[derive(Clone, Debug, Serialize, Deserialize)]
1795#[serde(default)]
1796pub struct RateLimit {
1797    pub max_requests: u32,
1798    pub window_seconds: u64,
1799    pub identifier: Option<String>,
1800    pub trust_hops: Option<u32>,
1801}
1802
1803#[derive(Debug, Clone, Serialize, Deserialize)]
1804#[serde(default)]
1805pub struct RateLimiterConfig {
1806    pub enabled: bool,
1807    pub driver: CacheDriver,
1808    pub api_rate_limit: RateLimit,
1809    pub websocket_rate_limit: RateLimit,
1810    pub redis: RedisConfig,
1811}
1812
1813#[derive(Debug, Clone, Serialize, Deserialize)]
1814#[serde(default)]
1815pub struct SslConfig {
1816    pub enabled: bool,
1817    pub cert_path: String,
1818    pub key_path: String,
1819    pub passphrase: Option<String>,
1820    pub ca_path: Option<String>,
1821    pub redirect_http: bool,
1822    pub http_port: Option<u16>,
1823}
1824
1825#[derive(Debug, Clone, Serialize, Deserialize)]
1826#[serde(default)]
1827pub struct WebhooksConfig {
1828    pub batching: BatchingConfig,
1829    pub retry: WebhookRetryConfig,
1830    pub request_timeout_ms: u64,
1831}
1832
1833#[derive(Debug, Clone, Serialize, Deserialize)]
1834#[serde(default)]
1835pub struct BatchingConfig {
1836    pub enabled: bool,
1837    pub duration: u64,
1838    pub size: usize,
1839}
1840
1841#[derive(Debug, Clone, Serialize, Deserialize)]
1842#[serde(default)]
1843pub struct WebhookRetryConfig {
1844    pub enabled: bool,
1845    pub max_attempts: Option<u32>,
1846    pub max_elapsed_time_ms: u64,
1847    pub initial_backoff_ms: u64,
1848    pub max_backoff_ms: u64,
1849}
1850
1851impl Default for WebhooksConfig {
1852    fn default() -> Self {
1853        Self {
1854            batching: BatchingConfig::default(),
1855            retry: WebhookRetryConfig::default(),
1856            request_timeout_ms: 10_000,
1857        }
1858    }
1859}
1860
1861impl Default for WebhookRetryConfig {
1862    fn default() -> Self {
1863        Self {
1864            enabled: true,
1865            max_attempts: None,
1866            max_elapsed_time_ms: 300_000,
1867            initial_backoff_ms: 1_000,
1868            max_backoff_ms: 60_000,
1869        }
1870    }
1871}
1872
1873#[derive(Debug, Clone, Serialize, Deserialize)]
1874#[serde(default)]
1875pub struct ClusterHealthConfig {
1876    pub enabled: bool,
1877    pub heartbeat_interval_ms: u64,
1878    pub node_timeout_ms: u64,
1879    pub cleanup_interval_ms: u64,
1880}
1881
1882#[derive(Debug, Clone, Serialize, Deserialize)]
1883#[serde(default)]
1884pub struct UnixSocketConfig {
1885    pub enabled: bool,
1886    pub path: String,
1887    #[serde(deserialize_with = "deserialize_octal_permission")]
1888    pub permission_mode: u32,
1889}
1890
1891#[derive(Debug, Clone, Serialize, Deserialize)]
1892#[serde(default)]
1893pub struct DeltaCompressionOptionsConfig {
1894    pub enabled: bool,
1895    pub algorithm: String,
1896    pub full_message_interval: u32,
1897    pub min_message_size: usize,
1898    pub max_state_age_secs: u64,
1899    pub max_channel_states_per_socket: usize,
1900    pub max_conflation_states_per_channel: Option<usize>,
1901    pub conflation_key_path: Option<String>,
1902    pub cluster_coordination: bool,
1903    pub coordination_backend: DeltaCoordinationBackend,
1904    pub omit_delta_algorithm: bool,
1905}
1906
1907/// Cleanup system configuration (minimal version for options; full impl lives in sockudo crate)
1908#[derive(Debug, Clone, Serialize, Deserialize)]
1909#[serde(default)]
1910pub struct CleanupConfig {
1911    pub queue_buffer_size: usize,
1912    pub batch_size: usize,
1913    pub batch_timeout_ms: u64,
1914    pub worker_threads: WorkerThreadsConfig,
1915    pub max_retry_attempts: u32,
1916    pub async_enabled: bool,
1917    pub fallback_to_sync: bool,
1918}
1919
1920/// Worker threads configuration for the cleanup system
1921#[derive(Debug, Clone)]
1922pub enum WorkerThreadsConfig {
1923    Auto,
1924    Fixed(usize),
1925}
1926
1927impl serde::Serialize for WorkerThreadsConfig {
1928    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
1929    where
1930        S: serde::Serializer,
1931    {
1932        match self {
1933            WorkerThreadsConfig::Auto => serializer.serialize_str("auto"),
1934            WorkerThreadsConfig::Fixed(n) => serializer.serialize_u64(*n as u64),
1935        }
1936    }
1937}
1938
1939impl<'de> serde::Deserialize<'de> for WorkerThreadsConfig {
1940    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
1941    where
1942        D: serde::Deserializer<'de>,
1943    {
1944        use serde::de;
1945        struct WorkerThreadsVisitor;
1946        impl<'de> de::Visitor<'de> for WorkerThreadsVisitor {
1947            type Value = WorkerThreadsConfig;
1948
1949            fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
1950                formatter.write_str(r#""auto" or a positive integer"#)
1951            }
1952
1953            fn visit_str<E: de::Error>(self, value: &str) -> Result<Self::Value, E> {
1954                if value.eq_ignore_ascii_case("auto") {
1955                    Ok(WorkerThreadsConfig::Auto)
1956                } else if let Ok(n) = value.parse::<usize>() {
1957                    Ok(WorkerThreadsConfig::Fixed(n))
1958                } else {
1959                    Err(E::custom(format!(
1960                        "expected 'auto' or a number, got '{value}'"
1961                    )))
1962                }
1963            }
1964
1965            fn visit_u64<E: de::Error>(self, value: u64) -> Result<Self::Value, E> {
1966                Ok(WorkerThreadsConfig::Fixed(value as usize))
1967            }
1968
1969            fn visit_i64<E: de::Error>(self, value: i64) -> Result<Self::Value, E> {
1970                if value >= 0 {
1971                    Ok(WorkerThreadsConfig::Fixed(value as usize))
1972                } else {
1973                    Err(E::custom("worker_threads must be non-negative"))
1974                }
1975            }
1976        }
1977        deserializer.deserialize_any(WorkerThreadsVisitor)
1978    }
1979}
1980
1981impl Default for CleanupConfig {
1982    fn default() -> Self {
1983        Self {
1984            queue_buffer_size: 1024,
1985            batch_size: 64,
1986            batch_timeout_ms: 100,
1987            worker_threads: WorkerThreadsConfig::Auto,
1988            max_retry_attempts: 3,
1989            async_enabled: true,
1990            fallback_to_sync: true,
1991        }
1992    }
1993}
1994
1995impl CleanupConfig {
1996    pub fn validate(&self) -> Result<(), String> {
1997        if self.queue_buffer_size == 0 {
1998            return Err("queue_buffer_size must be greater than 0".to_string());
1999        }
2000        if self.batch_size == 0 {
2001            return Err("batch_size must be greater than 0".to_string());
2002        }
2003        if self.batch_timeout_ms == 0 {
2004            return Err("batch_timeout_ms must be greater than 0".to_string());
2005        }
2006        if let WorkerThreadsConfig::Fixed(n) = self.worker_threads
2007            && n == 0
2008        {
2009            return Err("worker_threads must be greater than 0 when using fixed count".to_string());
2010        }
2011        Ok(())
2012    }
2013}
2014
2015#[derive(Debug, Clone, Serialize, Deserialize, Default)]
2016#[serde(default)]
2017pub struct TagFilteringConfig {
2018    #[serde(default)]
2019    pub enabled: bool,
2020    #[serde(default = "default_true")]
2021    pub enable_tags: bool,
2022}
2023
2024fn default_true() -> bool {
2025    true
2026}
2027
2028// --- Default Implementations ---
2029
2030impl Default for ServerOptions {
2031    fn default() -> Self {
2032        Self {
2033            adapter: AdapterConfig::default(),
2034            app_manager: AppManagerConfig::default(),
2035            cache: CacheConfig::default(),
2036            channel_limits: ChannelLimits::default(),
2037            cors: CorsConfig::default(),
2038            database: DatabaseConfig::default(),
2039            database_pooling: DatabasePooling::default(),
2040            debug: false,
2041            tag_filtering: TagFilteringConfig::default(),
2042            event_limits: EventLimits::default(),
2043            host: "0.0.0.0".to_string(),
2044            http_api: HttpApiConfig::default(),
2045            instance: InstanceConfig::default(),
2046            logging: None,
2047            metrics: MetricsConfig::default(),
2048            mode: "production".to_string(),
2049            port: 6001,
2050            path_prefix: "/".to_string(),
2051            presence: PresenceConfig::default(),
2052            queue: QueueConfig::default(),
2053            rate_limiter: RateLimiterConfig::default(),
2054            shutdown_grace_period: 10,
2055            ssl: SslConfig::default(),
2056            user_authentication_timeout: 3600,
2057            webhooks: WebhooksConfig::default(),
2058            websocket_max_payload_kb: 64,
2059            cleanup: CleanupConfig::default(),
2060            activity_timeout: 120,
2061            cluster_health: ClusterHealthConfig::default(),
2062            unix_socket: UnixSocketConfig::default(),
2063            delta_compression: DeltaCompressionOptionsConfig::default(),
2064            websocket: WebSocketConfig::default(),
2065            connection_recovery: ConnectionRecoveryConfig::default(),
2066            history: HistoryConfig::default(),
2067            presence_history: PresenceHistoryConfig::default(),
2068            idempotency: IdempotencyConfig::default(),
2069            ephemeral: EphemeralConfig::default(),
2070            echo_control: EchoControlConfig::default(),
2071            event_name_filtering: EventNameFilteringConfig::default(),
2072            versioned_messages: VersionedMessagesConfig::default(),
2073            annotations: AnnotationsConfig::default(),
2074            push: PushConfig::default(),
2075            health_check_timeout_ms: 400,
2076        }
2077    }
2078}
2079
2080impl Default for SqsQueueConfig {
2081    fn default() -> Self {
2082        Self {
2083            region: "us-east-1".to_string(),
2084            queue_url_prefix: None,
2085            visibility_timeout: 30,
2086            endpoint_url: None,
2087            max_messages: 10,
2088            wait_time_seconds: 5,
2089            concurrency: 5,
2090            fifo: false,
2091            message_group_id: Some("default".to_string()),
2092        }
2093    }
2094}
2095
2096impl Default for SnsQueueConfig {
2097    fn default() -> Self {
2098        Self {
2099            region: "us-east-1".to_string(),
2100            topic_arn: String::new(),
2101            endpoint_url: None,
2102        }
2103    }
2104}
2105
2106impl Default for RedisAdapterConfig {
2107    fn default() -> Self {
2108        Self {
2109            requests_timeout: 5000,
2110            prefix: "sockudo_adapter:".to_string(),
2111            redis_pub_options: AHashMap::new(),
2112            redis_sub_options: AHashMap::new(),
2113            cluster_mode: false,
2114        }
2115    }
2116}
2117
2118impl Default for RedisClusterAdapterConfig {
2119    fn default() -> Self {
2120        Self {
2121            nodes: vec![],
2122            prefix: "sockudo_adapter:".to_string(),
2123            request_timeout_ms: 1000,
2124            use_connection_manager: true,
2125            use_sharded_pubsub: false,
2126        }
2127    }
2128}
2129
2130impl Default for NatsAdapterConfig {
2131    fn default() -> Self {
2132        Self {
2133            servers: vec!["nats://localhost:4222".to_string()],
2134            prefix: "sockudo_adapter:".to_string(),
2135            request_timeout_ms: 5000,
2136            username: None,
2137            password: None,
2138            token: None,
2139            connection_timeout_ms: 5000,
2140            nodes_number: None,
2141            discovery_max_wait_ms: 1000,
2142            discovery_idle_wait_ms: 150,
2143            subscription_capacity: None,
2144            client_capacity: None,
2145            max_reconnects: None,
2146            presence_sync_chunk_size: None,
2147        }
2148    }
2149}
2150
2151impl Default for PulsarAdapterConfig {
2152    fn default() -> Self {
2153        Self {
2154            url: "pulsar://127.0.0.1:6650".to_string(),
2155            prefix: "sockudo-adapter".to_string(),
2156            request_timeout_ms: 5000,
2157            token: None,
2158            nodes_number: None,
2159        }
2160    }
2161}
2162
2163impl Default for RabbitMqAdapterConfig {
2164    fn default() -> Self {
2165        Self {
2166            url: "amqp://guest:guest@127.0.0.1:5672/%2f".to_string(),
2167            prefix: "sockudo_adapter".to_string(),
2168            request_timeout_ms: 5000,
2169            connection_timeout_ms: 5000,
2170            nodes_number: None,
2171        }
2172    }
2173}
2174
2175impl Default for GooglePubSubAdapterConfig {
2176    fn default() -> Self {
2177        Self {
2178            project_id: "".to_string(),
2179            prefix: "sockudo-adapter".to_string(),
2180            request_timeout_ms: 5000,
2181            emulator_host: None,
2182            nodes_number: None,
2183        }
2184    }
2185}
2186
2187impl Default for KafkaAdapterConfig {
2188    fn default() -> Self {
2189        Self {
2190            brokers: vec!["localhost:9092".to_string()],
2191            prefix: "sockudo_adapter".to_string(),
2192            request_timeout_ms: 5000,
2193            security_protocol: None,
2194            sasl_mechanism: None,
2195            sasl_username: None,
2196            sasl_password: None,
2197            nodes_number: None,
2198        }
2199    }
2200}
2201
2202impl Default for IggyConfig {
2203    fn default() -> Self {
2204        Self {
2205            connection_string: "iggy://iggy:iggy@127.0.0.1:8090".to_string(),
2206            username: None,
2207            password: None,
2208            consumer_name: None,
2209            stream: "sockudo".to_string(),
2210            topic_prefix: "sockudo-adapter".to_string(),
2211            queue_topic_prefix: "sockudo-queue".to_string(),
2212            consumer_group_prefix: "sockudo-workers".to_string(),
2213            request_timeout_ms: 5000,
2214            poll_interval_ms: 50,
2215            poll_batch_size: 100,
2216            partitions_count: 1,
2217            partition_id: 0,
2218            auto_create: true,
2219            start_from_latest: true,
2220            nodes_number: None,
2221        }
2222    }
2223}
2224
2225impl Default for CacheSettings {
2226    fn default() -> Self {
2227        Self {
2228            enabled: true,
2229            ttl: 300,
2230        }
2231    }
2232}
2233
2234impl Default for MemoryCacheOptions {
2235    fn default() -> Self {
2236        Self {
2237            ttl: 300,
2238            cleanup_interval: 60,
2239            max_capacity: 10000,
2240        }
2241    }
2242}
2243
2244impl Default for CacheConfig {
2245    fn default() -> Self {
2246        Self {
2247            driver: CacheDriver::default(),
2248            redis: RedisConfig {
2249                prefix: Some("sockudo_cache:".to_string()),
2250                url_override: None,
2251                cluster_mode: false,
2252            },
2253            memory: MemoryCacheOptions::default(),
2254        }
2255    }
2256}
2257
2258impl Default for ChannelLimits {
2259    fn default() -> Self {
2260        Self {
2261            max_name_length: 200,
2262            cache_ttl: 3600,
2263        }
2264    }
2265}
2266impl Default for CorsConfig {
2267    fn default() -> Self {
2268        Self {
2269            credentials: true,
2270            origin: vec!["*".to_string()],
2271            methods: vec!["GET".to_string(), "POST".to_string(), "OPTIONS".to_string()],
2272            allowed_headers: vec![
2273                "Authorization".to_string(),
2274                "Content-Type".to_string(),
2275                "X-Requested-With".to_string(),
2276                "Accept".to_string(),
2277            ],
2278        }
2279    }
2280}
2281
2282impl Default for DatabaseConnection {
2283    fn default() -> Self {
2284        Self {
2285            host: "localhost".to_string(),
2286            port: 3306,
2287            username: "root".to_string(),
2288            password: "".to_string(),
2289            database: "sockudo".to_string(),
2290            table_name: "applications".to_string(),
2291            connection_pool_size: 10,
2292            pool_min: None,
2293            pool_max: None,
2294            cache_ttl: 300,
2295            cache_cleanup_interval: 60,
2296            cache_max_capacity: 100,
2297        }
2298    }
2299}
2300
2301impl Default for RedisConnection {
2302    fn default() -> Self {
2303        Self {
2304            host: "127.0.0.1".to_string(),
2305            port: 6379,
2306            db: 0,
2307            username: None,
2308            password: None,
2309            key_prefix: "sockudo:".to_string(),
2310            sentinels: Vec::new(),
2311            sentinel_password: None,
2312            name: "mymaster".to_string(),
2313            cluster: RedisClusterConnection::default(),
2314            cluster_nodes: Vec::new(),
2315        }
2316    }
2317}
2318
2319impl Default for RedisSentinel {
2320    fn default() -> Self {
2321        Self {
2322            host: "localhost".to_string(),
2323            port: 26379,
2324        }
2325    }
2326}
2327
2328impl Default for ClusterNode {
2329    fn default() -> Self {
2330        Self {
2331            host: "127.0.0.1".to_string(),
2332            port: 7000,
2333        }
2334    }
2335}
2336
2337impl Default for DatabasePooling {
2338    fn default() -> Self {
2339        Self {
2340            enabled: true,
2341            min: 2,
2342            max: 10,
2343        }
2344    }
2345}
2346
2347impl Default for EventLimits {
2348    fn default() -> Self {
2349        Self {
2350            max_channels_at_once: 100,
2351            max_name_length: 200,
2352            max_payload_in_kb: 100,
2353            max_batch_size: 10,
2354        }
2355    }
2356}
2357
2358impl Default for IdempotencyConfig {
2359    fn default() -> Self {
2360        Self {
2361            enabled: true,
2362            ttl_seconds: 120,
2363            max_key_length: 128,
2364        }
2365    }
2366}
2367
2368impl Default for EphemeralConfig {
2369    fn default() -> Self {
2370        Self { enabled: true }
2371    }
2372}
2373
2374impl Default for EchoControlConfig {
2375    fn default() -> Self {
2376        Self {
2377            enabled: true,
2378            default_echo_messages: true,
2379        }
2380    }
2381}
2382
2383impl Default for EventNameFilteringConfig {
2384    fn default() -> Self {
2385        Self {
2386            enabled: true,
2387            max_events_per_filter: 50,
2388            max_event_name_length: 200,
2389        }
2390    }
2391}
2392
2393impl Default for ConnectionRecoveryConfig {
2394    fn default() -> Self {
2395        Self {
2396            enabled: false,
2397            buffer_ttl_seconds: 120,
2398            max_buffer_size: 100,
2399        }
2400    }
2401}
2402
2403impl Default for HttpApiConfig {
2404    fn default() -> Self {
2405        Self {
2406            request_limit_in_mb: 10,
2407            accept_traffic: AcceptTraffic::default(),
2408            usage_enabled: true,
2409        }
2410    }
2411}
2412
2413impl Default for AcceptTraffic {
2414    fn default() -> Self {
2415        Self {
2416            memory_threshold: 0.90,
2417        }
2418    }
2419}
2420
2421impl Default for InstanceConfig {
2422    fn default() -> Self {
2423        Self {
2424            process_id: uuid::Uuid::new_v4().to_string(),
2425        }
2426    }
2427}
2428
2429impl Default for LoggingConfig {
2430    fn default() -> Self {
2431        Self {
2432            colors_enabled: true,
2433            include_target: true,
2434        }
2435    }
2436}
2437
2438impl Default for MetricsConfig {
2439    fn default() -> Self {
2440        Self {
2441            enabled: true,
2442            driver: MetricsDriver::default(),
2443            host: "0.0.0.0".to_string(),
2444            prometheus: PrometheusConfig::default(),
2445            port: 9601,
2446        }
2447    }
2448}
2449
2450impl Default for PrometheusConfig {
2451    fn default() -> Self {
2452        Self {
2453            prefix: "sockudo_".to_string(),
2454        }
2455    }
2456}
2457
2458impl Default for PresenceConfig {
2459    fn default() -> Self {
2460        Self {
2461            max_members_per_channel: 100,
2462            max_member_size_in_kb: 2,
2463        }
2464    }
2465}
2466
2467impl Default for RedisQueueConfig {
2468    fn default() -> Self {
2469        Self {
2470            concurrency: 5,
2471            prefix: Some("sockudo_queue:".to_string()),
2472            url_override: None,
2473            cluster_mode: false,
2474        }
2475    }
2476}
2477
2478impl Default for RateLimit {
2479    fn default() -> Self {
2480        Self {
2481            max_requests: 60,
2482            window_seconds: 60,
2483            identifier: Some("default".to_string()),
2484            trust_hops: Some(0),
2485        }
2486    }
2487}
2488
2489impl Default for RateLimiterConfig {
2490    fn default() -> Self {
2491        Self {
2492            enabled: true,
2493            driver: CacheDriver::Memory,
2494            api_rate_limit: RateLimit {
2495                max_requests: 100,
2496                window_seconds: 60,
2497                identifier: Some("api".to_string()),
2498                trust_hops: Some(0),
2499            },
2500            websocket_rate_limit: RateLimit {
2501                max_requests: 20,
2502                window_seconds: 60,
2503                identifier: Some("websocket_connect".to_string()),
2504                trust_hops: Some(0),
2505            },
2506            redis: RedisConfig {
2507                prefix: Some("sockudo_rl:".to_string()),
2508                url_override: None,
2509                cluster_mode: false,
2510            },
2511        }
2512    }
2513}
2514
2515impl Default for SslConfig {
2516    fn default() -> Self {
2517        Self {
2518            enabled: false,
2519            cert_path: "".to_string(),
2520            key_path: "".to_string(),
2521            passphrase: None,
2522            ca_path: None,
2523            redirect_http: false,
2524            http_port: Some(80),
2525        }
2526    }
2527}
2528
2529impl Default for BatchingConfig {
2530    fn default() -> Self {
2531        Self {
2532            enabled: true,
2533            duration: 50,
2534            size: 100,
2535        }
2536    }
2537}
2538
2539impl Default for ClusterHealthConfig {
2540    fn default() -> Self {
2541        Self {
2542            enabled: true,
2543            heartbeat_interval_ms: 10000,
2544            node_timeout_ms: 30000,
2545            cleanup_interval_ms: 10000,
2546        }
2547    }
2548}
2549
2550impl Default for UnixSocketConfig {
2551    fn default() -> Self {
2552        Self {
2553            enabled: false,
2554            path: "/var/run/sockudo/sockudo.sock".to_string(),
2555            permission_mode: 0o660,
2556        }
2557    }
2558}
2559
2560impl Default for DeltaCompressionOptionsConfig {
2561    fn default() -> Self {
2562        Self {
2563            enabled: true,
2564            algorithm: "fossil".to_string(),
2565            full_message_interval: 10,
2566            min_message_size: 100,
2567            max_state_age_secs: 300,
2568            max_channel_states_per_socket: 100,
2569            max_conflation_states_per_channel: Some(100),
2570            conflation_key_path: None,
2571            cluster_coordination: false,
2572            coordination_backend: DeltaCoordinationBackend::Auto,
2573            omit_delta_algorithm: false,
2574        }
2575    }
2576}
2577
2578#[cfg(test)]
2579mod tests {
2580    use super::{
2581        DeltaCoordinationBackend, PushQueueDriver, PushStorageDriver, QueueDriver, ServerOptions,
2582        VersionStoreDriver,
2583    };
2584    use crate::app::{App, AppPolicy};
2585    use std::str::FromStr;
2586
2587    const APP_BOOTSTRAP_ENV_KEYS: &[&str] = &[
2588        "SOCKUDO_DEFAULT_APP_ID",
2589        "SOCKUDO_DEFAULT_APP_KEY",
2590        "SOCKUDO_DEFAULT_APP_SECRET",
2591        "SOCKUDO_DEFAULT_APP_ENABLED",
2592        "SOCKUDO_SKIP_INLINE_APPS",
2593        "APP_MANAGER_REGISTER_INLINE_APPS",
2594    ];
2595
2596    struct EnvGuard {
2597        previous: Vec<(&'static str, Option<String>)>,
2598    }
2599
2600    impl EnvGuard {
2601        fn app_bootstrap(overrides: &[(&'static str, &'static str)]) -> Self {
2602            let previous = APP_BOOTSTRAP_ENV_KEYS
2603                .iter()
2604                .map(|key| (*key, std::env::var(key).ok()))
2605                .collect();
2606
2607            // SAFETY: These tests isolate the app-bootstrap environment keys
2608            // before applying per-test overrides and restore them in Drop.
2609            unsafe {
2610                for key in APP_BOOTSTRAP_ENV_KEYS {
2611                    std::env::remove_var(key);
2612                }
2613                for (key, value) in overrides {
2614                    std::env::set_var(key, value);
2615                }
2616            }
2617
2618            Self { previous }
2619        }
2620    }
2621
2622    impl Drop for EnvGuard {
2623        fn drop(&mut self) {
2624            // SAFETY: Restores each key to its pre-test value or removes it if
2625            // it did not exist before the test.
2626            unsafe {
2627                for (key, value) in &self.previous {
2628                    if let Some(value) = value {
2629                        std::env::set_var(key, value);
2630                    } else {
2631                        std::env::remove_var(key);
2632                    }
2633                }
2634            }
2635        }
2636    }
2637
2638    fn inline_test_app() -> App {
2639        App::from_policy(
2640            "app-id".to_string(),
2641            "app-key".to_string(),
2642            "app-secret".to_string(),
2643            true,
2644            AppPolicy::default(),
2645        )
2646    }
2647
2648    #[test]
2649    fn queue_driver_parses_broker_backends() {
2650        assert_eq!(
2651            QueueDriver::from_str("rabbitmq").unwrap(),
2652            QueueDriver::RabbitMq
2653        );
2654        assert_eq!(QueueDriver::from_str("kafka").unwrap(), QueueDriver::Kafka);
2655        assert_eq!(
2656            QueueDriver::from_str("pulsar").unwrap(),
2657            QueueDriver::Pulsar
2658        );
2659        assert_eq!(
2660            QueueDriver::from_str("google-pubsub").unwrap(),
2661            QueueDriver::GooglePubSub
2662        );
2663    }
2664
2665    #[test]
2666    fn delta_coordination_backend_parses_expected_values() {
2667        assert_eq!(
2668            DeltaCoordinationBackend::from_str("auto").unwrap(),
2669            DeltaCoordinationBackend::Auto
2670        );
2671        assert_eq!(
2672            DeltaCoordinationBackend::from_str("redis-cluster").unwrap(),
2673            DeltaCoordinationBackend::RedisCluster
2674        );
2675        assert_eq!(
2676            DeltaCoordinationBackend::from_str("nats").unwrap(),
2677            DeltaCoordinationBackend::Nats
2678        );
2679    }
2680
2681    #[tokio::test]
2682    async fn app_bootstrap_env_overrides_inline_apps() {
2683        {
2684            let _env = EnvGuard::app_bootstrap(&[("SOCKUDO_DEFAULT_APP_ENABLED", "false")]);
2685            let mut options = ServerOptions::default();
2686            options.app_manager.array.apps.push(inline_test_app());
2687
2688            options.override_from_env().await.unwrap();
2689
2690            assert!(options.app_manager.array.apps.is_empty());
2691        }
2692
2693        {
2694            let _env = EnvGuard::app_bootstrap(&[("SOCKUDO_DEFAULT_APP_ENABLED", "true")]);
2695            let mut options = ServerOptions::default();
2696            options.app_manager.array.apps.push(inline_test_app());
2697
2698            options.override_from_env().await.unwrap();
2699
2700            assert_eq!(options.app_manager.array.apps.len(), 1);
2701            assert_eq!(options.app_manager.array.apps[0].id, "app-id");
2702        }
2703
2704        {
2705            let _env = EnvGuard::app_bootstrap(&[
2706                ("SOCKUDO_DEFAULT_APP_ID", "prod-app"),
2707                ("SOCKUDO_DEFAULT_APP_KEY", "prod-key"),
2708                ("SOCKUDO_DEFAULT_APP_SECRET", "prod-secret"),
2709                ("SOCKUDO_DEFAULT_APP_ENABLED", "true"),
2710            ]);
2711            let mut options = ServerOptions::default();
2712            options.app_manager.array.apps.push(inline_test_app());
2713
2714            options.override_from_env().await.unwrap();
2715
2716            assert_eq!(options.app_manager.array.apps.len(), 1);
2717            let app = &options.app_manager.array.apps[0];
2718            assert_eq!(app.id, "prod-app");
2719            assert_eq!(app.key, "prod-key");
2720            assert_eq!(app.secret, "prod-secret");
2721            assert!(app.enabled);
2722        }
2723
2724        {
2725            let _env = EnvGuard::app_bootstrap(&[("APP_MANAGER_REGISTER_INLINE_APPS", "false")]);
2726            let mut options = ServerOptions::default();
2727            options.app_manager.array.apps.push(inline_test_app());
2728
2729            options.override_from_env().await.unwrap();
2730
2731            assert!(options.app_manager.array.apps.is_empty());
2732        }
2733    }
2734
2735    #[tokio::test]
2736    async fn versioned_messages_driver_overrides_from_env() {
2737        let previous = std::env::var("VERSIONED_MESSAGES_DRIVER").ok();
2738        // SAFETY: This test controls the environment variable lifecycle for a
2739        // single key and restores the prior value before it returns.
2740        unsafe { std::env::set_var("VERSIONED_MESSAGES_DRIVER", "postgres") };
2741
2742        let mut options = ServerOptions::default();
2743        options.override_from_env().await.unwrap();
2744
2745        if let Some(previous) = previous {
2746            // SAFETY: Restoring the pre-test value for the same key.
2747            unsafe { std::env::set_var("VERSIONED_MESSAGES_DRIVER", previous) };
2748        } else {
2749            // SAFETY: Removing the test-only environment variable before exit.
2750            unsafe { std::env::remove_var("VERSIONED_MESSAGES_DRIVER") };
2751        }
2752
2753        assert_eq!(
2754            options.versioned_messages.driver,
2755            VersionStoreDriver::Postgres
2756        );
2757    }
2758
2759    #[tokio::test]
2760    async fn push_storage_driver_overrides_from_env() {
2761        let previous = std::env::var("PUSH_STORAGE_DRIVER").ok();
2762        // SAFETY: This test controls the environment variable lifecycle for a
2763        // single key and restores the prior value before it returns.
2764        unsafe { std::env::set_var("PUSH_STORAGE_DRIVER", "mysql") };
2765
2766        let mut options = ServerOptions::default();
2767        options.override_from_env().await.unwrap();
2768
2769        if let Some(previous) = previous {
2770            // SAFETY: Restoring the pre-test value for the same key.
2771            unsafe { std::env::set_var("PUSH_STORAGE_DRIVER", previous) };
2772        } else {
2773            // SAFETY: Removing the test-only environment variable before exit.
2774            unsafe { std::env::remove_var("PUSH_STORAGE_DRIVER") };
2775        }
2776
2777        assert_eq!(options.push.storage_driver, PushStorageDriver::Mysql);
2778    }
2779
2780    #[tokio::test]
2781    async fn push_queue_driver_overrides_from_env() {
2782        let previous = std::env::var("PUSH_QUEUE_DRIVER").ok();
2783        // SAFETY: This test controls the environment variable lifecycle for a
2784        // single key and restores the prior value before it returns.
2785        unsafe { std::env::set_var("PUSH_QUEUE_DRIVER", "redis-cluster") };
2786
2787        let mut options = ServerOptions::default();
2788        options.override_from_env().await.unwrap();
2789
2790        if let Some(previous) = previous {
2791            // SAFETY: Restoring the pre-test value for the same key.
2792            unsafe { std::env::set_var("PUSH_QUEUE_DRIVER", previous) };
2793        } else {
2794            // SAFETY: Removing the test-only environment variable before exit.
2795            unsafe { std::env::remove_var("PUSH_QUEUE_DRIVER") };
2796        }
2797
2798        assert_eq!(options.push.queue_driver, PushQueueDriver::RedisCluster);
2799    }
2800
2801    #[test]
2802    fn push_config_defaults_follow_capacity_model() {
2803        let options = ServerOptions::default();
2804
2805        assert_eq!(options.push.storage_driver, PushStorageDriver::Memory);
2806        assert_eq!(options.push.queue_driver, PushQueueDriver::Memory);
2807        assert!(!options.push.fcm_enabled);
2808        assert!(!options.push.apns_enabled);
2809        assert!(!options.push.webpush_enabled);
2810        assert!(!options.push.hms_enabled);
2811        assert!(!options.push.wns_enabled);
2812        assert_eq!(options.push.fanout_fast_threshold, 10_000);
2813        assert_eq!(options.push.fanout_shard_size, 100_000);
2814        assert_eq!(options.push.publish_status_ttl_days, 30);
2815        assert_eq!(options.push.default_quotas.acceptance_rps, 100);
2816        assert_eq!(options.push.circuit_breaker.failure_threshold, 5);
2817        assert!(options.push.payload_redaction.redact_payload);
2818    }
2819
2820    #[tokio::test]
2821    async fn push_release_env_overrides_are_parsed() {
2822        let keys = [
2823            "PUSH_FCM_ENABLED",
2824            "PUSH_APNS_ENABLED",
2825            "PUSH_WEBPUSH_ENABLED",
2826            "PUSH_HMS_ENABLED",
2827            "PUSH_WNS_ENABLED",
2828            "PUSH_CREDENTIAL_ENCRYPTION_KEY",
2829            "PUSH_FANOUT_FAST_THRESHOLD",
2830            "PUSH_FANOUT_SHARD_SIZE",
2831            "PUSH_FANOUT_SYNC_THRESHOLD",
2832            "PUSH_BACKPRESSURE_LAG_THRESHOLD_SECS",
2833            "PUSH_PUBLISH_STATUS_TTL_DAYS",
2834            "PUSH_FAILURE_THRESHOLD",
2835            "PUSH_SCHEDULER_INTERVAL_SECS",
2836            "PUSH_STALE_DEVICE_MAX_AGE_DAYS",
2837            "PUSH_ANALYTICS_ENABLED",
2838            "PUSH_DEFAULT_ACCEPTANCE_RPS",
2839            "PUSH_DEFAULT_DELIVERY_QUOTA_DAILY",
2840            "PUSH_DEFAULT_FANOUT_MAX",
2841            "PUSH_DEFAULT_INFLIGHT_MAX",
2842        ];
2843        let previous: Vec<_> = keys
2844            .iter()
2845            .map(|key| (*key, std::env::var(key).ok()))
2846            .collect();
2847
2848        // SAFETY: This test owns the listed environment variables and restores
2849        // their previous values before returning.
2850        unsafe {
2851            std::env::set_var("PUSH_FCM_ENABLED", "true");
2852            std::env::set_var("PUSH_APNS_ENABLED", "true");
2853            std::env::set_var("PUSH_WEBPUSH_ENABLED", "true");
2854            std::env::set_var("PUSH_HMS_ENABLED", "true");
2855            std::env::set_var("PUSH_WNS_ENABLED", "true");
2856            std::env::set_var("PUSH_CREDENTIAL_ENCRYPTION_KEY", "env:key:v1");
2857            std::env::set_var("PUSH_FANOUT_FAST_THRESHOLD", "12345");
2858            std::env::set_var("PUSH_FANOUT_SHARD_SIZE", "54321");
2859            std::env::set_var("PUSH_FANOUT_SYNC_THRESHOLD", "250");
2860            std::env::set_var("PUSH_BACKPRESSURE_LAG_THRESHOLD_SECS", "42");
2861            std::env::set_var("PUSH_PUBLISH_STATUS_TTL_DAYS", "14");
2862            std::env::set_var("PUSH_FAILURE_THRESHOLD", "9");
2863            std::env::set_var("PUSH_SCHEDULER_INTERVAL_SECS", "11");
2864            std::env::set_var("PUSH_STALE_DEVICE_MAX_AGE_DAYS", "120");
2865            std::env::set_var("PUSH_ANALYTICS_ENABLED", "true");
2866            std::env::set_var("PUSH_DEFAULT_ACCEPTANCE_RPS", "700");
2867            std::env::set_var("PUSH_DEFAULT_DELIVERY_QUOTA_DAILY", "8000");
2868            std::env::set_var("PUSH_DEFAULT_FANOUT_MAX", "9000");
2869            std::env::set_var("PUSH_DEFAULT_INFLIGHT_MAX", "1000");
2870        }
2871
2872        let mut options = ServerOptions::default();
2873        options.override_from_env().await.unwrap();
2874
2875        for (key, value) in previous {
2876            // SAFETY: Restoring each pre-test value or removing the test-only
2877            // variable for the same key.
2878            unsafe {
2879                if let Some(value) = value {
2880                    std::env::set_var(key, value);
2881                } else {
2882                    std::env::remove_var(key);
2883                }
2884            }
2885        }
2886
2887        assert!(options.push.fcm_enabled);
2888        assert!(options.push.apns_enabled);
2889        assert!(options.push.webpush_enabled);
2890        assert!(options.push.hms_enabled);
2891        assert!(options.push.wns_enabled);
2892        assert_eq!(
2893            options.push.credential_encryption_key.as_deref(),
2894            Some("env:key:v1")
2895        );
2896        assert_eq!(options.push.fanout_fast_threshold, 12_345);
2897        assert_eq!(options.push.fanout_shard_size, 54_321);
2898        assert_eq!(options.push.fanout_sync_threshold, 250);
2899        assert_eq!(options.push.backpressure_lag_threshold_secs, 42);
2900        assert_eq!(options.push.publish_status_ttl_days, 14);
2901        assert_eq!(options.push.circuit_breaker.failure_threshold, 9);
2902        assert_eq!(options.push.scheduler_interval_secs, 11);
2903        assert_eq!(options.push.stale_device_max_age_days, 120);
2904        assert!(options.push.analytics_enabled);
2905        assert_eq!(options.push.default_quotas.acceptance_rps, 700);
2906        assert_eq!(options.push.default_quotas.delivery_quota_daily, 8_000);
2907        assert_eq!(options.push.default_quotas.fanout_max, 9_000);
2908        assert_eq!(options.push.default_quotas.inflight_max, 1_000);
2909    }
2910}
2911
2912impl ClusterHealthConfig {
2913    pub fn validate(&self) -> Result<(), String> {
2914        if self.heartbeat_interval_ms == 0 {
2915            return Err("heartbeat_interval_ms must be greater than 0".to_string());
2916        }
2917        if self.node_timeout_ms == 0 {
2918            return Err("node_timeout_ms must be greater than 0".to_string());
2919        }
2920        if self.cleanup_interval_ms == 0 {
2921            return Err("cleanup_interval_ms must be greater than 0".to_string());
2922        }
2923
2924        if self.heartbeat_interval_ms > self.node_timeout_ms / 3 {
2925            return Err(format!(
2926                "heartbeat_interval_ms ({}) should be at least 3x smaller than node_timeout_ms ({}) to avoid false positive dead node detection. Recommended: heartbeat_interval_ms <= {}",
2927                self.heartbeat_interval_ms,
2928                self.node_timeout_ms,
2929                self.node_timeout_ms / 3
2930            ));
2931        }
2932
2933        if self.cleanup_interval_ms > self.node_timeout_ms {
2934            return Err(format!(
2935                "cleanup_interval_ms ({}) should not be larger than node_timeout_ms ({}) to ensure timely dead node detection",
2936                self.cleanup_interval_ms, self.node_timeout_ms
2937            ));
2938        }
2939
2940        Ok(())
2941    }
2942}
2943
2944impl ServerOptions {
2945    pub async fn load_from_file(path: &str) -> Result<Self, Box<dyn std::error::Error>> {
2946        let content = tokio::fs::read_to_string(path).await?;
2947        let options: Self = if path.ends_with(".toml") {
2948            toml::from_str(&content)?
2949        } else {
2950            // Legacy JSON support
2951            sonic_rs::from_str(&content)?
2952        };
2953        Ok(options)
2954    }
2955
2956    pub async fn override_from_env(&mut self) -> Result<(), Box<dyn std::error::Error>> {
2957        // --- General Settings ---
2958        if let Ok(mode) = std::env::var("ENVIRONMENT") {
2959            self.mode = mode;
2960        }
2961        self.debug = parse_bool_env("DEBUG_MODE", self.debug);
2962        if parse_bool_env("DEBUG", false) {
2963            self.debug = true;
2964            info!("DEBUG environment variable forces debug mode ON");
2965        }
2966
2967        self.activity_timeout = parse_env::<u64>("ACTIVITY_TIMEOUT", self.activity_timeout);
2968
2969        if let Ok(host) = std::env::var("HOST") {
2970            self.host = host;
2971        }
2972        self.port = parse_env::<u16>("PORT", self.port);
2973        self.shutdown_grace_period =
2974            parse_env::<u64>("SHUTDOWN_GRACE_PERIOD", self.shutdown_grace_period);
2975        self.user_authentication_timeout = parse_env::<u64>(
2976            "USER_AUTHENTICATION_TIMEOUT",
2977            self.user_authentication_timeout,
2978        );
2979        self.websocket_max_payload_kb =
2980            parse_env::<u32>("WEBSOCKET_MAX_PAYLOAD_KB", self.websocket_max_payload_kb);
2981        if let Ok(id) = std::env::var("INSTANCE_PROCESS_ID") {
2982            self.instance.process_id = id;
2983        }
2984
2985        // --- Driver Configuration ---
2986        if let Ok(driver_str) = std::env::var("ADAPTER_DRIVER") {
2987            self.adapter.driver =
2988                parse_driver_enum(driver_str, self.adapter.driver.clone(), "Adapter");
2989        }
2990        self.adapter.buffer_multiplier_per_cpu = parse_env::<usize>(
2991            "ADAPTER_BUFFER_MULTIPLIER_PER_CPU",
2992            self.adapter.buffer_multiplier_per_cpu,
2993        );
2994        self.adapter.enable_socket_counting = parse_env::<bool>(
2995            "ADAPTER_ENABLE_SOCKET_COUNTING",
2996            self.adapter.enable_socket_counting,
2997        );
2998        self.adapter.fallback_to_local =
2999            parse_env::<bool>("ADAPTER_FALLBACK_TO_LOCAL", self.adapter.fallback_to_local);
3000        if let Ok(driver_str) = std::env::var("CACHE_DRIVER") {
3001            self.cache.driver = parse_driver_enum(driver_str, self.cache.driver.clone(), "Cache");
3002        }
3003        if let Ok(driver_str) = std::env::var("QUEUE_DRIVER") {
3004            self.queue.driver = parse_driver_enum(driver_str, self.queue.driver.clone(), "Queue");
3005        }
3006        if let Ok(driver_str) = std::env::var("APP_MANAGER_DRIVER") {
3007            self.app_manager.driver =
3008                parse_driver_enum(driver_str, self.app_manager.driver.clone(), "AppManager");
3009        }
3010        if let Ok(driver_str) = std::env::var("RATE_LIMITER_DRIVER") {
3011            self.rate_limiter.driver = parse_driver_enum(
3012                driver_str,
3013                self.rate_limiter.driver.clone(),
3014                "RateLimiter Backend",
3015            );
3016        }
3017
3018        // --- Database: Redis ---
3019        if let Ok(host) = std::env::var("DATABASE_REDIS_HOST") {
3020            self.database.redis.host = host;
3021        }
3022        self.database.redis.port =
3023            parse_env::<u16>("DATABASE_REDIS_PORT", self.database.redis.port);
3024        if let Ok(username) = std::env::var("DATABASE_REDIS_USERNAME") {
3025            self.database.redis.username = if username.is_empty() {
3026                None
3027            } else {
3028                Some(username)
3029            };
3030        }
3031        if let Ok(password) = std::env::var("DATABASE_REDIS_PASSWORD") {
3032            self.database.redis.password = Some(password);
3033        }
3034        self.database.redis.db = parse_env::<u32>("DATABASE_REDIS_DB", self.database.redis.db);
3035        if let Ok(prefix) = std::env::var("DATABASE_REDIS_KEY_PREFIX") {
3036            self.database.redis.key_prefix = prefix;
3037        }
3038        if let Ok(cluster_username) = std::env::var("DATABASE_REDIS_CLUSTER_USERNAME") {
3039            self.database.redis.cluster.username = if cluster_username.is_empty() {
3040                None
3041            } else {
3042                Some(cluster_username)
3043            };
3044        }
3045        if let Ok(cluster_password) = std::env::var("DATABASE_REDIS_CLUSTER_PASSWORD") {
3046            self.database.redis.cluster.password = Some(cluster_password);
3047        }
3048        self.database.redis.cluster.use_tls = parse_bool_env(
3049            "DATABASE_REDIS_CLUSTER_USE_TLS",
3050            self.database.redis.cluster.use_tls,
3051        );
3052
3053        // --- Database: MySQL ---
3054        if let Ok(host) = std::env::var("DATABASE_MYSQL_HOST") {
3055            self.database.mysql.host = host;
3056        }
3057        self.database.mysql.port =
3058            parse_env::<u16>("DATABASE_MYSQL_PORT", self.database.mysql.port);
3059        if let Ok(user) = std::env::var("DATABASE_MYSQL_USERNAME") {
3060            self.database.mysql.username = user;
3061        }
3062        if let Ok(pass) = std::env::var("DATABASE_MYSQL_PASSWORD") {
3063            self.database.mysql.password = pass;
3064        }
3065        if let Ok(db) = std::env::var("DATABASE_MYSQL_DATABASE") {
3066            self.database.mysql.database = db;
3067        }
3068        if let Ok(table) = std::env::var("DATABASE_MYSQL_TABLE_NAME") {
3069            self.database.mysql.table_name = table;
3070        }
3071        override_db_pool_settings(&mut self.database.mysql, "DATABASE_MYSQL");
3072
3073        // --- Database: PostgreSQL ---
3074        if let Ok(host) = std::env::var("DATABASE_POSTGRES_HOST") {
3075            self.database.postgres.host = host;
3076        }
3077        self.database.postgres.port =
3078            parse_env::<u16>("DATABASE_POSTGRES_PORT", self.database.postgres.port);
3079        if let Ok(user) = std::env::var("DATABASE_POSTGRES_USERNAME") {
3080            self.database.postgres.username = user;
3081        }
3082        if let Ok(pass) = std::env::var("DATABASE_POSTGRES_PASSWORD") {
3083            self.database.postgres.password = pass;
3084        }
3085        if let Ok(db) = std::env::var("DATABASE_POSTGRES_DATABASE") {
3086            self.database.postgres.database = db;
3087        }
3088        override_db_pool_settings(&mut self.database.postgres, "DATABASE_POSTGRES");
3089
3090        // --- Database: DynamoDB ---
3091        if let Ok(region) = std::env::var("DATABASE_DYNAMODB_REGION") {
3092            self.database.dynamodb.region = region;
3093        }
3094        if let Ok(table) = std::env::var("DATABASE_DYNAMODB_TABLE_NAME") {
3095            self.database.dynamodb.table_name = table;
3096        }
3097        if let Ok(endpoint) = std::env::var("DATABASE_DYNAMODB_ENDPOINT_URL") {
3098            self.database.dynamodb.endpoint_url = Some(endpoint);
3099        }
3100        if let Ok(key_id) = std::env::var("AWS_ACCESS_KEY_ID") {
3101            self.database.dynamodb.aws_access_key_id = Some(key_id);
3102        }
3103        if let Ok(secret) = std::env::var("AWS_SECRET_ACCESS_KEY") {
3104            self.database.dynamodb.aws_secret_access_key = Some(secret);
3105        }
3106
3107        // --- Database: SurrealDB ---
3108        if let Ok(url) = std::env::var("DATABASE_SURREALDB_URL") {
3109            self.database.surrealdb.url = url;
3110        }
3111        if let Ok(namespace) = std::env::var("DATABASE_SURREALDB_NAMESPACE") {
3112            self.database.surrealdb.namespace = namespace;
3113        }
3114        if let Ok(database) = std::env::var("DATABASE_SURREALDB_DATABASE") {
3115            self.database.surrealdb.database = database;
3116        }
3117        if let Ok(username) = std::env::var("DATABASE_SURREALDB_USERNAME") {
3118            self.database.surrealdb.username = username;
3119        }
3120        if let Ok(password) = std::env::var("DATABASE_SURREALDB_PASSWORD") {
3121            self.database.surrealdb.password = password;
3122        }
3123        if let Ok(table) = std::env::var("DATABASE_SURREALDB_TABLE_NAME") {
3124            self.database.surrealdb.table_name = table;
3125        }
3126
3127        // --- Redis Cluster ---
3128        let apply_redis_cluster_nodes = |options: &mut Self, nodes: &str| {
3129            let node_list: Vec<String> = nodes
3130                .split(',')
3131                .map(|s| s.trim())
3132                .filter(|s| !s.is_empty())
3133                .map(ToString::to_string)
3134                .collect();
3135
3136            options.adapter.cluster.nodes = node_list.clone();
3137            options.queue.redis_cluster.nodes = node_list.clone();
3138
3139            let parsed_nodes: Vec<ClusterNode> = node_list
3140                .iter()
3141                .filter_map(|seed| ClusterNode::from_seed(seed))
3142                .collect();
3143            options.database.redis.cluster.nodes = parsed_nodes.clone();
3144            options.database.redis.cluster_nodes = parsed_nodes;
3145        };
3146
3147        if let Ok(nodes) = std::env::var("REDIS_CLUSTER_NODES") {
3148            apply_redis_cluster_nodes(self, &nodes);
3149        }
3150        if let Ok(nodes) = std::env::var("DATABASE_REDIS_CLUSTER_NODES") {
3151            apply_redis_cluster_nodes(self, &nodes);
3152        }
3153        self.queue.redis_cluster.concurrency = parse_env::<u32>(
3154            "REDIS_CLUSTER_QUEUE_CONCURRENCY",
3155            self.queue.redis_cluster.concurrency,
3156        );
3157        if let Ok(prefix) = std::env::var("REDIS_CLUSTER_QUEUE_PREFIX") {
3158            self.queue.redis_cluster.prefix = Some(prefix);
3159        }
3160
3161        // --- SSL Configuration ---
3162        self.ssl.enabled = parse_bool_env("SSL_ENABLED", self.ssl.enabled);
3163        if let Ok(val) = std::env::var("SSL_CERT_PATH") {
3164            self.ssl.cert_path = val;
3165        }
3166        if let Ok(val) = std::env::var("SSL_KEY_PATH") {
3167            self.ssl.key_path = val;
3168        }
3169        self.ssl.redirect_http = parse_bool_env("SSL_REDIRECT_HTTP", self.ssl.redirect_http);
3170        if let Some(port) = parse_env_optional::<u16>("SSL_HTTP_PORT") {
3171            self.ssl.http_port = Some(port);
3172        }
3173
3174        // --- Unix Socket Configuration ---
3175        self.unix_socket.enabled = parse_bool_env("UNIX_SOCKET_ENABLED", self.unix_socket.enabled);
3176        if let Ok(path) = std::env::var("UNIX_SOCKET_PATH") {
3177            self.unix_socket.path = path;
3178        }
3179        if let Ok(mode_str) = std::env::var("UNIX_SOCKET_PERMISSION_MODE") {
3180            if mode_str.chars().all(|c| c.is_digit(8)) {
3181                if let Ok(mode) = u32::from_str_radix(&mode_str, 8) {
3182                    if mode <= 0o777 {
3183                        self.unix_socket.permission_mode = mode;
3184                    } else {
3185                        warn!(
3186                            "UNIX_SOCKET_PERMISSION_MODE '{}' exceeds maximum value 777. Using default: {:o}",
3187                            mode_str, self.unix_socket.permission_mode
3188                        );
3189                    }
3190                } else {
3191                    warn!(
3192                        "Failed to parse UNIX_SOCKET_PERMISSION_MODE '{}' as octal. Using default: {:o}",
3193                        mode_str, self.unix_socket.permission_mode
3194                    );
3195                }
3196            } else {
3197                warn!(
3198                    "UNIX_SOCKET_PERMISSION_MODE '{}' must contain only octal digits (0-7). Using default: {:o}",
3199                    mode_str, self.unix_socket.permission_mode
3200                );
3201            }
3202        }
3203
3204        // --- Metrics ---
3205        if let Ok(driver_str) = std::env::var("METRICS_DRIVER") {
3206            self.metrics.driver =
3207                parse_driver_enum(driver_str, self.metrics.driver.clone(), "Metrics");
3208        }
3209        self.metrics.enabled = parse_bool_env("METRICS_ENABLED", self.metrics.enabled);
3210        if let Ok(val) = std::env::var("METRICS_HOST") {
3211            self.metrics.host = val;
3212        }
3213        self.metrics.port = parse_env::<u16>("METRICS_PORT", self.metrics.port);
3214        if let Ok(val) = std::env::var("METRICS_PROMETHEUS_PREFIX") {
3215            self.metrics.prometheus.prefix = val;
3216        }
3217
3218        // --- HTTP API ---
3219        self.http_api.usage_enabled =
3220            parse_bool_env("HTTP_API_USAGE_ENABLED", self.http_api.usage_enabled);
3221
3222        // --- Rate Limiter ---
3223        self.rate_limiter.enabled =
3224            parse_bool_env("RATE_LIMITER_ENABLED", self.rate_limiter.enabled);
3225        self.rate_limiter.api_rate_limit.max_requests = parse_env::<u32>(
3226            "RATE_LIMITER_API_MAX_REQUESTS",
3227            self.rate_limiter.api_rate_limit.max_requests,
3228        );
3229        self.rate_limiter.api_rate_limit.window_seconds = parse_env::<u64>(
3230            "RATE_LIMITER_API_WINDOW_SECONDS",
3231            self.rate_limiter.api_rate_limit.window_seconds,
3232        );
3233        if let Some(hops) = parse_env_optional::<u32>("RATE_LIMITER_API_TRUST_HOPS") {
3234            self.rate_limiter.api_rate_limit.trust_hops = Some(hops);
3235        }
3236        self.rate_limiter.websocket_rate_limit.max_requests = parse_env::<u32>(
3237            "RATE_LIMITER_WS_MAX_REQUESTS",
3238            self.rate_limiter.websocket_rate_limit.max_requests,
3239        );
3240        self.rate_limiter.websocket_rate_limit.window_seconds = parse_env::<u64>(
3241            "RATE_LIMITER_WS_WINDOW_SECONDS",
3242            self.rate_limiter.websocket_rate_limit.window_seconds,
3243        );
3244        if let Ok(prefix) = std::env::var("RATE_LIMITER_REDIS_PREFIX") {
3245            self.rate_limiter.redis.prefix = Some(prefix);
3246        }
3247
3248        // --- Queue: Redis ---
3249        self.queue.redis.concurrency =
3250            parse_env::<u32>("QUEUE_REDIS_CONCURRENCY", self.queue.redis.concurrency);
3251        if let Ok(prefix) = std::env::var("QUEUE_REDIS_PREFIX") {
3252            self.queue.redis.prefix = Some(prefix);
3253        }
3254
3255        // --- Queue: SQS ---
3256        if let Ok(region) = std::env::var("QUEUE_SQS_REGION") {
3257            self.queue.sqs.region = region;
3258        }
3259        self.queue.sqs.visibility_timeout = parse_env::<i32>(
3260            "QUEUE_SQS_VISIBILITY_TIMEOUT",
3261            self.queue.sqs.visibility_timeout,
3262        );
3263        self.queue.sqs.max_messages =
3264            parse_env::<i32>("QUEUE_SQS_MAX_MESSAGES", self.queue.sqs.max_messages);
3265        self.queue.sqs.wait_time_seconds = parse_env::<i32>(
3266            "QUEUE_SQS_WAIT_TIME_SECONDS",
3267            self.queue.sqs.wait_time_seconds,
3268        );
3269        self.queue.sqs.concurrency =
3270            parse_env::<u32>("QUEUE_SQS_CONCURRENCY", self.queue.sqs.concurrency);
3271        self.queue.sqs.fifo = parse_bool_env("QUEUE_SQS_FIFO", self.queue.sqs.fifo);
3272        if let Ok(endpoint) = std::env::var("QUEUE_SQS_ENDPOINT_URL") {
3273            self.queue.sqs.endpoint_url = Some(endpoint);
3274        }
3275
3276        // --- Queue: SNS ---
3277        if let Ok(region) = std::env::var("QUEUE_SNS_REGION") {
3278            self.queue.sns.region = region;
3279        }
3280        if let Ok(topic_arn) = std::env::var("QUEUE_SNS_TOPIC_ARN") {
3281            self.queue.sns.topic_arn = topic_arn;
3282        }
3283        if let Ok(endpoint) = std::env::var("QUEUE_SNS_ENDPOINT_URL") {
3284            self.queue.sns.endpoint_url = Some(endpoint);
3285        }
3286
3287        // --- Webhooks ---
3288        self.webhooks.batching.enabled =
3289            parse_bool_env("WEBHOOK_BATCHING_ENABLED", self.webhooks.batching.enabled);
3290        self.webhooks.batching.duration =
3291            parse_env::<u64>("WEBHOOK_BATCHING_DURATION", self.webhooks.batching.duration);
3292        self.webhooks.batching.size =
3293            parse_env::<usize>("WEBHOOK_BATCHING_SIZE", self.webhooks.batching.size);
3294
3295        // --- NATS Adapter ---
3296        if let Ok(servers) = std::env::var("NATS_SERVERS") {
3297            self.adapter.nats.servers = servers.split(',').map(|s| s.trim().to_string()).collect();
3298        }
3299        if let Ok(user) = std::env::var("NATS_USERNAME") {
3300            self.adapter.nats.username = Some(user);
3301        }
3302        if let Ok(pass) = std::env::var("NATS_PASSWORD") {
3303            self.adapter.nats.password = Some(pass);
3304        }
3305        if let Ok(token) = std::env::var("NATS_TOKEN") {
3306            self.adapter.nats.token = Some(token);
3307        }
3308        if let Ok(prefix) = std::env::var("NATS_PREFIX") {
3309            self.adapter.nats.prefix = prefix;
3310        }
3311        self.adapter.nats.connection_timeout_ms = parse_env::<u64>(
3312            "NATS_CONNECTION_TIMEOUT_MS",
3313            self.adapter.nats.connection_timeout_ms,
3314        );
3315        self.adapter.nats.request_timeout_ms = parse_env::<u64>(
3316            "NATS_REQUEST_TIMEOUT_MS",
3317            self.adapter.nats.request_timeout_ms,
3318        );
3319        self.adapter.nats.discovery_max_wait_ms = parse_env::<u64>(
3320            "NATS_DISCOVERY_MAX_WAIT_MS",
3321            self.adapter.nats.discovery_max_wait_ms,
3322        );
3323        self.adapter.nats.discovery_idle_wait_ms = parse_env::<u64>(
3324            "NATS_DISCOVERY_IDLE_WAIT_MS",
3325            self.adapter.nats.discovery_idle_wait_ms,
3326        );
3327        if let Some(nodes) = parse_env_optional::<u32>("NATS_NODES_NUMBER") {
3328            self.adapter.nats.nodes_number = Some(nodes);
3329        }
3330        if let Some(v) = parse_env_optional::<usize>("NATS_SUBSCRIPTION_CAPACITY") {
3331            self.adapter.nats.subscription_capacity = Some(v);
3332        }
3333        if let Some(v) = parse_env_optional::<usize>("NATS_CLIENT_CAPACITY") {
3334            self.adapter.nats.client_capacity = Some(v);
3335        }
3336        if let Some(v) = parse_env_optional::<usize>("NATS_MAX_RECONNECTS") {
3337            self.adapter.nats.max_reconnects = Some(v);
3338        }
3339        if let Some(v) = parse_env_optional::<usize>("NATS_PRESENCE_SYNC_CHUNK_SIZE") {
3340            self.adapter.nats.presence_sync_chunk_size = Some(v);
3341        }
3342
3343        // --- Pulsar Adapter ---
3344        if let Ok(url) = std::env::var("PULSAR_URL") {
3345            self.adapter.pulsar.url = url;
3346        }
3347        if let Ok(prefix) = std::env::var("PULSAR_PREFIX") {
3348            self.adapter.pulsar.prefix = prefix;
3349        }
3350        if let Ok(token) = std::env::var("PULSAR_TOKEN") {
3351            self.adapter.pulsar.token = Some(token);
3352        }
3353        self.adapter.pulsar.request_timeout_ms = parse_env::<u64>(
3354            "PULSAR_REQUEST_TIMEOUT_MS",
3355            self.adapter.pulsar.request_timeout_ms,
3356        );
3357        if let Some(nodes) = parse_env_optional::<u32>("PULSAR_NODES_NUMBER") {
3358            self.adapter.pulsar.nodes_number = Some(nodes);
3359        }
3360
3361        // --- RabbitMQ Adapter ---
3362        if let Ok(url) = std::env::var("RABBITMQ_URL") {
3363            self.adapter.rabbitmq.url = url;
3364        }
3365        if let Ok(prefix) = std::env::var("RABBITMQ_PREFIX") {
3366            self.adapter.rabbitmq.prefix = prefix;
3367        }
3368        self.adapter.rabbitmq.connection_timeout_ms = parse_env::<u64>(
3369            "RABBITMQ_CONNECTION_TIMEOUT_MS",
3370            self.adapter.rabbitmq.connection_timeout_ms,
3371        );
3372        self.adapter.rabbitmq.request_timeout_ms = parse_env::<u64>(
3373            "RABBITMQ_REQUEST_TIMEOUT_MS",
3374            self.adapter.rabbitmq.request_timeout_ms,
3375        );
3376        if let Some(nodes) = parse_env_optional::<u32>("RABBITMQ_NODES_NUMBER") {
3377            self.adapter.rabbitmq.nodes_number = Some(nodes);
3378        }
3379
3380        // --- Google Pub/Sub Adapter ---
3381        if let Ok(project_id) = std::env::var("GOOGLE_PUBSUB_PROJECT_ID") {
3382            self.adapter.google_pubsub.project_id = project_id;
3383        }
3384        if let Ok(prefix) = std::env::var("GOOGLE_PUBSUB_PREFIX") {
3385            self.adapter.google_pubsub.prefix = prefix;
3386        }
3387        if let Ok(emulator_host) = std::env::var("PUBSUB_EMULATOR_HOST") {
3388            self.adapter.google_pubsub.emulator_host = Some(emulator_host);
3389        }
3390        self.adapter.google_pubsub.request_timeout_ms = parse_env::<u64>(
3391            "GOOGLE_PUBSUB_REQUEST_TIMEOUT_MS",
3392            self.adapter.google_pubsub.request_timeout_ms,
3393        );
3394        if let Some(nodes) = parse_env_optional::<u32>("GOOGLE_PUBSUB_NODES_NUMBER") {
3395            self.adapter.google_pubsub.nodes_number = Some(nodes);
3396        }
3397
3398        // --- Kafka Adapter ---
3399        if let Ok(brokers) = std::env::var("KAFKA_BROKERS") {
3400            self.adapter.kafka.brokers = brokers.split(',').map(|s| s.trim().to_string()).collect();
3401        }
3402        if let Ok(prefix) = std::env::var("KAFKA_PREFIX") {
3403            self.adapter.kafka.prefix = prefix;
3404        }
3405        if let Ok(protocol) = std::env::var("KAFKA_SECURITY_PROTOCOL") {
3406            self.adapter.kafka.security_protocol = Some(protocol);
3407        }
3408        if let Ok(mechanism) = std::env::var("KAFKA_SASL_MECHANISM") {
3409            self.adapter.kafka.sasl_mechanism = Some(mechanism);
3410        }
3411        if let Ok(username) = std::env::var("KAFKA_SASL_USERNAME") {
3412            self.adapter.kafka.sasl_username = Some(username);
3413        }
3414        if let Ok(password) = std::env::var("KAFKA_SASL_PASSWORD") {
3415            self.adapter.kafka.sasl_password = Some(password);
3416        }
3417        self.adapter.kafka.request_timeout_ms = parse_env::<u64>(
3418            "KAFKA_REQUEST_TIMEOUT_MS",
3419            self.adapter.kafka.request_timeout_ms,
3420        );
3421        if let Some(nodes) = parse_env_optional::<u32>("KAFKA_NODES_NUMBER") {
3422            self.adapter.kafka.nodes_number = Some(nodes);
3423        }
3424
3425        // --- Apache Iggy Adapter / Queue ---
3426        if let Ok(connection_string) = std::env::var("IGGY_CONNECTION_STRING") {
3427            self.adapter.iggy.connection_string = connection_string.clone();
3428            self.queue.iggy.connection_string = connection_string;
3429        }
3430        if let Ok(username) = std::env::var("IGGY_USERNAME") {
3431            let username = (!username.is_empty()).then_some(username);
3432            self.adapter.iggy.username = username.clone();
3433            self.queue.iggy.username = username;
3434        }
3435        if let Ok(password) = std::env::var("IGGY_PASSWORD") {
3436            let password = (!password.is_empty()).then_some(password);
3437            self.adapter.iggy.password = password.clone();
3438            self.queue.iggy.password = password;
3439        }
3440        if let Ok(consumer_name) = std::env::var("IGGY_CONSUMER_NAME") {
3441            let consumer_name = (!consumer_name.is_empty()).then_some(consumer_name);
3442            self.adapter.iggy.consumer_name = consumer_name.clone();
3443            self.queue.iggy.consumer_name = consumer_name;
3444        } else if let Ok(process_id) = std::env::var("INSTANCE_PROCESS_ID") {
3445            let process_id = (!process_id.is_empty()).then_some(process_id);
3446            self.adapter.iggy.consumer_name = process_id.clone();
3447            self.queue.iggy.consumer_name = process_id;
3448        }
3449        if let Ok(stream) = std::env::var("IGGY_STREAM") {
3450            self.adapter.iggy.stream = stream.clone();
3451            self.queue.iggy.stream = stream;
3452        }
3453        if let Ok(prefix) = std::env::var("IGGY_TOPIC_PREFIX") {
3454            self.adapter.iggy.topic_prefix = prefix;
3455        }
3456        if let Ok(prefix) = std::env::var("IGGY_QUEUE_TOPIC_PREFIX") {
3457            self.queue.iggy.queue_topic_prefix = prefix;
3458        }
3459        if let Ok(prefix) = std::env::var("IGGY_CONSUMER_GROUP_PREFIX") {
3460            self.queue.iggy.consumer_group_prefix = prefix;
3461        }
3462        self.adapter.iggy.request_timeout_ms = parse_env::<u64>(
3463            "IGGY_REQUEST_TIMEOUT_MS",
3464            self.adapter.iggy.request_timeout_ms,
3465        );
3466        self.queue.iggy.request_timeout_ms = parse_env::<u64>(
3467            "IGGY_REQUEST_TIMEOUT_MS",
3468            self.queue.iggy.request_timeout_ms,
3469        );
3470        self.adapter.iggy.poll_interval_ms =
3471            parse_env::<u64>("IGGY_POLL_INTERVAL_MS", self.adapter.iggy.poll_interval_ms);
3472        self.queue.iggy.poll_interval_ms =
3473            parse_env::<u64>("IGGY_POLL_INTERVAL_MS", self.queue.iggy.poll_interval_ms);
3474        self.adapter.iggy.poll_batch_size =
3475            parse_env::<u32>("IGGY_POLL_BATCH_SIZE", self.adapter.iggy.poll_batch_size);
3476        self.queue.iggy.poll_batch_size =
3477            parse_env::<u32>("IGGY_POLL_BATCH_SIZE", self.queue.iggy.poll_batch_size);
3478        self.adapter.iggy.partitions_count = parse_env::<u32>(
3479            "ADAPTER_IGGY_PARTITIONS_COUNT",
3480            parse_env::<u32>("IGGY_PARTITIONS_COUNT", self.adapter.iggy.partitions_count),
3481        );
3482        self.queue.iggy.partitions_count = parse_env::<u32>(
3483            "QUEUE_IGGY_PARTITIONS_COUNT",
3484            parse_env::<u32>("IGGY_PARTITIONS_COUNT", self.queue.iggy.partitions_count),
3485        );
3486        self.adapter.iggy.partition_id = parse_env::<u32>(
3487            "ADAPTER_IGGY_PARTITION_ID",
3488            parse_env::<u32>("IGGY_PARTITION_ID", self.adapter.iggy.partition_id),
3489        );
3490        self.queue.iggy.partition_id = parse_env::<u32>(
3491            "QUEUE_IGGY_PARTITION_ID",
3492            parse_env::<u32>("IGGY_PARTITION_ID", self.queue.iggy.partition_id),
3493        );
3494        self.adapter.iggy.auto_create =
3495            parse_env::<bool>("IGGY_AUTO_CREATE", self.adapter.iggy.auto_create);
3496        self.queue.iggy.auto_create =
3497            parse_env::<bool>("IGGY_AUTO_CREATE", self.queue.iggy.auto_create);
3498        self.adapter.iggy.start_from_latest = parse_env::<bool>(
3499            "IGGY_START_FROM_LATEST",
3500            self.adapter.iggy.start_from_latest,
3501        );
3502        if let Some(nodes) = parse_env_optional::<u32>("IGGY_NODES_NUMBER") {
3503            self.adapter.iggy.nodes_number = Some(nodes);
3504            self.queue.iggy.nodes_number = Some(nodes);
3505        }
3506
3507        // --- CORS ---
3508        if let Ok(origins) = std::env::var("CORS_ORIGINS") {
3509            let parsed: Vec<String> = origins.split(',').map(|s| s.trim().to_string()).collect();
3510            if let Err(e) = crate::origin_validation::OriginValidator::validate_patterns(&parsed) {
3511                warn!(
3512                    "CORS_ORIGINS contains invalid patterns: {}. Keeping previous CORS origins.",
3513                    e
3514                );
3515            } else {
3516                self.cors.origin = parsed;
3517            }
3518        }
3519        if let Ok(methods) = std::env::var("CORS_METHODS") {
3520            self.cors.methods = methods.split(',').map(|s| s.trim().to_string()).collect();
3521        }
3522        if let Ok(headers) = std::env::var("CORS_HEADERS") {
3523            self.cors.allowed_headers = headers.split(',').map(|s| s.trim().to_string()).collect();
3524        }
3525        self.cors.credentials = parse_bool_env("CORS_CREDENTIALS", self.cors.credentials);
3526
3527        // --- Performance Tuning ---
3528        self.database_pooling.enabled =
3529            parse_bool_env("DATABASE_POOLING_ENABLED", self.database_pooling.enabled);
3530        if let Some(min) = parse_env_optional::<u32>("DATABASE_POOL_MIN") {
3531            self.database_pooling.min = min;
3532        }
3533        if let Some(max) = parse_env_optional::<u32>("DATABASE_POOL_MAX") {
3534            self.database_pooling.max = max;
3535        }
3536
3537        if let Some(pool_size) = parse_env_optional::<u32>("DATABASE_CONNECTION_POOL_SIZE") {
3538            self.database.mysql.connection_pool_size = pool_size;
3539            self.database.postgres.connection_pool_size = pool_size;
3540        }
3541        if let Some(cache_ttl) = parse_env_optional::<u64>("CACHE_TTL_SECONDS") {
3542            self.app_manager.cache.ttl = cache_ttl;
3543            self.channel_limits.cache_ttl = cache_ttl;
3544            self.database.mysql.cache_ttl = cache_ttl;
3545            self.database.postgres.cache_ttl = cache_ttl;
3546            self.database.surrealdb.cache_ttl = cache_ttl;
3547            self.cache.memory.ttl = cache_ttl;
3548        }
3549        if let Some(cleanup_interval) = parse_env_optional::<u64>("CACHE_CLEANUP_INTERVAL") {
3550            self.database.mysql.cache_cleanup_interval = cleanup_interval;
3551            self.database.postgres.cache_cleanup_interval = cleanup_interval;
3552            self.cache.memory.cleanup_interval = cleanup_interval;
3553        }
3554        if let Some(max_capacity) = parse_env_optional::<u64>("CACHE_MAX_CAPACITY") {
3555            self.database.mysql.cache_max_capacity = max_capacity;
3556            self.database.postgres.cache_max_capacity = max_capacity;
3557            self.database.surrealdb.cache_max_capacity = max_capacity;
3558            self.cache.memory.max_capacity = max_capacity;
3559        }
3560
3561        let skip_inline_apps = parse_bool_env("SOCKUDO_SKIP_INLINE_APPS", false)
3562            || !parse_bool_env("APP_MANAGER_REGISTER_INLINE_APPS", true);
3563        if skip_inline_apps {
3564            let app_count = self.app_manager.array.apps.len();
3565            self.app_manager.array.apps.clear();
3566            if app_count > 0 {
3567                info!(
3568                    "Skipping {} inline app(s) from configuration due to environment override",
3569                    app_count
3570                );
3571            }
3572        }
3573
3574        let default_app_id = std::env::var("SOCKUDO_DEFAULT_APP_ID").ok();
3575        let default_app_key = std::env::var("SOCKUDO_DEFAULT_APP_KEY").ok();
3576        let default_app_secret = std::env::var("SOCKUDO_DEFAULT_APP_SECRET").ok();
3577        let default_app_enabled_env = std::env::var("SOCKUDO_DEFAULT_APP_ENABLED").ok();
3578        let default_app_enabled = parse_bool_env("SOCKUDO_DEFAULT_APP_ENABLED", true);
3579        let default_app_credentials_configured =
3580            default_app_id.is_some() || default_app_key.is_some() || default_app_secret.is_some();
3581        let default_app_env_configured =
3582            default_app_credentials_configured || default_app_enabled_env.is_some();
3583        let default_app_should_override_inline = default_app_credentials_configured
3584            || default_app_enabled_env.is_some_and(|_| !default_app_enabled);
3585
3586        if default_app_should_override_inline {
3587            let app_count = self.app_manager.array.apps.len();
3588            self.app_manager.array.apps.clear();
3589            if app_count > 0 {
3590                info!(
3591                    "Replacing {} inline app(s) from configuration with SOCKUDO_DEFAULT_APP_* settings",
3592                    app_count
3593                );
3594            }
3595        }
3596
3597        if let (Some(app_id), Some(app_key), Some(app_secret)) =
3598            (default_app_id, default_app_key, default_app_secret)
3599            && default_app_enabled
3600        {
3601            let default_app = App::from_policy(
3602                app_id,
3603                app_key,
3604                app_secret,
3605                default_app_enabled,
3606                crate::app::AppPolicy {
3607                    limits: crate::app::AppLimitsPolicy {
3608                        max_connections: parse_env::<u32>(
3609                            "SOCKUDO_DEFAULT_APP_MAX_CONNECTIONS",
3610                            100,
3611                        ),
3612                        max_backend_events_per_second: Some(parse_env::<u32>(
3613                            "SOCKUDO_DEFAULT_APP_MAX_BACKEND_EVENTS_PER_SECOND",
3614                            100,
3615                        )),
3616                        max_client_events_per_second: parse_env::<u32>(
3617                            "SOCKUDO_DEFAULT_APP_MAX_CLIENT_EVENTS_PER_SECOND",
3618                            100,
3619                        ),
3620                        max_read_requests_per_second: Some(parse_env::<u32>(
3621                            "SOCKUDO_DEFAULT_APP_MAX_READ_REQUESTS_PER_SECOND",
3622                            100,
3623                        )),
3624                        max_presence_members_per_channel: Some(parse_env::<u32>(
3625                            "SOCKUDO_DEFAULT_APP_MAX_PRESENCE_MEMBERS_PER_CHANNEL",
3626                            100,
3627                        )),
3628                        max_presence_member_size_in_kb: Some(parse_env::<u32>(
3629                            "SOCKUDO_DEFAULT_APP_MAX_PRESENCE_MEMBER_SIZE_IN_KB",
3630                            100,
3631                        )),
3632                        max_channel_name_length: Some(parse_env::<u32>(
3633                            "SOCKUDO_DEFAULT_APP_MAX_CHANNEL_NAME_LENGTH",
3634                            100,
3635                        )),
3636                        max_event_channels_at_once: Some(parse_env::<u32>(
3637                            "SOCKUDO_DEFAULT_APP_MAX_EVENT_CHANNELS_AT_ONCE",
3638                            100,
3639                        )),
3640                        max_event_name_length: Some(parse_env::<u32>(
3641                            "SOCKUDO_DEFAULT_APP_MAX_EVENT_NAME_LENGTH",
3642                            100,
3643                        )),
3644                        max_event_payload_in_kb: Some(parse_env::<u32>(
3645                            "SOCKUDO_DEFAULT_APP_MAX_EVENT_PAYLOAD_IN_KB",
3646                            100,
3647                        )),
3648                        max_event_batch_size: Some(parse_env::<u32>(
3649                            "SOCKUDO_DEFAULT_APP_MAX_EVENT_BATCH_SIZE",
3650                            100,
3651                        )),
3652                        decay_seconds: None,
3653                        terminate_on_limit: false,
3654                        message_rate_limit: None,
3655                    },
3656                    features: crate::app::AppFeaturesPolicy {
3657                        enable_client_messages: std::env::var(
3658                            "SOCKUDO_DEFAULT_APP_ENABLE_CLIENT_MESSAGES",
3659                        )
3660                        .ok()
3661                        .map(|value| {
3662                            matches!(
3663                                value.trim().to_ascii_lowercase().as_str(),
3664                                "true" | "1" | "yes" | "on"
3665                            )
3666                        })
3667                        .unwrap_or_else(|| parse_bool_env("SOCKUDO_ENABLE_CLIENT_MESSAGES", false)),
3668                        enable_user_authentication: Some(parse_bool_env(
3669                            "SOCKUDO_DEFAULT_APP_ENABLE_USER_AUTHENTICATION",
3670                            false,
3671                        )),
3672                        enable_watchlist_events: Some(parse_bool_env(
3673                            "SOCKUDO_DEFAULT_APP_ENABLE_WATCHLIST_EVENTS",
3674                            false,
3675                        )),
3676                    },
3677                    channels: crate::app::AppChannelsPolicy {
3678                        allowed_origins: {
3679                            if let Ok(origins_str) =
3680                                std::env::var("SOCKUDO_DEFAULT_APP_ALLOWED_ORIGINS")
3681                            {
3682                                if !origins_str.is_empty() {
3683                                    Some(
3684                                        origins_str
3685                                            .split(',')
3686                                            .map(|s| s.trim().to_string())
3687                                            .collect(),
3688                                    )
3689                                } else {
3690                                    None
3691                                }
3692                            } else {
3693                                None
3694                            }
3695                        },
3696                        annotations_enabled: Some(parse_bool_env(
3697                            "SOCKUDO_DEFAULT_APP_ANNOTATIONS_ENABLED",
3698                            false,
3699                        )),
3700                        channel_delta_compression: None,
3701                        channel_namespaces: None,
3702                    },
3703                    webhooks: None,
3704                    idempotency: None,
3705                    connection_recovery: None,
3706                    history: None,
3707                    presence_history: None,
3708                },
3709            );
3710
3711            self.app_manager.array.apps.push(default_app);
3712            info!("Successfully registered default app from env");
3713        } else if default_app_env_configured && !default_app_enabled {
3714            info!("Default app registration disabled by SOCKUDO_DEFAULT_APP_ENABLED=false");
3715        } else if default_app_credentials_configured {
3716            warn!(
3717                "SOCKUDO_DEFAULT_APP_* environment was configured but id, key, and secret were not all provided; no default app registered"
3718            );
3719        }
3720
3721        // Special handling for REDIS_URL
3722        if let Ok(redis_url_env) = std::env::var("REDIS_URL") {
3723            info!("Applying REDIS_URL environment variable override");
3724
3725            let redis_url_json = sonic_rs::json!(redis_url_env);
3726
3727            self.adapter
3728                .redis
3729                .redis_pub_options
3730                .insert("url".to_string(), redis_url_json.clone());
3731            self.adapter
3732                .redis
3733                .redis_sub_options
3734                .insert("url".to_string(), redis_url_json);
3735
3736            self.cache.redis.url_override = Some(redis_url_env.clone());
3737            self.queue.redis.url_override = Some(redis_url_env.clone());
3738            self.rate_limiter.redis.url_override = Some(redis_url_env);
3739        }
3740
3741        // --- Logging Configuration ---
3742        let has_colors_env = std::env::var("LOG_COLORS_ENABLED").is_ok();
3743        let has_target_env = std::env::var("LOG_INCLUDE_TARGET").is_ok();
3744        if has_colors_env || has_target_env {
3745            let logging_config = self.logging.get_or_insert_with(Default::default);
3746            if has_colors_env {
3747                logging_config.colors_enabled =
3748                    parse_bool_env("LOG_COLORS_ENABLED", logging_config.colors_enabled);
3749            }
3750            if has_target_env {
3751                logging_config.include_target =
3752                    parse_bool_env("LOG_INCLUDE_TARGET", logging_config.include_target);
3753            }
3754        }
3755
3756        // --- Cleanup Configuration ---
3757        self.cleanup.async_enabled =
3758            parse_bool_env("CLEANUP_ASYNC_ENABLED", self.cleanup.async_enabled);
3759        self.cleanup.fallback_to_sync =
3760            parse_bool_env("CLEANUP_FALLBACK_TO_SYNC", self.cleanup.fallback_to_sync);
3761        self.cleanup.queue_buffer_size =
3762            parse_env::<usize>("CLEANUP_QUEUE_BUFFER_SIZE", self.cleanup.queue_buffer_size);
3763        self.cleanup.batch_size = parse_env::<usize>("CLEANUP_BATCH_SIZE", self.cleanup.batch_size);
3764        self.cleanup.batch_timeout_ms =
3765            parse_env::<u64>("CLEANUP_BATCH_TIMEOUT_MS", self.cleanup.batch_timeout_ms);
3766        if let Ok(worker_threads_str) = std::env::var("CLEANUP_WORKER_THREADS") {
3767            self.cleanup.worker_threads = if worker_threads_str.to_lowercase() == "auto" {
3768                WorkerThreadsConfig::Auto
3769            } else if let Ok(n) = worker_threads_str.parse::<usize>() {
3770                WorkerThreadsConfig::Fixed(n)
3771            } else {
3772                warn!(
3773                    "Invalid CLEANUP_WORKER_THREADS value '{}', keeping current setting",
3774                    worker_threads_str
3775                );
3776                self.cleanup.worker_threads.clone()
3777            };
3778        }
3779        self.cleanup.max_retry_attempts = parse_env::<u32>(
3780            "CLEANUP_MAX_RETRY_ATTEMPTS",
3781            self.cleanup.max_retry_attempts,
3782        );
3783
3784        // Cluster health configuration
3785        self.cluster_health.enabled =
3786            parse_bool_env("CLUSTER_HEALTH_ENABLED", self.cluster_health.enabled);
3787        self.cluster_health.heartbeat_interval_ms = parse_env::<u64>(
3788            "CLUSTER_HEALTH_HEARTBEAT_INTERVAL",
3789            self.cluster_health.heartbeat_interval_ms,
3790        );
3791        self.cluster_health.node_timeout_ms = parse_env::<u64>(
3792            "CLUSTER_HEALTH_NODE_TIMEOUT",
3793            self.cluster_health.node_timeout_ms,
3794        );
3795        self.cluster_health.cleanup_interval_ms = parse_env::<u64>(
3796            "CLUSTER_HEALTH_CLEANUP_INTERVAL",
3797            self.cluster_health.cleanup_interval_ms,
3798        );
3799
3800        // Health check endpoint timeout
3801        self.health_check_timeout_ms =
3802            parse_env::<u64>("HEALTH_CHECK_TIMEOUT_MS", self.health_check_timeout_ms);
3803
3804        // Tag filtering configuration
3805        self.tag_filtering.enabled =
3806            parse_bool_env("TAG_FILTERING_ENABLED", self.tag_filtering.enabled);
3807
3808        // WebSocket buffer configuration
3809        if let Ok(val) = std::env::var("WEBSOCKET_MAX_MESSAGES") {
3810            if val.to_lowercase() == "none" || val == "0" {
3811                self.websocket.max_messages = None;
3812            } else if let Ok(n) = val.parse::<usize>() {
3813                self.websocket.max_messages = Some(n);
3814            }
3815        }
3816        if let Ok(val) = std::env::var("WEBSOCKET_MAX_BYTES") {
3817            if val.to_lowercase() == "none" || val == "0" {
3818                self.websocket.max_bytes = None;
3819            } else if let Ok(n) = val.parse::<usize>() {
3820                self.websocket.max_bytes = Some(n);
3821            }
3822        }
3823        self.websocket.disconnect_on_buffer_full = parse_bool_env(
3824            "WEBSOCKET_DISCONNECT_ON_BUFFER_FULL",
3825            self.websocket.disconnect_on_buffer_full,
3826        );
3827        self.websocket.max_message_size = parse_env::<usize>(
3828            "WEBSOCKET_MAX_MESSAGE_SIZE",
3829            self.websocket.max_message_size,
3830        );
3831        self.websocket.max_frame_size =
3832            parse_env::<usize>("WEBSOCKET_MAX_FRAME_SIZE", self.websocket.max_frame_size);
3833        self.websocket.write_buffer_size = parse_env::<usize>(
3834            "WEBSOCKET_WRITE_BUFFER_SIZE",
3835            self.websocket.write_buffer_size,
3836        );
3837        self.websocket.max_backpressure = parse_env::<usize>(
3838            "WEBSOCKET_MAX_BACKPRESSURE",
3839            self.websocket.max_backpressure,
3840        );
3841        self.websocket.auto_ping = parse_bool_env("WEBSOCKET_AUTO_PING", self.websocket.auto_ping);
3842        self.websocket.ping_interval =
3843            parse_env::<u32>("WEBSOCKET_PING_INTERVAL", self.websocket.ping_interval);
3844        self.websocket.idle_timeout =
3845            parse_env::<u32>("WEBSOCKET_IDLE_TIMEOUT", self.websocket.idle_timeout);
3846        if let Ok(mode) = std::env::var("WEBSOCKET_COMPRESSION") {
3847            self.websocket.compression = mode;
3848        }
3849
3850        // Connection recovery (includes serial + message_id + replay buffer)
3851        self.connection_recovery.enabled = parse_bool_env(
3852            "CONNECTION_RECOVERY_ENABLED",
3853            self.connection_recovery.enabled,
3854        );
3855        self.connection_recovery.buffer_ttl_seconds = parse_env::<u64>(
3856            "CONNECTION_RECOVERY_BUFFER_TTL",
3857            self.connection_recovery.buffer_ttl_seconds,
3858        );
3859        self.connection_recovery.max_buffer_size = parse_env::<usize>(
3860            "CONNECTION_RECOVERY_MAX_BUFFER_SIZE",
3861            self.connection_recovery.max_buffer_size,
3862        );
3863
3864        self.history.enabled = parse_bool_env("HISTORY_ENABLED", self.history.enabled);
3865        self.history.rewind_enabled =
3866            parse_bool_env("HISTORY_REWIND_ENABLED", self.history.rewind_enabled);
3867        self.history.retention_window_seconds = parse_env::<u64>(
3868            "HISTORY_RETENTION_WINDOW_SECONDS",
3869            self.history.retention_window_seconds,
3870        );
3871        self.history.max_page_size =
3872            parse_env::<usize>("HISTORY_MAX_PAGE_SIZE", self.history.max_page_size);
3873        self.history.writer_shards =
3874            parse_env::<usize>("HISTORY_WRITER_SHARDS", self.history.writer_shards);
3875        self.history.writer_queue_capacity = parse_env::<usize>(
3876            "HISTORY_WRITER_QUEUE_CAPACITY",
3877            self.history.writer_queue_capacity,
3878        );
3879        if let Ok(backend) = std::env::var("HISTORY_BACKEND") {
3880            self.history.backend = HistoryBackend::from_str(&backend)?;
3881        }
3882        if let Ok(max_messages) = std::env::var("HISTORY_MAX_MESSAGES_PER_CHANNEL") {
3883            self.history.max_messages_per_channel = Some(
3884                max_messages
3885                    .parse::<usize>()
3886                    .map_err(|e| format!("Invalid HISTORY_MAX_MESSAGES_PER_CHANNEL: {e}"))?,
3887            );
3888        }
3889        if let Ok(max_bytes) = std::env::var("HISTORY_MAX_BYTES_PER_CHANNEL") {
3890            self.history.max_bytes_per_channel = Some(
3891                max_bytes
3892                    .parse::<u64>()
3893                    .map_err(|e| format!("Invalid HISTORY_MAX_BYTES_PER_CHANNEL: {e}"))?,
3894            );
3895        }
3896        if let Ok(table_prefix) = std::env::var("HISTORY_POSTGRES_TABLE_PREFIX") {
3897            self.history.postgres.table_prefix = table_prefix;
3898        }
3899        self.history.postgres.write_timeout_ms = parse_env::<u64>(
3900            "HISTORY_POSTGRES_WRITE_TIMEOUT_MS",
3901            self.history.postgres.write_timeout_ms,
3902        );
3903        self.history.purge_interval_seconds = parse_env::<u64>(
3904            "HISTORY_PURGE_INTERVAL_SECONDS",
3905            self.history.purge_interval_seconds,
3906        );
3907        self.history.purge_batch_size =
3908            parse_env::<usize>("HISTORY_PURGE_BATCH_SIZE", self.history.purge_batch_size);
3909        self.history.max_purge_per_tick = parse_env::<usize>(
3910            "HISTORY_MAX_PURGE_PER_TICK",
3911            self.history.max_purge_per_tick,
3912        );
3913
3914        self.presence_history.enabled =
3915            parse_bool_env("PRESENCE_HISTORY_ENABLED", self.presence_history.enabled);
3916        self.presence_history.retention_window_seconds = parse_env::<u64>(
3917            "PRESENCE_HISTORY_RETENTION_WINDOW_SECONDS",
3918            self.presence_history.retention_window_seconds,
3919        );
3920        self.presence_history.max_page_size = parse_env::<usize>(
3921            "PRESENCE_HISTORY_MAX_PAGE_SIZE",
3922            self.presence_history.max_page_size,
3923        );
3924        if let Ok(max_events) = std::env::var("PRESENCE_HISTORY_MAX_EVENTS_PER_CHANNEL") {
3925            self.presence_history.max_events_per_channel =
3926                Some(max_events.parse::<usize>().map_err(|e| {
3927                    format!("Invalid PRESENCE_HISTORY_MAX_EVENTS_PER_CHANNEL: {e}")
3928                })?);
3929        }
3930        if let Ok(max_bytes) = std::env::var("PRESENCE_HISTORY_MAX_BYTES_PER_CHANNEL") {
3931            self.presence_history.max_bytes_per_channel = Some(
3932                max_bytes
3933                    .parse::<u64>()
3934                    .map_err(|e| format!("Invalid PRESENCE_HISTORY_MAX_BYTES_PER_CHANNEL: {e}"))?,
3935            );
3936        }
3937        self.idempotency.enabled = parse_bool_env("IDEMPOTENCY_ENABLED", self.idempotency.enabled);
3938        self.idempotency.ttl_seconds =
3939            parse_env::<u64>("IDEMPOTENCY_TTL_SECONDS", self.idempotency.ttl_seconds);
3940        self.idempotency.max_key_length = parse_env::<usize>(
3941            "IDEMPOTENCY_MAX_KEY_LENGTH",
3942            self.idempotency.max_key_length,
3943        );
3944
3945        self.ephemeral.enabled = parse_bool_env("EPHEMERAL_ENABLED", self.ephemeral.enabled);
3946
3947        self.echo_control.enabled =
3948            parse_bool_env("ECHO_CONTROL_ENABLED", self.echo_control.enabled);
3949        self.echo_control.default_echo_messages = parse_bool_env(
3950            "ECHO_CONTROL_DEFAULT_ECHO_MESSAGES",
3951            self.echo_control.default_echo_messages,
3952        );
3953
3954        self.event_name_filtering.enabled = parse_bool_env(
3955            "EVENT_NAME_FILTERING_ENABLED",
3956            self.event_name_filtering.enabled,
3957        );
3958        self.event_name_filtering.max_events_per_filter = parse_env::<usize>(
3959            "EVENT_NAME_FILTERING_MAX_EVENTS_PER_FILTER",
3960            self.event_name_filtering.max_events_per_filter,
3961        );
3962        self.event_name_filtering.max_event_name_length = parse_env::<usize>(
3963            "EVENT_NAME_FILTERING_MAX_EVENT_NAME_LENGTH",
3964            self.event_name_filtering.max_event_name_length,
3965        );
3966        self.versioned_messages.enabled = parse_bool_env(
3967            "VERSIONED_MESSAGES_ENABLED",
3968            self.versioned_messages.enabled,
3969        );
3970        if let Ok(driver_str) = std::env::var("VERSIONED_MESSAGES_DRIVER") {
3971            self.versioned_messages.driver = parse_driver_enum(
3972                driver_str,
3973                self.versioned_messages.driver.clone(),
3974                "VersionedMessages Backend",
3975            );
3976        }
3977        if let Ok(driver_str) = std::env::var("PUSH_STORAGE_DRIVER") {
3978            self.push.storage_driver = parse_driver_enum(
3979                driver_str,
3980                self.push.storage_driver.clone(),
3981                "Push Storage Backend",
3982            );
3983        }
3984        if let Ok(driver_str) = std::env::var("PUSH_QUEUE_DRIVER") {
3985            self.push.queue_driver = parse_driver_enum(
3986                driver_str,
3987                self.push.queue_driver.clone(),
3988                "Push Queue Backend",
3989            );
3990        }
3991        self.push.fcm_enabled = parse_bool_env("PUSH_FCM_ENABLED", self.push.fcm_enabled);
3992        self.push.apns_enabled = parse_bool_env("PUSH_APNS_ENABLED", self.push.apns_enabled);
3993        self.push.webpush_enabled =
3994            parse_bool_env("PUSH_WEBPUSH_ENABLED", self.push.webpush_enabled);
3995        self.push.hms_enabled = parse_bool_env("PUSH_HMS_ENABLED", self.push.hms_enabled);
3996        self.push.wns_enabled = parse_bool_env("PUSH_WNS_ENABLED", self.push.wns_enabled);
3997        if let Some(key) = parse_env_optional::<String>("PUSH_CREDENTIAL_ENCRYPTION_KEY") {
3998            self.push.credential_encryption_key = Some(key);
3999        }
4000        self.push.fanout_fast_threshold = parse_env::<u64>(
4001            "PUSH_FANOUT_FAST_THRESHOLD",
4002            self.push.fanout_fast_threshold,
4003        );
4004        self.push.fanout_shard_size =
4005            parse_env::<u64>("PUSH_FANOUT_SHARD_SIZE", self.push.fanout_shard_size);
4006        self.push.fanout_sync_threshold = parse_env::<u64>(
4007            "PUSH_FANOUT_SYNC_THRESHOLD",
4008            self.push.fanout_sync_threshold,
4009        );
4010        self.push.backpressure_lag_threshold_secs = parse_env::<u64>(
4011            "PUSH_BACKPRESSURE_LAG_THRESHOLD_SECS",
4012            self.push.backpressure_lag_threshold_secs,
4013        );
4014        self.push.publish_status_ttl_days = parse_env::<u64>(
4015            "PUSH_PUBLISH_STATUS_TTL_DAYS",
4016            self.push.publish_status_ttl_days,
4017        );
4018        self.push.circuit_breaker.failure_threshold = parse_env::<u32>(
4019            "PUSH_FAILURE_THRESHOLD",
4020            self.push.circuit_breaker.failure_threshold,
4021        );
4022        self.push.scheduler_interval_secs = parse_env::<u64>(
4023            "PUSH_SCHEDULER_INTERVAL_SECS",
4024            self.push.scheduler_interval_secs,
4025        );
4026        self.push.stale_device_max_age_days = parse_env::<u64>(
4027            "PUSH_STALE_DEVICE_MAX_AGE_DAYS",
4028            self.push.stale_device_max_age_days,
4029        );
4030        self.push.analytics_enabled =
4031            parse_bool_env("PUSH_ANALYTICS_ENABLED", self.push.analytics_enabled);
4032        self.push.default_quotas.acceptance_rps = parse_env::<u64>(
4033            "PUSH_DEFAULT_ACCEPTANCE_RPS",
4034            self.push.default_quotas.acceptance_rps,
4035        );
4036        self.push.default_quotas.delivery_quota_daily = parse_env::<u64>(
4037            "PUSH_DEFAULT_DELIVERY_QUOTA_DAILY",
4038            self.push.default_quotas.delivery_quota_daily,
4039        );
4040        self.push.default_quotas.fanout_max = parse_env::<u64>(
4041            "PUSH_DEFAULT_FANOUT_MAX",
4042            self.push.default_quotas.fanout_max,
4043        );
4044        self.push.default_quotas.inflight_max = parse_env::<u64>(
4045            "PUSH_DEFAULT_INFLIGHT_MAX",
4046            self.push.default_quotas.inflight_max,
4047        );
4048        self.versioned_messages.max_page_size = parse_env::<usize>(
4049            "VERSIONED_MESSAGES_MAX_PAGE_SIZE",
4050            self.versioned_messages.max_page_size,
4051        );
4052        self.versioned_messages.retention_window_seconds = parse_env::<u64>(
4053            "VERSIONED_MESSAGES_RETENTION_WINDOW_SECONDS",
4054            self.versioned_messages.retention_window_seconds,
4055        );
4056        self.versioned_messages.purge_interval_seconds = parse_env::<u64>(
4057            "VERSIONED_MESSAGES_PURGE_INTERVAL_SECONDS",
4058            self.versioned_messages.purge_interval_seconds,
4059        );
4060        self.versioned_messages.purge_batch_size = parse_env::<usize>(
4061            "VERSIONED_MESSAGES_PURGE_BATCH_SIZE",
4062            self.versioned_messages.purge_batch_size,
4063        );
4064        self.versioned_messages.max_purge_per_tick = parse_env::<usize>(
4065            "VERSIONED_MESSAGES_MAX_PURGE_PER_TICK",
4066            self.versioned_messages.max_purge_per_tick,
4067        );
4068        self.annotations.enabled = parse_bool_env("ANNOTATIONS_ENABLED", self.annotations.enabled);
4069
4070        Ok(())
4071    }
4072
4073    pub fn validate(&self) -> Result<(), String> {
4074        if self.unix_socket.enabled {
4075            if self.unix_socket.path.is_empty() {
4076                return Err(
4077                    "Unix socket path cannot be empty when Unix socket is enabled".to_string(),
4078                );
4079            }
4080
4081            self.validate_unix_socket_security()?;
4082
4083            if self.ssl.enabled {
4084                tracing::warn!(
4085                    "Both Unix socket and SSL are enabled. This is unusual as Unix sockets are typically used behind reverse proxies that handle SSL termination."
4086                );
4087            }
4088
4089            if self.unix_socket.permission_mode > 0o777 {
4090                return Err(format!(
4091                    "Unix socket permission_mode ({:o}) is invalid. Must be a valid octal mode (0o000 to 0o777)",
4092                    self.unix_socket.permission_mode
4093                ));
4094            }
4095        }
4096
4097        if let Err(e) = self.cleanup.validate() {
4098            return Err(format!("Invalid cleanup configuration: {}", e));
4099        }
4100
4101        if self.history.enabled {
4102            if self.history.max_page_size == 0 {
4103                return Err("history.max_page_size must be greater than 0".to_string());
4104            }
4105            if self.history.writer_shards == 0 {
4106                return Err("history.writer_shards must be greater than 0".to_string());
4107            }
4108            if self.history.writer_queue_capacity == 0 {
4109                return Err("history.writer_queue_capacity must be greater than 0".to_string());
4110            }
4111            if self.history.retention_window_seconds == 0 {
4112                return Err("history.retention_window_seconds must be greater than 0".to_string());
4113            }
4114            if self.history.postgres.table_prefix.trim().is_empty() {
4115                return Err("history.postgres.table_prefix must not be empty".to_string());
4116            }
4117        }
4118
4119        if self.presence_history.enabled {
4120            if self.presence_history.max_page_size == 0 {
4121                return Err("presence_history.max_page_size must be greater than 0".to_string());
4122            }
4123            if self.presence_history.retention_window_seconds == 0 {
4124                return Err(
4125                    "presence_history.retention_window_seconds must be greater than 0".to_string(),
4126                );
4127            }
4128        }
4129
4130        if self.versioned_messages.enabled && self.versioned_messages.max_page_size == 0 {
4131            return Err("versioned_messages.max_page_size must be greater than 0".to_string());
4132        }
4133        if self.annotations.enabled && !self.versioned_messages.enabled {
4134            return Err("annotations require versioned_messages.enabled".to_string());
4135        }
4136
4137        if self.adapter.nats.presence_sync_chunk_size == Some(0) {
4138            return Err("nats.presence_sync_chunk_size must be > 0 when set".to_string());
4139        }
4140
4141        Ok(())
4142    }
4143
4144    fn validate_unix_socket_security(&self) -> Result<(), String> {
4145        let path = &self.unix_socket.path;
4146
4147        if path.contains("../") || path.contains("..\\") {
4148            return Err(
4149                "Unix socket path contains directory traversal sequences (../). This is not allowed for security reasons.".to_string()
4150            );
4151        }
4152
4153        if self.unix_socket.permission_mode & 0o002 != 0 {
4154            warn!(
4155                "Unix socket permission mode ({:o}) allows world write access. This may be a security risk. Consider using more restrictive permissions like 0o660 or 0o750.",
4156                self.unix_socket.permission_mode
4157            );
4158        }
4159
4160        if self.unix_socket.permission_mode & 0o007 > 0o005 {
4161            warn!(
4162                "Unix socket permission mode ({:o}) grants write permissions to others. Consider using more restrictive permissions.",
4163                self.unix_socket.permission_mode
4164            );
4165        }
4166
4167        if self.mode == "production" && path.starts_with("/tmp/") {
4168            warn!(
4169                "Unix socket path '{}' is in /tmp directory. In production, consider using a more permanent location like /var/run/sockudo/ for better security and persistence.",
4170                path
4171            );
4172        }
4173
4174        if !path.starts_with('/') {
4175            return Err(
4176                "Unix socket path must be absolute (start with /) for security and reliability."
4177                    .to_string(),
4178            );
4179        }
4180
4181        Ok(())
4182    }
4183}
4184
4185#[cfg(test)]
4186mod redis_connection_tests {
4187    use super::{ClusterNode, RedisClusterConnection, RedisConnection, RedisSentinel};
4188
4189    #[test]
4190    fn test_standard_url_basic() {
4191        let conn = RedisConnection {
4192            host: "127.0.0.1".to_string(),
4193            port: 6379,
4194            db: 0,
4195            username: None,
4196            password: None,
4197            key_prefix: "sockudo:".to_string(),
4198            sentinels: Vec::new(),
4199            sentinel_password: None,
4200            name: "mymaster".to_string(),
4201            cluster: RedisClusterConnection::default(),
4202            cluster_nodes: Vec::new(),
4203        };
4204        assert_eq!(conn.to_url(), "redis://127.0.0.1:6379/0");
4205    }
4206
4207    #[test]
4208    fn test_standard_url_with_password() {
4209        let conn = RedisConnection {
4210            host: "127.0.0.1".to_string(),
4211            port: 6379,
4212            db: 2,
4213            username: None,
4214            password: Some("secret".to_string()),
4215            key_prefix: "sockudo:".to_string(),
4216            sentinels: Vec::new(),
4217            sentinel_password: None,
4218            name: "mymaster".to_string(),
4219            cluster: RedisClusterConnection::default(),
4220            cluster_nodes: Vec::new(),
4221        };
4222        assert_eq!(conn.to_url(), "redis://:secret@127.0.0.1:6379/2");
4223    }
4224
4225    #[test]
4226    fn test_standard_url_with_username_and_password() {
4227        let conn = RedisConnection {
4228            host: "redis.example.com".to_string(),
4229            port: 6380,
4230            db: 1,
4231            username: Some("admin".to_string()),
4232            password: Some("pass123".to_string()),
4233            key_prefix: "sockudo:".to_string(),
4234            sentinels: Vec::new(),
4235            sentinel_password: None,
4236            name: "mymaster".to_string(),
4237            cluster: RedisClusterConnection::default(),
4238            cluster_nodes: Vec::new(),
4239        };
4240        assert_eq!(
4241            conn.to_url(),
4242            "redis://admin:pass123@redis.example.com:6380/1"
4243        );
4244    }
4245
4246    #[test]
4247    fn test_standard_url_with_special_chars_in_password() {
4248        let conn = RedisConnection {
4249            host: "127.0.0.1".to_string(),
4250            port: 6379,
4251            db: 0,
4252            username: None,
4253            password: Some("pass@word#123".to_string()),
4254            key_prefix: "sockudo:".to_string(),
4255            sentinels: Vec::new(),
4256            sentinel_password: None,
4257            name: "mymaster".to_string(),
4258            cluster: RedisClusterConnection::default(),
4259            cluster_nodes: Vec::new(),
4260        };
4261        assert_eq!(conn.to_url(), "redis://:pass%40word%23123@127.0.0.1:6379/0");
4262    }
4263
4264    #[test]
4265    fn test_is_sentinel_configured_false() {
4266        let conn = RedisConnection::default();
4267        assert!(!conn.is_sentinel_configured());
4268    }
4269
4270    #[test]
4271    fn test_is_sentinel_configured_true() {
4272        let conn = RedisConnection {
4273            sentinels: vec![RedisSentinel {
4274                host: "sentinel1".to_string(),
4275                port: 26379,
4276            }],
4277            ..Default::default()
4278        };
4279        assert!(conn.is_sentinel_configured());
4280    }
4281
4282    #[test]
4283    fn test_sentinel_url_basic() {
4284        let conn = RedisConnection {
4285            host: "127.0.0.1".to_string(),
4286            port: 6379,
4287            db: 0,
4288            username: None,
4289            password: None,
4290            key_prefix: "sockudo:".to_string(),
4291            sentinels: vec![
4292                RedisSentinel {
4293                    host: "sentinel1".to_string(),
4294                    port: 26379,
4295                },
4296                RedisSentinel {
4297                    host: "sentinel2".to_string(),
4298                    port: 26379,
4299                },
4300            ],
4301            sentinel_password: None,
4302            name: "mymaster".to_string(),
4303            cluster: RedisClusterConnection::default(),
4304            cluster_nodes: Vec::new(),
4305        };
4306        assert_eq!(
4307            conn.to_url(),
4308            "redis+sentinel://sentinel1:26379,sentinel2:26379/mymaster/0"
4309        );
4310    }
4311
4312    #[test]
4313    fn test_sentinel_url_with_sentinel_password() {
4314        let conn = RedisConnection {
4315            host: "127.0.0.1".to_string(),
4316            port: 6379,
4317            db: 0,
4318            username: None,
4319            password: None,
4320            key_prefix: "sockudo:".to_string(),
4321            sentinels: vec![RedisSentinel {
4322                host: "sentinel1".to_string(),
4323                port: 26379,
4324            }],
4325            sentinel_password: Some("sentinelpass".to_string()),
4326            name: "mymaster".to_string(),
4327            cluster: RedisClusterConnection::default(),
4328            cluster_nodes: Vec::new(),
4329        };
4330        assert_eq!(
4331            conn.to_url(),
4332            "redis+sentinel://:sentinelpass@sentinel1:26379/mymaster/0"
4333        );
4334    }
4335
4336    #[test]
4337    fn test_sentinel_url_with_master_password() {
4338        let conn = RedisConnection {
4339            host: "127.0.0.1".to_string(),
4340            port: 6379,
4341            db: 1,
4342            username: None,
4343            password: Some("masterpass".to_string()),
4344            key_prefix: "sockudo:".to_string(),
4345            sentinels: vec![RedisSentinel {
4346                host: "sentinel1".to_string(),
4347                port: 26379,
4348            }],
4349            sentinel_password: None,
4350            name: "mymaster".to_string(),
4351            cluster: RedisClusterConnection::default(),
4352            cluster_nodes: Vec::new(),
4353        };
4354        assert_eq!(
4355            conn.to_url(),
4356            "redis+sentinel://sentinel1:26379/mymaster/1?password=masterpass"
4357        );
4358    }
4359
4360    #[test]
4361    fn test_sentinel_url_with_all_auth() {
4362        let conn = RedisConnection {
4363            host: "127.0.0.1".to_string(),
4364            port: 6379,
4365            db: 2,
4366            username: Some("redisuser".to_string()),
4367            password: Some("redispass".to_string()),
4368            key_prefix: "sockudo:".to_string(),
4369            sentinels: vec![
4370                RedisSentinel {
4371                    host: "sentinel1".to_string(),
4372                    port: 26379,
4373                },
4374                RedisSentinel {
4375                    host: "sentinel2".to_string(),
4376                    port: 26380,
4377                },
4378            ],
4379            sentinel_password: Some("sentinelauth".to_string()),
4380            name: "production-master".to_string(),
4381            cluster: RedisClusterConnection::default(),
4382            cluster_nodes: Vec::new(),
4383        };
4384        assert_eq!(
4385            conn.to_url(),
4386            "redis+sentinel://:sentinelauth@sentinel1:26379,sentinel2:26380/production-master/2?password=redispass&username=redisuser"
4387        );
4388    }
4389
4390    #[test]
4391    fn test_sentinel_to_host_port() {
4392        let sentinel = RedisSentinel {
4393            host: "sentinel.example.com".to_string(),
4394            port: 26379,
4395        };
4396        assert_eq!(sentinel.to_host_port(), "sentinel.example.com:26379");
4397    }
4398
4399    #[test]
4400    fn test_cluster_node_urls_with_shared_cluster_auth_and_tls() {
4401        let conn = RedisConnection {
4402            cluster: RedisClusterConnection {
4403                nodes: vec![
4404                    ClusterNode {
4405                        host: "node1.secure-cluster.com".to_string(),
4406                        port: 7000,
4407                    },
4408                    ClusterNode {
4409                        host: "redis://node2.secure-cluster.com:7001".to_string(),
4410                        port: 7001,
4411                    },
4412                    ClusterNode {
4413                        host: "rediss://node3.secure-cluster.com".to_string(),
4414                        port: 7002,
4415                    },
4416                ],
4417                username: None,
4418                password: Some("cluster-secret".to_string()),
4419                use_tls: true,
4420            },
4421            ..Default::default()
4422        };
4423
4424        assert_eq!(
4425            conn.cluster_node_urls(),
4426            vec![
4427                "rediss://:cluster-secret@node1.secure-cluster.com:7000",
4428                "redis://:cluster-secret@node2.secure-cluster.com:7001",
4429                "rediss://:cluster-secret@node3.secure-cluster.com:7002",
4430            ]
4431        );
4432    }
4433
4434    #[test]
4435    fn test_cluster_node_urls_fallback_to_legacy_nodes() {
4436        let conn = RedisConnection {
4437            password: Some("fallback-secret".to_string()),
4438            cluster_nodes: vec![ClusterNode {
4439                host: "legacy-node.example.com".to_string(),
4440                port: 7000,
4441            }],
4442            ..Default::default()
4443        };
4444
4445        assert_eq!(
4446            conn.cluster_node_urls(),
4447            vec!["redis://:fallback-secret@legacy-node.example.com:7000"]
4448        );
4449    }
4450
4451    #[test]
4452    fn test_normalize_cluster_seed_urls() {
4453        let conn = RedisConnection {
4454            cluster: RedisClusterConnection {
4455                nodes: Vec::new(),
4456                username: Some("svc-user".to_string()),
4457                password: Some("svc-pass".to_string()),
4458                use_tls: true,
4459            },
4460            ..Default::default()
4461        };
4462
4463        let seeds = vec![
4464            "node1.example.com:7000".to_string(),
4465            "redis://node2.example.com:7001".to_string(),
4466            "rediss://node3.example.com".to_string(),
4467        ];
4468
4469        assert_eq!(
4470            conn.normalize_cluster_seed_urls(&seeds),
4471            vec![
4472                "rediss://svc-user:svc-pass@node1.example.com:7000",
4473                "redis://svc-user:svc-pass@node2.example.com:7001",
4474                "rediss://svc-user:svc-pass@node3.example.com:6379",
4475            ]
4476        );
4477    }
4478}
4479
4480#[cfg(test)]
4481mod cluster_node_tests {
4482    use super::ClusterNode;
4483
4484    #[test]
4485    fn test_to_url_basic_host() {
4486        let node = ClusterNode {
4487            host: "localhost".to_string(),
4488            port: 6379,
4489        };
4490        assert_eq!(node.to_url(), "redis://localhost:6379");
4491    }
4492
4493    #[test]
4494    fn test_to_url_ip_address() {
4495        let node = ClusterNode {
4496            host: "127.0.0.1".to_string(),
4497            port: 6379,
4498        };
4499        assert_eq!(node.to_url(), "redis://127.0.0.1:6379");
4500    }
4501
4502    #[test]
4503    fn test_to_url_with_redis_protocol() {
4504        let node = ClusterNode {
4505            host: "redis://example.com".to_string(),
4506            port: 6379,
4507        };
4508        assert_eq!(node.to_url(), "redis://example.com:6379");
4509    }
4510
4511    #[test]
4512    fn test_to_url_with_rediss_protocol() {
4513        let node = ClusterNode {
4514            host: "rediss://secure.example.com".to_string(),
4515            port: 6379,
4516        };
4517        assert_eq!(node.to_url(), "rediss://secure.example.com:6379");
4518    }
4519
4520    #[test]
4521    fn test_to_url_with_rediss_protocol_and_port_in_url() {
4522        let node = ClusterNode {
4523            host: "rediss://secure.example.com:7000".to_string(),
4524            port: 6379,
4525        };
4526        assert_eq!(node.to_url(), "rediss://secure.example.com:7000");
4527    }
4528
4529    #[test]
4530    fn test_to_url_with_redis_protocol_and_port_in_url() {
4531        let node = ClusterNode {
4532            host: "redis://example.com:7001".to_string(),
4533            port: 6379,
4534        };
4535        assert_eq!(node.to_url(), "redis://example.com:7001");
4536    }
4537
4538    #[test]
4539    fn test_to_url_with_trailing_whitespace() {
4540        let node = ClusterNode {
4541            host: "  rediss://secure.example.com  ".to_string(),
4542            port: 6379,
4543        };
4544        assert_eq!(node.to_url(), "rediss://secure.example.com:6379");
4545    }
4546
4547    #[test]
4548    fn test_to_url_custom_port() {
4549        let node = ClusterNode {
4550            host: "redis-cluster.example.com".to_string(),
4551            port: 7000,
4552        };
4553        assert_eq!(node.to_url(), "redis://redis-cluster.example.com:7000");
4554    }
4555
4556    #[test]
4557    fn test_to_url_plain_host_with_port_in_host_field() {
4558        let node = ClusterNode {
4559            host: "redis-cluster.example.com:7010".to_string(),
4560            port: 7000,
4561        };
4562        assert_eq!(node.to_url(), "redis://redis-cluster.example.com:7010");
4563    }
4564
4565    #[test]
4566    fn test_to_url_with_options_adds_auth_and_tls() {
4567        let node = ClusterNode {
4568            host: "node.example.com".to_string(),
4569            port: 7000,
4570        };
4571        assert_eq!(
4572            node.to_url_with_options(true, Some("svc-user"), Some("secret")),
4573            "rediss://svc-user:secret@node.example.com:7000"
4574        );
4575    }
4576
4577    #[test]
4578    fn test_to_url_with_options_keeps_embedded_auth() {
4579        let node = ClusterNode {
4580            host: "rediss://:node-secret@node.example.com:7000".to_string(),
4581            port: 7000,
4582        };
4583        assert_eq!(
4584            node.to_url_with_options(true, Some("global-user"), Some("global-secret")),
4585            "rediss://:node-secret@node.example.com:7000"
4586        );
4587    }
4588
4589    #[test]
4590    fn test_from_seed_parses_plain_host_port() {
4591        let node = ClusterNode::from_seed("cluster-node-1:7005").expect("node should parse");
4592        assert_eq!(node.host, "cluster-node-1");
4593        assert_eq!(node.port, 7005);
4594    }
4595
4596    #[test]
4597    fn test_from_seed_keeps_scheme_urls() {
4598        let node =
4599            ClusterNode::from_seed("rediss://secure.example.com:7005").expect("node should parse");
4600        assert_eq!(node.host, "rediss://secure.example.com:7005");
4601        assert_eq!(node.port, 7005);
4602    }
4603
4604    #[test]
4605    fn test_to_url_aws_elasticache_hostname() {
4606        let node = ClusterNode {
4607            host: "rediss://my-cluster.use1.cache.amazonaws.com".to_string(),
4608            port: 6379,
4609        };
4610        assert_eq!(
4611            node.to_url(),
4612            "rediss://my-cluster.use1.cache.amazonaws.com:6379"
4613        );
4614    }
4615
4616    #[test]
4617    fn test_to_url_with_ipv6_no_port() {
4618        let node = ClusterNode {
4619            host: "rediss://[::1]".to_string(),
4620            port: 6379,
4621        };
4622        assert_eq!(node.to_url(), "rediss://[::1]:6379");
4623    }
4624
4625    #[test]
4626    fn test_to_url_with_ipv6_and_port_in_url() {
4627        let node = ClusterNode {
4628            host: "rediss://[::1]:7000".to_string(),
4629            port: 6379,
4630        };
4631        assert_eq!(node.to_url(), "rediss://[::1]:7000");
4632    }
4633
4634    #[test]
4635    fn test_to_url_with_ipv6_full_address_no_port() {
4636        let node = ClusterNode {
4637            host: "rediss://[2001:db8::1]".to_string(),
4638            port: 6379,
4639        };
4640        assert_eq!(node.to_url(), "rediss://[2001:db8::1]:6379");
4641    }
4642
4643    #[test]
4644    fn test_to_url_with_ipv6_full_address_with_port() {
4645        let node = ClusterNode {
4646            host: "rediss://[2001:db8::1]:7000".to_string(),
4647            port: 6379,
4648        };
4649        assert_eq!(node.to_url(), "rediss://[2001:db8::1]:7000");
4650    }
4651
4652    #[test]
4653    fn test_to_url_with_redis_protocol_ipv6() {
4654        let node = ClusterNode {
4655            host: "redis://[::1]".to_string(),
4656            port: 6379,
4657        };
4658        assert_eq!(node.to_url(), "redis://[::1]:6379");
4659    }
4660}
4661
4662#[cfg(test)]
4663mod cors_config_tests {
4664    use super::CorsConfig;
4665
4666    fn cors_from_json(json: &str) -> sonic_rs::Result<CorsConfig> {
4667        sonic_rs::from_str(json)
4668    }
4669
4670    #[test]
4671    fn test_deserialize_valid_exact_origins() {
4672        let config =
4673            cors_from_json(r#"{"origin": ["https://example.com", "https://other.com"]}"#).unwrap();
4674        assert_eq!(config.origin.len(), 2);
4675    }
4676
4677    #[test]
4678    fn test_deserialize_valid_wildcard_patterns() {
4679        let config =
4680            cors_from_json(r#"{"origin": ["*.example.com", "https://*.staging.example.com"]}"#)
4681                .unwrap();
4682        assert_eq!(config.origin.len(), 2);
4683    }
4684
4685    #[test]
4686    fn test_deserialize_allows_special_markers() {
4687        assert!(cors_from_json(r#"{"origin": ["*"]}"#).is_ok());
4688        assert!(cors_from_json(r#"{"origin": ["Any"]}"#).is_ok());
4689        assert!(cors_from_json(r#"{"origin": ["any"]}"#).is_ok());
4690        assert!(cors_from_json(r#"{"origin": ["*", "https://example.com"]}"#).is_ok());
4691    }
4692
4693    #[test]
4694    fn test_deserialize_rejects_invalid_patterns() {
4695        assert!(cors_from_json(r#"{"origin": ["*.*bad"]}"#).is_err());
4696        assert!(cors_from_json(r#"{"origin": ["*."]}"#).is_err());
4697        assert!(cors_from_json(r#"{"origin": [""]}"#).is_err());
4698        assert!(cors_from_json(r#"{"origin": ["https://"]}"#).is_err());
4699    }
4700
4701    #[test]
4702    fn test_deserialize_rejects_mixed_valid_and_invalid() {
4703        assert!(cors_from_json(r#"{"origin": ["https://good.com", "*.*bad"]}"#).is_err());
4704    }
4705}