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::*;
|
|
|
|
|
|
|
|
use garage_model::bucket_table::*;
|
|
|
|
use garage_model::key_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-10-19 14:16:10 +00:00
|
|
|
println!("Key name: {}", key.name.get());
|
|
|
|
println!("Key ID: {}", key.key_id);
|
|
|
|
println!("Secret key: {}", key.secret_key);
|
2021-12-14 12:55:11 +00:00
|
|
|
match &key.state {
|
|
|
|
Deletable::Present(p) => {
|
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() {
|
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 => {
|
|
|
|
println!("\nKey is deleted.");
|
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>) {
|
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 {
|
2021-12-16 15:17:51 +00:00
|
|
|
let key_name = relevant_keys
|
|
|
|
.get(key_id)
|
|
|
|
.map(|k| k.name.get().as_str())
|
|
|
|
.unwrap_or("");
|
|
|
|
table.push(format!("\t{}\t{} ({})", alias, key_id, key_name));
|
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() {
|
|
|
|
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 key_name = relevant_keys
|
|
|
|
.get(k)
|
|
|
|
.map(|k| k.name.get().as_str())
|
|
|
|
.unwrap_or("");
|
|
|
|
table.push(format!(
|
|
|
|
"\t{}{}{}\t{} ({})",
|
|
|
|
rflag, wflag, oflag, k, key_name
|
|
|
|
));
|
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 format_table(data: Vec<String>) {
|
|
|
|
let data = data
|
|
|
|
.iter()
|
|
|
|
.map(|s| s.split('\t').collect::<Vec<_>>())
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
|
|
|
|
let columns = data.iter().map(|row| row.len()).fold(0, std::cmp::max);
|
|
|
|
let mut column_size = vec![0; columns];
|
|
|
|
|
|
|
|
let mut out = String::new();
|
|
|
|
|
|
|
|
for row in data.iter() {
|
|
|
|
for (i, col) in row.iter().enumerate() {
|
|
|
|
column_size[i] = std::cmp::max(column_size[i], col.chars().count());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for row in data.iter() {
|
|
|
|
for (col, col_len) in row[..row.len() - 1].iter().zip(column_size.iter()) {
|
|
|
|
out.push_str(col);
|
|
|
|
(0..col_len - col.chars().count() + 2).for_each(|_| out.push(' '));
|
|
|
|
}
|
2021-10-26 08:20:05 +00:00
|
|
|
out.push_str(row[row.len() - 1]);
|
2021-10-19 14:16:10 +00:00
|
|
|
out.push('\n');
|
|
|
|
}
|
|
|
|
|
|
|
|
print!("{}", out);
|
|
|
|
}
|
|
|
|
|
|
|
|
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 {
|
|
|
|
if hex::encode(&c).starts_with(&pattern) {
|
|
|
|
candidates.push(c);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if candidates.len() != 1 {
|
|
|
|
Err(Error::Message(format!(
|
|
|
|
"{} nodes match '{}'",
|
|
|
|
candidates.len(),
|
|
|
|
pattern,
|
|
|
|
)))
|
|
|
|
} else {
|
|
|
|
Ok(candidates[0])
|
|
|
|
}
|
|
|
|
}
|