use std::fs::File;
use std::io::Read;
use std::path::Path;
use bytes::Bytes;
use serde::{Serialize, Deserialize};
use serde_json::Value;
use crate::{session::SessionStore, LabradorResult, RequestBody, RequestType, WechatCommonResponse, request, get_nonce_str};
use crate::wechat::miniapp::method::{MaMediaMethod, WechatMaMethod};
use crate::wechat::miniapp::{WechatMaClient, WechatRequest};
#[derive(Debug, Clone)]
pub struct WechatMaMedia<'a, T: SessionStore> {
client: &'a WechatMaClient<T>,
}
#[allow(unused)]
impl<'a, T: SessionStore> WechatMaMedia<'a, T> {
#[inline]
pub fn new(client: &WechatMaClient<T>) -> WechatMaMedia<T> {
WechatMaMedia {
client,
}
}
pub async fn upload_media(&self, media_type: &str, file_name: Option<&str>, data: Vec<u8>) -> LabradorResult<WechatMaMediaResponse> {
let default_file_name = format!("{}.png", get_nonce_str());
let req = WechatMaMediaRequest {
media_type: media_type.to_string(),
file_name: file_name.map(|v| v.to_string()).unwrap_or(default_file_name),
media_data: data
};
let v= self.client.execute::<WechatMaMediaRequest, String>(req).await?.json::<Value>()?;
WechatCommonResponse::parse::<WechatMaMediaResponse>(v)
}
pub async fn upload_media_with_file(&self, media_type: &str, file_path: &str) -> LabradorResult<WechatMaMediaResponse> {
let path = Path::new(file_path);
let file_name = path.file_name().map(|v| v.to_str().unwrap_or_default()).unwrap_or_default();
let mut f = File::open(path)?;
let mut content: Vec<u8> = Vec::new();
let _ = f.read_to_end(&mut content)?;
self.upload_media(media_type, file_name.into(),content).await
}
pub async fn upload_media_with_url(&self, media_type: &str, url: &str) -> LabradorResult<WechatMaMediaResponse> {
let result = request(|client| client.get(url)).await?;
let content = result.bytes()?;
self.upload_media(media_type, None,content.to_vec()).await
}
pub async fn get_media(&self, media_id: &str) -> LabradorResult<Bytes> {
let response = self.client.post(WechatMaMethod::Media(MaMediaMethod::GetMedia), vec![("media_id".to_string(), media_id.to_string())], serde_json::Value::Null, RequestType::Json).await?;
response.bytes()
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WechatMaMediaRequest {
pub media_type: String,
pub file_name: String,
pub media_data: Vec<u8>
}
impl WechatRequest for WechatMaMediaRequest {
fn get_api_method_name(&self) -> String {
MaMediaMethod::UploadMedia(self.media_type.to_string()).get_method()
}
fn get_request_body<T: Serialize>(&self) -> RequestBody<T> {
let form = reqwest::multipart::Form::new().part("media", reqwest::multipart::Part::stream(self.media_data.to_owned()).file_name(self.file_name.to_string()));
form.into()
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct WechatMaMediaResponse {
pub url: Option<String>,
pub media_id: Option<String>,
#[serde(rename="type")]
pub r#type: Option<String>,
pub thumb_media_id: Option<String>,
pub created_at: Option<i64>,
}