clap_log_flag/
lib.rs

1#![cfg_attr(feature = "nightly", deny(missing_docs))]
2#![cfg_attr(feature = "nightly", feature(external_doc))]
3#![cfg_attr(feature = "nightly", doc(include = "../README.md"))]
4#![cfg_attr(test, deny(warnings))]
5
6extern crate env_logger;
7extern crate failure;
8extern crate log;
9extern crate pretty_env_logger;
10#[macro_use]
11extern crate structopt;
12
13use env_logger::Builder as LoggerBuilder;
14use failure::Error;
15use log::Level;
16use pretty_env_logger::formatted_builder;
17
18/// Add log functionality to Structopt.
19#[derive(StructOpt, Debug)]
20pub struct Log {
21  /// Enable pretty printing
22  #[structopt(short = "P", long = "pretty")]
23  pretty: bool,
24}
25
26impl Log {
27  /// Initialize `env_logger` and set the log level for the given package.
28  ///
29  /// All other modules default to printing warnings.
30  pub fn log(
31    &self,
32    level: Option<Level>,
33    own_pkg_name: &str,
34  ) -> Result<(), Error> {
35    let level_filter = match level {
36      Some(level) => level.to_level_filter(),
37      None => return Ok(()),
38    };
39
40    init_builder(self.pretty)?
41      .filter(Some(&own_pkg_name.replace("-", "_")), level_filter)
42      .filter(None, Level::Warn.to_level_filter())
43      .try_init()?;
44    Ok(())
45  }
46
47  /// Initialize `env_logger` and set the log level for all packages. No
48  /// additional filtering is applied.
49  pub fn log_all(&self, level: Option<Level>) -> Result<(), Error> {
50    let level_filter = match level {
51      Some(level) => level.to_level_filter(),
52      None => return Ok(()),
53    };
54
55    init_builder(self.pretty)?
56      .filter(None, level_filter)
57      .try_init()?;
58    Ok(())
59  }
60}
61
62fn init_builder(pretty: bool) -> Result<LoggerBuilder, Error> {
63  if pretty {
64    Ok(formatted_builder()?)
65  } else {
66    Ok(LoggerBuilder::new())
67  }
68}
clap_log_flag/lib.rs

clap_log_flag/
lib.rs
