use std::sync::Arc;
use crate::core::client::{VimClient, Result};
#[derive(Clone)]
pub struct Task {
client: Arc<dyn VimClient>,
mo_id: String,
}
impl Task {
pub fn new(client: Arc<dyn VimClient>, mo_id: &str) -> Self {
Self {
client,
mo_id: mo_id.to_string(),
}
}
pub async fn set_task_description(&self, description: &crate::types::structs::LocalizableMessage) -> Result<()> {
let input = SetTaskDescriptionRequestType {description, };
self.client.invoke_void("", "Task", &self.mo_id, "SetTaskDescription", Some(&input)).await
}
pub async fn update_progress(&self, percent_done: i32) -> Result<()> {
let input = UpdateProgressRequestType {percent_done, };
self.client.invoke_void("", "Task", &self.mo_id, "UpdateProgress", Some(&input)).await
}
pub async fn cancel_task(&self) -> Result<()> {
self.client.invoke_void("", "Task", &self.mo_id, "CancelTask", None).await
}
pub async fn set_custom_value(&self, key: &str, value: &str) -> Result<()> {
let input = SetCustomValueRequestType {key, value, };
self.client.invoke_void("", "Task", &self.mo_id, "setCustomValue", Some(&input)).await
}
pub async fn set_task_state(&self, state: crate::types::enums::TaskInfoStateEnum, result: Option<crate::types::vim_any::VimAny>, fault: Option<&crate::types::structs::MethodFault>) -> Result<()> {
let input = SetTaskStateRequestType {state, result, fault, };
self.client.invoke_void("", "Task", &self.mo_id, "SetTaskState", Some(&input)).await
}
pub async fn available_field(&self) -> Result<Option<Vec<crate::types::structs::CustomFieldDef>>> {
let pv_opt = self.client.fetch_property_raw("", "Task", &self.mo_id, "availableField").await?;
match pv_opt {
Some(pv) => Ok(Some(crate::core::client::extract_property(pv)?)),
None => Ok(None),
}
}
pub async fn info(&self) -> Result<crate::types::structs::TaskInfo> {
let pv_opt = self.client.fetch_property_raw("", "Task", &self.mo_id, "info").await?;
let pv = pv_opt.ok_or_else(|| crate::core::client::VimError::ParseError("property info was empty".to_string()))?;
let result: crate::types::structs::TaskInfo = crate::core::client::extract_property(pv)?;
Ok(result)
}
pub async fn value(&self) -> Result<Option<Vec<Box<dyn crate::types::traits::CustomFieldValueTrait>>>> {
let pv_opt = self.client.fetch_property_raw("", "Task", &self.mo_id, "value").await?;
match pv_opt {
Some(pv) => Ok(Some(crate::core::client::extract_property(pv)?)),
None => Ok(None),
}
}
}
struct SetTaskDescriptionRequestType<'a> {
description: &'a crate::types::structs::LocalizableMessage,
}
impl<'a> miniserde::Serialize for SetTaskDescriptionRequestType<'a> {
fn begin(&self) -> miniserde::ser::Fragment<'_> {
miniserde::ser::Fragment::Map(Box::new(SetTaskDescriptionRequestTypeSer { data: self, seq: 0 }))
}
}
struct SetTaskDescriptionRequestTypeSer<'b, 'a> {
data: &'b SetTaskDescriptionRequestType<'a>,
seq: usize,
}
impl<'b, 'a> miniserde::ser::Map for SetTaskDescriptionRequestTypeSer<'b, 'a> {
fn next(&mut self) -> Option<(std::borrow::Cow<'_, str>, &dyn miniserde::Serialize)> {
let seq = self.seq;
self.seq += 1;
match seq {
0 => return Some((std::borrow::Cow::Borrowed("_typeName"), &"SetTaskDescriptionRequestType")),
1 => return Some((std::borrow::Cow::Borrowed("description"), &self.data.description as &dyn miniserde::Serialize)),
_ => return None,
}
}
}
struct UpdateProgressRequestType {
percent_done: i32,
}
impl miniserde::Serialize for UpdateProgressRequestType {
fn begin(&self) -> miniserde::ser::Fragment<'_> {
miniserde::ser::Fragment::Map(Box::new(UpdateProgressRequestTypeSer { data: self, seq: 0 }))
}
}
struct UpdateProgressRequestTypeSer<'b> {
data: &'b UpdateProgressRequestType,
seq: usize,
}
impl<'b> miniserde::ser::Map for UpdateProgressRequestTypeSer<'b> {
fn next(&mut self) -> Option<(std::borrow::Cow<'_, str>, &dyn miniserde::Serialize)> {
let seq = self.seq;
self.seq += 1;
match seq {
0 => return Some((std::borrow::Cow::Borrowed("_typeName"), &"UpdateProgressRequestType")),
1 => return Some((std::borrow::Cow::Borrowed("percentDone"), &self.data.percent_done as &dyn miniserde::Serialize)),
_ => return None,
}
}
}
struct SetCustomValueRequestType<'a> {
key: &'a str,
value: &'a str,
}
impl<'a> miniserde::Serialize for SetCustomValueRequestType<'a> {
fn begin(&self) -> miniserde::ser::Fragment<'_> {
miniserde::ser::Fragment::Map(Box::new(SetCustomValueRequestTypeSer { data: self, seq: 0 }))
}
}
struct SetCustomValueRequestTypeSer<'b, 'a> {
data: &'b SetCustomValueRequestType<'a>,
seq: usize,
}
impl<'b, 'a> miniserde::ser::Map for SetCustomValueRequestTypeSer<'b, 'a> {
fn next(&mut self) -> Option<(std::borrow::Cow<'_, str>, &dyn miniserde::Serialize)> {
let seq = self.seq;
self.seq += 1;
match seq {
0 => return Some((std::borrow::Cow::Borrowed("_typeName"), &"setCustomValueRequestType")),
1 => return Some((std::borrow::Cow::Borrowed("key"), &self.data.key as &dyn miniserde::Serialize)),
2 => return Some((std::borrow::Cow::Borrowed("value"), &self.data.value as &dyn miniserde::Serialize)),
_ => return None,
}
}
}
struct SetTaskStateRequestType<'a> {
state: crate::types::enums::TaskInfoStateEnum,
result: Option<crate::types::vim_any::VimAny>,
fault: Option<&'a crate::types::structs::MethodFault>,
}
impl<'a> miniserde::Serialize for SetTaskStateRequestType<'a> {
fn begin(&self) -> miniserde::ser::Fragment<'_> {
miniserde::ser::Fragment::Map(Box::new(SetTaskStateRequestTypeSer { data: self, seq: 0 }))
}
}
struct SetTaskStateRequestTypeSer<'b, 'a> {
data: &'b SetTaskStateRequestType<'a>,
seq: usize,
}
impl<'b, 'a> miniserde::ser::Map for SetTaskStateRequestTypeSer<'b, 'a> {
fn next(&mut self) -> Option<(std::borrow::Cow<'_, str>, &dyn miniserde::Serialize)> {
loop {
let seq = self.seq;
self.seq += 1;
match seq {
0 => return Some((std::borrow::Cow::Borrowed("_typeName"), &"SetTaskStateRequestType")),
1 => return Some((std::borrow::Cow::Borrowed("state"), &self.data.state as &dyn miniserde::Serialize)),
2 => {
let Some(ref val) = self.data.result else { continue; };
return Some((std::borrow::Cow::Borrowed("result"), val as &dyn miniserde::Serialize));
}
3 => {
let Some(ref val) = self.data.fault else { continue; };
return Some((std::borrow::Cow::Borrowed("fault"), val as &dyn miniserde::Serialize));
}
_ => return None,
}
}
}
}