diff --git a/frontend/src/pages/DatabaseSchema.tsx b/frontend/src/pages/DatabaseSchema.tsx index cc60014..8d869db 100644 --- a/frontend/src/pages/DatabaseSchema.tsx +++ b/frontend/src/pages/DatabaseSchema.tsx @@ -15,9 +15,10 @@ import { } from '@xyflow/react'; import '@xyflow/react/dist/style.css'; import Dagre from '@dagrejs/dagre'; -import { Database as DatabaseIcon, Loader2, Key, Link, RefreshCw, Search, X, Table2, MessageSquare, Columns } from 'lucide-react'; +import { Database as DatabaseIcon, Loader2, Key, Link, Search, X, Table2, MessageSquare, Columns } from 'lucide-react'; import { databasesApi, schemaApi, TableInfo, SchemaData } from '@/services/api'; import { Database } from '@/types'; +import { useAuthStore } from '@/stores/authStore'; // Search result type interface SearchResult { @@ -273,6 +274,7 @@ function getLayoutedElements(schema: SchemaData): { nodes: Node[]; edges: Edge[] export default function DatabaseSchema() { const queryClient = useQueryClient(); + const user = useAuthStore((state) => state.user); const [selectedDbId, setSelectedDbId] = useState(''); const [searchQuery, setSearchQuery] = useState(''); const [selectedTables, setSelectedTables] = useState>(new Set()); @@ -581,22 +583,25 @@ export default function DatabaseSchema() { )} - {/* Refresh button hidden but mutation available for future use - {selectedDbId && ( + {selectedDbId && user?.is_superadmin && ( )} - */} {schema && (