moonfire-nvr/src/main.rs

// This file is part of Moonfire NVR, a security camera digital video recorder.
// Copyright (C) 2016 Scott Lamb <slamb@slamb.org>
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// In addition, as a special exception, the copyright holders give
// permission to link the code of portions of this program with the
// OpenSSL library under certain conditions as described in each
// individual source file, and distribute linked combinations including
// the two.
//
// You must obey the GNU General Public License in all respects for all
// of the code used other than OpenSSL. If you modify file(s) with this
// exception, you may extend this exception to your version of the
// file(s), but you are not obligated to do so. If you do not wish to do
// so, delete this exception statement from your version. If you delete
// this exception statement from all source files in the program, then
// also delete it here.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.

#![cfg_attr(all(feature="nightly", test), feature(test))]

extern crate byteorder;
extern crate core;
#[macro_use] extern crate chan;
extern crate chan_signal;
extern crate docopt;
#[macro_use] extern crate ffmpeg;
extern crate ffmpeg_sys;
extern crate fnv;
extern crate http_entity;
extern crate hyper;
#[macro_use] extern crate lazy_static;
extern crate libc;
#[macro_use] extern crate log;
extern crate lru_cache;
extern crate rusqlite;
extern crate memmap;
#[macro_use] extern crate mime;
extern crate openssl;
extern crate regex;
extern crate rustc_serialize;
extern crate serde;
extern crate serde_json;
extern crate slog;
extern crate slog_envlogger;
extern crate slog_stdlog;
extern crate slog_term;
extern crate smallvec;
extern crate time;
extern crate url;
extern crate uuid;

mod clock;
mod coding;
mod cmds;
mod db;
mod dir;
mod error;
mod h264;
mod mmapfile;
mod mp4;
mod pieces;
mod recording;
mod stream;
mod streamer;
mod strutil;
#[cfg(test)] mod testutil;
mod web;

/// Commandline usage string. This is in the particular format expected by the `docopt` crate.
/// Besides being printed on --help or argument parsing error, it's actually parsed to define the
/// allowed commandline arguments and their defaults.
const USAGE: &'static str = "
Usage: moonfire-nvr <command> [<args>...]
       moonfire-nvr (--help | --version)

Options:
    -h, --help             Show this message.
    --version              Show the version of moonfire-nvr.

Commands:
    check                  Check database integrity
    init                   Initialize a database
    run                    Run the daemon: record from cameras and handle HTTP requests
    ts                     Translate between human-readable and numeric timestamps
    upgrade                Upgrade the database to the latest schema
";

/// Commandline arguments corresponding to `USAGE`; automatically filled by the `docopt` crate.
#[derive(Debug, RustcDecodable)]
struct Args {
    arg_command: Option<cmds::Command>,
}

fn version() -> String {
    let major = option_env!("CARGO_PKG_VERSION_MAJOR");
    let minor = option_env!("CARGO_PKG_VERSION_MAJOR");
    let patch = option_env!("CARGO_PKG_VERSION_MAJOR");
    match (major, minor, patch) {
        (Some(major), Some(minor), Some(patch)) => format!("{}.{}.{}", major, minor, patch),
        _ => "".to_owned(),
    }
}

fn main() {
    // Parse commandline arguments.
    // (Note this differs from cmds::parse_args in that it specifies options_first.)
    let args: Args = docopt::Docopt::new(USAGE)
                                    .and_then(|d| d.options_first(true)
                                                   .version(Some(version()))
                                                   .decode())
                                    .unwrap_or_else(|e| e.exit());

    if let Err(e) = args.arg_command.unwrap().run() {
        use std::io::Write;
        writeln!(&mut ::std::io::stderr(), "{}", e).unwrap();
        ::std::process::exit(1);
    }
}