use alloc::format;
use io_http::rfc6750::bearer::HttpAuthBearer;
use log::{debug, trace};
use serde_variant::to_variant_name;
use url::Url;
use crate::{
coroutine::*,
gmail_try,
v1::rest::messages::{GmailMessage, GmailMessageFormat},
v1::send::{GMAIL_API_BASE, GmailSend, GmailSendError, GmailSendOutput},
};
pub struct GmailMessageGet {
send: GmailSend<GmailMessage>,
}
impl GmailMessageGet {
pub fn new(
auth: &HttpAuthBearer,
user_id: &str,
id: &str,
format: GmailMessageFormat,
metadata_headers: &[&str],
) -> Result<Self, GmailSendError> {
debug!("prepare gmail message retrieval");
trace!("id: {id:?}");
let mut url =
Url::parse(GMAIL_API_BASE)?.join(&format!("users/{user_id}/messages/{id}"))?;
{
let mut query = url.query_pairs_mut();
query.append_pair("format", to_variant_name(&format).unwrap_or_default());
if matches!(format, GmailMessageFormat::Metadata) {
for header in metadata_headers {
query.append_pair("metadataHeaders", header);
}
}
}
let send = GmailSend::get(auth, url);
Ok(Self { send })
}
}
impl GmailCoroutine for GmailMessageGet {
type Yield = GmailYield;
type Return = Result<GmailSendOutput<GmailMessage>, GmailSendError>;
fn resume(&mut self, arg: Option<&[u8]>) -> GmailCoroutineState<Self::Yield, Self::Return> {
let out = gmail_try!(&mut self.send, arg);
debug!("message retrieved");
trace!("out: {out:?}");
GmailCoroutineState::Complete(Ok(out))
}
}