Add Vss KVStore api.

Author: G8XSU

rust/api/Cargo.toml

@@ -4,6 +4,7 @@ version = "0.1.0"
 edition = "2021"
 
 [dependencies]
+async-trait = "0.1.77"
 prost = { version = "0.11.6", default-features = false, features = ["std", "prost-derive"] }
 bytes = "1.4.0"
 

rust/api/src/error.rs

@@ -0,0 +1,54 @@
+use std::error::Error;
+use std::fmt::{Display, Formatter};
+use std::io;
+
+/// When there is an error while writing to VSS storage, the response contains a relevant error code.
+/// A mapping from a VSS server error codes. Refer to [`ErrorResponse`] docs for more
+/// information regarding each error code and corresponding use-cases.
+#[derive(Debug)]
+pub enum VssError {
+	/// Please refer to [`ErrorCode::NoSuchKeyException`].
+	NoSuchKeyError(String),
+
+	/// Please refer to [`ErrorCode::InvalidRequestException`].
+	InvalidRequestError(String),
+
+	/// Please refer to [`ErrorCode::ConflictException`].
+	ConflictError(String),
+
+	/// Please refer to [`ErrorCode::AuthException`].
+	AuthError(String),
+
+	/// Please refer to [`ErrorCode::InternalServerException`].
+	InternalServerError(String),
+}
+
+impl Display for VssError {
+	fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
+		match self {
+			VssError::NoSuchKeyError(message) => {
+				write!(f, "Requested key does not exist: {}", message)
+			},
+			VssError::InvalidRequestError(message) => {
+				write!(f, "Request sent to VSS Storage was invalid: {}", message)
+			},
+			VssError::ConflictError(message) => {
+				write!(f, "Potential version conflict in write operation: {}", message)
+			},
+			VssError::AuthError(message) => {
+				write!(f, "Authentication or Authorization failure: {}", message)
+			},
+			VssError::InternalServerError(message) => {
+				write!(f, "InternalServerError: {}", message)
+			},
+		}
+	}
+}
+
+impl Error for VssError {}
+
+impl From<io::Error> for VssError {
+	fn from(err: io::Error) -> Self {
+		VssError::InternalServerError(err.to_string())
+	}
+}

rust/api/src/kv_store.rs

@@ -0,0 +1,24 @@
+use crate::error::VssError;
+use crate::types::{
+	DeleteObjectRequest, DeleteObjectResponse, GetObjectRequest, GetObjectResponse,
+	ListKeyVersionsRequest, ListKeyVersionsResponse, PutObjectRequest, PutObjectResponse,
+};
+use async_trait::async_trait;
+
+pub const GLOBAL_VERSION_KEY: &str = "global_version";
+pub const INITIAL_RECORD_VERSION: i32 = 1;
+#[async_trait]
+pub trait KvStore: Send + Sync {
+	async fn get(
+		&self, user_token: String, request: GetObjectRequest,
+	) -> Result<GetObjectResponse, VssError>;
+	async fn put(
+		&self, user_token: String, request: PutObjectRequest,
+	) -> Result<PutObjectResponse, VssError>;
+	async fn delete(
+		&self, user_token: String, request: DeleteObjectRequest,
+	) -> Result<DeleteObjectResponse, VssError>;
+	async fn list_key_versions(
+		&self, user_token: String, request: ListKeyVersionsRequest,
+	) -> Result<ListKeyVersionsResponse, VssError>;
+}

rust/api/src/lib.rs

@@ -1 +1,3 @@
+pub mod error;
+pub mod kv_store;
 pub mod types;