feat(query): expand get_tables to return table type and inclide foreign tables and views

Author: alexander-zuev

supabase_mcp/main.py

@@ -32,9 +32,11 @@ async def get_db_schemas():
     return result
 
 
-@mcp.tool(description="List all tables in a schema with their sizes, row counts, and metadata.")
+@mcp.tool(
+    description="List all tables, foreign tables, and views in a schema with their sizes, row counts, and metadata."
+)
 async def get_tables(schema_name: str):
-    """Get all tables from a schema with size, row count, column count, and index information."""
+    """Get all tables, foreign tables, and views from a schema with size, row count, column count, and index information."""
     schema_name = validate_schema_name(schema_name)
     query = PreBuiltQueries.get_tables_in_schema_query(schema_name)
     return supabase.execute_query(query)

supabase_mcp/queries.py

@@ -32,13 +32,14 @@ def get_tables_in_schema_query(schema_name: str) -> str:
                 pg_stat_get_live_tuples(pc.oid) as row_count,
                 (SELECT COUNT(*) FROM information_schema.columns c WHERE c.table_schema = t.table_schema AND c.table_name = t.table_name) as column_count,
                 (SELECT COUNT(*) FROM pg_indexes i WHERE i.schemaname = t.table_schema AND i.tablename = t.table_name) as index_count,
-                pg_total_relation_size(quote_ident(t.table_schema) || '.' || quote_ident(t.table_name)) as size_bytes  -- Added for ordering
+                pg_total_relation_size(quote_ident(t.table_schema) || '.' || quote_ident(t.table_name)) as size_bytes,  -- Added for ordering
+                t.table_type  -- Added to distinguish between regular and foreign tables
             FROM information_schema.tables t
             JOIN pg_class pc
                 ON pc.relname = t.table_name
                 AND pc.relnamespace = (SELECT oid FROM pg_namespace WHERE nspname = '{schema_name}')
             WHERE t.table_schema = '{schema_name}'
-                AND t.table_type = 'BASE TABLE'
+                AND t.table_type IN ('BASE TABLE', 'FOREIGN TABLE', 'VIEW')
             ORDER BY size_bytes DESC;
         """