use argparse::{ArgumentParser, Store, StoreOption, StoreTrue};
use std::fs::File;
use a2::{
client::ClientConfig, Client, DefaultNotificationBuilder, Endpoint, NotificationBuilder, NotificationOptions,
};
#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync>> {
tracing_subscriber::fmt().init();
let mut key_file = String::new();
let mut team_id = String::new();
let mut key_id = String::new();
let mut device_token = String::new();
let mut message = String::from("Ch-check it out!");
let mut sandbox = false;
let mut topic: Option<String> = None;
{
let mut ap = ArgumentParser::new();
ap.set_description("APNs token-based push");
ap.refer(&mut key_file)
.add_option(&["-p", "--pkcs8"], Store, "Private key PKCS8");
ap.refer(&mut team_id)
.add_option(&["-t", "--team_id"], Store, "APNs team ID");
ap.refer(&mut key_id)
.add_option(&["-k", "--key_id"], Store, "APNs key ID");
ap.refer(&mut device_token)
.add_option(&["-d", "--device_token"], Store, "APNs device token");
ap.refer(&mut message)
.add_option(&["-m", "--message"], Store, "Notification message");
ap.refer(&mut sandbox)
.add_option(&["-s", "--sandbox"], StoreTrue, "Use the development APNs servers");
ap.refer(&mut topic)
.add_option(&["-o", "--topic"], StoreOption, "APNS topic");
ap.parse_args_or_exit();
}
let mut private_key = File::open(key_file).unwrap();
let endpoint = if sandbox {
Endpoint::Sandbox
} else {
Endpoint::Production
};
let client_config = ClientConfig::new(endpoint);
let client = Client::token(&mut private_key, key_id, team_id, client_config).unwrap();
let options = NotificationOptions {
apns_topic: topic.as_deref(),
..Default::default()
};
let builder = DefaultNotificationBuilder::new()
.set_body(message.as_ref())
.set_sound("default")
.set_badge(1u32);
let payload = builder.build(device_token.as_ref(), options);
let response = client.send(payload).await?;
println!("Sent: {:?}", response);
Ok(())
}