Move the wasm lib, and cleanup rust directory and all references (#5585)

* git mv src/wasm-lib rust Signed-off-by: Jess Frazelle <github@jessfraz.com> * mv wasm-lib to workspace Signed-off-by: Jess Frazelle <github@jessfraz.com> * mv kcl-lib Signed-off-by: Jess Frazelle <github@jessfraz.com> * mv derive docs Signed-off-by: Jess Frazelle <github@jessfraz.com> * resolve file paths Signed-off-by: Jess Frazelle <github@jessfraz.com> * clippy Signed-off-by: Jess Frazelle <github@jessfraz.com> * move more shit Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix more paths Signed-off-by: Jess Frazelle <github@jessfraz.com> * make yarn build:wasm work Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix scripts Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixups Signed-off-by: Jess Frazelle <github@jessfraz.com> * better references Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix cargo ci Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix reference Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix more ci Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix tests Signed-off-by: Jess Frazelle <github@jessfraz.com> * cargo sort Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix script Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix Signed-off-by: Jess Frazelle <github@jessfraz.com> * fmt Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix a dep Signed-off-by: Jess Frazelle <github@jessfraz.com> * sort Signed-off-by: Jess Frazelle <github@jessfraz.com> * remove unused deps Signed-off-by: Jess Frazelle <github@jessfraz.com> * Revert "remove unused deps" This reverts commit fbabdb062e275fd5cbc1476f8480a1afee15d972. * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * deps; Signed-off-by: Jess Frazelle <github@jessfraz.com> * fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> --------- Signed-off-by: Jess Frazelle <github@jessfraz.com>
2025-03-01 13:59:01 -08:00
parent 0a2bf4b55f
commit c3bdc6f106
1443 changed files with 509 additions and 4274 deletions
--- a/rust/kcl-lib/src/test_server.rs
+++ b/rust/kcl-lib/src/test_server.rs
@ -0,0 +1,131 @@
+//! Types used to send data to the test server.
+
+use std::path::PathBuf;
+
+use crate::{
+    engine::new_zoo_client,
+    errors::ExecErrorWithState,
+    execution::{EnvironmentRef, ExecState, ExecutorContext, ExecutorSettings},
+    settings::types::UnitLength,
+    ConnectionError, ExecError, KclError, KclErrorWithOutputs, Program,
+};
+
+#[derive(serde::Deserialize, serde::Serialize)]
+pub struct RequestBody {
+    pub kcl_program: String,
+    #[serde(default)]
+    pub test_name: String,
+}
+
+/// Executes a kcl program and takes a snapshot of the result.
+/// This returns the bytes of the snapshot.
+pub async fn execute_and_snapshot(
+    code: &str,
+    units: UnitLength,
+    current_file: Option<PathBuf>,
+) -> Result<image::DynamicImage, ExecError> {
+    let ctx = new_context(units, true, current_file).await?;
+    let program = Program::parse_no_errs(code).map_err(KclErrorWithOutputs::no_outputs)?;
+    let res = do_execute_and_snapshot(&ctx, program)
+        .await
+        .map(|(_, _, snap)| snap)
+        .map_err(|err| err.error);
+    ctx.close().await;
+    res
+}
+
+/// Executes a kcl program and takes a snapshot of the result.
+/// This returns the bytes of the snapshot.
+pub async fn execute_and_snapshot_ast(
+    ast: Program,
+    units: UnitLength,
+    current_file: Option<PathBuf>,
+) -> Result<(ExecState, EnvironmentRef, image::DynamicImage), ExecErrorWithState> {
+    let ctx = new_context(units, true, current_file).await?;
+    let res = do_execute_and_snapshot(&ctx, ast).await;
+    ctx.close().await;
+    res
+}
+
+pub async fn execute_and_snapshot_no_auth(
+    code: &str,
+    units: UnitLength,
+    current_file: Option<PathBuf>,
+) -> Result<(image::DynamicImage, EnvironmentRef), ExecError> {
+    let ctx = new_context(units, false, current_file).await?;
+    let program = Program::parse_no_errs(code).map_err(KclErrorWithOutputs::no_outputs)?;
+    let res = do_execute_and_snapshot(&ctx, program)
+        .await
+        .map(|(_, env_ref, snap)| (snap, env_ref))
+        .map_err(|err| err.error);
+    ctx.close().await;
+    res
+}
+
+async fn do_execute_and_snapshot(
+    ctx: &ExecutorContext,
+    program: Program,
+) -> Result<(ExecState, EnvironmentRef, image::DynamicImage), ExecErrorWithState> {
+    let mut exec_state = ExecState::new(&ctx.settings);
+    let result = ctx
+        .run_with_ui_outputs(&program, &mut exec_state)
+        .await
+        .map_err(|err| ExecErrorWithState::new(err.into(), exec_state.clone()))?;
+    for e in exec_state.errors() {
+        if e.severity.is_err() {
+            return Err(ExecErrorWithState::new(
+                KclErrorWithOutputs::no_outputs(KclError::Semantic(e.clone().into())).into(),
+                exec_state.clone(),
+            ));
+        }
+    }
+    let snapshot_png_bytes = ctx
+        .prepare_snapshot()
+        .await
+        .map_err(|err| ExecErrorWithState::new(err, exec_state.clone()))?
+        .contents
+        .0;
+
+    // Decode the snapshot, return it.
+    let img = image::ImageReader::new(std::io::Cursor::new(snapshot_png_bytes))
+        .with_guessed_format()
+        .map_err(|e| ExecError::BadPng(e.to_string()))
+        .and_then(|x| x.decode().map_err(|e| ExecError::BadPng(e.to_string())))
+        .map_err(|err| ExecErrorWithState::new(err, exec_state.clone()))?;
+
+    ctx.close().await;
+
+    Ok((exec_state, result.0, img))
+}
+
+pub async fn new_context(
+    units: UnitLength,
+    with_auth: bool,
+    current_file: Option<PathBuf>,
+) -> Result<ExecutorContext, ConnectionError> {
+    let mut client = new_zoo_client(if with_auth { None } else { Some("bad_token".to_string()) }, None)
+        .map_err(ConnectionError::CouldNotMakeClient)?;
+    if !with_auth {
+        // Use prod, don't override based on env vars.
+        // We do this so even in the engine repo, tests that need to run with
+        // no auth can fail in the same way as they would in prod.
+        client.set_base_url("https://api.zoo.dev".to_string());
+    }
+
+    let mut settings = ExecutorSettings {
+        units,
+        highlight_edges: true,
+        enable_ssao: false,
+        show_grid: false,
+        replay: None,
+        project_directory: None,
+        current_file: None,
+    };
+    if let Some(current_file) = current_file {
+        settings.with_current_file(current_file);
+    }
+    let ctx = ExecutorContext::new(&client, settings)
+        .await
+        .map_err(ConnectionError::Establishing)?;
+    Ok(ctx)
+}