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>
This commit is contained in:
Jess Frazelle
GitHub
495
rust/kcl-to-core/src/conn_mock_core.rs
Normal file
495
rust/kcl-to-core/src/conn_mock_core.rs
Normal file
@ -0,0 +1,495 @@
|
||||
use std::{collections::HashMap, sync::Arc};
|
||||
|
||||
use anyhow::Result;
|
||||
use indexmap::IndexMap;
|
||||
use kcl_lib::{
|
||||
exec::{ArtifactCommand, DefaultPlanes, IdGenerator},
|
||||
ExecutionKind, KclError,
|
||||
};
|
||||
use kittycad_modeling_cmds::{
|
||||
self as kcmc,
|
||||
id::ModelingCmdId,
|
||||
ok_response::OkModelingCmdResponse,
|
||||
shared::PathSegment::{self, *},
|
||||
websocket::{ModelingBatch, ModelingCmdReq, OkWebSocketResponseData, WebSocketRequest, WebSocketResponse},
|
||||
};
|
||||
use tokio::sync::RwLock;
|
||||
use uuid::Uuid;
|
||||
|
||||
const CPP_PREFIX: &str = "const double scaleFactor = 100;\n";
|
||||
const NEED_PLANES: bool = true;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub struct EngineConnection {
|
||||
batch: Arc<RwLock<Vec<(WebSocketRequest, kcl_lib::SourceRange)>>>,
|
||||
batch_end: Arc<RwLock<IndexMap<uuid::Uuid, (WebSocketRequest, kcl_lib::SourceRange)>>>,
|
||||
core_test: Arc<RwLock<String>>,
|
||||
default_planes: Arc<RwLock<Option<DefaultPlanes>>>,
|
||||
execution_kind: Arc<RwLock<ExecutionKind>>,
|
||||
}
|
||||
|
||||
impl EngineConnection {
|
||||
pub async fn new(result: Arc<RwLock<String>>) -> Result<EngineConnection> {
|
||||
result.write().await.push_str(CPP_PREFIX);
|
||||
|
||||
Ok(EngineConnection {
|
||||
batch: Arc::new(RwLock::new(Vec::new())),
|
||||
batch_end: Arc::new(RwLock::new(IndexMap::new())),
|
||||
core_test: result,
|
||||
default_planes: Default::default(),
|
||||
execution_kind: Default::default(),
|
||||
})
|
||||
}
|
||||
|
||||
fn handle_command(&self, cmd_id: &ModelingCmdId, cmd: &kcmc::ModelingCmd) -> (String, OkModelingCmdResponse) {
|
||||
let cpp_id = id_to_cpp(cmd_id);
|
||||
let cmd_id = format!("{}", cmd_id);
|
||||
let mut this_response = OkModelingCmdResponse::Empty {};
|
||||
|
||||
let new_code = match cmd {
|
||||
kcmc::ModelingCmd::ObjectVisible(kcmc::ObjectVisible { hidden, object_id }) => {
|
||||
format!(r#"scene->getSceneObject(Utils::UUID("{object_id}"))->setHidden({hidden});"#)
|
||||
}
|
||||
kcmc::ModelingCmd::EnableSketchMode(kcmc::EnableSketchMode {
|
||||
entity_id,
|
||||
animated: _,
|
||||
ortho: _,
|
||||
adjust_camera: _,
|
||||
planar_normal,
|
||||
}) => {
|
||||
if let Some(normal) = planar_normal {
|
||||
format!(
|
||||
r#"
|
||||
if(!scene->enableSketchMode(Utils::UUID("{entity_id}"), glm::dvec3 {{ {}, {}, {}, }}, nullopt))
|
||||
{{
|
||||
Utils::Plane plane_{cpp_id}(glm::dvec3 {{ 0, 0, 0 }}, glm::dvec3 {{ 1, 0, 0 }}, glm::dvec3 {{ 0, 1, 0 }});
|
||||
scene->enableSketchMode(plane_{cpp_id}, nullopt, nullopt, false);
|
||||
}}
|
||||
"#,
|
||||
normal.x, normal.y, normal.z
|
||||
)
|
||||
} else {
|
||||
"".into()
|
||||
}
|
||||
}
|
||||
kcmc::ModelingCmd::SketchModeDisable(kcmc::SketchModeDisable { .. }) => {
|
||||
"scene->disableSketchMode();".into()
|
||||
}
|
||||
kcmc::ModelingCmd::MakePlane(kcmc::MakePlane {
|
||||
origin,
|
||||
x_axis,
|
||||
y_axis,
|
||||
size,
|
||||
..
|
||||
}) => {
|
||||
let plane_id = format!("plane_{}", cpp_id);
|
||||
format!(
|
||||
r#"
|
||||
auto {plane_id} = make_shared<Object>("plane", glm::dvec3 {{ 0, 0, 0 }});
|
||||
{plane_id}->setUUID(Utils::UUID("{cmd_id}"));
|
||||
{plane_id}->makePlane(glm::dvec3 {{ {}, {}, {} }} * scaleFactor, glm::dvec3 {{ {}, {}, {} }}, glm::dvec3 {{ {}, {}, {} }}, {}, false);
|
||||
{plane_id}->setHidden();
|
||||
scene->addSceneObject({plane_id});
|
||||
"#,
|
||||
origin.x.0,
|
||||
origin.y.0,
|
||||
origin.z.0,
|
||||
x_axis.x,
|
||||
x_axis.y,
|
||||
x_axis.z,
|
||||
y_axis.x,
|
||||
y_axis.y,
|
||||
y_axis.z,
|
||||
size.0
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::StartPath(kcmc::StartPath { .. }) => {
|
||||
let sketch_id = format!("sketch_{}", cpp_id);
|
||||
let path_id = format!("path_{}", cpp_id);
|
||||
format!(
|
||||
r#"
|
||||
auto {sketch_id} = make_shared<Object>("sketch", glm::dvec3 {{ 0, 0, 0 }});
|
||||
{sketch_id}->setUUID(Utils::UUID("{cmd_id}"));
|
||||
{sketch_id}->makePath(true);
|
||||
auto {path_id} = {sketch_id}->get<Model::Brep::Path>();
|
||||
scene->addSceneObject({sketch_id});
|
||||
"#
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::MovePathPen(kcmc::MovePathPen { path, to }) => {
|
||||
format!(
|
||||
r#"
|
||||
path_{}->moveTo(glm::dvec3 {{ {}, {}, 0.0 }} * scaleFactor);
|
||||
"#,
|
||||
id_to_cpp(path),
|
||||
to.x.0,
|
||||
to.y.0
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::ExtendPath(kcmc::ExtendPath { path, segment }) => match segment {
|
||||
Line { end, relative } => {
|
||||
format!(
|
||||
r#"
|
||||
path_{}->lineTo(glm::dvec3 {{ {}, {}, 0.0 }} * scaleFactor, {{ {} }});
|
||||
"#,
|
||||
id_to_cpp(path),
|
||||
end.x.0,
|
||||
end.y.0,
|
||||
relative
|
||||
)
|
||||
}
|
||||
PathSegment::Arc {
|
||||
center,
|
||||
radius,
|
||||
start,
|
||||
end,
|
||||
relative,
|
||||
} => {
|
||||
let start = start.value;
|
||||
let end = end.value;
|
||||
let radius = radius.0;
|
||||
|
||||
format!(
|
||||
r#"
|
||||
path_{}->addArc(glm::dvec2 {{ {}, {} }} * scaleFactor, {radius} * scaleFactor, {start}, {end}, {{ {} }});
|
||||
"#,
|
||||
id_to_cpp(path),
|
||||
center.x.0,
|
||||
center.y.0,
|
||||
relative
|
||||
)
|
||||
}
|
||||
PathSegment::TangentialArcTo {
|
||||
angle_snap_increment: _,
|
||||
to,
|
||||
} => {
|
||||
format!(
|
||||
r#"
|
||||
path_{}->tangentialArcTo(glm::dvec3 {{ {}, {}, {} }} * scaleFactor, nullopt, {{ true }});
|
||||
"#,
|
||||
id_to_cpp(path),
|
||||
to.x.0,
|
||||
to.y.0,
|
||||
to.z.0,
|
||||
)
|
||||
}
|
||||
_ => {
|
||||
format!("//{:?}", cmd)
|
||||
}
|
||||
},
|
||||
kcmc::ModelingCmd::ClosePath(kcmc::ClosePath { path_id }) => {
|
||||
format!(
|
||||
r#"
|
||||
path_{}->close();
|
||||
sketch_{}->toSolid2D();
|
||||
"#,
|
||||
uuid_to_cpp(path_id),
|
||||
uuid_to_cpp(path_id)
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::Extrude(kcmc::Extrude {
|
||||
distance,
|
||||
target,
|
||||
faces: _, // Engine team: start using this once the frontend and engine both use it.
|
||||
}) => {
|
||||
format!(
|
||||
r#"
|
||||
scene->getSceneObject(Utils::UUID("{target}"))->extrudeToSolid3D({} * scaleFactor, true);
|
||||
"#,
|
||||
distance.0
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::Revolve(kcmc::Revolve {
|
||||
angle,
|
||||
axis,
|
||||
axis_is_2d,
|
||||
origin,
|
||||
target,
|
||||
tolerance,
|
||||
}) => {
|
||||
let ox = origin.x.0;
|
||||
let oy = origin.y.0;
|
||||
let oz = origin.z.0;
|
||||
let ax = axis.x;
|
||||
let ay = axis.y;
|
||||
let az = axis.z;
|
||||
let angle = angle.value;
|
||||
let tolerance = tolerance.0;
|
||||
format!(
|
||||
r#"
|
||||
scene->getSceneObject(Utils::UUID("{target}"))->revolveToSolid3D(nullopt, glm::dvec3 {{ {ox}, {oy}, {oz} }} * scaleFactor, glm::dvec3 {{ {ax}, {ay}, {az} }}, {axis_is_2d}, {angle}, {tolerance});
|
||||
"#
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::Solid2dAddHole(kcmc::Solid2dAddHole { hole_id, object_id }) => {
|
||||
format!(
|
||||
r#"scene->getSceneObject(Utils::UUID("{object_id}"))->get<Model::Brep::Solid2D>()->addHole(
|
||||
make_shared<Model::Brep::Path>(*scene->getSceneObject(Utils::UUID("{hole_id}"))->get<Model::Brep::Solid2D>()->getPath())
|
||||
);"#
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::Solid3dGetExtrusionFaceInfo(kcmc::Solid3dGetExtrusionFaceInfo {
|
||||
object_id,
|
||||
edge_id,
|
||||
}) => {
|
||||
format!(
|
||||
r#"
|
||||
//face info get {} {}
|
||||
"#,
|
||||
object_id, edge_id
|
||||
)
|
||||
}
|
||||
kcmc::ModelingCmd::EntityCircularPattern(kcmc::EntityCircularPattern {
|
||||
entity_id,
|
||||
axis,
|
||||
center,
|
||||
num_repetitions,
|
||||
arc_degrees,
|
||||
rotate_duplicates,
|
||||
}) => {
|
||||
let entity_ids = generate_repl_uuids(*num_repetitions as usize);
|
||||
|
||||
this_response = OkModelingCmdResponse::EntityCircularPattern(kcmc::output::EntityCircularPattern {
|
||||
entity_ids: entity_ids.clone(),
|
||||
});
|
||||
|
||||
let mut base_code: String = format!(
|
||||
r#"
|
||||
auto reps_{cpp_id} = scene->entityCircularPattern(Utils::UUID("{}"), {num_repetitions}, glm::dvec3 {{ {}, {}, {} }} * scaleFactor, glm::dvec3 {{ {}, {}, {} }} * scaleFactor, {arc_degrees}, {rotate_duplicates});
|
||||
"#,
|
||||
entity_id, axis.x, axis.y, axis.z, center.x.0, center.y.0, center.z.0
|
||||
);
|
||||
|
||||
let repl_uuid_fix_code = codegen_cpp_repl_uuid_setters(&cpp_id, &entity_ids);
|
||||
base_code.push_str(&repl_uuid_fix_code);
|
||||
|
||||
base_code
|
||||
}
|
||||
kcmc::ModelingCmd::EntityLinearPattern(kcmc::EntityLinearPattern {
|
||||
entity_id: _,
|
||||
axis: _,
|
||||
num_repetitions: _,
|
||||
spacing: _,
|
||||
}) => {
|
||||
// let num_transforms = transforms.len();
|
||||
// let num_repetitions = transform.iter().map(|t| if t.replicate { 1 } else { 0 } ).sum();
|
||||
|
||||
// let mut base_code: String = format!(
|
||||
// r#"
|
||||
// std::vector<std::optional<Scene::Scene::LinearPatternTransform>> transforms_{cpp_id}({num_transforms});
|
||||
// "#);
|
||||
|
||||
// for t in transform {
|
||||
// translations_xyz.push(t.translate.x.to_millimeters(state.units));
|
||||
// translations_xyz.push(t.translate.y.to_millimeters(state.units));
|
||||
// translations_xyz.push(t.translate.z.to_millimeters(state.units));
|
||||
// scale_xyz.push(t.scale.x);
|
||||
// scale_xyz.push(t.scale.y);
|
||||
// scale_xyz.push(t.scale.z);
|
||||
// }
|
||||
|
||||
// let entity_ids = generate_repl_uuids(*num_repetitions as usize);
|
||||
|
||||
// this_response = OkModelingCmdResponse::EntityLinearPattern {
|
||||
// data: kittycad::types::EntityLinearPattern {
|
||||
// entity_ids: entity_ids.clone(),
|
||||
// },
|
||||
// };
|
||||
|
||||
// let mut base_code: String = format!(
|
||||
// r#"
|
||||
// auto reps_{cpp_id} = scene->entityCircularPattern(Utils::UUID("{}"), {num_repetitions}, glm::dvec3 {{ {}, {}, {} }} * scaleFactor, glm::dvec3 {{ {}, {}, {} }} * scaleFactor, {arc_degrees}, {rotate_duplicates});
|
||||
// "#,
|
||||
// entity_id, axis.x, axis.y, axis.z, center.x, center.y, center.z
|
||||
// );
|
||||
|
||||
// let repl_uuid_fix_code = codegen_cpp_repl_uuid_setters(&cpp_id, &entity_ids);
|
||||
// base_code.push_str(&repl_uuid_fix_code);
|
||||
|
||||
// base_code
|
||||
format!("//{:?}", cmd)
|
||||
}
|
||||
_ => {
|
||||
//helps us follow along with the currently unhandled engine commands
|
||||
format!("//{:?}", cmd)
|
||||
}
|
||||
};
|
||||
|
||||
(new_code, this_response)
|
||||
}
|
||||
}
|
||||
|
||||
fn id_to_cpp(id: &ModelingCmdId) -> String {
|
||||
uuid_to_cpp(&id.0)
|
||||
}
|
||||
|
||||
fn uuid_to_cpp(id: &uuid::Uuid) -> String {
|
||||
let str = format!("{}", id);
|
||||
str::replace(&str, "-", "_")
|
||||
}
|
||||
|
||||
fn generate_repl_uuids(count: usize) -> Vec<uuid::Uuid> {
|
||||
let mut repl_ids: Vec<uuid::Uuid> = Vec::new();
|
||||
|
||||
repl_ids.resize_with(count, uuid::Uuid::new_v4);
|
||||
repl_ids
|
||||
}
|
||||
|
||||
fn codegen_cpp_repl_uuid_setters(reps_id: &str, entity_ids: &[uuid::Uuid]) -> String {
|
||||
let mut codegen = String::new();
|
||||
|
||||
for (i, id) in entity_ids.iter().enumerate() {
|
||||
let cpp_id = uuid_to_cpp(id);
|
||||
let iter = format!(
|
||||
r#"
|
||||
//change object id -> {id}
|
||||
auto repl_{cpp_id} = scene->getSceneObject(reps_{reps_id}[{i}]);
|
||||
scene->removeSceneObject(repl_{cpp_id}->getUUID(), false);
|
||||
repl_{cpp_id}->setUUID(Utils::UUID("{id}"));
|
||||
scene->addSceneObject(repl_{cpp_id});
|
||||
"#
|
||||
);
|
||||
codegen.push_str(&iter);
|
||||
}
|
||||
|
||||
codegen
|
||||
}
|
||||
|
||||
#[async_trait::async_trait]
|
||||
impl kcl_lib::EngineManager for EngineConnection {
|
||||
fn batch(&self) -> Arc<RwLock<Vec<(WebSocketRequest, kcl_lib::SourceRange)>>> {
|
||||
self.batch.clone()
|
||||
}
|
||||
|
||||
fn batch_end(&self) -> Arc<RwLock<IndexMap<uuid::Uuid, (WebSocketRequest, kcl_lib::SourceRange)>>> {
|
||||
self.batch_end.clone()
|
||||
}
|
||||
|
||||
fn responses(&self) -> Arc<RwLock<IndexMap<Uuid, WebSocketResponse>>> {
|
||||
Arc::new(RwLock::new(IndexMap::new()))
|
||||
}
|
||||
|
||||
fn artifact_commands(&self) -> Arc<RwLock<Vec<ArtifactCommand>>> {
|
||||
Arc::new(RwLock::new(Vec::new()))
|
||||
}
|
||||
|
||||
async fn execution_kind(&self) -> ExecutionKind {
|
||||
let guard = self.execution_kind.read().await;
|
||||
*guard
|
||||
}
|
||||
|
||||
async fn replace_execution_kind(&self, execution_kind: ExecutionKind) -> ExecutionKind {
|
||||
let mut guard = self.execution_kind.write().await;
|
||||
let original = *guard;
|
||||
*guard = execution_kind;
|
||||
original
|
||||
}
|
||||
|
||||
async fn default_planes(
|
||||
&self,
|
||||
id_generator: &mut IdGenerator,
|
||||
source_range: kcl_lib::SourceRange,
|
||||
) -> Result<DefaultPlanes, KclError> {
|
||||
if NEED_PLANES {
|
||||
{
|
||||
let opt = self.default_planes.read().await.as_ref().cloned();
|
||||
if let Some(planes) = opt {
|
||||
return Ok(planes);
|
||||
}
|
||||
} // drop the read lock
|
||||
|
||||
let new_planes = self.new_default_planes(id_generator, source_range).await?;
|
||||
*self.default_planes.write().await = Some(new_planes.clone());
|
||||
|
||||
Ok(new_planes)
|
||||
} else {
|
||||
Ok(DefaultPlanes::default())
|
||||
}
|
||||
}
|
||||
|
||||
async fn clear_scene_post_hook(
|
||||
&self,
|
||||
_id_generator: &mut IdGenerator,
|
||||
_source_range: kcl_lib::SourceRange,
|
||||
) -> Result<(), KclError> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn inner_send_modeling_cmd(
|
||||
&self,
|
||||
id: uuid::Uuid,
|
||||
_source_range: kcl_lib::SourceRange,
|
||||
cmd: WebSocketRequest,
|
||||
_id_to_source_range: HashMap<uuid::Uuid, kcl_lib::SourceRange>,
|
||||
) -> Result<WebSocketResponse, KclError> {
|
||||
match cmd {
|
||||
WebSocketRequest::ModelingCmdBatchReq(ModelingBatch {
|
||||
ref requests,
|
||||
batch_id: _,
|
||||
responses: _,
|
||||
}) => {
|
||||
let mut responses = HashMap::new();
|
||||
for request in requests {
|
||||
let (new_code, this_response) = self.handle_command(&request.cmd_id, &request.cmd);
|
||||
|
||||
if !new_code.is_empty() {
|
||||
let new_code = new_code
|
||||
.trim()
|
||||
.split(' ')
|
||||
.filter(|s| !s.is_empty())
|
||||
.collect::<Vec<_>>()
|
||||
.join(" ")
|
||||
+ "\n";
|
||||
//println!("{new_code}");
|
||||
self.core_test.write().await.push_str(&new_code);
|
||||
}
|
||||
|
||||
responses.insert(
|
||||
request.cmd_id,
|
||||
kcmc::websocket::BatchResponse::Success {
|
||||
response: this_response,
|
||||
},
|
||||
);
|
||||
}
|
||||
Ok(WebSocketResponse::Success(kcmc::websocket::SuccessWebSocketResponse {
|
||||
success: true,
|
||||
request_id: Some(id),
|
||||
resp: OkWebSocketResponseData::ModelingBatch { responses },
|
||||
}))
|
||||
}
|
||||
WebSocketRequest::ModelingCmdReq(ModelingCmdReq { cmd, cmd_id }) => {
|
||||
//also handle unbatched requests inline
|
||||
let (new_code, this_response) = self.handle_command(&cmd_id, &cmd);
|
||||
|
||||
if !new_code.is_empty() {
|
||||
let new_code = new_code
|
||||
.trim()
|
||||
.split(' ')
|
||||
.filter(|s| !s.is_empty())
|
||||
.collect::<Vec<_>>()
|
||||
.join(" ")
|
||||
+ "\n";
|
||||
//println!("{new_code}");
|
||||
self.core_test.write().await.push_str(&new_code);
|
||||
}
|
||||
|
||||
Ok(WebSocketResponse::Success(kcmc::websocket::SuccessWebSocketResponse {
|
||||
success: true,
|
||||
request_id: Some(id),
|
||||
resp: OkWebSocketResponseData::Modeling {
|
||||
modeling_response: this_response,
|
||||
},
|
||||
}))
|
||||
}
|
||||
_ => Ok(WebSocketResponse::Success(kcmc::websocket::SuccessWebSocketResponse {
|
||||
success: true,
|
||||
request_id: Some(id),
|
||||
resp: OkWebSocketResponseData::Modeling {
|
||||
modeling_response: OkModelingCmdResponse::Empty {},
|
||||
},
|
||||
})),
|
||||
}
|
||||
}
|
||||
|
||||
async fn close(&self) {}
|
||||
}
|
||||
24
rust/kcl-to-core/src/lib.rs
Normal file
24
rust/kcl-to-core/src/lib.rs
Normal file
@ -0,0 +1,24 @@
|
||||
use std::sync::Arc;
|
||||
|
||||
use anyhow::Result;
|
||||
use kcl_lib::{ExecState, ExecutorContext};
|
||||
use tokio::sync::RwLock;
|
||||
|
||||
#[cfg(not(target_arch = "wasm32"))]
|
||||
mod conn_mock_core;
|
||||
|
||||
///Converts the given kcl code to an engine test
|
||||
pub async fn kcl_to_engine_core(code: &str) -> Result<String> {
|
||||
let program = kcl_lib::Program::parse_no_errs(code)?;
|
||||
|
||||
let result = Arc::new(RwLock::new("".into()));
|
||||
let ref_result = Arc::clone(&result);
|
||||
|
||||
let ctx = ExecutorContext::new_forwarded_mock(Arc::new(Box::new(
|
||||
crate::conn_mock_core::EngineConnection::new(ref_result).await?,
|
||||
)));
|
||||
ctx.run(&program, &mut ExecState::new(&ctx.settings)).await?;
|
||||
|
||||
let result = result.read().await.clone();
|
||||
Ok(result)
|
||||
}
|
||||
20
rust/kcl-to-core/src/tool.rs
Normal file
20
rust/kcl-to-core/src/tool.rs
Normal file
@ -0,0 +1,20 @@
|
||||
use std::{env, fs};
|
||||
|
||||
use kcl_to_core::*;
|
||||
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
let args: Vec<String> = env::args().collect();
|
||||
|
||||
if args.len() < 2 {
|
||||
println!("Usage: kcl-to-core path/to/file.kcl");
|
||||
return;
|
||||
}
|
||||
|
||||
let file_path = &args[1];
|
||||
let kcl = fs::read_to_string(file_path).expect("read file");
|
||||
|
||||
let result = kcl_to_engine_core(&kcl).await.expect("kcl conversion");
|
||||
|
||||
println!("{}", result);
|
||||
}
|
||||
Reference in New Issue
Block a user