use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
req_option::RequestOption,
validate_required,
};
use serde::{Deserialize, Serialize};
use crate::ccm::wiki::v2::models::WikiSpaceMember;
use crate::common::{api_endpoints::WikiApiV2, api_utils::*};
pub struct CreateWikiSpaceMemberRequest {
space_id: String,
need_notification: Option<bool>,
config: Config,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CreateWikiSpaceMemberParams {
pub member_type: String,
pub member_id: String,
pub member_role: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CreateWikiSpaceMemberResponse {
pub member: Option<WikiSpaceMember>,
}
impl ApiResponseTrait for CreateWikiSpaceMemberResponse {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl CreateWikiSpaceMemberRequest {
pub fn new(config: Config) -> Self {
Self {
space_id: String::new(),
need_notification: None,
config,
}
}
pub fn space_id(mut self, space_id: impl Into<String>) -> Self {
self.space_id = space_id.into();
self
}
pub fn need_notification(mut self, need_notification: bool) -> Self {
self.need_notification = Some(need_notification);
self
}
pub async fn execute(
self,
params: CreateWikiSpaceMemberParams,
) -> SDKResult<CreateWikiSpaceMemberResponse> {
self.execute_with_options(params, RequestOption::default())
.await
}
pub async fn execute_with_options(
self,
params: CreateWikiSpaceMemberParams,
option: RequestOption,
) -> SDKResult<CreateWikiSpaceMemberResponse> {
validate_required!(self.space_id, "知识空间ID不能为空");
validate_required!(params.member_type, "成员类型不能为空");
validate_required!(params.member_id, "成员ID不能为空");
validate_required!(params.member_role, "成员角色不能为空");
let api_endpoint = WikiApiV2::SpaceMemberCreate(self.space_id.clone());
let mut api_request: ApiRequest<CreateWikiSpaceMemberResponse> = api_endpoint
.to_request()
.body(serialize_params(¶ms, "添加知识空间成员")?);
if let Some(need_notification) = self.need_notification {
api_request = api_request.query("need_notification", &need_notification.to_string());
}
Transport::request_typed(api_request, &self.config, Some(option), "添加知识空间成员").await
}
}
#[cfg(test)]
mod tests {
use super::*;
use serde_json::json;
use wiremock::MockServer;
use wiremock::matchers::{method, path};
use wiremock::{Mock, ResponseTemplate};
#[tokio::test]
async fn test_create_wiki_space_member_returns_data_on_success() {
let server = MockServer::start().await;
Mock::given(method("POST"))
.and(path("/open-apis/wiki/v2/spaces/sp001/members"))
.respond_with(ResponseTemplate::new(200).set_body_json(json!({
"code": 0, "msg": "success", "data": {}
})))
.mount(&server)
.await;
let config = Config::builder()
.app_id("ci_app_id")
.app_secret("ci_app_secret")
.base_url(server.uri())
.enable_token_cache(false)
.build();
let resp = CreateWikiSpaceMemberRequest::new(config)
.space_id("sp001")
.execute(CreateWikiSpaceMemberParams {
member_type: "openid".into(),
member_id: "ou001".into(),
member_role: "member".into(),
})
.await
.expect("添加知识空间成员应成功");
assert!(resp.member.is_none());
let received = server.received_requests().await.unwrap_or_default();
assert_eq!(received.len(), 1);
assert_eq!(
received[0].url.path(),
"/open-apis/wiki/v2/spaces/sp001/members"
);
}
}