use reqwest::Method;
use serde::{Deserialize, Serialize};
use crate::{
core::{
api_req::ApiRequest,
api_resp::{ApiResponseTrait, BaseResponse, ResponseFormat},
config::Config,
constants::AccessTokenType,
endpoints::{EndpointBuilder, LINGO_DRAFT_CREATE, LINGO_DRAFT_UPDATE},
http::Transport,
req_option::RequestOption,
trait_system::AsyncServiceOperation,
SDKResult,
},
impl_basic_service, impl_service_constructor,
service::lingo::models::{Draft, OuterInfo, RelatedMeta},
};
#[derive(Debug, Clone)]
pub struct DraftService {
pub config: Config,
}
impl DraftService {
pub async fn create_draft(
&self,
request: DraftCreateRequest,
option: Option<RequestOption>,
) -> SDKResult<BaseResponse<DraftCreateResponse>> {
<Self as AsyncServiceOperation<DraftCreateRequest, BaseResponse<DraftCreateResponse>>>::execute_with_observability(
self, "create_draft", || async {
let api_req = ApiRequest {
http_method: Method::POST,
api_path: LINGO_DRAFT_CREATE.to_string(),
supported_access_token_types: vec![AccessTokenType::Tenant, AccessTokenType::User],
body: serde_json::to_vec(&request)?,
..Default::default()
};
Transport::request(api_req, &self.config, option).await
}).await
}
pub async fn update_draft(
&self,
draft_id: &str,
request: DraftUpdateRequest,
option: Option<RequestOption>,
) -> SDKResult<BaseResponse<DraftUpdateResponse>> {
let draft_id = draft_id.to_string(); <Self as AsyncServiceOperation<DraftUpdateRequest, BaseResponse<DraftUpdateResponse>>>::execute_with_observability(
self, "update_draft", move || async move {
let api_req = ApiRequest {
http_method: Method::PUT,
api_path: EndpointBuilder::replace_param(LINGO_DRAFT_UPDATE, "{draft_id}", &draft_id),
supported_access_token_types: vec![AccessTokenType::Tenant, AccessTokenType::User],
body: serde_json::to_vec(&request)?,
..Default::default()
};
Transport::request(api_req, &self.config, option).await
}).await
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DraftCreateRequest {
#[serde(skip_serializing_if = "Option::is_none")]
pub entity_id: Option<String>,
pub main_keys: Vec<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub aliases: Option<Vec<String>>,
pub description: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub classification_id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub outer_info: Option<OuterInfo>,
#[serde(skip_serializing_if = "Option::is_none")]
pub related_meta: Option<RelatedMeta>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct DraftCreateResponse {
pub draft: Draft,
}
impl ApiResponseTrait for DraftCreateResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DraftUpdateRequest {
#[serde(skip_serializing_if = "Option::is_none")]
pub main_keys: Option<Vec<String>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub aliases: Option<Vec<String>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub description: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub classification_id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub outer_info: Option<OuterInfo>,
#[serde(skip_serializing_if = "Option::is_none")]
pub related_meta: Option<RelatedMeta>,
}
#[derive(Debug, Serialize, Deserialize)]
pub struct DraftUpdateResponse {
pub draft: Draft,
}
impl ApiResponseTrait for DraftUpdateResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl_basic_service!(DraftService, "lingo.draft", "v1");
impl_service_constructor!(DraftService);
impl AsyncServiceOperation<DraftCreateRequest, BaseResponse<DraftCreateResponse>> for DraftService {}
impl AsyncServiceOperation<DraftUpdateRequest, BaseResponse<DraftUpdateResponse>> for DraftService {}