rnacos 0.8.3

Nacos server re-implemented in Rust.
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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145

use crate::common::model::privilege::{PrivilegeGroup, PrivilegeGroupFlags};
use crate::user::permission::UserRoleHelper;
use serde::{Deserialize, Serialize};
use std::collections::HashSet;
use std::{collections::HashMap, sync::Arc};

#[derive(Clone, Debug, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub enum UserSourceType {
    Inner,
    Ldap,
}

impl Default for UserSourceType {
    fn default() -> Self {
        UserSourceType::Inner
    }
}

impl UserSourceType {
    pub fn is_inner(&self) -> bool {
        matches!(self, UserSourceType::Inner)
    }

    pub fn from_name(name: &str) -> Option<Self> {
        match name {
            "" => Some(UserSourceType::Inner),
            "INNER" => Some(UserSourceType::Inner),
            "LDAP" => Some(UserSourceType::Ldap),
            _ => None,
        }
    }

    pub fn to_str(&self) -> &'static str {
        match self {
            UserSourceType::Inner => "INNER",
            UserSourceType::Ldap => "LDAP",
        }
    }
}

#[derive(Clone, prost::Message, Serialize, Deserialize)]
pub struct UserDo {
    #[prost(string, tag = "1")]
    pub username: String,
    #[prost(string, tag = "2")]
    pub password: String,
    #[prost(string, tag = "3")]
    pub nickname: String,
    #[prost(uint32, tag = "4")]
    pub gmt_create: u32,
    #[prost(uint32, tag = "5")]
    pub gmt_modified: u32,
    #[prost(bool, tag = "6")]
    pub enable: bool,
    #[prost(string, repeated, tag = "7")]
    pub roles: ::prost::alloc::vec::Vec<String>,
    #[prost(map = "string, string", tag = "8")]
    pub extend_info:
        ::std::collections::HashMap<::prost::alloc::string::String, ::prost::alloc::string::String>,
    #[prost(string, optional, tag = "9")]
    pub password_hash: Option<String>,
    #[prost(uint32, optional, tag = "10")]
    pub namespace_privilege_flags: Option<u32>,
    #[prost(string, repeated, tag = "11")]
    pub namespace_white_list: ::prost::alloc::vec::Vec<String>,
    #[prost(string, repeated, tag = "12")]
    pub namespace_black_list: ::prost::alloc::vec::Vec<String>,
    #[prost(string, optional, tag = "13")]
    pub source: Option<String>,
}

impl UserDo {
    pub fn to_bytes(&self) -> Vec<u8> {
        let mut v = Vec::new();
        prost::Message::encode(self, &mut v).unwrap_or_default();
        v
    }

    pub fn from_bytes(v: &[u8]) -> anyhow::Result<Self> {
        Ok(prost::Message::decode(v)?)
    }

    pub fn build_namespace_privilege(&self) -> PrivilegeGroup<Arc<String>> {
        let namespace_privilege_flags = self.namespace_privilege_flags.unwrap_or_default() as u8;
        if namespace_privilege_flags & PrivilegeGroupFlags::ENABLE.bits() > 0 {
            let mut namespace_whitelist = HashSet::new();
            for item in &self.namespace_white_list {
                namespace_whitelist.insert(Arc::new(item.clone()));
            }
            let mut namespace_black_list = HashSet::new();
            for item in &self.namespace_black_list {
                namespace_black_list.insert(Arc::new(item.clone()));
            }
            PrivilegeGroup::new(
                namespace_privilege_flags,
                Some(Arc::new(namespace_whitelist)),
                Some(Arc::new(namespace_black_list)),
            )
        } else {
            PrivilegeGroup::all()
        }
    }
}

#[derive(Clone, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
pub struct UserDto {
    pub username: Arc<String>,
    pub nickname: Option<String>,
    pub password: Option<String>,
    pub password_hash: Option<String>,
    pub gmt_create: Option<i64>,
    pub gmt_modified: Option<i64>,
    pub enable: Option<bool>,
    pub roles: Option<Vec<Arc<String>>>,
    pub extend_info: Option<HashMap<String, String>>,
    pub namespace_privilege: Option<PrivilegeGroup<Arc<String>>>,
    pub source: Option<String>,
}

impl From<UserDo> for UserDto {
    fn from(value: UserDo) -> Self {
        let mut roles = vec![];
        for role in &value.roles {
            roles.push(UserRoleHelper::get_role(role));
        }
        let namespace_privilege = Some(value.build_namespace_privilege());
        Self {
            username: Arc::new(value.username),
            nickname: Some(value.nickname),
            //password: Some(value.password),
            //不直接返回密码
            password_hash: value.password_hash,
            password: None,
            gmt_create: Some(value.gmt_create as i64 * 1000),
            gmt_modified: Some(value.gmt_modified as i64 * 1000),
            enable: Some(value.enable),
            roles: Some(roles),
            extend_info: Some(value.extend_info),
            namespace_privilege,
            source: value.source,
        }
    }
}