kcl_lib/std/

clone.rs

//! Standard library clone.

use std::collections::HashMap;

use kcmc::ModelingCmd;
use kcmc::each_cmd as mcmd;
use kcmc::ok_response::OkModelingCmdResponse;
use kcmc::websocket::OkWebSocketResponseData;
use kittycad_modeling_cmds::{self as kcmc};

use super::extrude::do_post_extrude;
use crate::errors::KclError;
use crate::errors::KclErrorDetails;
use crate::execution::CreatorFace;
use crate::execution::EntityCloneInfo;
use crate::execution::ExecState;
use crate::execution::ExtrudeSurface;
use crate::execution::Geometry;
use crate::execution::GeometryWithImportedGeometry;
use crate::execution::KclValue;
use crate::execution::Metadata;
use crate::execution::ModelingCmdMeta;
use crate::execution::Sketch;
use crate::execution::Solid;
use crate::execution::SolidCreator;
use crate::execution::TagEngineInfo;
use crate::execution::TagIdentifier;
use crate::execution::types::ArrayLen;
use crate::execution::types::PrimitiveType;
use crate::execution::types::RuntimeType;
use crate::parsing::ast::types::TagNode;
use crate::std::Args;
use crate::std::extrude::BeingExtruded;
use crate::std::extrude::NamedCapTags;

type Result<T> = std::result::Result<T, KclError>;

/// Clone a sketch or solid.
///
/// This works essentially like a copy-paste operation.
pub async fn clone(exec_state: &mut ExecState, args: Args) -> Result<KclValue> {
    let geometries = args.get_unlabeled_kw_arg(
        "geometry",
        &RuntimeType::Array(
            Box::new(RuntimeType::Union(vec![
                RuntimeType::Primitive(PrimitiveType::Sketch),
                RuntimeType::Primitive(PrimitiveType::Solid),
                RuntimeType::imported(),
            ])),
            ArrayLen::Minimum(1),
        ),
        exec_state,
    )?;

    let cloned = inner_clone(geometries, exec_state, args).await?;
    Ok(cloned.into())
}

async fn inner_clone(
    geometries: Vec<GeometryWithImportedGeometry>,
    exec_state: &mut ExecState,
    args: Args,
) -> Result<Vec<GeometryWithImportedGeometry>> {
    let mut res = vec![];

    for g in geometries {
        let new_id = exec_state.next_uuid();
        let mut geometry = g.clone();
        let old_id = geometry.id(&args.ctx).await?;
        // Pattern copies have a new top-level entity ID, but their KCL
        // geometry still describes the source topology. Map that source
        // topology directly to the clone so paths and tagged faces receive
        // the clone's child IDs.
        let source_topology_id = match &geometry {
            GeometryWithImportedGeometry::Sketch(sketch) => sketch.original_id,
            GeometryWithImportedGeometry::Solid(solid) => solid.topology_id(),
            GeometryWithImportedGeometry::ImportedGeometry(_) => old_id,
        };
        let mut entity_clone_info = None;

        let mut new_geometry = match &geometry {
            GeometryWithImportedGeometry::ImportedGeometry(imported) => {
                let mut new_imported = imported.clone();
                new_imported.id = new_id;
                GeometryWithImportedGeometry::ImportedGeometry(new_imported)
            }
            GeometryWithImportedGeometry::Sketch(sketch) => {
                let mut new_sketch = sketch.clone();
                new_sketch.id = new_id;
                new_sketch.original_id = new_id;
                new_sketch.artifact_id = new_id.into();
                GeometryWithImportedGeometry::Sketch(new_sketch)
            }
            GeometryWithImportedGeometry::Solid(solid) => {
                // We flush before the clone so all the shit exists.
                exec_state
                    .flush_batch_for_solids(
                        ModelingCmdMeta::from_args(exec_state, &args),
                        std::slice::from_ref(solid),
                    )
                    .await?;

                let mut new_solid = solid.clone();
                // Sweep-backed solids have separate engine entity and body
                // artifact IDs. Preserve that split so the cloned Path and
                // Sweep can coexist. Pattern copies replace `artifact_id`
                // with their own entity ID, so consult their retained source
                // artifact to recover the same distinction.
                let source_artifact_id = solid.pattern_source_artifact_id.unwrap_or(solid.artifact_id);
                let result_artifact_id = if source_artifact_id == source_topology_id.into() {
                    new_id.into()
                } else {
                    exec_state.next_artifact_id()
                };
                entity_clone_info = Some(EntityCloneInfo {
                    source_artifact_id,
                    result_artifact_id,
                    source_topology_id: source_topology_id.into(),
                });
                new_solid.id = new_id;
                new_solid.value_id = new_id;
                new_solid.become_new_body(new_id, result_artifact_id);
                if let Some(sketch) = new_solid.sketch_mut() {
                    sketch.original_id = new_id;
                }
                GeometryWithImportedGeometry::Solid(new_solid)
            }
        };

        if args.ctx.no_engine_commands().await {
            res.push(new_geometry);
        } else {
            exec_state
                .batch_modeling_cmd_with_entity_clone_info(
                    ModelingCmdMeta::from_args_id(exec_state, &args, new_id),
                    ModelingCmd::from(mcmd::EntityClone::builder().entity_id(old_id).build()),
                    entity_clone_info,
                )
                .await?;

            fix_tags_and_references(&mut new_geometry, old_id, source_topology_id, exec_state, &args).await?;
            res.push(new_geometry)
        }
    }

    Ok(res)
}
/// Fix the tags and references of the cloned geometry.
pub(super) async fn fix_tags_and_references(
    new_geometry: &mut GeometryWithImportedGeometry,
    old_geometry_id: uuid::Uuid,
    source_topology_id: uuid::Uuid,
    exec_state: &mut ExecState,
    args: &Args,
) -> Result<()> {
    let new_geometry_id = new_geometry.id(&args.ctx).await?;
    let entity_id_map =
        get_old_new_child_map(new_geometry_id, old_geometry_id, source_topology_id, exec_state, args).await?;

    // Fix the path references in the new geometry.
    match new_geometry {
        GeometryWithImportedGeometry::ImportedGeometry(_) => {}
        GeometryWithImportedGeometry::Sketch(sketch) => {
            sketch.clone = Some(source_topology_id);
            fix_sketch_tags_and_references(sketch, &entity_id_map, exec_state, None).await?;
        }
        GeometryWithImportedGeometry::Solid(solid) => {
            let body_type = match solid.best_guess_body_type {
                Some(body_type) => body_type,
                None => super::surfaces::query_body_type(solid, exec_state, args).await?,
            };
            solid.best_guess_body_type = Some(body_type);

            let (start_tag, end_tag) = get_named_cap_tags(solid);
            let solid_value = solid.value.clone();
            let solid_artifact_id = solid.artifact_id;
            let old_face_tag_names = solid.faces.keys().cloned().collect::<Vec<_>>();
            let face_creator = match &solid.creator {
                SolidCreator::Face(face) => Some((
                    remap_id(face.face_id, &entity_id_map),
                    remap_id(face.solid_id, &entity_id_map),
                )),
                _ => None,
            };

            if solid.sketch().is_none() {
                remap_edge_cuts(solid, &entity_id_map);
                remap_sketchless_solid(solid, &entity_id_map);
                solid.faces.clear();
                restore_face_tags(solid, &old_face_tag_names, exec_state);
                return Ok(());
            }

            // Make the sketch id the new geometry id.
            let sketch = solid.sketch_mut().ok_or_else(|| {
                KclError::new_internal(KclErrorDetails::new(
                    "A sketch-backed clone lost its creator sketch during metadata reconstruction.".to_owned(),
                    vec![args.source_range],
                ))
            })?;
            sketch.id = new_geometry_id;
            sketch.original_id = new_geometry_id;
            sketch.artifact_id = new_geometry_id.into();
            sketch.clone = Some(source_topology_id);

            fix_sketch_tags_and_references(sketch, &entity_id_map, exec_state, Some(solid_value)).await?;
            let sketch_for_post = sketch.clone();

            // Do the after extrude things to update those ids, based on the new sketch
            // information.
            let mut new_solid = do_post_extrude(
                &sketch_for_post,
                solid_artifact_id,
                solid.sectional,
                &NamedCapTags {
                    start: start_tag.as_ref(),
                    end: end_tag.as_ref(),
                },
                kittycad_modeling_cmds::shared::ExtrudeMethod::New,
                exec_state,
                args,
                None,
                Some(&entity_id_map.clone()),
                body_type,
                BeingExtruded::Sketch,
            )
            .await?;

            if let Some((face_id, solid_id)) = face_creator {
                let rebuilt_sketch = new_solid.sketch().cloned().ok_or_else(|| {
                    KclError::new_internal(KclErrorDetails::new(
                        "A face-created clone lost its creator sketch during metadata reconstruction.".to_owned(),
                        vec![args.source_range],
                    ))
                })?;
                new_solid.creator = SolidCreator::Face(CreatorFace {
                    face_id,
                    solid_id,
                    sketch: rebuilt_sketch,
                });
            }

            restore_sketch_tag_surfaces(&mut new_solid);
            *solid = new_solid;

            restore_face_tags(solid, &old_face_tag_names, exec_state);
        }
    }

    Ok(())
}

