xui-rs 2.0.0

A modern, typed async Rust SDK for the 3x-ui panel API
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130

use serde::{Deserialize, Serialize};

/// Strategy used by a generated JSON-subscription balancer.
#[derive(Clone, Copy, Debug, Default, Deserialize, Eq, Hash, PartialEq, Serialize)]
#[serde(rename_all = "camelCase")]
#[non_exhaustive]
pub enum SubscriptionBalancerStrategy {
    /// Chooses the least-loaded eligible outbound.
    LeastLoad,
    /// Chooses the lowest-latency eligible outbound.
    LeastPing,
    /// Chooses an eligible outbound randomly.
    #[default]
    Random,
    /// Cycles through eligible outbounds.
    RoundRobin,
    /// A newer strategy not understood by this SDK version.
    #[serde(other)]
    Unknown,
}

impl SubscriptionBalancerStrategy {
    pub(crate) const fn as_str(self) -> Option<&'static str> {
        match self {
            Self::LeastLoad => Some("leastLoad"),
            Self::LeastPing => Some("leastPing"),
            Self::Random => Some("random"),
            Self::RoundRobin => Some("roundRobin"),
            Self::Unknown => None,
        }
    }
}

/// Persisted subscription balancer returned by 3x-ui.
#[derive(Clone, Debug, Default, Deserialize, PartialEq, Serialize)]
#[serde(default, rename_all = "camelCase")]
pub struct SubscriptionBalancer {
    /// Database identifier.
    pub id: i64,
    /// User-visible name.
    pub remark: String,
    /// Selection strategy.
    pub strategy: SubscriptionBalancerStrategy,
    /// Inbounds whose generated outbounds participate.
    pub inbound_ids: Vec<i64>,
    /// Relative least-load weights keyed by inbound identifier.
    #[serde(default, deserialize_with = "deserialize_weights")]
    pub member_weights: std::collections::BTreeMap<i64, f64>,
    /// One-based order in generated subscription documents.
    pub sort_order: i32,
    /// Whether the balancer is emitted.
    pub enabled: bool,
    /// Creation timestamp in Unix milliseconds.
    pub created_at: i64,
    /// Update timestamp in Unix milliseconds.
    pub updated_at: i64,
}

/// Writable subscription-balancer fields.
#[derive(Clone, Debug, PartialEq)]
pub struct SubscriptionBalancerInput {
    /// User-visible name.
    pub remark: String,
    /// Selection strategy.
    pub strategy: SubscriptionBalancerStrategy,
    /// Inbounds whose generated outbounds participate.
    pub inbound_ids: Vec<i64>,
    /// Relative least-load weights keyed by inbound identifier.
    pub member_weights: std::collections::BTreeMap<i64, f64>,
    /// One-based order in generated subscription documents.
    pub sort_order: i32,
    /// Explicit enabled state. `None` uses the create default or preserves it on update.
    pub enabled: Option<bool>,
}

impl SubscriptionBalancerInput {
    /// Creates an enabled random balancer at sort position one.
    pub fn new(remark: impl Into<String>, inbound_ids: Vec<i64>) -> Self {
        Self {
            remark: remark.into(),
            strategy: SubscriptionBalancerStrategy::Random,
            inbound_ids,
            member_weights: std::collections::BTreeMap::new(),
            sort_order: 1,
            enabled: Some(true),
        }
    }
}

/// Decodes numeric member weights, treating an explicit JSON null as an empty map.
/// Malformed maps still return the deserializer error.
fn deserialize_weights<'de, D: serde::Deserializer<'de>>(
    deserializer: D,
) -> Result<std::collections::BTreeMap<i64, f64>, D::Error> {
    Option::<std::collections::BTreeMap<i64, f64>>::deserialize(deserializer)
        .map(Option::unwrap_or_default)
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn strategies_and_constructor_match_wire_defaults() {
        assert_eq!(
            SubscriptionBalancerStrategy::LeastLoad.as_str(),
            Some("leastLoad")
        );
        assert_eq!(
            SubscriptionBalancerStrategy::LeastPing.as_str(),
            Some("leastPing")
        );
        assert_eq!(
            SubscriptionBalancerStrategy::Random.as_str(),
            Some("random")
        );
        assert_eq!(
            SubscriptionBalancerStrategy::RoundRobin.as_str(),
            Some("roundRobin")
        );
        assert_eq!(SubscriptionBalancerStrategy::Unknown.as_str(), None);

        let input = SubscriptionBalancerInput::new("EU pool", vec![7, 9]);
        assert_eq!(input.remark, "EU pool");
        assert_eq!(input.strategy, SubscriptionBalancerStrategy::Random);
        assert_eq!(input.inbound_ids, [7, 9]);
        assert_eq!(input.sort_order, 1);
        assert_eq!(input.enabled, Some(true));
    }
}