web: repair config source editor

crates/config-source-wasm/src/lib.rs

@@ -8,6 +8,7 @@ use wasm_bindgen::prelude::*;
 
 thread_local! {
     static SESSION: RefCell<Option<ConfigTreeSnapshot>> = const { RefCell::new(None) };
+    static SCHEMA_BUNDLE: RefCell<Option<WorkspaceConfigSchemaBundle>> = const { RefCell::new(None) };
 }
 
 #[wasm_bindgen]
@@ -23,6 +24,13 @@ pub fn set_snapshot(snapshot: JsValue) -> Result<(), JsValue> {
     Ok(())
 }
 
+#[wasm_bindgen]
+pub fn set_schema_bundle(schema_bundle: JsValue) -> Result<(), JsValue> {
+    let schema_bundle: WorkspaceConfigSchemaBundle = decode(schema_bundle)?;
+    SCHEMA_BUNDLE.with(|session| session.replace(Some(schema_bundle)));
+    Ok(())
+}
+
 #[wasm_bindgen]
 pub fn apply_changes(changes: JsValue) -> Result<JsValue, JsValue> {
     let changes: Vec<ConfigTreeChange> = decode(changes)?;
@@ -91,8 +99,8 @@ pub fn complete_current(
             .as_ref()
             .ok_or_else(|| JsValue::from_str("config source snapshot is not initialized"))?;
         let utf8_byte_offset = utf16_to_utf8_offset(&source, utf16_offset)?;
-        let result = SnapshotEnvironment::new(snapshot.clone())
-            .complete(&entrypoint, &source, utf8_byte_offset, explicit)
+        let result = session_environment(snapshot.clone())
+            .complete_config(&entrypoint, &source, utf8_byte_offset, explicit)
             .map_err(|error| JsValue::from_str(&format!("{error:?}")))?
             .map(|result| WasmCompletionResult {
                 from: result.from,
@@ -132,7 +140,16 @@ pub fn analyze_snapshot(
 ) -> Result<JsValue, JsValue> {
     let snapshot: ConfigTreeSnapshot = decode(snapshot)?;
     let entrypoint = VirtualPath::parse(entrypoint).map_err(js_error)?;
-    encode(SnapshotEnvironment::new(snapshot).analyze(&entrypoint, source_override.as_deref()))
+    encode(session_environment(snapshot).analyze(&entrypoint, source_override.as_deref()))
+}
+
+fn session_environment(snapshot: ConfigTreeSnapshot) -> SnapshotEnvironment {
+    let schema_bundle = SCHEMA_BUNDLE.with(|schema_bundle| schema_bundle.borrow().clone());
+    let mut environment = SnapshotEnvironment::new(snapshot);
+    if let Some(schema_bundle) = schema_bundle {
+        environment = environment.with_schema_bundle(schema_bundle);
+    }
+    environment
 }
 
 #[wasm_bindgen]