hxuanyu/cs2-dumper
1
0
Fork 0
mirror of https://github.com/a2x/cs2-dumper.git synced 2025-04-23 21:15:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add CLI option to disable log file (#228)

Browse Source
This commit is contained in:
a2x 2024-10-10 12:00:07 +10:00
parent a2c5b73625
commit 1da70ebd46
1 changed files with 23 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/main.rs
View File

@ -11,7 +11,9 @@ use log::{info, LevelFilter};
use memflow::prelude::v1::*; use memflow::prelude::v1::*;
use simplelog::{ColorChoice, CombinedLogger, Config, TermLogger, TerminalMode, WriteLogger}; use simplelog::{
ColorChoice, CombinedLogger, Config, SharedLogger, TermLogger, TerminalMode, WriteLogger,
};
use output::Output; use output::Output;
@ -48,8 +50,12 @@ struct Args {
process_name: String, process_name: String,
/// Increase logging verbosity. Can be specified multiple times. /// Increase logging verbosity. Can be specified multiple times.
#[arg(short, action = ArgAction::Count)] #[arg(short, long, action = ArgAction::Count)]
verbose: u8, verbose: u8,
/// Prevent creation of the cs2-dumper.log file.
#[arg(short, long)]
no_log_file: bool,
} }
fn main() -> Result<()> { fn main() -> Result<()> {
@ -63,20 +69,23 @@ fn main() -> Result<()> {
_ => LevelFilter::Trace, _ => LevelFilter::Trace,
}; };
CombinedLogger::init(vec![ let mut loggers: Vec<Box<dyn SharedLogger>> = vec![TermLogger::new(
TermLogger::new(
level_filter, level_filter,
Config::default(), Config::default(),
TerminalMode::Mixed, TerminalMode::Mixed,
ColorChoice::Auto, ColorChoice::Auto,
), )];
WriteLogger::new(
// Create the log file by default.
if !args.no_log_file {
loggers.push(WriteLogger::new(
LevelFilter::Info, LevelFilter::Info,
Config::default(), Config::default(),
File::create("cs2-dumper.log").unwrap(), File::create("cs2-dumper.log")?,
), ));
]) }
.unwrap();
CombinedLogger::init(loggers)?;
let conn_args = args let conn_args = args
.connector_args .connector_args