use harmos_domain::{MemorySize, schema::Described};
#[harmos::operation(harmos_path = crate, codec = manual, id = "harmos.initialize", response = (), policy = super::POLICY, )]
#[derive(Clone, PartialEq, Debug, Default)]
pub struct Initialize {
pub config_json: Vec<u8>,
pub generation_context: Vec<u8>,
pub assignments: Vec<u8>,
}
#[cfg(feature = "lifecycle")]
pub struct Control {
#[cfg(feature = "lifecycle")]
instance: Box<dyn harmos_sdk::operations::lifecycle::LifecycleInstance>,
}
#[cfg(feature = "lifecycle")]
#[derive(Debug, thiserror::Error)]
pub enum ControlError {
#[error(transparent)]
Dependency(#[from] harmos_sdk::ResourceError),
#[error("plugin lifecycle access failed: {0}")]
Access(#[source] harmos_sdk::operations::access::AccessError),
}
harmos_domain::postcard_product!(Initialize { config_json: Vec<u8>, generation_context: Vec<u8>, assignments: Vec<u8> });
impl Described for Initialize {
const NAME: &'static str = <harmos_domain::model::operations::lifecycle::InitializationSchema as Described>::NAME;
const FIELDS: &'static [harmos_domain::schema::declaration::Field] =
<harmos_domain::model::operations::lifecycle::InitializationSchema as Described>::FIELDS;
const MARKER: &'static [u8] = <harmos_domain::model::operations::lifecycle::InitializationSchema as Described>::MARKER;
}
impl MemorySize for Initialize {
fn memory_size(&self) -> usize {
std::mem::size_of::<Self>()
.saturating_add(self.config_json.capacity())
.saturating_add(self.generation_context.capacity())
.saturating_add(self.assignments.capacity())
}
}
#[cfg(feature = "lifecycle")]
impl harmos_sdk::operations::Carrier for Control {
type Declaration = Initialize;
type Error = super::ControlError;
type StreamEndpoints = ();
async fn bind(context: crate::ExecutionContext) -> Result<Self, Self::Error> {
let access = context.resource::<harmos_sdk::operations::lifecycle::LifecycleAccess>()?;
Ok(Self {
instance: access.0.clone().begin(super::Action::Initialize),
})
}
async fn attempt(
&mut self, context: crate::ExecutionContext, request: &mut Initialize, _: &mut (),
) -> Result<crate::Response<()>, Self::Error> {
let _ = &request;
self.instance
.run(context, &request.config_json, &request.generation_context, &request.assignments)
.await
.map_err(super::ControlError::Access)?;
Ok(().into())
}
async fn cleanup(&mut self) -> Result<(), Self::Error> {
self.instance.cleanup().await.map_err(super::ControlError::Access)
}
}
#[cfg(feature = "lifecycle")]
impl crate::Executable for Initialize {
type Carrier = Control;
}