feat(native): analyse models natively behind the preview engine

crates/sqlbuild-analysis/src/assembly/analysis_session/_helpers/catalog_state.rs

@@ -0,0 +1,106 @@
+//! A session copy of the compile's binding catalog that records what Python's would gain.
+
+use std::collections::{HashMap, HashSet};
+
+use crate::assembly::analysis_session::_helpers::mappings::{
+    ShapeTable, catalog_columns, catalog_relations, same_mapping,
+};
+use crate::assembly::analysis_session::types::{Pairs, Shapes};
+use crate::semantic_validation::models::{Columns, ProjectCatalog};
+use crate::semantic_validation::types::{BindingRequest, Relations};
+
+/// Python's `BindingCatalog` state that model analysis reads and changes.
+#[derive(Debug)]
+pub(crate) struct SessionCatalog {
+    pub(crate) native: ProjectCatalog,
+    schemas: HashMap<String, Pairs>,
+    additions: Shapes,
+    analysis_names: Vec<String>,
+    recorded_analysis: HashSet<String>,
+}
+
+impl SessionCatalog {
+    /// Copy `native`, whose Python wrapper holds `schemas`.
+    pub(crate) fn new(native: &ProjectCatalog, schemas: &Shapes) -> Self {
+        Self {
+            native: native.with_relations(Relations::new()),
+            schemas: schemas.iter().cloned().collect(),
+            additions: Vec::new(),
+            analysis_names: Vec::new(),
+            recorded_analysis: HashSet::new(),
+        }
+    }
+
+    /// Python's `BindingCatalog.prepare`: add unknown closed relations, then override the rest.
+    pub(crate) fn prepare(&mut self, requests: &[(&str, &Shapes)]) -> Vec<BindingRequest> {
+        let mut additions: Shapes = Vec::new();
+        for (_, schema) in requests {
+            for (name, columns) in schema.iter() {
+                if !columns.is_empty() && !self.schemas.contains_key(name) {
+                    self.schemas.insert(name.clone(), columns.clone());
+                    additions.push((name.clone(), columns.clone()));
+                }
+            }
+        }
+        if !additions.is_empty() {
+            self.native.update_relations(catalog_relations(&additions));
+            self.additions.extend(additions);
+        }
+        requests
+            .iter()
+            .map(|(sql, schema)| self.binding_request(sql, schema))
+            .collect()
+    }
+
+    fn binding_request(&self, sql: &str, schema: &Shapes) -> BindingRequest {
+        let mut overrides: Relations = Relations::new();
+        let mut references: Vec<(String, bool)> = Vec::with_capacity(schema.len());
+        for (name, columns) in schema {
+            references.push((name.clone(), !columns.is_empty()));
+            let known: bool = self
+                .schemas
+                .get(name)
+                .is_some_and(|known| same_mapping(known, columns));
+            if !columns.is_empty() && !known {
+                overrides.insert(name.clone(), catalog_columns(columns));
+            }
+        }
+        references.sort();
+        (sql.to_owned(), references, overrides)
+    }
+
+    /// Python's `BindingCatalog.prepare_analysis` for the relations `names` a batch reads.
+    pub(crate) fn prepare_analysis(
+        &mut self,
+        names: &[String],
+        types: &ShapeTable,
+        nullability: &ShapeTable,
+    ) {
+        let mut updates: HashMap<String, (Columns, Columns)> = HashMap::new();
+        for name in names {
+            let Some(columns) = nullability.get(name) else {
+                continue;
+            };
+            let column_types: Pairs = types.get(name).cloned().unwrap_or_default();
+            updates.insert(
+                name.clone(),
+                (catalog_columns(&column_types), catalog_columns(columns)),
+            );
+            if self.recorded_analysis.insert(name.clone()) {
+                self.analysis_names.push(name.clone());
+            }
+        }
+        if !updates.is_empty() {
+            self.native.update_analysis(updates);
+        }
+    }
+
+    pub(crate) fn register_override(&mut self, overrides: Relations) -> usize {
+        self.native.register_override(overrides)
+    }
+
+    /// The `schemas` additions and analysis-shape names Python's catalog records.
+    pub(crate) fn into_changes(self) -> (Shapes, Vec<String>) {
+        (self.additions, self.analysis_names)
+    }
+}