62 lines
1.6 KiB
Rust
62 lines
1.6 KiB
Rust
#[macro_use] extern crate clap;
|
|
extern crate futures;
|
|
extern crate hyper;
|
|
extern crate tokio;
|
|
extern crate tokio_core;
|
|
extern crate tokio_io;
|
|
extern crate webmetro;
|
|
|
|
mod commands;
|
|
|
|
use clap::{App, AppSettings};
|
|
use futures::prelude::*;
|
|
use tokio_core::reactor::Core;
|
|
use webmetro::error::WebmetroError;
|
|
|
|
use commands::{
|
|
relay,
|
|
filter,
|
|
dump
|
|
};
|
|
|
|
fn options() -> App<'static, 'static> {
|
|
App::new("webmetro")
|
|
.version(crate_version!())
|
|
.about("Utilities for broadcasting & relaying live WebM video/audio streams")
|
|
.setting(AppSettings::VersionlessSubcommands)
|
|
.subcommand(relay::options())
|
|
.subcommand(filter::options())
|
|
.subcommand(dump::options())
|
|
}
|
|
|
|
fn main() {
|
|
let args = options().get_matches();
|
|
|
|
let core = Core::new().unwrap();
|
|
let handle = core.handle();
|
|
|
|
tokio_run(core, match args.subcommand() {
|
|
("filter", Some(sub_args)) => box_up(filter::run(sub_args)),
|
|
("relay", Some(sub_args)) => box_up(relay::run(sub_args)),
|
|
("dump", Some(sub_args)) => box_up(dump::run(sub_args)),
|
|
_ => box_up(futures::lazy(|| {
|
|
options().print_help().unwrap();
|
|
println!("");
|
|
Ok(())
|
|
}))
|
|
});
|
|
}
|
|
|
|
fn tokio_run(mut core: Core, task: Box<Future<Item=(), Error=WebmetroError>>) {
|
|
core.run(task.into_future()).unwrap_or_else(|err| {
|
|
eprintln!("Error: {}", err);
|
|
::std::process::exit(1);
|
|
});
|
|
}
|
|
|
|
fn box_up<F: IntoFuture<Item=(), Error=WebmetroError>>(task: F) -> Box<Future<Item=(), Error=WebmetroError>>
|
|
where F::Future: 'static
|
|
{
|
|
Box::new(task.into_future())
|
|
}
|