From bdf23de657bb54097466c475e4b30ee4489a8ea9 Mon Sep 17 00:00:00 2001
From: Kohaku-Blueleaf <59680068+KohakuBlueleaf@users.noreply.github.com>
Date: Mon, 20 Oct 2025 01:51:33 +0800
Subject: [PATCH] fix admin portal cannot see org issue

---
 src/kohaku-hub-admin/src/components.d.ts | 13 -------------
 src/kohaku-hub-admin/src/pages/users.vue | 23 ++++++++++++++++++++---
 src/kohaku-hub-admin/src/utils/api.js    |  4 ++--
 src/kohakuhub/api/admin/routers/users.py |  9 ++++++++-
 4 files changed, 30 insertions(+), 19 deletions(-)

diff --git a/src/kohaku-hub-admin/src/components.d.ts b/src/kohaku-hub-admin/src/components.d.ts
index e73312d..2b51942 100644
--- a/src/kohaku-hub-admin/src/components.d.ts
+++ b/src/kohaku-hub-admin/src/components.d.ts
@@ -12,8 +12,6 @@ declare module 'vue' {
     ChartCard: typeof import('./components/ChartCard.vue')['default']
     ElAlert: typeof import('element-plus/es')['ElAlert']
     ElAside: typeof import('element-plus/es')['ElAside']
-    ElBreadcrumb: typeof import('element-plus/es')['ElBreadcrumb']
-    ElBreadcrumbItem: typeof import('element-plus/es')['ElBreadcrumbItem']
     ElButton: typeof import('element-plus/es')['ElButton']
     ElCard: typeof import('element-plus/es')['ElCard']
     ElContainer: typeof import('element-plus/es')['ElContainer']
@@ -26,23 +24,12 @@ declare module 'vue' {
     ElFormItem: typeof import('element-plus/es')['ElFormItem']
     ElHeader: typeof import('element-plus/es')['ElHeader']
     ElInput: typeof import('element-plus/es')['ElInput']
-    ElInputNumber: typeof import('element-plus/es')['ElInputNumber']
     ElMain: typeof import('element-plus/es')['ElMain']
     ElMenu: typeof import('element-plus/es')['ElMenu']
     ElMenuItem: typeof import('element-plus/es')['ElMenuItem']
-    ElOption: typeof import('element-plus/es')['ElOption']
-    ElPagination: typeof import('element-plus/es')['ElPagination']
-    ElProgress: typeof import('element-plus/es')['ElProgress']
-    ElRadioButton: typeof import('element-plus/es')['ElRadioButton']
-    ElRadioGroup: typeof import('element-plus/es')['ElRadioGroup']
-    ElResult: typeof import('element-plus/es')['ElResult']
-    ElSelect: typeof import('element-plus/es')['ElSelect']
-    ElSkeleton: typeof import('element-plus/es')['ElSkeleton']
     ElSwitch: typeof import('element-plus/es')['ElSwitch']
     ElTable: typeof import('element-plus/es')['ElTable']
     ElTableColumn: typeof import('element-plus/es')['ElTableColumn']
-    ElTabPane: typeof import('element-plus/es')['ElTabPane']
-    ElTabs: typeof import('element-plus/es')['ElTabs']
     ElTag: typeof import('element-plus/es')['ElTag']
     FileTree: typeof import('./components/FileTree.vue')['default']
     GlobalSearch: typeof import('./components/GlobalSearch.vue')['default']
diff --git a/src/kohaku-hub-admin/src/pages/users.vue b/src/kohaku-hub-admin/src/pages/users.vue
index c8115db..054a891 100644
--- a/src/kohaku-hub-admin/src/pages/users.vue
+++ b/src/kohaku-hub-admin/src/pages/users.vue
@@ -22,9 +22,10 @@ const dialogVisible = ref(false);
 const userDialogVisible = ref(false);
 const selectedUser = ref(null);
 
-// Search
+// Search and filters
 const searchQuery = ref("");
 const searchDebounceTimer = ref(null);
+const showOrgs = ref(true); // Toggle to show/hide organizations
 
 // Pagination
 const currentPage = ref(1);
@@ -105,6 +106,7 @@ async function loadUsers() {
       search: searchQuery.value || undefined,
       limit: pageSize.value,
       offset: (currentPage.value - 1) * pageSize.value,
+      include_orgs: showOrgs.value,
     });
     users.value = response.users;
   } catch (error) {
@@ -302,7 +304,7 @@ onMounted(() => {
 
       <!-- Search Bar -->
       <el-card class="mb-4">
-        <div class="flex gap-4 items-center">
+        <div class="flex gap-4 items-center flex-wrap">
           <el-input
             v-model="searchQuery"
             placeholder="Search users by username or email..."
@@ -315,6 +317,12 @@ onMounted(() => {
               <div class="i-carbon-search text-gray-400" />
             </template>
           </el-input>
+          <el-switch
+            v-model="showOrgs"
+            @change="loadUsers"
+            active-text="Show Organizations"
+            inactive-text="Users Only"
+          />
           <span v-if="searchQuery" class="text-sm text-gray-500">
             Searching for: "{{ searchQuery }}"
           </span>
@@ -341,7 +349,16 @@ onMounted(() => {
             label="Username"
             min-width="150"
             sortable="custom"
-          />
+          >
+            <template #default="{ row }">
+              <div class="flex items-center gap-2">
+                <span>{{ row.username }}</span>
+                <el-tag v-if="row.is_org" size="small" type="warning"
+                  >ORG</el-tag
+                >
+              </div>
+            </template>
+          </el-table-column>
           <el-table-column prop="email" label="Email" min-width="200" />
           <el-table-column
             prop="private_used"
diff --git a/src/kohaku-hub-admin/src/utils/api.js b/src/kohaku-hub-admin/src/utils/api.js
index 964968e..6599d94 100644
--- a/src/kohaku-hub-admin/src/utils/api.js
+++ b/src/kohaku-hub-admin/src/utils/api.js
@@ -32,11 +32,11 @@ function createAdminClient(token) {
  */
 export async function listUsers(
   token,
-  { search, limit = 100, offset = 0 } = {},
+  { search, limit = 100, offset = 0, include_orgs = false } = {},
 ) {
   const client = createAdminClient(token);
   const response = await client.get("/users", {
-    params: { search, limit, offset },
+    params: { search, limit, offset, include_orgs },
   });
   return response.data;
 }
diff --git a/src/kohakuhub/api/admin/routers/users.py b/src/kohakuhub/api/admin/routers/users.py
index 3dce0f5..590c321 100644
--- a/src/kohakuhub/api/admin/routers/users.py
+++ b/src/kohakuhub/api/admin/routers/users.py
@@ -88,6 +88,7 @@ async def list_users(
     search: str | None = None,
     limit: int = 100,
     offset: int = 0,
+    include_orgs: bool = False,
     _admin: bool = Depends(verify_admin_token),
 ):
     """List all users with quota information.
@@ -96,13 +97,18 @@ async def list_users(
         search: Search by username or email (optional)
         limit: Maximum number of users to return
         offset: Offset for pagination
+        include_orgs: Include organizations in results (default: False)
         _admin: Admin authentication (dependency)
 
     Returns:
         List of users with quota information
     """
 
-    users_query = User.select().where(User.is_org == False)
+    # Filter by type
+    if include_orgs:
+        users_query = User.select()  # Include both users and orgs
+    else:
+        users_query = User.select().where(User.is_org == False)  # Users only
 
     # Add search filter if provided
     if search:
@@ -119,6 +125,7 @@ async def list_users(
             "email": u.email,
             "email_verified": u.email_verified,
             "is_active": u.is_active,
+            "is_org": u.is_org,  # Add org flag
             "private_quota_bytes": u.private_quota_bytes,
             "public_quota_bytes": u.public_quota_bytes,
             "private_used_bytes": u.private_used_bytes,