fn remap_id(id: uuid::Uuid, entity_id_map: &HashMap<uuid::Uuid, uuid::Uuid>) -> uuid::Uuid {
    entity_id_map.get(&id).copied().unwrap_or(id)
}

fn remap_edge_cuts(solid: &mut Solid, entity_id_map: &HashMap<uuid::Uuid, uuid::Uuid>) {
    for edge_cut in &mut solid.edge_cuts {
        edge_cut.set_id(remap_id(edge_cut.id(), entity_id_map));
        edge_cut.set_edge_id(remap_id(edge_cut.edge_id(), entity_id_map));
    }
    for id in &mut solid.pending_edge_cut_ids {
        *id = remap_id(*id, entity_id_map);
    }
}

fn remap_sketchless_solid(solid: &mut Solid, entity_id_map: &HashMap<uuid::Uuid, uuid::Uuid>) {
    for surface in &mut solid.value {
        surface.set_id(remap_id(surface.get_id(), entity_id_map));
        surface.set_face_id(remap_id(surface.face_id(), entity_id_map));
    }

    solid.start_cap_id = solid.start_cap_id.map(|id| remap_id(id, entity_id_map));
    solid.end_cap_id = solid.end_cap_id.map(|id| remap_id(id, entity_id_map));

    if let SolidCreator::Edge(creator) = &mut solid.creator {
        creator.edge_id = remap_id(creator.edge_id, entity_id_map);
        creator.body_id = remap_id(creator.body_id, entity_id_map);
    }
}

/// Restore any sketch tag surfaces that could not be mapped from stale source
/// metadata before [`do_post_extrude`] rebuilt the cloned solid's surfaces.
fn restore_sketch_tag_surfaces(solid: &mut Solid) {
    let surfaces_by_tag = solid
        .value
        .iter()
        .filter_map(|surface| surface.get_tag().map(|tag| (tag.name.clone(), surface.clone())))
        .collect::<HashMap<_, _>>();
    let Some(sketch) = solid.sketch_mut() else {
        return;
    };

    for (name, tag) in &mut sketch.tags {
        let Some(surface) = surfaces_by_tag.get(name) else {
            continue;
        };
        let Some((_, info)) = tag.info.last_mut() else {
            continue;
        };
        if info.surface.is_none() {
            info.surface = Some(surface.clone());
        }
    }
}

/// Rebuild the face tag map of a cloned solid from its new surfaces.
///
/// [`do_post_extrude`] leaves `faces` empty, and we can't reuse the sketch's
/// tags like tagging at creation time does, because the cloned sketch still
/// carries the original solid's face info. Build fresh tag identifiers from
/// the new surfaces, which have the clone's face ids.
fn restore_face_tags(solid: &mut Solid, face_tag_names: &[String], exec_state: &ExecState) {
    let surfaces = solid.value.clone();
    for surface in surfaces {
        let Some(tag) = surface.get_tag() else {
            continue;
        };
        if !face_tag_names.iter().any(|tag_name| tag_name == &tag.name) {
            continue;
        }

        let mut solid_copy = solid.clone();
        if let Some(sketch) = solid_copy.sketch_mut() {
            // Avoid recursive tags.
            sketch.tags.clear();
        }
        solid_copy.faces.clear();

        let tag_id = TagIdentifier {
            value: tag.name.clone(),
            info: vec![(
                exec_state.stack().current_epoch(),
                TagEngineInfo {
                    id: surface.get_id(),
                    surface: Some(surface.clone()),
                    path: None,
                    geometry: Geometry::Solid(solid_copy),
                },
            )],
            meta: vec![Metadata {
                source_range: tag.clone().into(),
            }],
        };

        match solid.faces.get_mut(&tag.name) {
            Some(existing_tag) => existing_tag.merge_info(&tag_id),
            None => {
                solid.faces.insert(tag.name.clone(), tag_id);
            }
        }
    }

    for name in face_tag_names {
        if !solid.faces.contains_key(name) {
            crate::log::logln!("Failed to find new face for face tag: {name:?}");
        }
    }
}

async fn get_old_new_child_map(
    new_geometry_id: uuid::Uuid,
    old_geometry_id: uuid::Uuid,
    source_topology_id: uuid::Uuid,
    exec_state: &mut ExecState,
    args: &Args,
) -> Result<HashMap<uuid::Uuid, uuid::Uuid>> {
    // Artifact graph ID management expects the cloned entity's own children
    // to be queried first. Pattern copies retain the source topology in KCL,
    // though, so use that topology for the runtime old-to-new ID map.
    if old_geometry_id != source_topology_id {
        get_all_child_uuids(old_geometry_id, exec_state, args).await?;
    }

    // Get the old geometries entity ids.
    let old_entity_ids = get_all_child_uuids(source_topology_id, exec_state, args).await?;

    // Get the new geometries entity ids.
    let new_entity_ids = get_all_child_uuids(new_geometry_id, exec_state, args).await?;

    // Create a map of old entity ids to new entity ids.
    let mut entity_id_map = HashMap::from_iter(
        old_entity_ids
            .iter()
            .zip(new_entity_ids.iter())
            .map(|(old_id, new_id)| (*old_id, *new_id)),
    );
    entity_id_map.insert(old_geometry_id, new_geometry_id);
    entity_id_map.insert(source_topology_id, new_geometry_id);
    Ok(entity_id_map)
}

