CLI account management + Directory refactoring
This commit is contained in:
mdecimus
360
crates/cli/src/modules/account.rs
Normal file
360
crates/cli/src/modules/account.rs
Normal file
@@ -0,0 +1,360 @@
|
||||
/*
|
||||
* Copyright (c) 2020-2023, Stalwart Labs Ltd.
|
||||
*
|
||||
* This file is part of Stalwart Mail Server.
|
||||
*
|
||||
* This program is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU Affero General Public License as
|
||||
* published by the Free Software Foundation, either version 3 of
|
||||
* the License, or (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU Affero General Public License for more details.
|
||||
* in the LICENSE file at the top-level directory of this distribution.
|
||||
* You should have received a copy of the GNU Affero General Public License
|
||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*
|
||||
* You can be released from the requirements of the AGPLv3 license by
|
||||
* purchasing a commercial license. Please contact licensing@stalw.art
|
||||
* for more details.
|
||||
*/
|
||||
|
||||
use std::fmt::Display;
|
||||
|
||||
use prettytable::{Attr, Cell, Row, Table};
|
||||
use pwhash::sha512_crypt;
|
||||
use reqwest::Method;
|
||||
use serde_json::Value;
|
||||
|
||||
use super::{
|
||||
cli::{AccountCommands, Client},
|
||||
Principal, PrincipalField, PrincipalUpdate, PrincipalValue, Type,
|
||||
};
|
||||
|
||||
impl AccountCommands {
|
||||
pub async fn exec(self, client: Client) {
|
||||
match self {
|
||||
AccountCommands::Create {
|
||||
name,
|
||||
password,
|
||||
description,
|
||||
quota,
|
||||
is_admin,
|
||||
addresses,
|
||||
member_of,
|
||||
} => {
|
||||
let principal = Principal {
|
||||
id: None,
|
||||
typ: if is_admin.unwrap_or_default() {
|
||||
Type::Superuser
|
||||
} else {
|
||||
Type::Individual
|
||||
}
|
||||
.into(),
|
||||
quota,
|
||||
used_quota: None,
|
||||
name: name.clone().into(),
|
||||
secrets: vec![sha512_crypt::hash(password).unwrap()],
|
||||
emails: addresses.unwrap_or_default(),
|
||||
member_of: member_of.unwrap_or_default(),
|
||||
description,
|
||||
};
|
||||
let account_id = client
|
||||
.http_request::<u32, _>(Method::POST, "/admin/principal", Some(principal))
|
||||
.await;
|
||||
eprintln!("Successfully created account {name:?} with id {account_id}.");
|
||||
}
|
||||
AccountCommands::Update {
|
||||
name,
|
||||
new_name,
|
||||
password,
|
||||
description,
|
||||
quota,
|
||||
is_admin,
|
||||
addresses,
|
||||
member_of,
|
||||
} => {
|
||||
let mut changes = Vec::new();
|
||||
if let Some(new_name) = new_name {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::Name,
|
||||
PrincipalValue::String(new_name),
|
||||
));
|
||||
}
|
||||
if let Some(password) = password {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::Secrets,
|
||||
PrincipalValue::StringList(vec![sha512_crypt::hash(password).unwrap()]),
|
||||
));
|
||||
}
|
||||
if let Some(description) = description {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::Description,
|
||||
PrincipalValue::String(description),
|
||||
));
|
||||
}
|
||||
if let Some(quota) = quota {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::Quota,
|
||||
PrincipalValue::Integer(quota),
|
||||
));
|
||||
}
|
||||
if let Some(is_admin) = is_admin {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::Type,
|
||||
PrincipalValue::Type(if is_admin {
|
||||
Type::Superuser
|
||||
} else {
|
||||
Type::Individual
|
||||
}),
|
||||
));
|
||||
}
|
||||
if let Some(addresses) = addresses {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::Emails,
|
||||
PrincipalValue::StringList(addresses),
|
||||
));
|
||||
}
|
||||
if let Some(member_of) = member_of {
|
||||
changes.push(PrincipalUpdate::set(
|
||||
PrincipalField::MemberOf,
|
||||
PrincipalValue::StringList(member_of),
|
||||
));
|
||||
}
|
||||
|
||||
if !changes.is_empty() {
|
||||
client
|
||||
.http_request::<Value, _>(
|
||||
Method::PATCH,
|
||||
&format!("/admin/principal/{name}"),
|
||||
Some(changes),
|
||||
)
|
||||
.await;
|
||||
eprintln!("Successfully updated account {name:?}.");
|
||||
} else {
|
||||
eprintln!("No changes to apply.");
|
||||
}
|
||||
}
|
||||
AccountCommands::AddEmail { name, addresses } => {
|
||||
client
|
||||
.http_request::<Value, _>(
|
||||
Method::PATCH,
|
||||
&format!("/admin/principal/{name}"),
|
||||
Some(
|
||||
addresses
|
||||
.into_iter()
|
||||
.map(|address| {
|
||||
PrincipalUpdate::add_item(
|
||||
PrincipalField::Emails,
|
||||
PrincipalValue::String(address),
|
||||
)
|
||||
})
|
||||
.collect::<Vec<_>>(),
|
||||
),
|
||||
)
|
||||
.await;
|
||||
eprintln!("Successfully updated account {name:?}.");
|
||||
}
|
||||
AccountCommands::RemoveEmail { name, addresses } => {
|
||||
client
|
||||
.http_request::<Value, _>(
|
||||
Method::PATCH,
|
||||
&format!("/admin/principal/{name}"),
|
||||
Some(
|
||||
addresses
|
||||
.into_iter()
|
||||
.map(|address| {
|
||||
PrincipalUpdate::remove_item(
|
||||
PrincipalField::Emails,
|
||||
PrincipalValue::String(address),
|
||||
)
|
||||
})
|
||||
.collect::<Vec<_>>(),
|
||||
),
|
||||
)
|
||||
.await;
|
||||
eprintln!("Successfully updated account {name:?}.");
|
||||
}
|
||||
AccountCommands::AddToGroup { name, member_of } => {
|
||||
client
|
||||
.http_request::<Value, _>(
|
||||
Method::PATCH,
|
||||
&format!("/admin/principal/{name}"),
|
||||
Some(
|
||||
member_of
|
||||
.into_iter()
|
||||
.map(|group| {
|
||||
PrincipalUpdate::add_item(
|
||||
PrincipalField::MemberOf,
|
||||
PrincipalValue::String(group),
|
||||
)
|
||||
})
|
||||
.collect::<Vec<_>>(),
|
||||
),
|
||||
)
|
||||
.await;
|
||||
eprintln!("Successfully updated account {name:?}.");
|
||||
}
|
||||
AccountCommands::RemoveFromGroup { name, member_of } => {
|
||||
client
|
||||
.http_request::<Value, _>(
|
||||
Method::PATCH,
|
||||
&format!("/admin/principal/{name}"),
|
||||
Some(
|
||||
member_of
|
||||
.into_iter()
|
||||
.map(|group| {
|
||||
PrincipalUpdate::remove_item(
|
||||
PrincipalField::MemberOf,
|
||||
PrincipalValue::String(group),
|
||||
)
|
||||
})
|
||||
.collect::<Vec<_>>(),
|
||||
),
|
||||
)
|
||||
.await;
|
||||
eprintln!("Successfully updated account {name:?}.");
|
||||
}
|
||||
AccountCommands::Delete { name } => {
|
||||
client
|
||||
.http_request::<Value, String>(
|
||||
Method::DELETE,
|
||||
&format!("/admin/principal/{name}"),
|
||||
None,
|
||||
)
|
||||
.await;
|
||||
eprintln!("Successfully deleted account {name:?}.");
|
||||
}
|
||||
AccountCommands::Display { name } => {
|
||||
client.display_principal(&name).await;
|
||||
}
|
||||
AccountCommands::List { from, limit } => {
|
||||
client
|
||||
.list_principals("individual", "Account", from, limit)
|
||||
.await;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl Client {
|
||||
pub async fn display_principal(&self, name: &str) {
|
||||
let principal = self
|
||||
.http_request::<Principal, String>(
|
||||
Method::GET,
|
||||
&format!("/admin/principal/{name}"),
|
||||
None,
|
||||
)
|
||||
.await;
|
||||
let mut table = Table::new();
|
||||
if let Some(name) = principal.name {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new("Name").with_style(Attr::Bold),
|
||||
Cell::new(&name),
|
||||
]));
|
||||
}
|
||||
let is_list = if let Some(typ) = principal.typ {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new("Type").with_style(Attr::Bold),
|
||||
Cell::new(&typ.to_string()),
|
||||
]));
|
||||
matches!(typ, Type::List)
|
||||
} else {
|
||||
false
|
||||
};
|
||||
if let Some(description) = principal.description {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new("Description").with_style(Attr::Bold),
|
||||
Cell::new(&description),
|
||||
]));
|
||||
}
|
||||
if let Some(quota) = principal.quota {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new("Quota").with_style(Attr::Bold),
|
||||
Cell::new("a.to_string()),
|
||||
]));
|
||||
}
|
||||
if let Some(used_quota) = principal.used_quota {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new("Used Quota").with_style(Attr::Bold),
|
||||
Cell::new(&used_quota.to_string()),
|
||||
]));
|
||||
}
|
||||
if !principal.member_of.is_empty() {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new(if is_list { "List members" } else { "Member of" })
|
||||
.with_style(Attr::Bold),
|
||||
Cell::new(&principal.member_of.join(", ")),
|
||||
]));
|
||||
}
|
||||
if !principal.emails.is_empty() {
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new("E-mail address(es)").with_style(Attr::Bold),
|
||||
Cell::new(&principal.emails.join(", ")),
|
||||
]));
|
||||
}
|
||||
eprintln!();
|
||||
table.printstd();
|
||||
eprintln!();
|
||||
}
|
||||
|
||||
pub async fn list_principals(
|
||||
&self,
|
||||
record_type: &str,
|
||||
record_name: &str,
|
||||
from: Option<String>,
|
||||
limit: Option<usize>,
|
||||
) {
|
||||
let mut query = form_urlencoded::Serializer::new("/admin/principal?".to_string());
|
||||
|
||||
query.append_pair("type", record_type);
|
||||
|
||||
if let Some(from) = &from {
|
||||
query.append_pair("from", from);
|
||||
}
|
||||
if let Some(limit) = limit {
|
||||
query.append_pair("limit", &limit.to_string());
|
||||
}
|
||||
|
||||
let results = self
|
||||
.http_request::<Vec<String>, String>(Method::GET, &query.finish(), None)
|
||||
.await;
|
||||
if !results.is_empty() {
|
||||
let mut table = Table::new();
|
||||
table.add_row(Row::new(vec![
|
||||
Cell::new(&format!("{record_name} Name")).with_style(Attr::Bold)
|
||||
]));
|
||||
|
||||
for domain in &results {
|
||||
table.add_row(Row::new(vec![Cell::new(domain)]));
|
||||
}
|
||||
|
||||
eprintln!();
|
||||
table.printstd();
|
||||
eprintln!();
|
||||
}
|
||||
|
||||
eprintln!(
|
||||
"\n\n{} {}{} found.\n",
|
||||
results.len(),
|
||||
record_name.to_ascii_lowercase(),
|
||||
if results.len() == 1 { "" } else { "s" }
|
||||
);
|
||||
}
|
||||
}
|
||||
|
||||
impl Display for Type {
|
||||
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
|
||||
match self {
|
||||
Type::Superuser => write!(f, "Superuser"),
|
||||
Type::Individual => write!(f, "Individual"),
|
||||
Type::Group => write!(f, "Group"),
|
||||
Type::List => write!(f, "List"),
|
||||
Type::Resource => write!(f, "Resource"),
|
||||
Type::Location => write!(f, "Location"),
|
||||
Type::Other => write!(f, "Other"),
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user