From 196a9399d28e20c55cbb173ce4052845cfad5bf3 Mon Sep 17 00:00:00 2001
From: Frederick Muriuki Muriithi
Date: Fri, 22 Sep 2023 10:29:18 +0300
Subject: Update fetching a single role

---
 gn_auth/auth/authorisation/roles/models.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'gn_auth/auth/authorisation/roles/models.py')

diff --git a/gn_auth/auth/authorisation/roles/models.py b/gn_auth/auth/authorisation/roles/models.py
index 206b05e..4281446 100644
--- a/gn_auth/auth/authorisation/roles/models.py
+++ b/gn_auth/auth/authorisation/roles/models.py
@@ -115,7 +115,9 @@ def user_role(conn: db.DbConnection, user: User, role_id: UUID) -> Either:
     """Retrieve a specific non-resource role assigned to the user."""
     with db.cursor(conn) as cursor:
         cursor.execute(
-            "SELECT r.*, p.* FROM user_roles AS ur INNER JOIN roles AS r "
+            "SELECT res.resource_id, ur.user_id, r.*, p.* "
+            "FROM resources AS res INNER JOIN user_roles AS ur "
+            "ON res.resource_id=ur.resource_id INNER JOIN roles AS r "
             "ON ur.role_id=r.role_id INNER JOIN role_privileges AS rp "
             "ON r.role_id=rp.role_id INNER JOIN privileges AS p "
             "ON rp.privilege_id=p.privilege_id "
@@ -124,8 +126,10 @@ def user_role(conn: db.DbConnection, user: User, role_id: UUID) -> Either:
 
         results = cursor.fetchall()
         if results:
-            return Right(tuple(
-                reduce(__organise_privileges__, results, {}).values())[0])
+            res_role_obj = tuple(reduce(__organise_privileges__, results, {}).values())[0]
+            resource_id = res_role_obj["resource_id"]
+            role = tuple(res_role_obj["roles"].values())[0]
+            return Right((role, resource_id))
         return Left(NotFoundError(
             f"Could not find role with id '{role_id}'",))
 
-- 
cgit v1.2.3