async fn get_all_child_uuids(
    geometry_id: uuid::Uuid,
    exec_state: &mut ExecState,
    args: &Args,
) -> Result<Vec<uuid::Uuid>> {
    let response = exec_state
        .send_modeling_cmd(
            ModelingCmdMeta::from_args(exec_state, args),
            ModelingCmd::from(mcmd::EntityGetAllChildUuids::builder().entity_id(geometry_id).build()),
        )
        .await?;
    let OkWebSocketResponseData::Modeling {
        modeling_response: OkModelingCmdResponse::EntityGetAllChildUuids(resp),
    } = response
    else {
        return Err(KclError::new_engine(KclErrorDetails::new(
            format!("EntityGetAllChildUuids response was not as expected: {response:?}"),
            vec![args.source_range],
        )));
    };
    Ok(resp.entity_ids)
}

/// Fix the tags and references of a sketch.
async fn fix_sketch_tags_and_references(
    new_sketch: &mut Sketch,
    entity_id_map: &HashMap<uuid::Uuid, uuid::Uuid>,
    exec_state: &mut ExecState,
    surfaces: Option<Vec<ExtrudeSurface>>,
) -> Result<()> {
    // Fix the path references in the sketch.
    for path in new_sketch.paths.as_mut_slice() {
        if let Some(new_path_id) = entity_id_map.get(&path.get_id()) {
            path.set_id(*new_path_id);
        } else {
            // We log on these because we might have already flushed and the id is no longer
            // relevant since filleted or something.
            crate::log::logln!("Failed to find new path id for old path id: {:?}", path.get_id());
        }
    }

    // Map the surface tags to the new surface ids.
    let mut surface_id_map: HashMap<String, &ExtrudeSurface> = HashMap::new();
    let surfaces = surfaces.unwrap_or_default();
    for surface in surfaces.iter() {
        if let Some(tag) = surface.get_tag() {
            surface_id_map.insert(tag.name.clone(), surface);
        }
    }

    // Fix the tags
    // This is annoying, in order to fix the tags we need to iterate over the paths again, but not
    // mutable borrow the paths.
    for path in new_sketch.paths.clone() {
        // Check if this path has a tag.
        if let Some(tag) = path.get_tag() {
            let mut surface = None;
            if let Some(found_surface) = surface_id_map.get(&tag.name) {
                let mut new_surface = (*found_surface).clone();
                if let Some(new_face_id) = entity_id_map.get(&new_surface.face_id()).copied() {
                    new_surface.set_face_id(new_face_id);
                    surface = Some(new_surface);
                } else {
                    // A boolean can retain a tagged path while replacing or
                    // removing its old face. `do_post_extrude` queries the
                    // live topology and rebuilds this optional surface data.
                    crate::log::logln!(
                        "Failed to find new face id for stale old face id: {:?}",
                        new_surface.face_id()
                    );
                }
            }

            new_sketch.add_tag(&tag, &path, exec_state, surface.as_ref());
        }
    }

    // Fix the base path.
    if let Some(new_base_path) = entity_id_map.get(&new_sketch.start.geo_meta.id) {
        new_sketch.start.geo_meta.id = *new_base_path;
    } else {
        crate::log::logln!(
            "Failed to find new base path id for old base path id: {:?}",
            new_sketch.start.geo_meta.id
        );
    }

    Ok(())
}

// Return the named cap tags for the original solid.
fn get_named_cap_tags(solid: &Solid) -> (Option<TagNode>, Option<TagNode>) {
    let mut start_tag = None;
    let mut end_tag = None;
    // Check the start cap.
    if let Some(start_cap_id) = solid.start_cap_id {
        // Check if we had a value for that cap.
        for value in &solid.value {
            if value.get_id() == start_cap_id {
                start_tag = value.get_tag();
                break;
            }
        }
    }

    // Check the end cap.
    if let Some(end_cap_id) = solid.end_cap_id {
        // Check if we had a value for that cap.
        for value in &solid.value {
            if value.get_id() == end_cap_id {
                end_tag = value.get_tag();
                break;
            }
        }
    }

    (start_tag, end_tag)
}

#[cfg(test)]
mod tests {
    use kcl_api::artifact::SweepSubType;
    use kittycad_modeling_cmds::shared::BodyType;
    use pretty_assertions::assert_eq;
    use pretty_assertions::assert_ne;

    use crate::exec::KclValueView;
    use crate::execution::Artifact;
    use crate::execution::ArtifactGraph;
    use crate::execution::ArtifactId;
    use crate::execution::Solid;
    use crate::execution::SolidCreator;

    fn assert_cloned_composite_topology(artifact_graph: &ArtifactGraph, cloned_composite: &Solid) {
        let Some(Artifact::CompositeSolid(cloned_artifact)) = artifact_graph.get(&cloned_composite.artifact_id) else {
            panic!("Expected a cloned composite solid artifact at the engine entity ID");
        };
        assert_eq!(cloned_artifact.id, cloned_composite.artifact_id);
        assert!(!cloned_artifact.consumed);

        let cloned_face_sweep_ids = artifact_graph
            .values()
            .filter_map(|artifact| match artifact {
                Artifact::Wall(wall) if wall.cmd_id == cloned_composite.id => Some(wall.sweep_id),
                Artifact::Cap(cap) if cap.cmd_id == cloned_composite.id => Some(cap.sweep_id),
                _ => None,
            })
            .collect::<Vec<_>>();
        assert!(!cloned_face_sweep_ids.is_empty());

        for sweep_id in cloned_face_sweep_ids {
            let Some(Artifact::Sweep(sweep)) = artifact_graph.get(&sweep_id) else {
                panic!("Expected every cloned composite face to reference a sweep");
            };
            assert_eq!(sweep.code_ref, cloned_artifact.code_ref);
            let source_sweep_id = sweep.source_sweep_id.expect("Expected cloned sweep provenance");
            assert_ne!(sweep.id, source_sweep_id);
            assert!(matches!(artifact_graph.get(&source_sweep_id), Some(Artifact::Sweep(_))));
        }
    }

    // Ensure the clone function returns a sketch with different ids for all the internal paths and
    // the resulting sketch.
    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_sketch() {
        let code = r#"cube = startSketchOn(XY)
    |> startProfile(at = [0,0])
    |> line(end = [0, 10])
    |> line(end = [10, 0])
    |> line(end = [0, -10])
    |> close()

clonedCube = clone(cube)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        // Execute the program.
        let result = ctx.run_with_caching(program.clone()).await.unwrap();
        let cube = result.variables.get("cube").unwrap();
        let cloned_cube = result.variables.get("clonedCube").unwrap();

        assert_ne!(cube, cloned_cube);

        let KclValueView::Sketch { value: cube } = cube else {
            panic!("Expected a sketch, got: {cube:?}");
        };
        let KclValueView::Sketch { value: cloned_cube } = cloned_cube else {
            panic!("Expected a sketch, got: {cloned_cube:?}");
        };

        assert_ne!(cube.id, cloned_cube.id);
        assert_ne!(cube.original_id, cloned_cube.original_id);
        assert_ne!(cube.artifact_id, cloned_cube.artifact_id);

        assert_eq!(cloned_cube.artifact_id, cloned_cube.id.into());
        assert_eq!(cloned_cube.original_id, cloned_cube.id);

        for (path, cloned_path) in cube.paths.iter().zip(cloned_cube.paths.iter()) {
            assert_ne!(path.get_id(), cloned_path.get_id());
            assert_eq!(path.get_tag(), cloned_path.get_tag());
        }

        assert_eq!(cube.tags.len(), 0);
        assert_eq!(cloned_cube.tags.len(), 0);

        ctx.close().await;
    }

