From df35feba18787cac06b2a87e3426752fb78da2d5 Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Mon, 3 Jan 2022 19:06:04 +0100
Subject: New buckets for 0.6.0: make bucket id a SK and not a HK, CLI updates

---
 src/model/bucket_table.rs  | 28 +++++++++++++++++++++-------
 src/model/helper/bucket.rs |  6 +++---
 2 files changed, 24 insertions(+), 10 deletions(-)

(limited to 'src/model')

diff --git a/src/model/bucket_table.rs b/src/model/bucket_table.rs
index 52c2316c..d687e774 100644
--- a/src/model/bucket_table.rs
+++ b/src/model/bucket_table.rs
@@ -105,15 +105,29 @@ impl Bucket {
 			crdt::Deletable::Present(state) => state.authorized_keys.items(),
 		}
 	}
-}
 
-impl Entry<Uuid, EmptyKey> for Bucket {
-	fn partition_key(&self) -> &Uuid {
-		&self.id
+	pub fn aliases(&self) -> &[(String, u64, bool)] {
+		match &self.state {
+			crdt::Deletable::Deleted => &[],
+			crdt::Deletable::Present(state) => state.aliases.items(),
+		}
+	}
+
+	pub fn local_aliases(&self) -> &[((String, String), u64, bool)] {
+		match &self.state {
+			crdt::Deletable::Deleted => &[],
+			crdt::Deletable::Present(state) => state.local_aliases.items(),
+		}
 	}
-	fn sort_key(&self) -> &EmptyKey {
+}
+
+impl Entry<EmptyKey, Uuid> for Bucket {
+	fn partition_key(&self) -> &EmptyKey {
 		&EmptyKey
 	}
+	fn sort_key(&self) -> &Uuid {
+		&self.id
+	}
 }
 
 impl Crdt for Bucket {
@@ -127,8 +141,8 @@ pub struct BucketTable;
 impl TableSchema for BucketTable {
 	const TABLE_NAME: &'static str = "bucket_v2";
 
-	type P = Uuid;
-	type S = EmptyKey;
+	type P = EmptyKey;
+	type S = Uuid;
 	type E = Bucket;
 	type Filter = DeletedFilter;
 
diff --git a/src/model/helper/bucket.rs b/src/model/helper/bucket.rs
index 52cedb12..6f171c8b 100644
--- a/src/model/helper/bucket.rs
+++ b/src/model/helper/bucket.rs
@@ -36,7 +36,7 @@ impl<'a> BucketHelper<'a> {
 			Ok(self
 				.0
 				.bucket_table
-				.get(&bucket_id, &EmptyKey)
+				.get(&EmptyKey, &bucket_id)
 				.await?
 				.filter(|x| !x.state.is_deleted())
 				.map(|_| bucket_id))
@@ -58,7 +58,7 @@ impl<'a> BucketHelper<'a> {
 		Ok(self
 			.0
 			.bucket_table
-			.get(&bucket_id, &EmptyKey)
+			.get(&EmptyKey, &bucket_id)
 			.await?
 			.ok_or_message(format!("Bucket {:?} does not exist", bucket_id))?)
 	}
@@ -70,7 +70,7 @@ impl<'a> BucketHelper<'a> {
 	pub async fn get_existing_bucket(&self, bucket_id: Uuid) -> Result<Bucket, Error> {
 		self.0
 			.bucket_table
-			.get(&bucket_id, &EmptyKey)
+			.get(&EmptyKey, &bucket_id)
 			.await?
 			.filter(|b| !b.is_deleted())
 			.ok_or_bad_request(format!(
-- 
cgit v1.2.3