89 lines
2.3 KiB
Rust
89 lines
2.3 KiB
Rust
// Copyright (C) 2021 Scott Lamb <slamb@slamb.org>
|
|
// SPDX-License-Identifier: MIT OR Apache-2.0
|
|
|
|
//! RTSP client examples.
|
|
|
|
mod metadata;
|
|
mod mp4;
|
|
|
|
use anyhow::Error;
|
|
use log::{error, info};
|
|
use std::str::FromStr;
|
|
use structopt::StructOpt;
|
|
|
|
#[derive(StructOpt)]
|
|
struct Source {
|
|
/// `rtsp://` URL to connect to.
|
|
#[structopt(long, parse(try_from_str))]
|
|
url: url::Url,
|
|
|
|
/// Username to send if the server requires authentication.
|
|
#[structopt(long)]
|
|
username: Option<String>,
|
|
|
|
/// Password; requires username.
|
|
#[structopt(long, requires = "username")]
|
|
password: Option<String>,
|
|
}
|
|
|
|
#[derive(StructOpt)]
|
|
enum Cmd {
|
|
/// Write available audio and video streams to mp4 file
|
|
Mp4(mp4::Opts),
|
|
/// Get realtime metadata of onvif stream, use Ctrl+C to stop
|
|
Metadata(metadata::Opts),
|
|
/// Get info about available streams and exit
|
|
Info(metadata::Opts),
|
|
}
|
|
|
|
fn init_logging() -> mylog::Handle {
|
|
let h = mylog::Builder::new()
|
|
.set_format(
|
|
::std::env::var("MOONFIRE_FORMAT")
|
|
.map_err(|_| ())
|
|
.and_then(|s| mylog::Format::from_str(&s))
|
|
.unwrap_or(mylog::Format::Google),
|
|
)
|
|
.set_spec(::std::env::var("MOONFIRE_LOG").as_deref().unwrap_or("info"))
|
|
.build();
|
|
h.clone().install().unwrap();
|
|
h
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
let mut h = init_logging();
|
|
if let Err(e) = {
|
|
let _a = h.async_scope();
|
|
main_inner().await
|
|
} {
|
|
error!("Fatal: {}", itertools::join(e.chain(), "\ncaused by: "));
|
|
std::process::exit(1);
|
|
}
|
|
info!("Done");
|
|
}
|
|
|
|
/// Interpets the `username` and `password` of a [Source].
|
|
fn creds(
|
|
username: Option<String>,
|
|
password: Option<String>,
|
|
) -> Option<retina::client::Credentials> {
|
|
match (username, password) {
|
|
(Some(username), password) => Some(retina::client::Credentials {
|
|
username,
|
|
password: password.unwrap_or_default(),
|
|
}),
|
|
(None, None) => None,
|
|
_ => unreachable!(), // structopt/clap enforce that password requires username.
|
|
}
|
|
}
|
|
|
|
async fn main_inner() -> Result<(), Error> {
|
|
let cmd = Cmd::from_args();
|
|
match cmd {
|
|
Cmd::Mp4(opts) => mp4::run(opts).await,
|
|
Cmd::Metadata(opts) => metadata::run(opts, false).await,
|
|
Cmd::Info(opts) => metadata::run(opts, true).await,
|
|
}
|
|
}
|