From 131fa446ee0ce5c755d96ed81d2c4da6c386bed7 Mon Sep 17 00:00:00 2001
From: John Chilton <jmchilton@gmail.com>
Date: Wed, 4 Oct 2023 13:35:52 -0400
Subject: [PATCH] Initial bug fixes for tool shed 2.

---
 lib/tool_shed/managers/users.py                    |  4 +++-
 lib/tool_shed/webapp/api2/users.py                 |  2 +-
 .../webapp/frontend/src/components/ShedToolbar.vue |  4 +---
 .../src/components/pages/RepositoryPage.vue        |  2 +-
 lib/tool_shed/webapp/frontend/src/schema/schema.ts | 14 ++++++++------
 lib/tool_shed_client/schema/__init__.py            |  4 ++++
 6 files changed, 18 insertions(+), 12 deletions(-)

diff --git a/lib/tool_shed/managers/users.py b/lib/tool_shed/managers/users.py
index 055aaf1945d1..7d9012cf938b 100644
--- a/lib/tool_shed/managers/users.py
+++ b/lib/tool_shed/managers/users.py
@@ -12,7 +12,7 @@
 from tool_shed.webapp.model import User
 from tool_shed_client.schema import (
     CreateUserRequest,
-    User as ApiUser,
+    UserV2 as ApiUser,
 )
 
 
@@ -58,9 +58,11 @@ def api_create_user(trans: ProvidesUserContext, request: CreateUserRequest) -> A
 
 
 def get_api_user(app: ToolShedApp, user: User) -> ApiUser:
+    admin = user.email in app.config.admin_users
     return ApiUser(
         id=app.security.encode_id(user.id),
         username=user.username,
+        is_admin=admin,
     )
 
 
diff --git a/lib/tool_shed/webapp/api2/users.py b/lib/tool_shed/webapp/api2/users.py
index 8873d8b9800f..fefef4e71a7d 100644
--- a/lib/tool_shed/webapp/api2/users.py
+++ b/lib/tool_shed/webapp/api2/users.py
@@ -34,7 +34,7 @@
 from tool_shed.webapp.model import User as SaUser
 from tool_shed_client.schema import (
     CreateUserRequest,
-    User,
+    UserV2 as User,
 )
 from . import (
     depends,
diff --git a/lib/tool_shed/webapp/frontend/src/components/ShedToolbar.vue b/lib/tool_shed/webapp/frontend/src/components/ShedToolbar.vue
index 631d2349f8ac..3338c42984c1 100644
--- a/lib/tool_shed/webapp/frontend/src/components/ShedToolbar.vue
+++ b/lib/tool_shed/webapp/frontend/src/components/ShedToolbar.vue
@@ -8,13 +8,11 @@ defineProps({
         required: true,
     },
 })
-// TODO: recover from the API
-const ADMINS = ["jmchilton"]
 
 const authStore = useAuthStore()
 void authStore.setup()
 
-const admin = computed(() => authStore.user && ADMINS.indexOf(authStore.user.username) > -1)
+const admin = computed(() => authStore.user && authStore.user.is_admin)
 </script>
 <template>
     <q-toolbar class="bg-primary glossy text-white">
diff --git a/lib/tool_shed/webapp/frontend/src/components/pages/RepositoryPage.vue b/lib/tool_shed/webapp/frontend/src/components/pages/RepositoryPage.vue
index 6e31d63fc2d0..51929fc44d31 100644
--- a/lib/tool_shed/webapp/frontend/src/components/pages/RepositoryPage.vue
+++ b/lib/tool_shed/webapp/frontend/src/components/pages/RepositoryPage.vue
@@ -242,7 +242,7 @@ const toolsYaml = computed(
                         <revision-actions
                             :repository-id="repositoryId"
                             :current-metadata="currentMetadata"
-                            v-if="currentMetadata"
+                            v-if="currentMetadata && can_manage"
                             @update="onUpdate"
                         />
                     </revision-select>
diff --git a/lib/tool_shed/webapp/frontend/src/schema/schema.ts b/lib/tool_shed/webapp/frontend/src/schema/schema.ts
index 5b2011c4db4f..91fd9d54ea0c 100644
--- a/lib/tool_shed/webapp/frontend/src/schema/schema.ts
+++ b/lib/tool_shed/webapp/frontend/src/schema/schema.ts
@@ -1007,10 +1007,12 @@ export interface components {
             /** Email */
             email: string
         }
-        /** User */
-        User: {
+        /** UserV2 */
+        UserV2: {
             /** Id */
             id: string
+            /** Is Admin */
+            is_admin: boolean
             /** Username */
             username: string
         }
@@ -1786,7 +1788,7 @@ export interface operations {
             /** @description Successful Response */
             200: {
                 content: {
-                    "application/json": components["schemas"]["User"][]
+                    "application/json": components["schemas"]["UserV2"][]
                 }
             }
         }
@@ -1805,7 +1807,7 @@ export interface operations {
             /** @description Successful Response */
             200: {
                 content: {
-                    "application/json": components["schemas"]["User"]
+                    "application/json": components["schemas"]["UserV2"]
                 }
             }
             /** @description Validation Error */
@@ -1825,7 +1827,7 @@ export interface operations {
             /** @description Successful Response */
             200: {
                 content: {
-                    "application/json": components["schemas"]["User"]
+                    "application/json": components["schemas"]["UserV2"]
                 }
             }
         }
@@ -1845,7 +1847,7 @@ export interface operations {
             /** @description Successful Response */
             200: {
                 content: {
-                    "application/json": components["schemas"]["User"]
+                    "application/json": components["schemas"]["UserV2"]
                 }
             }
             /** @description Validation Error */
diff --git a/lib/tool_shed_client/schema/__init__.py b/lib/tool_shed_client/schema/__init__.py
index 2e7e1d7a9014..a97ca4f3401b 100644
--- a/lib/tool_shed_client/schema/__init__.py
+++ b/lib/tool_shed_client/schema/__init__.py
@@ -61,6 +61,10 @@ class User(BaseModel):
     username: str
 
 
+class UserV2(User):
+    is_admin: bool
+
+
 class Category(BaseModel):
     id: str
     name: str