2021-12-16 15:17:51 +00:00
|
|
|
use std::collections::HashMap;
|
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
use garage_util::crdt::*;
|
2021-10-19 14:16:10 +00:00
|
|
|
use garage_util::data::Uuid;
|
|
|
|
use garage_util::error::*;
|
2022-05-18 20:24:09 +00:00
|
|
|
use garage_util::formater::format_table;
|
2021-10-19 14:16:10 +00:00
|
|
|
|
|
|
|
use garage_model::bucket_table::*;
|
|
|
|
use garage_model::key_table::*;
|
|
|
|
|
2022-01-03 18:06:04 +00:00
|
|
|
pub fn print_bucket_list(bl: Vec<Bucket>) {
|
|
|
|
println!("List of buckets:");
|
|
|
|
|
|
|
|
let mut table = vec![];
|
|
|
|
for bucket in bl {
|
|
|
|
let aliases = bucket
|
|
|
|
.aliases()
|
|
|
|
.iter()
|
|
|
|
.filter(|(_, _, active)| *active)
|
|
|
|
.map(|(name, _, _)| name.to_string())
|
|
|
|
.collect::<Vec<_>>();
|
2022-01-05 15:28:46 +00:00
|
|
|
let local_aliases_n = match &bucket
|
2022-01-03 18:06:04 +00:00
|
|
|
.local_aliases()
|
|
|
|
.iter()
|
|
|
|
.filter(|(_, _, active)| *active)
|
2022-01-05 15:28:46 +00:00
|
|
|
.collect::<Vec<_>>()[..]
|
2022-01-03 18:06:04 +00:00
|
|
|
{
|
2022-01-05 15:28:46 +00:00
|
|
|
[] => "".into(),
|
|
|
|
[((k, n), _, _)] => format!("{}:{}", k, n),
|
|
|
|
s => format!("[{} local aliases]", s.len()),
|
2022-01-03 18:06:04 +00:00
|
|
|
};
|
|
|
|
table.push(format!(
|
|
|
|
"\t{}\t{}\t{}",
|
|
|
|
aliases.join(","),
|
|
|
|
local_aliases_n,
|
|
|
|
hex::encode(bucket.id)
|
|
|
|
));
|
|
|
|
}
|
|
|
|
format_table(table);
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn print_key_list(kl: Vec<(String, String)>) {
|
|
|
|
println!("List of keys:");
|
|
|
|
let mut table = vec![];
|
|
|
|
for key in kl {
|
|
|
|
table.push(format!("\t{}\t{}", key.0, key.1));
|
|
|
|
}
|
|
|
|
format_table(table);
|
|
|
|
}
|
|
|
|
|
2021-12-16 15:17:51 +00:00
|
|
|
pub fn print_key_info(key: &Key, relevant_buckets: &HashMap<Uuid, Bucket>) {
|
|
|
|
let bucket_global_aliases = |b: &Uuid| {
|
|
|
|
if let Some(bucket) = relevant_buckets.get(b) {
|
|
|
|
if let Some(p) = bucket.state.as_option() {
|
|
|
|
return p
|
|
|
|
.aliases
|
|
|
|
.items()
|
|
|
|
.iter()
|
|
|
|
.filter(|(_, _, active)| *active)
|
|
|
|
.map(|(a, _, _)| a.clone())
|
|
|
|
.collect::<Vec<_>>()
|
|
|
|
.join(", ");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
"".to_string()
|
|
|
|
};
|
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
match &key.state {
|
|
|
|
Deletable::Present(p) => {
|
2022-01-04 17:59:17 +00:00
|
|
|
println!("Key name: {}", p.name.get());
|
|
|
|
println!("Key ID: {}", key.key_id);
|
|
|
|
println!("Secret key: {}", p.secret_key);
|
2021-12-16 10:47:58 +00:00
|
|
|
println!("Can create buckets: {}", p.allow_create_bucket.get());
|
2021-12-14 12:55:11 +00:00
|
|
|
println!("\nKey-specific bucket aliases:");
|
2021-12-15 17:36:15 +00:00
|
|
|
let mut table = vec![];
|
2021-12-14 12:55:11 +00:00
|
|
|
for (alias_name, _, alias) in p.local_aliases.items().iter() {
|
2022-01-03 17:32:15 +00:00
|
|
|
if let Some(bucket_id) = alias {
|
2021-12-16 15:17:51 +00:00
|
|
|
table.push(format!(
|
|
|
|
"\t{}\t{}\t{}",
|
|
|
|
alias_name,
|
|
|
|
bucket_global_aliases(bucket_id),
|
|
|
|
hex::encode(bucket_id)
|
|
|
|
));
|
2021-12-14 12:55:11 +00:00
|
|
|
}
|
|
|
|
}
|
2021-12-15 17:36:15 +00:00
|
|
|
format_table(table);
|
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
println!("\nAuthorized buckets:");
|
2021-12-15 17:36:15 +00:00
|
|
|
let mut table = vec![];
|
2021-12-16 15:17:51 +00:00
|
|
|
for (bucket_id, perm) in p.authorized_buckets.items().iter() {
|
2022-01-05 15:28:46 +00:00
|
|
|
if !perm.is_any() {
|
|
|
|
continue;
|
|
|
|
}
|
2021-12-14 12:55:11 +00:00
|
|
|
let rflag = if perm.allow_read { "R" } else { " " };
|
|
|
|
let wflag = if perm.allow_write { "W" } else { " " };
|
2021-12-16 10:47:58 +00:00
|
|
|
let oflag = if perm.allow_owner { "O" } else { " " };
|
2021-12-16 15:17:51 +00:00
|
|
|
let local_aliases = p
|
|
|
|
.local_aliases
|
|
|
|
.items()
|
|
|
|
.iter()
|
2022-01-03 17:32:15 +00:00
|
|
|
.filter(|(_, _, a)| *a == Some(*bucket_id))
|
2021-12-16 15:17:51 +00:00
|
|
|
.map(|(a, _, _)| a.clone())
|
|
|
|
.collect::<Vec<_>>()
|
|
|
|
.join(", ");
|
|
|
|
table.push(format!(
|
|
|
|
"\t{}{}{}\t{}\t{}\t{:?}",
|
|
|
|
rflag,
|
|
|
|
wflag,
|
|
|
|
oflag,
|
|
|
|
bucket_global_aliases(bucket_id),
|
|
|
|
local_aliases,
|
|
|
|
bucket_id
|
|
|
|
));
|
2021-12-14 12:55:11 +00:00
|
|
|
}
|
2021-12-15 17:36:15 +00:00
|
|
|
format_table(table);
|
2021-12-14 12:55:11 +00:00
|
|
|
}
|
|
|
|
Deletable::Deleted => {
|
2022-01-04 17:59:17 +00:00
|
|
|
println!("Key {} is deleted.", key.key_id);
|
2021-10-19 14:16:10 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2021-12-16 15:17:51 +00:00
|
|
|
pub fn print_bucket_info(bucket: &Bucket, relevant_keys: &HashMap<String, Key>) {
|
2022-01-04 17:59:17 +00:00
|
|
|
let key_name = |k| {
|
|
|
|
relevant_keys
|
|
|
|
.get(k)
|
|
|
|
.map(|k| k.params().unwrap().name.get().as_str())
|
|
|
|
.unwrap_or("<deleted>")
|
|
|
|
};
|
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
println!("Bucket: {}", hex::encode(bucket.id));
|
|
|
|
match &bucket.state {
|
|
|
|
Deletable::Deleted => println!("Bucket is deleted."),
|
|
|
|
Deletable::Present(p) => {
|
2021-12-17 10:53:13 +00:00
|
|
|
println!("Website access: {}", p.website_config.get().is_some());
|
2021-12-16 15:17:51 +00:00
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
println!("\nGlobal aliases:");
|
|
|
|
for (alias, _, active) in p.aliases.items().iter() {
|
|
|
|
if *active {
|
2021-12-16 15:17:51 +00:00
|
|
|
println!(" {}", alias);
|
2021-12-14 12:55:11 +00:00
|
|
|
}
|
|
|
|
}
|
2021-12-15 17:36:15 +00:00
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
println!("\nKey-specific aliases:");
|
2021-12-16 15:17:51 +00:00
|
|
|
let mut table = vec![];
|
2021-12-14 12:55:11 +00:00
|
|
|
for ((key_id, alias), _, active) in p.local_aliases.items().iter() {
|
|
|
|
if *active {
|
2022-01-04 17:59:17 +00:00
|
|
|
table.push(format!("\t{} ({})\t{}", key_id, key_name(key_id), alias));
|
2021-12-14 12:55:11 +00:00
|
|
|
}
|
|
|
|
}
|
2021-12-16 15:17:51 +00:00
|
|
|
format_table(table);
|
2021-12-15 17:36:15 +00:00
|
|
|
|
2021-12-14 12:55:11 +00:00
|
|
|
println!("\nAuthorized keys:");
|
2021-12-16 15:17:51 +00:00
|
|
|
let mut table = vec![];
|
2021-12-14 12:55:11 +00:00
|
|
|
for (k, perm) in p.authorized_keys.items().iter() {
|
2022-02-02 16:12:48 +00:00
|
|
|
if !perm.is_any() {
|
|
|
|
continue;
|
|
|
|
}
|
2021-12-14 12:55:11 +00:00
|
|
|
let rflag = if perm.allow_read { "R" } else { " " };
|
|
|
|
let wflag = if perm.allow_write { "W" } else { " " };
|
2021-12-16 10:47:58 +00:00
|
|
|
let oflag = if perm.allow_owner { "O" } else { " " };
|
2021-12-16 15:17:51 +00:00
|
|
|
table.push(format!(
|
2022-01-03 18:06:04 +00:00
|
|
|
"\t{}{}{}\t{}\t{}",
|
2022-01-04 17:59:17 +00:00
|
|
|
rflag,
|
|
|
|
wflag,
|
|
|
|
oflag,
|
|
|
|
k,
|
|
|
|
key_name(k)
|
2021-12-16 15:17:51 +00:00
|
|
|
));
|
2021-10-19 14:16:10 +00:00
|
|
|
}
|
2021-12-16 15:17:51 +00:00
|
|
|
format_table(table);
|
2021-10-19 14:16:10 +00:00
|
|
|
}
|
|
|
|
};
|
|
|
|
}
|
|
|
|
|
|
|
|
pub fn find_matching_node(
|
|
|
|
cand: impl std::iter::Iterator<Item = Uuid>,
|
|
|
|
pattern: &str,
|
|
|
|
) -> Result<Uuid, Error> {
|
|
|
|
let mut candidates = vec![];
|
|
|
|
for c in cand {
|
2022-05-09 09:14:55 +00:00
|
|
|
if hex::encode(&c).starts_with(&pattern) && !candidates.contains(&c) {
|
2021-10-19 14:16:10 +00:00
|
|
|
candidates.push(c);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if candidates.len() != 1 {
|
|
|
|
Err(Error::Message(format!(
|
|
|
|
"{} nodes match '{}'",
|
|
|
|
candidates.len(),
|
|
|
|
pattern,
|
|
|
|
)))
|
|
|
|
} else {
|
|
|
|
Ok(candidates[0])
|
|
|
|
}
|
|
|
|
}
|