From 83fdd59b13d4bf45bd35d9693ae361ff896636ab Mon Sep 17 00:00:00 2001
From: mrw1593 <botahamec@outlook.com>
Date: Tue, 6 Jun 2023 19:10:46 -0400
Subject: Add new endpoints for allowed and default scopes

---
 src/services/db/client.rs | 50 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

(limited to 'src/services/db')

diff --git a/src/services/db/client.rs b/src/services/db/client.rs
index ecf98a3..c25ad0d 100644
--- a/src/services/db/client.rs
+++ b/src/services/db/client.rs
@@ -94,6 +94,28 @@ pub async fn get_client_type<'c>(
 	Ok(ty)
 }
 
+pub async fn get_client_allowed_scopes<'c>(
+	executor: impl Executor<'c, Database = MySql>,
+	id: Uuid,
+) -> Result<Option<Box<str>>, RawUnexpected> {
+	let scopes = query_scalar!("SELECT allowed_scopes FROM clients WHERE id = ?", id)
+		.fetch_optional(executor)
+		.await?;
+
+	Ok(scopes.map(Box::from))
+}
+
+pub async fn get_client_default_scopes<'c>(
+	executor: impl Executor<'c, Database = MySql>,
+	id: Uuid,
+) -> Result<Option<Option<Box<str>>>, RawUnexpected> {
+	let scopes = query_scalar!("SELECT default_scopes FROM clients WHERE id = ?", id)
+		.fetch_optional(executor)
+		.await?;
+
+	Ok(scopes.map(|s| s.map(Box::from)))
+}
+
 pub async fn get_client_redirect_uris<'c>(
 	executor: impl Executor<'c, Database = MySql>,
 	id: Uuid,
@@ -236,6 +258,34 @@ pub async fn update_client_type<'c>(
 		.await
 }
 
+pub async fn update_client_allowed_scopes<'c>(
+	executor: impl Executor<'c, Database = MySql>,
+	id: Uuid,
+	allowed_scopes: &str,
+) -> Result<MySqlQueryResult, sqlx::Error> {
+	query!(
+		"UPDATE clients SET allowed_scopes = ? WHERE id = ?",
+		allowed_scopes,
+		id
+	)
+	.execute(executor)
+	.await
+}
+
+pub async fn update_client_default_scopes<'c>(
+	executor: impl Executor<'c, Database = MySql>,
+	id: Uuid,
+	default_scopes: Option<String>,
+) -> Result<MySqlQueryResult, sqlx::Error> {
+	query!(
+		"UPDATE clients SET default_scopes = ? WHERE id = ?",
+		default_scopes,
+		id
+	)
+	.execute(executor)
+	.await
+}
+
 pub async fn update_client_redirect_uris<'c>(
 	mut transaction: Transaction<'c, MySql>,
 	id: Uuid,
-- 
cgit v1.2.3