full database extracton

src/database/mod.rs

@@ -29,12 +29,15 @@ use std::collections::HashMap;
 use std::sync::Arc;
 use r#type::Type;
 
+#[derive(serde::Serialize)]
 pub struct Database {
   pub enums: HashMap<String, Enum>,
   pub types: HashMap<String, Type>,
   pub puncs: HashMap<String, Punc>,
   pub relations: HashMap<String, Relation>,
+  #[serde(skip)]
   pub schemas: HashMap<String, Arc<Schema>>,
+  #[serde(skip)]
   pub executor: Box<dyn DatabaseExecutor + Send + Sync>,
 }
 
@@ -238,23 +241,30 @@ impl Database {
 
     // Phase 2: Synthesize Composed Filter References
     let mut filter_schemas = Vec::new();
-    for type_def in self.types.values() {
+    for (type_name, type_def) in &self.types {
       for (id, schema_arc) in &type_def.schemas {
         // Only run synthesis on actual structured, table-backed boundaries. Exclude subschemas!
         let base_name = id.split('.').last().unwrap_or(id);
         let is_table_backed = base_name == type_def.name;
         if is_table_backed && !id.contains('/') {
           if let Some(filter_schema) = schema_arc.compile_filter(self, id, errors) {
-            filter_schemas.push((format!("{}.filter", id), Arc::new(filter_schema)));
+            filter_schemas.push((
+              type_name.clone(),
+              format!("{}.filter", id),
+              Arc::new(filter_schema),
+            ));
           }
         }
       }
     }
 
     let mut filter_ids = Vec::new();
-    for (id, filter_arc) in filter_schemas {
+    for (type_name, id, filter_arc) in filter_schemas {
       filter_ids.push(id.clone());
-      self.schemas.insert(id, filter_arc);
+      self.schemas.insert(id.clone(), filter_arc.clone());
+      if let Some(t) = self.types.get_mut(&type_name) {
+        t.schemas.insert(id, filter_arc);
+      }
     }
 
     // Now actively compile the newly injected filters to lock all nested compose references natively
@@ -269,50 +279,88 @@ impl Database {
   }
 
   fn collect_schemas(&mut self, errors: &mut Vec<crate::drop::Error>) {
-    let mut to_insert = Vec::new();
+    let mut type_insert = Vec::new();
+    let mut punc_insert = Vec::new();
+    let mut enum_insert = Vec::new();
+    let mut global_insert = Vec::new();
 
     // Pass 1: Extract all Schemas structurally off top level definitions into the master registry.
     // Validate every node recursively via string filters natively!
-    for type_def in self.types.values() {
+    for (type_name, type_def) in &self.types {
       for (id, schema_arc) in &type_def.schemas {
-        to_insert.push((id.clone(), Arc::clone(schema_arc)));
+        global_insert.push((id.clone(), Arc::clone(schema_arc)));
+        let mut local_insert = Vec::new();
         crate::database::schema::Schema::collect_schemas(
           schema_arc,
           id,
           id.clone(),
-          &mut to_insert,
-          errors,
-        );
-      }
-    }
-    for punc_def in self.puncs.values() {
-      for (id, schema_arc) in &punc_def.schemas {
-        to_insert.push((id.clone(), Arc::clone(schema_arc)));
-        crate::database::schema::Schema::collect_schemas(
-          schema_arc,
-          id,
-          id.clone(),
-          &mut to_insert,
-          errors,
-        );
-      }
-    }
-    for enum_def in self.enums.values() {
-      for (id, schema_arc) in &enum_def.schemas {
-        to_insert.push((id.clone(), Arc::clone(schema_arc)));
-        crate::database::schema::Schema::collect_schemas(
-          schema_arc,
-          id,
-          id.clone(),
-          &mut to_insert,
+          &mut local_insert,
           errors,
         );
+        for entry in &local_insert {
+          type_insert.push((type_name.clone(), entry.0.clone(), Arc::clone(&entry.1)));
+          global_insert.push((entry.0.clone(), Arc::clone(&entry.1)));
+        }
       }
     }
 
-    for (id, schema_arc) in to_insert {
+    for (punc_name, punc_def) in &self.puncs {
+      for (id, schema_arc) in &punc_def.schemas {
+        global_insert.push((id.clone(), Arc::clone(schema_arc)));
+        let mut local_insert = Vec::new();
+        crate::database::schema::Schema::collect_schemas(
+          schema_arc,
+          id,
+          id.clone(),
+          &mut local_insert,
+          errors,
+        );
+        for entry in &local_insert {
+          punc_insert.push((punc_name.clone(), entry.0.clone(), Arc::clone(&entry.1)));
+          global_insert.push((entry.0.clone(), Arc::clone(&entry.1)));
+        }
+      }
+    }
+
+    for (enum_name, enum_def) in &self.enums {
+      for (id, schema_arc) in &enum_def.schemas {
+        global_insert.push((id.clone(), Arc::clone(schema_arc)));
+        let mut local_insert = Vec::new();
+        crate::database::schema::Schema::collect_schemas(
+          schema_arc,
+          id,
+          id.clone(),
+          &mut local_insert,
+          errors,
+        );
+        for entry in &local_insert {
+          enum_insert.push((enum_name.clone(), entry.0.clone(), Arc::clone(&entry.1)));
+          global_insert.push((entry.0.clone(), Arc::clone(&entry.1)));
+        }
+      }
+    }
+
+    // Apply global inserts
+    for (id, schema_arc) in global_insert {
       self.schemas.insert(id, schema_arc);
     }
+
+    // Apply local scopes
+    for (origin_name, id, schema_arc) in type_insert {
+      if let Some(t) = self.types.get_mut(&origin_name) {
+        t.schemas.insert(id, schema_arc);
+      }
+    }
+    for (origin_name, id, schema_arc) in punc_insert {
+      if let Some(p) = self.puncs.get_mut(&origin_name) {
+        p.schemas.insert(id, schema_arc);
+      }
+    }
+    for (origin_name, id, schema_arc) in enum_insert {
+      if let Some(e) = self.enums.get_mut(&origin_name) {
+        e.schemas.insert(id, schema_arc);
+      }
+    }
   }
 
   /// Inspects the Postgres pg_constraint relations catalog to securely identify