xmppftw/cusku
Watch
1
0
Fork
You've already forked cusku
 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add webhook cli arguments

Browse source 
Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>
This commit is contained in:
Maxime “pep” Buquet 2023-05-22 00:37:40 +02:00
commit 179bd2e395
No known key found for this signature in database
GPG key ID: DEDA74AEECA9D0F2
2 changed files with 32 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

42
src/main.rs
View file

@ -54,6 +54,14 @@ struct Args {
/// Nickname to use in rooms
#[arg(short, long, default_value = "bot")]
nickname: String,
/// Token to match the one provided by the Webhook service
#[arg(short, long)]
webhook_token: Option<String>,
/// HTTP Webhook listening address and port, e.g., 127.0.0.1:1234 or [::1]:1234
#[arg(long, default_value = "[::1]:3000")]
addr: SocketAddr,
}
#[tokio::main]
@ -61,23 +69,27 @@ async fn main() {
pretty_env_logger::init();
let args = Args::parse();
let addr = SocketAddr::from(([127, 0, 0, 1], 3000));
let (value_tx, mut value_rx) = mpsc::unbounded_channel::<WebHook>();
let value_tx = Arc::new(Mutex::new(value_tx));
let make_svc = make_service_fn(move |_conn| {
let value_tx = value_tx.clone();
async move {
Ok::<_, Infallible>(service_fn(move |req| {
let value_tx = value_tx.clone();
webhooks(req, value_tx)
}))
}
});
let server = Server::bind(&addr).serve(make_svc);
println!("Listening on http://{}", addr);
let _join = tokio::spawn(server);
if let Some(token) = args.webhook_token {
let value_tx = Arc::new(Mutex::new(value_tx));
let make_svc = make_service_fn(move |_conn| {
let value_tx = value_tx.clone();
let token = token.clone();
async move {
Ok::<_, Infallible>(service_fn(move |req| {
let value_tx = value_tx.clone();
let token = token.clone();
webhooks(req, token, value_tx)
}))
}
});
let server = Server::bind(&args.addr).serve(make_svc);
println!("Listening on http://{}", &args.addr);
let _join = tokio::spawn(server);
}
let mut client = XmppClient::new(
&String::from(args.jid),
args.password.as_str(),