webmetro/src/main.rs

#[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())
}