rscale 0.1.0

Self-hosted Rust control plane for operating a single tailnet with Tailscale clients
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

use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct Node {
    pub id: u64,
    pub stable_id: String,
    pub name: String,
    pub hostname: String,
    pub auth_key_id: Option<String>,
    pub principal_id: Option<u64>,
    pub ipv4: Option<String>,
    pub ipv6: Option<String>,
    pub status: NodeStatus,
    pub tags: Vec<String>,
    #[serde(default)]
    pub tag_source: NodeTagSource,
    pub last_seen_unix_secs: Option<u64>,
}

#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum NodeStatus {
    Pending,
    Online,
    Offline,
    Expired,
    Disabled,
}

impl NodeStatus {
    pub fn as_str(&self) -> &'static str {
        match self {
            Self::Pending => "pending",
            Self::Online => "online",
            Self::Offline => "offline",
            Self::Expired => "expired",
            Self::Disabled => "disabled",
        }
    }

    pub fn parse(value: &str) -> Option<Self> {
        match value {
            "pending" => Some(Self::Pending),
            "online" => Some(Self::Online),
            "offline" => Some(Self::Offline),
            "expired" => Some(Self::Expired),
            "disabled" => Some(Self::Disabled),
            _ => None,
        }
    }
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq, Default)]
#[serde(rename_all = "snake_case")]
pub enum NodeTagSource {
    #[default]
    None,
    Request,
    AuthKey,
    Admin,
}

impl NodeTagSource {
    pub fn as_str(self) -> &'static str {
        match self {
            Self::None => "none",
            Self::Request => "request",
            Self::AuthKey => "auth_key",
            Self::Admin => "admin",
        }
    }

    pub fn parse(value: &str) -> Option<Self> {
        match value {
            "none" => Some(Self::None),
            "request" => Some(Self::Request),
            "auth_key" => Some(Self::AuthKey),
            "admin" => Some(Self::Admin),
            _ => None,
        }
    }

    pub fn is_server_forced(self) -> bool {
        matches!(self, Self::AuthKey | Self::Admin)
    }
}

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

    #[test]
    fn node_status_round_trips_to_strings() {
        for (status, expected) in [
            (NodeStatus::Pending, "pending"),
            (NodeStatus::Online, "online"),
            (NodeStatus::Offline, "offline"),
            (NodeStatus::Expired, "expired"),
            (NodeStatus::Disabled, "disabled"),
        ] {
            assert_eq!(status.as_str(), expected);
            assert_eq!(NodeStatus::parse(expected), Some(status));
        }
        assert_eq!(NodeStatus::parse("unknown"), None);
    }

    #[test]
    fn node_tag_source_round_trips_and_marks_server_forced_sources() {
        for (source, expected, is_server_forced) in [
            (NodeTagSource::None, "none", false),
            (NodeTagSource::Request, "request", false),
            (NodeTagSource::AuthKey, "auth_key", true),
            (NodeTagSource::Admin, "admin", true),
        ] {
            assert_eq!(source.as_str(), expected);
            assert_eq!(NodeTagSource::parse(expected), Some(source));
            assert_eq!(source.is_server_forced(), is_server_forced);
        }
        assert_eq!(NodeTagSource::parse("unknown"), None);
    }
}