use alien_error::AlienError;
use serde::{Deserialize, Serialize};
use crate::{parse_bundle_uri, BundleUri, ErrorData, Result, Sandbox, SandboxCode, Stack};
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
#[cfg_attr(feature = "openapi", derive(utoipa::ToSchema))]
#[cfg_attr(feature = "jsonschema", derive(schemars::JsonSchema))]
#[serde(rename_all = "camelCase")]
pub struct AwsSandboxImportData {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub image_identifier: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub image_arn: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub image_version: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub build_role_arn: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub bundle_uri: Option<String>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub egress_connector_arns: Vec<String>,
#[serde(
default,
skip_serializing_if = "Vec::is_empty",
deserialize_with = "crate::import::data::deserialize_u16_vec_from_numbers_or_strings"
)]
pub preview_ports: Vec<u16>,
#[serde(
default,
skip_serializing_if = "std::ops::Not::not",
deserialize_with = "crate::import::data::deserialize_bool_from_bool_or_string"
)]
pub allow_egress: bool,
}
impl AwsSandboxImportData {
pub fn runtime_built(
stack: &Stack,
sandbox: &Sandbox,
build_role_arn: String,
region: &str,
recorded_connector_arn: Option<&str>,
) -> Result<Self> {
let refuse = |reason: String| {
AlienError::new(ErrorData::OperationNotSupported {
operation: format!("register sandbox '{}'", sandbox.id),
reason,
})
};
let SandboxCode::Image { image } = &sandbox.code else {
return Err(refuse(
"an AWS sandbox is built from a prebuilt s3:// bundle, not from source".to_string(),
));
};
let bundle_uri = match parse_bundle_uri(image).map_err(refuse)? {
BundleUri::Literal(uri) => uri.to_string(),
BundleUri::Regional { before, after } => format!("{before}{region}{after}"),
};
let isolated = crate::sandbox_egress::sandbox_needs_connector(stack, sandbox);
let egress_connector_arns = if isolated {
vec![recorded_connector_arn
.ok_or_else(|| refuse("sandbox has no egress connector recorded yet".to_string()))?
.to_string()]
} else {
Vec::new()
};
Ok(Self {
image_identifier: None,
image_arn: None,
image_version: None,
build_role_arn: Some(build_role_arn),
bundle_uri: Some(bundle_uri),
egress_connector_arns,
preview_ports: sandbox.preview_ports.clone(),
allow_egress: !isolated,
})
}
}