feat: user management

src/apps/v1/users/users_repository.rs

@@ -1,5 +1,5 @@
 use super::{UsersActiveInactiveSchema, UsersSchema, UsersSetNewPasswordSchema};
-use crate::{AppState, ResourceEnum};
+use crate::{AppState, AuthOtpSchema, ResourceEnum};
 use anyhow::{bail, Result};
 
 pub struct UsersRepository<'a> {
@@ -12,9 +12,28 @@ impl<'a> UsersRepository<'a> {
 	}
 
 	pub async fn query_user_by_email(&self, email: String) -> Result<UsersSchema> {
-		let db = &self.state.surrealdb;
+		let db = &self.state.surrealdb_ws;
+		let sql = format!(
+			"SELECT * FROM {} WHERE email = $email",
+			ResourceEnum::Users.to_string()
+		);
+		let mut response: Vec<UsersSchema> = db
+			.query(sql)
+			.bind(("email", email.clone()))
+			.await?
+			.take(0)?;
+
+		if let Some(user) = response.pop() {
+			Ok(user)
+		} else {
+			bail!("User not found")
+		}
+	}
+
+	pub async fn query_user_by_id(&self, id: String) -> Result<UsersSchema> {
+		let db = &self.state.surrealdb_ws;
 		let result = db
-			.select((ResourceEnum::Users.to_string(), email.clone()))
+			.select((ResourceEnum::Users.to_string(), id.clone()))
 			.await?;
 		match result {
 			Some(response) => Ok(response),
@@ -23,9 +42,9 @@ impl<'a> UsersRepository<'a> {
 	}
 
 	pub async fn query_create_user(&self, data: UsersSchema) -> Result<String> {
-		let db = &self.state.surrealdb;
+		let db = &self.state.surrealdb_ws;
 		let record: Option<UsersSchema> = db
-			.create((ResourceEnum::Users.to_string(), &data.id))
+			.create(ResourceEnum::Users.to_string())
 			.content(data)
 			.await?;
 		match record {
@@ -34,17 +53,11 @@ impl<'a> UsersRepository<'a> {
 		}
 	}
 
-	pub async fn query_active_inactive_user(
-		&self,
-		data: UsersActiveInactiveSchema,
-	) -> Result<String> {
-		let db = &self.state.surrealdb;
-		let record: Option<UsersActiveInactiveSchema> = db
-			.update((ResourceEnum::Users.to_string(), &data.email))
-			.merge(UsersActiveInactiveSchema {
-				email: data.email.clone(),
-				is_active: data.is_active.clone(),
-			})
+	pub async fn query_update_user(&self, data: UsersSchema) -> Result<String> {
+		let db = &self.state.surrealdb_ws;
+		let record: Option<UsersSchema> = db
+			.update((ResourceEnum::Users.to_string(), &data.id.id.to_string()))
+			.merge(data)
 			.await?;
 		match record {
 			Some(_) => Ok("Success update user".into()),
@@ -52,11 +65,32 @@ impl<'a> UsersRepository<'a> {
 		}
 	}
 
+	pub async fn query_active_inactive_user(
+		&self,
+		email: String,
+		data: UsersActiveInactiveSchema,
+	) -> Result<String> {
+		let db = &self.state.surrealdb_ws;
+		let user = self.query_user_by_email(email.clone()).await?;
+		let table = user.id.tb.as_str();
+		let id = user.id.id.to_string();
+		let result: Option<AuthOtpSchema> = db
+			.update((table, id))
+			.merge(UsersActiveInactiveSchema {
+				is_active: data.is_active,
+			})
+			.await?;
+		match result {
+			Some(_) => Ok("Success update user".to_string()),
+			None => bail!("Failed to update user"),
+		}
+	}
+
 	pub async fn query_update_password_user(
 		&self,
 		data: UsersSetNewPasswordSchema,
 	) -> Result<String> {
-		let db = &self.state.surrealdb;
+		let db = &self.state.surrealdb_ws;
 		let record: Option<UsersSetNewPasswordSchema> = db
 			.update((ResourceEnum::Users.to_string(), &data.email))
 			.merge(UsersSetNewPasswordSchema {