use crate::models::authen::UserAccessTokenResponse;
use openlark_core::{
SDKResult,
api::{ApiRequest, ApiResponseTrait, ResponseFormat},
config::Config,
http::Transport,
req_option::RequestOption,
validate_required,
};
use serde::{Deserialize, Serialize};
pub struct OidcRefreshAccessTokenBuilder {
refresh_token: String,
client_id: Option<String>,
client_secret: Option<String>,
grant_type: Option<String>,
config: Config,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq)]
pub struct OidcRefreshAccessTokenResponseData {
pub data: UserAccessTokenResponse,
}
impl ApiResponseTrait for OidcRefreshAccessTokenResponseData {
fn data_format() -> ResponseFormat {
ResponseFormat::Data
}
}
impl OidcRefreshAccessTokenBuilder {
pub fn new(config: Config) -> Self {
Self {
refresh_token: String::new(),
client_id: None,
client_secret: None,
grant_type: Some("refresh_token".to_string()),
config,
}
}
pub fn refresh_token(mut self, refresh_token: impl Into<String>) -> Self {
self.refresh_token = refresh_token.into();
self
}
pub fn client_id(mut self, client_id: impl Into<String>) -> Self {
self.client_id = Some(client_id.into());
self
}
pub fn client_secret(mut self, client_secret: impl Into<String>) -> Self {
self.client_secret = Some(client_secret.into());
self
}
pub fn grant_type(mut self, grant_type: impl Into<String>) -> Self {
self.grant_type = Some(grant_type.into());
self
}
pub async fn execute(self) -> SDKResult<OidcRefreshAccessTokenResponseData> {
self.execute_with_options(RequestOption::default()).await
}
pub async fn execute_with_options(
self,
option: RequestOption,
) -> SDKResult<OidcRefreshAccessTokenResponseData> {
validate_required!(self.refresh_token, "刷新令牌不能为空");
use crate::common::api_endpoints::AuthenApiV1;
let api_endpoint = AuthenApiV1::OidcRefreshAccessToken;
let mut form_data = std::collections::HashMap::new();
form_data.insert("refresh_token".to_string(), self.refresh_token.clone());
if let Some(ref client_id) = self.client_id {
form_data.insert("client_id".to_string(), client_id.clone());
}
if let Some(ref client_secret) = self.client_secret {
form_data.insert("client_secret".to_string(), client_secret.clone());
}
if let Some(ref grant_type) = self.grant_type {
form_data.insert("grant_type".to_string(), grant_type.clone());
}
let api_request: ApiRequest<OidcRefreshAccessTokenResponseData> =
ApiRequest::post(api_endpoint.path())
.header("Content-Type", "application/x-www-form-urlencoded")
.body(openlark_core::api::RequestData::Form(form_data));
let response = Transport::request(api_request, &self.config, Some(option)).await?;
response.data.ok_or_else(|| {
openlark_core::error::validation_error("刷新 OIDC user_access_token", "响应数据为空")
})
}
}
#[cfg(test)]
#[allow(unused_imports)]
mod tests {
use super::*;
use openlark_core::config::Config;
fn create_test_config() -> Config {
Config::builder()
.app_id("test_app")
.app_secret("test_secret")
.build()
}
#[test]
fn test_oidc_refresh_access_token_builder_new() {
let config = create_test_config();
let builder = OidcRefreshAccessTokenBuilder::new(config);
assert!(builder.refresh_token.is_empty());
assert!(builder.client_id.is_none());
assert!(builder.client_secret.is_none());
assert_eq!(builder.grant_type, Some("refresh_token".to_string()));
}
#[test]
fn test_oidc_refresh_access_token_builder_chain() {
let config = create_test_config();
let builder = OidcRefreshAccessTokenBuilder::new(config)
.refresh_token("my_refresh_token")
.client_id("my_client_id")
.client_secret("my_client_secret")
.grant_type("refresh_token");
assert_eq!(builder.refresh_token, "my_refresh_token");
assert_eq!(builder.client_id, Some("my_client_id".to_string()));
assert_eq!(builder.client_secret, Some("my_client_secret".to_string()));
assert_eq!(builder.grant_type, Some("refresh_token".to_string()));
}
#[test]
fn test_oidc_refresh_access_token_builder_refresh_token_chained() {
let config = create_test_config();
let builder =
OidcRefreshAccessTokenBuilder::new(config).refresh_token("chained_refresh_token");
assert_eq!(builder.refresh_token, "chained_refresh_token");
}
#[test]
fn test_oidc_refresh_access_token_builder_client_id_chained() {
let config = create_test_config();
let builder = OidcRefreshAccessTokenBuilder::new(config).client_id("chained_client_id");
assert_eq!(builder.client_id, Some("chained_client_id".to_string()));
}
#[test]
fn test_oidc_refresh_access_token_builder_client_secret_chained() {
let config = create_test_config();
let builder =
OidcRefreshAccessTokenBuilder::new(config).client_secret("chained_client_secret");
assert_eq!(
builder.client_secret,
Some("chained_client_secret".to_string())
);
}
#[test]
fn test_oidc_refresh_access_token_builder_grant_type_chained() {
let config = create_test_config();
let builder = OidcRefreshAccessTokenBuilder::new(config).grant_type("refresh_token");
assert_eq!(builder.grant_type, Some("refresh_token".to_string()));
}
#[test]
fn test_oidc_refresh_access_token_response_data_deserialization() {
let json = r#"{"data":{"user_access_token":"token123","expires_in":7200,"refresh_token":"refresh456"}}"#;
let response: OidcRefreshAccessTokenResponseData =
serde_json::from_str(json).expect("JSON 反序列化失败");
assert_eq!(response.data.user_access_token, "token123");
assert_eq!(response.data.expires_in, 7200);
assert_eq!(response.data.refresh_token, Some("refresh456".to_string()));
}
#[test]
fn test_oidc_refresh_access_token_response_data_format() {
assert_eq!(
OidcRefreshAccessTokenResponseData::data_format(),
ResponseFormat::Data
);
}
}