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::{GmailInternalDateSource, GmailMessage},
v1::send::{GMAIL_API_BASE, GmailSend, GmailSendError, GmailSendOutput},
};
pub struct GmailMessageImport {
send: GmailSend<GmailMessage>,
}
impl GmailMessageImport {
pub fn new(
auth: &HttpAuthBearer,
user_id: &str,
message: &GmailMessage,
internal_date_source: Option<GmailInternalDateSource>,
deleted: bool,
never_mark_spam: bool,
process_for_calendar: bool,
) -> Result<Self, GmailSendError> {
debug!("prepare gmail message import");
trace!("message: {message:?}");
let mut url =
Url::parse(GMAIL_API_BASE)?.join(&format!("users/{user_id}/messages/import"))?;
{
let mut query = url.query_pairs_mut();
if let Some(ids) = internal_date_source {
query.append_pair(
"internalDateSource",
to_variant_name(&ids).unwrap_or_default(),
);
}
if deleted {
query.append_pair("deleted", "true");
}
if never_mark_spam {
query.append_pair("neverMarkSpam", "true");
}
if process_for_calendar {
query.append_pair("processForCalendar", "true");
}
}
let send = GmailSend::post_json(auth, url, message)?;
Ok(Self { send })
}
}
impl GmailCoroutine for GmailMessageImport {
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 imported");
trace!("out: {out:?}");
GmailCoroutineState::Complete(Ok(out))
}
}