    // Ensure the clone function returns a solid with different ids for all the internal paths and
    // references.
    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_solid() {
        let code = r#"cube = startSketchOn(XY)
    |> startProfile(at = [0,0])
    |> line(end = [0, 10])
    |> line(end = [10, 0])
    |> line(end = [0, -10])
    |> close()
    |> extrude(length = 5)

clonedCube = clone(cube)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        // Execute the program.
        let result = ctx.run_with_caching(program.clone()).await.unwrap();
        let cube = result.variables.get("cube").unwrap();
        let cloned_cube = result.variables.get("clonedCube").unwrap();

        assert_ne!(cube, cloned_cube);

        let KclValueView::Solid { value: cube } = cube else {
            panic!("Expected a solid, got: {cube:?}");
        };
        let KclValueView::Solid { value: cloned_cube } = cloned_cube else {
            panic!("Expected a solid, got: {cloned_cube:?}");
        };
        let cube_sketch = cube.sketch().expect("Expected cube to have a sketch");
        let cloned_cube_sketch = cloned_cube.sketch().expect("Expected cloned cube to have a sketch");

        assert_ne!(cube.id, cloned_cube.id);
        assert_ne!(cube_sketch.id, cloned_cube_sketch.id);
        assert_ne!(cube_sketch.original_id, cloned_cube_sketch.original_id);
        assert_ne!(cube.artifact_id, cloned_cube.artifact_id);
        assert_ne!(cube_sketch.artifact_id, cloned_cube_sketch.artifact_id);

        assert_ne!(cloned_cube.artifact_id, cloned_cube.id.into());

        for (path, cloned_path) in cube_sketch.paths.iter().zip(cloned_cube_sketch.paths.iter()) {
            assert_ne!(path.get_id(), cloned_path.get_id());
            assert_eq!(path.get_tag(), cloned_path.get_tag());
        }

        for (value, cloned_value) in cube.value.iter().zip(cloned_cube.value.iter()) {
            assert_ne!(value.get_id(), cloned_value.get_id());
            assert_eq!(value.get_tag(), cloned_value.get_tag());
        }

        assert_eq!(cube_sketch.tags.len(), 0);
        assert_eq!(cloned_cube_sketch.tags.len(), 0);

        assert_eq!(cube.edge_cuts.len(), 0);
        assert_eq!(cloned_cube.edge_cuts.len(), 0);

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_sketch_backed_surface_operations() {
        let code = r#"extrudeProfile = startSketchOn(XZ)
  |> startProfile(at = [0, 0])
  |> line(end = [4, 0])
  |> line(end = [2, 3])
surfaceExtrude = extrude(extrudeProfile, length = 2, bodyType = SURFACE)
surfaceExtrudeClone = clone(surfaceExtrude)

revolveProfile = startSketchOn(XZ)
  |> startProfile(at = [5, 0])
  |> line(end = [2, 3])
surfaceRevolve = revolve(revolveProfile, axis = Y, angle = 180deg, bodyType = SURFACE)
surfaceRevolveClone = clone(surfaceRevolve)

sweepProfile = startSketchOn(XY)
  |> startProfile(at = [-10, 10])
  |> line(end = [4, 0])
sweepPath = startSketchOn(XY)
  |> startProfile(at = [0, 0])
  |> line(end = [10, 0])
  |> tangentialArc(end = [4, -4])
surfaceSweep = sweep(sweepProfile, path = sweepPath, bodyType = SURFACE)
surfaceSweepClone = clone(surfaceSweep)

loftProfileA = startSketchOn(offsetPlane(XZ, offset = -10))
  |> startProfile(at = [-2, -2])
  |> line(end = [4, 0])
loftProfileB = startSketchOn(offsetPlane(XZ, offset = -15))
  |> startProfile(at = [-1, -1])
  |> line(end = [2, 0])
surfaceLoft = loft([loftProfileA, loftProfileB], bodyType = SURFACE)
surfaceLoftClone = clone(surfaceLoft)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        for (source_name, clone_name) in [
            ("surfaceExtrude", "surfaceExtrudeClone"),
            ("surfaceRevolve", "surfaceRevolveClone"),
            ("surfaceSweep", "surfaceSweepClone"),
            ("surfaceLoft", "surfaceLoftClone"),
        ] {
            let KclValueView::Solid { value: source } = result.variables.get(source_name).unwrap() else {
                panic!("Expected {source_name} to be a surface body");
            };
            let KclValueView::Solid { value: cloned } = result.variables.get(clone_name).unwrap() else {
                panic!("Expected {clone_name} to be a surface body");
            };

            assert_ne!(source.id, cloned.id);
            assert_eq!(cloned.best_guess_body_type, Some(BodyType::Surface));
            assert!(matches!(cloned.creator, SolidCreator::Sketch(_)));
            assert!(!cloned.value.is_empty());
        }

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_edge_created_surface() {
        let code = r#"@settings(kclVersion = 2.0)

baseSketch = sketch(on = XY) {
  bottom = line(start = [var 0mm, var 0mm], end = [var 10mm, var 0mm])
  right = line(start = [var 10mm, var 0mm], end = [var 10mm, var 10mm])
  top = line(start = [var 10mm, var 10mm], end = [var 0mm, var 10mm])
  left = line(start = [var 0mm, var 10mm], end = [var 0mm, var 0mm])
  coincident([bottom.end, right.start])
  coincident([right.end, top.start])
  coincident([top.end, left.start])
  coincident([left.end, bottom.start])
}
base = extrude(region(segments = [baseSketch.bottom, baseSketch.right]), length = 5mm)
source = extrude(
  getOppositeEdge(base.sketch.tags.bottom),
  length = 3mm,
  bodyType = SURFACE,
  method = NEW,
)
cloned = clone(source)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: source } = result.variables.get("source").unwrap() else {
            panic!("Expected an edge-created source surface");
        };
        let KclValueView::Solid { value: cloned } = result.variables.get("cloned").unwrap() else {
            panic!("Expected a cloned edge-created surface");
        };
        let SolidCreator::Edge(source_creator) = &source.creator else {
            panic!("Expected the source surface to retain its edge creator");
        };
        let SolidCreator::Edge(cloned_creator) = &cloned.creator else {
            panic!("Expected the cloned surface to retain its edge creator");
        };

        assert_ne!(source.id, cloned.id);
        assert_eq!(cloned.best_guess_body_type, Some(BodyType::Surface));
        assert_eq!(cloned_creator.body_id, cloned.id);
        assert_ne!(source_creator.edge_id, cloned_creator.edge_id);
        assert_ne!(source.value[0].get_id(), cloned.value[0].get_id());
        assert_ne!(source.value[0].face_id(), cloned.value[0].face_id());

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_preserves_face_creator() {
        let code = r#"profile = startSketchOn(XY)
  |> startProfile(at = [0, 0])
  |> yLine(length = 1, tag = $a)
  |> xLine(length = 1, tag = $b)
  |> close(tag = $c)
base = extrude(profile, length = 1)
source = extrude(c, length = 4, method = NEW)
cloned = clone(source)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: source } = result.variables.get("source").unwrap() else {
            panic!("Expected a face-created source body");
        };
        let KclValueView::Solid { value: cloned } = result.variables.get("cloned").unwrap() else {
            panic!("Expected a cloned face-created body");
        };
        let SolidCreator::Face(source_creator) = &source.creator else {
            panic!("Expected the source body to retain its face creator");
        };
        let SolidCreator::Face(cloned_creator) = &cloned.creator else {
            panic!("Expected the cloned body to retain its face creator");
        };

        assert_ne!(source.id, cloned.id);
        assert_ne!(source_creator.face_id, cloned_creator.face_id);
        assert_ne!(source_creator.solid_id, cloned_creator.solid_id);
        assert_ne!(source_creator.sketch.id, cloned_creator.sketch.id);
        assert_ne!(source_creator.sketch.original_id, cloned_creator.sketch.original_id);

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_procedural_blend_surface() {
        let code = r#"@settings(defaultLengthUnit = mm, kclVersion = 2.0)

sketchA = sketch(on = YZ) {
  line1 = line(start = [var 4.1mm, var -0.1mm], end = [var 5.5mm, var 0mm])
  line2 = line(start = [var 5.5mm, var 0mm], end = [var 5.5mm, var 3mm])
  line3 = line(start = [var 5.5mm, var 3mm], end = [var 3.9mm, var 2.8mm])
  line4 = line(start = [var 4.1mm, var 3mm], end = [var 4.5mm, var -0.2mm])
  coincident([line1.end, line2.start])
  coincident([line2.end, line3.start])
  coincident([line3.end, line4.start])
  coincident([line4.end, line1.start])
}

sketchB = sketch(on = -XZ) {
  line5 = line(start = [var -5.3mm, var -0.1mm], end = [var -3.5mm, var -0.1mm])
  line6 = line(start = [var -3.5mm, var -0.1mm], end = [var -3.5mm, var 3.1mm])
  line7 = line(start = [var -3.5mm, var 4.5mm], end = [var -5.4mm, var 4.5mm])
  line8 = line(start = [var -5.3mm, var 3.1mm], end = [var -5.3mm, var -0.1mm])
  coincident([line5.end, line6.start])
  coincident([line6.end, line7.start])
  coincident([line7.end, line8.start])
  coincident([line8.end, line5.start])
}

surfaceA = extrude(region(segments = [sketchB.line5, sketchB.line6]), length = -2mm, bodyType = SURFACE)
surfaceB = extrude(region(segments = [sketchA.line1, sketchA.line2]), length = -2mm, bodyType = SURFACE)
bridge = blend([surfaceA.sketch.tags.line7, surfaceB.sketch.tags.line3])
bridgeClone = clone(bridge)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: bridge } = result.variables.get("bridge").unwrap() else {
            panic!("Expected blend to create a procedural surface");
        };
        let KclValueView::Solid { value: cloned } = result.variables.get("bridgeClone").unwrap() else {
            panic!("Expected blend clone to create a procedural surface");
        };

        assert_ne!(bridge.id, cloned.id);
        assert_eq!(cloned.best_guess_body_type, Some(BodyType::Surface));
        assert!(matches!(cloned.creator, SolidCreator::Procedural));
        let Some(Artifact::Sweep(cloned_sweep)) = result.artifact_graph.get(&cloned.artifact_id) else {
            panic!("Expected the blend clone to have a sweep artifact");
        };
        assert_eq!(cloned_sweep.sub_type, SweepSubType::Blend);
        assert_eq!(cloned_sweep.source_sweep_id, Some(bridge.artifact_id));

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_multi_body_join_queries_body_type() {
        let code = r#"@settings(kclVersion = 2.0)

targetSketch = sketch(on = XY) {
  bottom = line(start = [var -10, var -10], end = [var 10, var -10])
  right = line(start = [var 10, var -10], end = [var 10, var 10])
  top = line(start = [var 10, var 10], end = [var -10, var 10])
  left = line(start = [var -10, var 10], end = [var -10, var -10])
  coincident([bottom.end, right.start])
  coincident([right.end, top.start])
  coincident([top.end, left.start])
  coincident([left.end, bottom.start])
}
target = extrude(region(point = [0, 0], sketch = targetSketch), length = 10)

cutterSketch = sketch(on = XY) {
  bottom = line(start = [var -1, var -12], end = [var 1, var -12])
  right = line(start = [var 1, var -12], end = [var 1, var 12])
  top = line(start = [var 1, var 12], end = [var -1, var 12])
  left = line(start = [var -1, var 12], end = [var -1, var -12])
  coincident([bottom.end, right.start])
  coincident([right.end, top.start])
  coincident([top.end, left.start])
  coincident([left.end, bottom.start])
}
cutter = extrude(region(point = [0, 0], sketch = cutterSketch), length = 10)

pieces = split([target], tools = [cutter], keepTools = true)
joined = joinSurfaces(pieces)
joinedClone = clone(joined)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: joined } = result.variables.get("joined").unwrap() else {
            panic!("Expected joinSurfaces to create a procedural body");
        };
        let KclValueView::Solid { value: cloned } = result.variables.get("joinedClone").unwrap() else {
            panic!("Expected joinSurfaces body to clone");
        };

        assert_ne!(joined.id, cloned.id);
        assert!(joined.best_guess_body_type.is_none());
        assert!(cloned.best_guess_body_type.is_some());
        assert!(matches!(cloned.creator, SolidCreator::Procedural));

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_loft() {
        let code = r#"@settings(kclVersion = 2.0)

firstSketch = sketch(on = XY) {
  circle1 = circle(start = [var 10, var 0], center = [var 0, var 0])
}
secondSketch = sketch(on = offsetPlane(XY, offset = 10)) {
  circle1 = circle(start = [var 5, var 0], center = [var 0, var 0])
}

lofted = loft([
  region(segments = [firstSketch.circle1]),
  region(segments = [secondSketch.circle1]),
])
clonedLoft = clone(lofted)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: lofted } = result.variables.get("lofted").unwrap() else {
            panic!("Expected a solid loft");
        };
        let KclValueView::Solid { value: cloned_loft } = result.variables.get("clonedLoft").unwrap() else {
            panic!("Expected a cloned solid loft");
        };

        assert_eq!(lofted.topology_id(), lofted.id);
        assert_eq!(lofted.original_id(), lofted.id);
        assert_eq!(lofted.artifact_id, lofted.id.into());

        assert_ne!(lofted.id, cloned_loft.id);
        assert_ne!(lofted.artifact_id, cloned_loft.artifact_id);
        assert_eq!(cloned_loft.topology_id(), cloned_loft.id);
        assert_eq!(cloned_loft.original_id(), cloned_loft.id);
        assert_eq!(cloned_loft.artifact_id, cloned_loft.id.into());

        let loft_sketch = lofted.sketch().expect("Expected loft to retain its base sketch");
        let cloned_sketch = cloned_loft
            .sketch()
            .expect("Expected cloned loft to retain its base sketch");
        for (path, cloned_path) in loft_sketch.paths.iter().zip(cloned_sketch.paths.iter()) {
            assert_ne!(path.get_id(), cloned_path.get_id());
            assert_eq!(path.get_tag(), cloned_path.get_tag());
        }

        assert!(!cloned_loft.value.is_empty());
        for (surface, cloned_surface) in lofted.value.iter().zip(cloned_loft.value.iter()) {
            assert_ne!(surface.get_id(), cloned_surface.get_id());
            assert_eq!(surface.get_tag(), cloned_surface.get_tag());
        }

        let Some(Artifact::Sweep(source_sweep)) = result.artifact_graph.get(&lofted.artifact_id) else {
            panic!("Expected the source loft to be represented by a sweep artifact");
        };
        assert_eq!(source_sweep.sub_type, SweepSubType::Loft);

        let Some(Artifact::Sweep(cloned_sweep)) = result.artifact_graph.get(&cloned_loft.artifact_id) else {
            panic!("Expected the cloned loft to be represented by a sweep artifact");
        };
        assert_eq!(cloned_sweep.sub_type, SweepSubType::Loft);
        assert_eq!(cloned_sweep.source_sweep_id, Some(lofted.artifact_id));
        assert_eq!(cloned_sweep.path_id, source_sweep.path_id);
        assert!(!cloned_sweep.consumed);

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_composite_solid_keeps_engine_artifact_id() {
        let code = r#"left = startSketchOn(XY)
    |> startProfile(at = [0, 0])
    |> line(end = [10, 0])
    |> line(end = [0, 10])
    |> line(end = [-10, 0])
    |> close()
    |> extrude(length = 5)

right = startSketchOn(XY)
    |> startProfile(at = [5, 0])
    |> line(end = [10, 0])
    |> line(end = [0, 10])
    |> line(end = [-10, 0])
    |> close()
    |> extrude(length = 5)

composite = union([left, right])
clonedComposite = clone(composite)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: composite } = result.variables.get("composite").unwrap() else {
            panic!("Expected composite to be a solid");
        };
        let KclValueView::Solid {
            value: cloned_composite,
        } = result.variables.get("clonedComposite").unwrap()
        else {
            panic!("Expected clonedComposite to be a solid");
        };

        assert_eq!(composite.artifact_id, composite.id.into());
        assert_eq!(cloned_composite.artifact_id, cloned_composite.id.into());
        assert_ne!(composite.id, cloned_composite.id);
        assert_ne!(composite.original_id(), composite.id);
        assert_eq!(composite.topology_id(), composite.id);
        assert_eq!(cloned_composite.original_id(), cloned_composite.id);
        assert_eq!(cloned_composite.topology_id(), cloned_composite.id);

        assert_cloned_composite_topology(&result.artifact_graph, cloned_composite);

        ctx.close().await;
    }

    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_imported_patterned_composite_uses_composite_topology() {
        let module_code = r#"left = startSketchOn(XY)
    |> startProfile(at = [0, 0])
    |> line(end = [10, 0])
    |> line(end = [0, 10])
    |> line(end = [-10, 0])
    |> close()
    |> extrude(length = 5)

right = startSketchOn(XY)
    |> startProfile(at = [5, 0])
    |> line(end = [10, 0])
    |> line(end = [0, 10])
    |> line(end = [-10, 0])
    |> close()
    |> extrude(length = 5)

export composite = union([left, right])
"#;
        let code = r#"import composite from 'composite.kcl'

patterned = patternLinear3d(
    composite,
    instances = 2,
    distance = 20,
    axis = [1, 0, 0],
)
patternCopy = patterned[1]
clonedCopy = clone(patternCopy)
"#;
        let tmpdir = tempfile::TempDir::with_prefix("clone_imported_patterned_composite").unwrap();
        let main_path = tmpdir.path().join("main.kcl");
        std::fs::write(tmpdir.path().join("composite.kcl"), module_code).unwrap();
        std::fs::write(&main_path, code).unwrap();

        let ctx = crate::test_server::new_context(true, Some(main_path)).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let KclValueView::Solid { value: composite } = result.variables.get("composite").unwrap() else {
            panic!("Expected composite to be a solid");
        };
        let KclValueView::Solid { value: pattern_copy } = result.variables.get("patternCopy").unwrap() else {
            panic!("Expected patternCopy to be a solid");
        };
        let KclValueView::Solid { value: cloned_copy } = result.variables.get("clonedCopy").unwrap() else {
            panic!("Expected clonedCopy to be a solid");
        };

        assert_eq!(composite.topology_id(), composite.id);
        assert_eq!(pattern_copy.topology_id(), composite.id);
        assert_eq!(cloned_copy.original_id(), cloned_copy.id);
        assert_eq!(cloned_copy.topology_id(), cloned_copy.id);
        assert_eq!(cloned_copy.artifact_id, cloned_copy.id.into());
        assert_ne!(pattern_copy.id, cloned_copy.id);
        assert!(result.artifact_graph.get(&pattern_copy.artifact_id).is_none());
        assert_cloned_composite_topology(&result.artifact_graph, cloned_copy);

        ctx.close().await;
    }

    // Ensure the clone function returns a sketch with different ids for all the internal paths and
    // the resulting sketch.
    // AND TAGS.
    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_sketch_with_tags() {
        let code = r#"cube = startSketchOn(XY)
    |> startProfile(at = [0,0]) // tag this one
    |> line(end = [0, 10], tag = $tag02)
    |> line(end = [10, 0], tag = $tag03)
    |> line(end = [0, -10], tag = $tag04)
    |> close(tag = $tag05)

clonedCube = clone(cube)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        // Execute the program.
        let result = ctx.run_with_caching(program.clone()).await.unwrap();
        let cube = result.variables.get("cube").unwrap();
        let cloned_cube = result.variables.get("clonedCube").unwrap();

        assert_ne!(cube, cloned_cube);

        let KclValueView::Sketch { value: cube } = cube else {
            panic!("Expected a sketch, got: {cube:?}");
        };
        let KclValueView::Sketch { value: cloned_cube } = cloned_cube else {
            panic!("Expected a sketch, got: {cloned_cube:?}");
        };

        assert_ne!(cube.id, cloned_cube.id);
        assert_ne!(cube.original_id, cloned_cube.original_id);

        for (path, cloned_path) in cube.paths.iter().zip(cloned_cube.paths.iter()) {
            assert_ne!(path.get_id(), cloned_path.get_id());
            assert_eq!(path.get_tag(), cloned_path.get_tag());
        }

        for (tag_name, tag) in &cube.tags {
            let cloned_tag = cloned_cube.tags.get(tag_name).unwrap();

            let tag_info = tag.get_cur_info().unwrap();
            let cloned_tag_info = cloned_tag.get_cur_info().unwrap();

            assert_ne!(tag_info.id, cloned_tag_info.id);
            assert_ne!(tag_info.geometry.id(), cloned_tag_info.geometry.id());
            assert_ne!(tag_info.path, cloned_tag_info.path);
            assert_eq!(tag_info.surface, None);
            assert_eq!(cloned_tag_info.surface, None);
        }

        ctx.close().await;
    }

    // Ensure the clone function returns a solid with different ids for all the internal paths and
    // references.
    // WITH TAGS.
    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_solid_with_tags() {
        let code = r#"cube = startSketchOn(XY)
    |> startProfile(at = [0,0]) // tag this one
    |> line(end = [0, 10], tag = $tag02)
    |> line(end = [10, 0], tag = $tag03)
    |> line(end = [0, -10], tag = $tag04)
    |> close(tag = $tag05)
    |> extrude(length = 5, tagEnd = $endCap)

clonedCube = clone(cube)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        // Execute the program.
        let result = ctx.run_with_caching(program.clone()).await.unwrap();
        let cube = result.variables.get("cube").unwrap();
        let cloned_cube = result.variables.get("clonedCube").unwrap();

        assert_ne!(cube, cloned_cube);

        let KclValueView::Solid { value: cube } = cube else {
            panic!("Expected a solid, got: {cube:?}");
        };
        let KclValueView::Solid { value: cloned_cube } = cloned_cube else {
            panic!("Expected a solid, got: {cloned_cube:?}");
        };
        let cube_sketch = cube.sketch().expect("Expected cube to have a sketch");
        let cloned_cube_sketch = cloned_cube.sketch().expect("Expected cloned cube to have a sketch");

        assert_ne!(cube.id, cloned_cube.id);
        assert_ne!(cube_sketch.id, cloned_cube_sketch.id);
        assert_ne!(cube_sketch.original_id, cloned_cube_sketch.original_id);
        assert_ne!(cube.artifact_id, cloned_cube.artifact_id);
        assert_ne!(cube_sketch.artifact_id, cloned_cube_sketch.artifact_id);

        assert_ne!(cloned_cube.artifact_id, cloned_cube.id.into());

        for (path, cloned_path) in cube_sketch.paths.iter().zip(cloned_cube_sketch.paths.iter()) {
            assert_ne!(path.get_id(), cloned_path.get_id());
            assert_eq!(path.get_tag(), cloned_path.get_tag());
        }

        for (value, cloned_value) in cube.value.iter().zip(cloned_cube.value.iter()) {
            assert_ne!(value.get_id(), cloned_value.get_id());
            assert_eq!(value.get_tag(), cloned_value.get_tag());
        }

        for (tag_name, tag) in &cube_sketch.tags {
            let cloned_tag = cloned_cube_sketch.tags.get(tag_name).unwrap();

            let tag_info = tag.get_cur_info().unwrap();
            let cloned_tag_info = cloned_tag.get_cur_info().unwrap();

            assert_ne!(tag_info.id, cloned_tag_info.id);
            assert_ne!(tag_info.geometry.id(), cloned_tag_info.geometry.id());
            assert_eq!(tag_info.path.is_some(), cloned_tag_info.path.is_some());
            if let (Some(path), Some(cloned_path)) = (&tag_info.path, &cloned_tag_info.path) {
                assert_ne!(path, cloned_path);
            }
            assert_eq!(tag_info.surface.is_some(), cloned_tag_info.surface.is_some());
            if let (Some(surface), Some(cloned_surface)) = (&tag_info.surface, &cloned_tag_info.surface) {
                assert_ne!(surface, cloned_surface);
            }
        }

        for (tag_name, tag) in &cube.faces {
            let cloned_tag = cloned_cube.faces.get(tag_name).unwrap();

            let tag_info = tag.get_cur_info().unwrap();
            let cloned_tag_info = cloned_tag.get_cur_info().unwrap();

            assert_ne!(tag_info.id, cloned_tag_info.id);
            assert_ne!(tag_info.geometry.id(), cloned_tag_info.geometry.id());
            assert_ne!(tag_info.surface, cloned_tag_info.surface);
        }
        assert!(cube.faces.contains_key("endCap"));
        assert!(cloned_cube.faces.contains_key("endCap"));

        assert_eq!(cube.edge_cuts.len(), 0);
        assert_eq!(cloned_cube.edge_cuts.len(), 0);

        ctx.close().await;
    }

    // Pattern copies retain the source topology in program memory. Cloning a
    // copy must map that source topology directly onto the clone so both wall
    // and cap tags refer to the clone's faces.
    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_pattern_copy_with_face_tags() {
        let code = r#"source = startSketchOn(XY)
    |> startProfile(at = [0, 0])
    |> line(end = [0, 10], tag = $wall)
    |> line(end = [10, 0])
    |> line(end = [0, -10])
    |> close()
    |> extrude(length = 5, tagEnd = $endCap)

patterned = patternLinear3d(
    source,
    instances = 2,
    distance = 15,
    axis = [1, 0, 0],
)
patternCopy = patterned[1]
clonedCopy = clone(patternCopy)
"#;
        let program = crate::Program::parse_no_errs(code).unwrap();
        let ctx = crate::test_server::new_context(true, None).await.unwrap();

        let result = ctx.run_with_caching(program).await.unwrap();
        let source = result.variables.get("source").unwrap();
        let pattern_copy = result.variables.get("patternCopy").unwrap();
        let cloned_copy = result.variables.get("clonedCopy").unwrap();

        let KclValueView::Solid { value: source } = source else {
            panic!("Expected a solid, got: {source:?}");
        };
        let KclValueView::Solid { value: pattern_copy } = pattern_copy else {
            panic!("Expected a solid, got: {pattern_copy:?}");
        };
        let KclValueView::Solid { value: cloned_copy } = cloned_copy else {
            panic!("Expected a solid, got: {cloned_copy:?}");
        };
        let pattern_sketch = pattern_copy.sketch().expect("Expected pattern copy to have a sketch");
        let cloned_sketch = cloned_copy.sketch().expect("Expected cloned copy to have a sketch");
        assert_eq!(pattern_copy.original_id(), source.id);
        assert_eq!(cloned_copy.original_id(), cloned_copy.id);
        assert!(result.artifact_graph.get(&pattern_copy.artifact_id).is_none());
        assert_ne!(cloned_copy.artifact_id, cloned_copy.id.into());

        let pattern_wall = pattern_sketch.tags.get("wall").unwrap().get_cur_info().unwrap();
        let cloned_wall = cloned_sketch.tags.get("wall").unwrap().get_cur_info().unwrap();
        assert_ne!(pattern_wall.id, cloned_wall.id);
        assert_ne!(pattern_wall.surface, cloned_wall.surface);
        let cloned_wall_id = ArtifactId::new(
            cloned_wall
                .surface
                .as_ref()
                .expect("Expected cloned wall tag to reference a surface")
                .face_id(),
        );

        let pattern_cap = pattern_copy.faces.get("endCap").unwrap().get_cur_info().unwrap();
        let cloned_cap = cloned_copy.faces.get("endCap").unwrap().get_cur_info().unwrap();
        assert_ne!(pattern_cap.id, cloned_cap.id);
        assert_ne!(pattern_cap.surface, cloned_cap.surface);
        let cloned_cap_id = ArtifactId::new(
            cloned_cap
                .surface
                .as_ref()
                .expect("Expected cloned cap tag to reference a surface")
                .face_id(),
        );

        assert!(matches!(
            result.artifact_graph.get(&cloned_copy.artifact_id),
            Some(Artifact::Sweep(sweep))
                if sweep.path_id == cloned_copy.id.into()
        ));
        assert!(matches!(
            result.artifact_graph.get(&cloned_copy.id.into()),
            Some(Artifact::Path(path))
                if path.sweep_id == Some(cloned_copy.artifact_id)
        ));
        assert!(matches!(
            result.artifact_graph.get(&cloned_wall_id),
            Some(Artifact::Wall(wall)) if wall.sweep_id == cloned_copy.artifact_id
        ));
        assert!(matches!(
            result.artifact_graph.get(&cloned_cap_id),
            Some(Artifact::Cap(cap)) if cap.sweep_id == cloned_copy.artifact_id
        ));

        ctx.close().await;
    }

    // Ensure we can get all paths even on a sketch where we closed it and it was already closed.
    #[tokio::test(flavor = "multi_thread")]
    #[ignore = "this test is not working yet, need to fix the getting of ids if sketch already closed"]
    async fn kcl_test_clone_cube_already_closed_sketch() {
        let code = r#"// Clone a basic solid and move it.

exampleSketch = startSketchOn(XY)
  |> startProfile(at = [0, 0])
  |> line(end = [10, 0])
  |> line(end = [0, 10])
  |> line(end = [-10, 0])
  |> line(end = [0, -10])
  |> close()

cube = extrude(exampleSketch, length = 5)
clonedCube = clone(cube)
    |> translate(
        x = 25.0,
    )"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        // Execute the program.
        let result = ctx.run_with_caching(program.clone()).await.unwrap();
        let cube = result.variables.get("cube").unwrap();
        let cloned_cube = result.variables.get("clonedCube").unwrap();

        assert_ne!(cube, cloned_cube);

        let KclValueView::Solid { value: cube } = cube else {
            panic!("Expected a solid, got: {cube:?}");
        };
        let KclValueView::Solid { value: cloned_cube } = cloned_cube else {
            panic!("Expected a solid, got: {cloned_cube:?}");
        };
        let cube_sketch = cube.sketch().expect("Expected cube to have a sketch");
        let cloned_cube_sketch = cloned_cube.sketch().expect("Expected cloned cube to have a sketch");

        assert_ne!(cube.id, cloned_cube.id);
        assert_ne!(cube_sketch.id, cloned_cube_sketch.id);
        assert_ne!(cube_sketch.original_id, cloned_cube_sketch.original_id);
        assert_ne!(cube.artifact_id, cloned_cube.artifact_id);
        assert_ne!(cube_sketch.artifact_id, cloned_cube_sketch.artifact_id);

        assert_ne!(cloned_cube.artifact_id, cloned_cube.id.into());

        for (path, cloned_path) in cube_sketch.paths.iter().zip(cloned_cube_sketch.paths.iter()) {
            assert_ne!(path.get_id(), cloned_path.get_id());
            assert_eq!(path.get_tag(), cloned_path.get_tag());
        }

        for (value, cloned_value) in cube.value.iter().zip(cloned_cube.value.iter()) {
            assert_ne!(value.get_id(), cloned_value.get_id());
            assert_eq!(value.get_tag(), cloned_value.get_tag());
        }

        for (tag_name, tag) in &cube_sketch.tags {
            let cloned_tag = cloned_cube_sketch.tags.get(tag_name).unwrap();

            let tag_info = tag.get_cur_info().unwrap();
            let cloned_tag_info = cloned_tag.get_cur_info().unwrap();

            assert_ne!(tag_info.id, cloned_tag_info.id);
            assert_ne!(tag_info.geometry.id(), cloned_tag_info.geometry.id());
            assert_ne!(tag_info.path, cloned_tag_info.path);
            assert_ne!(tag_info.surface, cloned_tag_info.surface);
        }

        for (edge_cut, cloned_edge_cut) in cube.edge_cuts.iter().zip(cloned_cube.edge_cuts.iter()) {
            assert_ne!(edge_cut.id(), cloned_edge_cut.id());
            assert_ne!(edge_cut.edge_id(), cloned_edge_cut.edge_id());
            assert_eq!(edge_cut.tag(), cloned_edge_cut.tag());
        }

        ctx.close().await;
    }

    // Ensure the clone function returns a solid with different ids for all the internal paths and
    // references.
    // WITH TAGS AND EDGE CUTS.
    #[tokio::test(flavor = "multi_thread")]
    async fn kcl_test_clone_solid_with_edge_cuts() {
        let code = r#"cube = startSketchOn(XY)
    |> startProfile(at = [0,0]) // tag this one
    |> line(end = [0, 10], tag = $tag02)
    |> line(end = [10, 0], tag = $tag03)
    |> line(end = [0, -10], tag = $tag04)
    |> close(tag = $tag05)
    |> extrude(length = 5) // TODO: Tag these
  |> fillet(
    radius = 2,
    tags = [
      getNextAdjacentEdge(tag02),
    ],
    tag = $fillet01,
  )
  |> fillet(
    radius = 2,
    tags = [
      getNextAdjacentEdge(tag04),
    ],
    tag = $fillet02,
  )
  |> chamfer(
    length = 2,
    tags = [
      getNextAdjacentEdge(tag03),
    ],
    tag = $chamfer01,
  )
  |> chamfer(
    length = 2,
    tags = [
      getNextAdjacentEdge(tag05),
    ],
    tag = $chamfer02,
  )

clonedCube = clone(cube)
"#;
        let ctx = crate::test_server::new_context(true, None).await.unwrap();
        let program = crate::Program::parse_no_errs(code).unwrap();

        // Execute the program.
        let result = ctx.run_with_caching(program.clone()).await.unwrap();
        let cube = result.variables.get("cube").unwrap();
        let cloned_cube = result.variables.get("clonedCube").unwrap();

        assert_ne!(cube, cloned_cube);

        let KclValueView::Solid { value: cube } = cube else {
            panic!("Expected a solid, got: {cube:?}");
        };
        let KclValueView::Solid { value: cloned_cube } = cloned_cube else {
            panic!("Expected a solid, got: {cloned_cube:?}");
        };
        let cube_sketch = cube.sketch().expect("Expected cube to have a sketch");
        let cloned_cube_sketch = cloned_cube.sketch().expect("Expected cloned cube to have a sketch");

        assert_ne!(cube.id, cloned_cube.id);
        assert_ne!(cube_sketch.id, cloned_cube_sketch.id);
        assert_ne!(cube_sketch.original_id, cloned_cube_sketch.original_id);
        assert_ne!(cube.artifact_id, cloned_cube.artifact_id);
        assert_ne!(cube_sketch.artifact_id, cloned_cube_sketch.artifact_id);

        assert_ne!(cloned_cube.artifact_id, cloned_cube.id.into());

        for (value, cloned_value) in cube.value.iter().zip(cloned_cube.value.iter()) {
            assert_ne!(value.get_id(), cloned_value.get_id());
            assert_eq!(value.get_tag(), cloned_value.get_tag());
        }

        for (edge_cut, cloned_edge_cut) in cube.edge_cuts.iter().zip(cloned_cube.edge_cuts.iter()) {
            assert_ne!(edge_cut.id(), cloned_edge_cut.id());
            assert_ne!(edge_cut.edge_id(), cloned_edge_cut.edge_id());
            assert_eq!(edge_cut.tag(), cloned_edge_cut.tag());
        }

        ctx.close().await;
    }
}