use std::sync::{Arc, Weak};
use async_trait::async_trait;
use wx_rust_common::bean::{CommonUploadData, CommonUploadParam};
use wx_rust_common::error::WxErrorException;
use wx_rust_common::util::http::MediaUploadRequestExecutor;
use wx_rust_miniapp::api::WxMaService;
use crate::api::WxOpenMaAuthService;
use crate::api::WxOpenService;
use crate::api::r#impl::WxOpenMaService;
use crate::api::r#impl::base_wx_open_service_impl;
use crate::bean::MaAuthQueryIdentityTreeResult;
use crate::bean::MaAuthQueryResult;
use crate::bean::MaAuthResubmitParam;
use crate::bean::MaAuthSubmitParam;
use crate::bean::MaAuthSubmitResult;
use crate::bean::MaAuthUploadResult;
use crate::enums::url_ma_domain::{
ma_auth_identity_url, ma_auth_query_url, ma_auth_resubmit_url, ma_auth_submit_url,
ma_auth_upload_url,
};
pub struct WxOpenMaAuthServiceImpl {
wx_open_service: Weak<dyn WxOpenService>,
app_id: String,
}
impl WxOpenMaAuthServiceImpl {
pub fn new(wx_open_service: Arc<dyn WxOpenService>, app_id: String) -> Self {
Self {
wx_open_service: Arc::downgrade(&wx_open_service),
app_id,
}
}
pub fn app_id(&self) -> &str {
&self.app_id
}
fn svc(&self) -> Result<Arc<dyn WxOpenService>, WxErrorException> {
self.wx_open_service
.upgrade()
.ok_or_else(|| WxErrorException::from_code(-99, "门面服务已被释放"))
}
fn ma_service(&self) -> Result<Arc<dyn WxMaService>, WxErrorException> {
let svc = self.svc()?;
let component = svc.wx_open_component_service().ok_or_else(|| {
WxErrorException::from_code(
-99,
"组件子服务未装配(getWxOpenComponentService 返回 null)",
)
})?;
let any = component
.get_wx_ma_service_by_appid(&self.app_id)
.ok_or_else(|| WxErrorException::from_code(-99, "getWxMaServiceByAppid 返回 None"))?;
let ma = any.downcast::<WxOpenMaService>().map_err(|_| {
WxErrorException::from_code(-99, "代 ma 服务 downcast 失败(缓存类型不匹配)")
})?;
Ok(ma as Arc<dyn WxMaService>)
}
}
#[async_trait]
impl WxOpenMaAuthService for WxOpenMaAuthServiceImpl {
async fn submit(
&self,
param: &MaAuthSubmitParam,
) -> Result<MaAuthSubmitResult, WxErrorException> {
let svc = self.svc()?;
let config = svc.wx_open_config_storage();
let ma = self.ma_service()?;
let body =
serde_json::to_string(param).map_err(|e| WxErrorException::Serde(e.to_string()))?;
let response = ma.post(&ma_auth_submit_url(config.as_ref()), &body).await?;
let response = base_wx_open_service_impl::normalize_errcode(&response)?;
serde_json::from_str(&response).map_err(|e| WxErrorException::Serde(e.to_string()))
}
async fn query(&self, task_id: &str) -> Result<MaAuthQueryResult, WxErrorException> {
let svc = self.svc()?;
let config = svc.wx_open_config_storage();
let ma = self.ma_service()?;
let body = serde_json::json!({ "taskid": task_id });
let response = ma
.post(&ma_auth_query_url(config.as_ref()), &body.to_string())
.await?;
let response = base_wx_open_service_impl::normalize_errcode(&response)?;
serde_json::from_str(&response).map_err(|e| WxErrorException::Serde(e.to_string()))
}
async fn upload(
&self,
data: &CommonUploadData,
) -> Result<MaAuthUploadResult, WxErrorException> {
let svc = self.svc()?;
let config = svc.wx_open_config_storage();
let ma = self.ma_service()?;
let executor = MediaUploadRequestExecutor::new(ma.http_client().clone());
let param = CommonUploadParam::new("media", data.clone());
let response = wx_rust_miniapp::api::r#impl::base_wx_ma_service_impl::execute_with_retry(
ma.as_ref(),
&executor,
&ma_auth_upload_url(config.as_ref()),
param,
)
.await?;
let response = base_wx_open_service_impl::normalize_errcode(&response)?;
serde_json::from_str(&response).map_err(|e| WxErrorException::Serde(e.to_string()))
}
async fn resubmit(
&self,
param: &MaAuthResubmitParam,
) -> Result<MaAuthSubmitResult, WxErrorException> {
let svc = self.svc()?;
let config = svc.wx_open_config_storage();
let ma = self.ma_service()?;
let body =
serde_json::to_string(param).map_err(|e| WxErrorException::Serde(e.to_string()))?;
let response = ma
.post(&ma_auth_resubmit_url(config.as_ref()), &body)
.await?;
let response = base_wx_open_service_impl::normalize_errcode(&response)?;
serde_json::from_str(&response).map_err(|e| WxErrorException::Serde(e.to_string()))
}
async fn query_identity_tree(&self) -> Result<MaAuthQueryIdentityTreeResult, WxErrorException> {
let svc = self.svc()?;
let config = svc.wx_open_config_storage();
let ma = self.ma_service()?;
let response = ma.get(&ma_auth_identity_url(config.as_ref()), "").await?;
let response = base_wx_open_service_impl::normalize_errcode(&response)?;
serde_json::from_str(&response).map_err(|e| WxErrorException::Serde(e.to_string()))
}
}