use std::collections::HashMap;
use axum::{
Json,
extract::Path,
};
use dal::{
ChangeSet,
ChangeSetId,
Component,
ComponentId,
WorkspacePk,
WsEvent,
diagram::view::{
View,
ViewId,
},
};
use serde::{
Deserialize,
Serialize,
};
use si_frontend_types::{
RawGeometry,
StringGeometry,
};
use ulid::Ulid;
use super::ViewResult;
use crate::{
extract::HandlerContext,
service::{
force_change_set_response::ForceChangeSetResponse,
v2::AccessBuilder,
},
};
#[derive(Deserialize, Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct SetComponentGeometryRequest {
pub data_by_component_id: HashMap<ComponentId, StringGeometry>,
pub client_ulid: Ulid,
pub request_ulid: Ulid,
}
#[derive(Deserialize, Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct Response {
pub request_ulid: Ulid,
}
pub async fn set_component_geometry(
HandlerContext(builder): HandlerContext,
AccessBuilder(access_builder): AccessBuilder,
Path((_workspace_pk, change_set_id, view_id)): Path<(WorkspacePk, ChangeSetId, ViewId)>,
Json(request): Json<SetComponentGeometryRequest>,
) -> ViewResult<ForceChangeSetResponse<Response>> {
let mut ctx = builder
.build(access_builder.build(change_set_id.into()))
.await?;
let force_change_set_id = ChangeSet::force_new(&mut ctx).await?;
let mut geometry_list = vec![];
for (id, string_geometry) in request.data_by_component_id {
let new_geometry: RawGeometry = string_geometry.try_into()?;
let mut component = Component::get_by_id(&ctx, id).await?;
let current_geometry = component.geometry(&ctx, view_id).await?;
let new_geometry_cache = new_geometry.clone();
let (width, height) = (
new_geometry.width.or_else(|| current_geometry.width()),
new_geometry.height.or_else(|| current_geometry.height()),
);
component
.set_geometry(
&ctx,
view_id,
new_geometry_cache.x,
new_geometry_cache.y,
width,
height,
)
.await?;
geometry_list.push((
id.into(),
RawGeometry {
x: new_geometry.x,
y: new_geometry.y,
width,
height,
},
))
}
WsEvent::set_component_position(
&ctx,
ctx.change_set_id(),
view_id,
geometry_list,
Some(request.client_ulid),
)
.await?
.publish_on_commit(&ctx)
.await?;
ctx.commit().await?;
Ok(ForceChangeSetResponse::new(
force_change_set_id,
Response {
request_ulid: request.request_ulid,
},
))
}
#[derive(Deserialize, Serialize, Debug)]
#[serde(rename_all = "camelCase")]
pub struct SetViewObjectGeometryRequest {
pub data_by_view_id: HashMap<ViewId, StringGeometry>,
pub client_ulid: Ulid,
pub request_ulid: Ulid,
}
pub async fn set_view_object_geometry(
HandlerContext(builder): HandlerContext,
AccessBuilder(access_builder): AccessBuilder,
Path((_workspace_pk, change_set_id, container_view_id)): Path<(
WorkspacePk,
ChangeSetId,
ViewId,
)>,
Json(request): Json<SetViewObjectGeometryRequest>,
) -> ViewResult<ForceChangeSetResponse<Response>> {
let mut ctx = builder
.build(access_builder.build(change_set_id.into()))
.await?;
let force_change_set_id = ChangeSet::force_new(&mut ctx).await?;
let mut geometry_list = vec![];
for (object_view_id, string_geometry) in request.data_by_view_id {
let new_geometry: RawGeometry = string_geometry.try_into()?;
let mut view = View::get_by_id(&ctx, object_view_id).await?;
let current_geometry = view.geometry(&ctx, container_view_id).await?;
let new_geometry_cache = new_geometry.clone();
let (width, height) = (
new_geometry.width.or_else(|| current_geometry.width()),
new_geometry.height.or_else(|| current_geometry.height()),
);
view.set_geometry(
&ctx,
container_view_id,
new_geometry_cache.x,
new_geometry_cache.y,
width,
height,
)
.await?;
geometry_list.push((
object_view_id.into(),
RawGeometry {
x: new_geometry.x,
y: new_geometry.y,
width,
height,
},
))
}
WsEvent::set_component_position(
&ctx,
ctx.change_set_id(),
container_view_id,
geometry_list,
Some(request.client_ulid),
)
.await?
.publish_on_commit(&ctx)
.await?;
ctx.commit().await?;
Ok(ForceChangeSetResponse::new(
force_change_set_id,
Response {
request_ulid: request.request_ulid,
},
))
}
