add catalog provider

Author: reluctanthero104

crates/cli/src/main.rs

@@ -81,6 +81,7 @@ struct LocalDataSource {
     path: Option<String>,
     connection_string: Option<String>,
     options: Option<HashMap<String, String>>,
+    hierarchy_level: Option<String>,
 }
 
 fn resolve_ctx_path(override_path: Option<PathBuf>) -> Result<PathBuf> {
@@ -498,11 +499,19 @@ async fn register_source(
                     source.name
                 )
             })?;
+            let mut postgres_options = source.options.clone().unwrap_or_default();
+            postgres_options.insert(
+                "hierarchy_level".to_string(),
+                source
+                    .hierarchy_level
+                    .clone()
+                    .unwrap_or_else(|| "table".to_string()),
+            );
             register_postgres_tables(
                 session_ctx,
                 &source.name,
                 conn_str,
-                source.options.as_ref(),
+                Some(&postgres_options),
                 false,
             )
             .await

crates/server/src/config.rs

@@ -66,6 +66,9 @@ pub struct DataSource {
     pub schema: Option<HashMap<String, String>>,
     /// Optional format-specific options
     pub options: Option<HashMap<String, String>>,
+    /// Registration hierarchy level for database sources (default: table)
+    #[serde(default)]
+    pub hierarchy_level: HierarchyLevel,
     /// Access mode: read_only (default) or read_write
     #[serde(default)]
     pub access_mode: AccessMode,
@@ -89,6 +92,24 @@ pub enum DataSourceType {
     Lance,
 }
 
+/// Registration hierarchy level for sources that can expose multiple objects.
+#[derive(Debug, Clone, Deserialize, Serialize, Hash, PartialEq, Eq, Default)]
+#[serde(rename_all = "lowercase")]
+pub enum HierarchyLevel {
+    #[default]
+    Table,
+    Catalog,
+}
+
+impl HierarchyLevel {
+    fn as_str(&self) -> &'static str {
+        match self {
+            Self::Table => "table",
+            Self::Catalog => "catalog",
+        }
+    }
+}
+
 /// Context configuration file structure
 #[derive(Debug, Deserialize)]
 struct ContextConfig {
@@ -743,12 +764,19 @@ async fn register_data_source(
                 source.options
             );
 
-            // Register PostgreSQL table using the sqlx-based provider
+            // Forward top-level hierarchy_level via options for provider handling.
+            let mut postgres_options = source.options.clone().unwrap_or_default();
+            postgres_options.insert(
+                "hierarchy_level".to_string(),
+                source.hierarchy_level.as_str().to_string(),
+            );
+
+            // Register PostgreSQL table(s) using the sqlx-based provider
             sources::providers::sqlx::postgres::register_postgres_tables(
                 session_ctx,
                 &source.name,
                 connection_string,
-                source.options.as_ref(),
+                Some(&postgres_options),
                 source.access_mode.is_read_write(),
             )
             .await