Method intercept_query

src/catalog/query_interceptor.rs:26–209

Check if a query is targeting pg_catalog and handle it

(query: &str, db: Arc<DbHandler>, session: Option<Arc<SessionState>>)

impl CatalogInterceptor {
    /// Check if a query is targeting pg_catalog and handle it
    pub async fn intercept_query(query: &str, db: Arc<DbHandler>, session: Option<Arc<SessionState>>) -> Option<Result<DbResponse, PgSqliteError>> {
        println!("INTERCEPT_QUERY: {}", query);
        // Quick check to avoid parsing if not a catalog query
        let lower_query = query.to_lowercase();
        println!("INTERCEPT: lower_query = {}", lower_query);
        
        // Check for cache status query
        if lower_query.contains("select * from pgsqlite_cache_status") {
            let (columns, rows) = crate::cache::format_cache_status_as_table();
            let rows_affected = rows.len();
            let response = DbResponse {
                columns,
                rows,
                rows_affected,
            };
            return Some(Ok(response));
        }
        
        // Special case: pg_catalog.version() should be handled by SQLite function, not catalog interceptor
        if lower_query.trim() == "select pg_catalog.version()" || 
           lower_query.trim() == "select version()" {
            return None;
        }
        
        // Check for catalog tables
        let has_catalog_tables = lower_query.contains("pg_catalog") || lower_query.contains("pg_type") ||
           lower_query.contains("pg_namespace") || lower_query.contains("pg_range") ||
           lower_query.contains("pg_tablespace") ||
           lower_query.contains("pg_class") || lower_query.contains("pg_attribute") ||
           lower_query.contains("pg_enum") ||
           lower_query.contains("pg_description") || lower_query.contains("pg_roles") ||
           lower_query.contains("pg_user") || lower_query.contains("pg_authid") ||
           lower_query.contains("pg_stats") || lower_query.contains("pg_constraint") ||
           lower_query.contains("pg_depend") || lower_query.contains("pg_sequence") ||
           lower_query.contains("pg_trigger") || lower_query.contains("pg_settings") ||
           lower_query.contains("pg_collation") ||
           lower_query.contains("pg_replication_slots") ||
           lower_query.contains("pg_shdepend") ||
           lower_query.contains("pg_statistic") ||
           lower_query.contains("information_schema") ||
           lower_query.contains("pg_stat_") || lower_query.contains("pg_database") ||
           lower_query.contains("pg_foreign_data_wrapper");

        // Check for system functions
        let has_system_functions = lower_query.contains("to_regtype") ||
           lower_query.contains("pg_get_constraintdef") || lower_query.contains("pg_table_is_visible") ||
           lower_query.contains("format_type") || lower_query.contains("pg_get_expr") ||
           lower_query.contains("pg_get_userbyid") || lower_query.contains("pg_get_indexdef") ||
           lower_query.contains("pg_size_pretty");

        println!("INTERCEPT: has_catalog_tables = {}, has_system_functions = {}", has_catalog_tables, has_system_functions);

        if !has_catalog_tables && !has_system_functions {
            println!("INTERCEPT: Returning None (no catalog tables or system functions)");
            return None;
        }
        
        debug!("Intercepting catalog query: {}", query);