mirror of
https://github.com/elicpeter/nyx.git
synced 2026-06-06 19:35:13 +02:00
a438886217
* refactor: Update comments for clarity and add expectations.json files for performance metrics * feat: Implement FP guard for JS/TS local-collection receivers to suppress missing ownership checks * feat: Enhance Rust parameter handling to classify local collections and prevent false ownership checks * refactor: Simplify code formatting for better readability in multiple files * refactor: Improve UTF-8 sequence length handling and enhance clarity in loop iteration * feat: Update Java and Python patterns to include new security rules * refactor: Improve comment clarity and consistency across multiple Rust files * refactor: Simplify code formatting for improved readability in integration tests and module files * refactor: Improve comment formatting and enhance clarity in assertions across multiple files
92 lines
2.9 KiB
Rust
92 lines
2.9 KiB
Rust
use clap::Parser;
|
|
use console::style;
|
|
use directories::ProjectDirs;
|
|
use nyx_scanner::cli::Cli;
|
|
use nyx_scanner::commands;
|
|
use nyx_scanner::errors::NyxResult;
|
|
use nyx_scanner::fmt;
|
|
use nyx_scanner::utils::Config;
|
|
use std::fs;
|
|
use std::time::Instant;
|
|
use tracing_subscriber::fmt::time;
|
|
use tracing_subscriber::prelude::*;
|
|
use tracing_subscriber::{EnvFilter, Registry, fmt as tracing_fmt};
|
|
// use tracing_appender::rolling::{RollingFileAppender, Rotation};
|
|
// use tracing_appender::non_blocking;
|
|
|
|
fn init_tracing() {
|
|
// let file_appender = RollingFileAppender::new(Rotation::HOURLY, "logs", "nyx-scanner.log");
|
|
// let (file_writer, guard) = non_blocking(file_appender);
|
|
|
|
let fmt_layer = tracing_fmt::layer()
|
|
.pretty()
|
|
.with_thread_ids(true)
|
|
.with_timer(time::UtcTime::rfc_3339());
|
|
|
|
// let file_layer = fmt::layer()
|
|
// .with_writer(file_writer)
|
|
// .without_time()
|
|
// .json();
|
|
|
|
Registry::default()
|
|
.with(EnvFilter::from_default_env())
|
|
.with(fmt_layer)
|
|
.init();
|
|
}
|
|
|
|
fn main() -> NyxResult<()> {
|
|
let now = Instant::now();
|
|
init_tracing();
|
|
|
|
tracing::debug!("CLI starting up");
|
|
|
|
if std::env::args().count() == 1 {
|
|
eprint!("{}", fmt::render_welcome());
|
|
return Ok(());
|
|
}
|
|
|
|
let cli = Cli::parse();
|
|
|
|
let proj_dirs =
|
|
ProjectDirs::from("", "", "nyx").ok_or("Unable to determine project directories")?;
|
|
|
|
// todo: check if we want to actually build a config file, maybe some environments will not want to have anything written
|
|
let config_dir = proj_dirs.config_dir();
|
|
fs::create_dir_all(config_dir)?;
|
|
|
|
let database_dir = proj_dirs.data_local_dir();
|
|
fs::create_dir_all(database_dir)?;
|
|
|
|
let (mut config, config_note) = Config::load(config_dir)?;
|
|
|
|
rayon::ThreadPoolBuilder::new()
|
|
.stack_size(config.performance.rayon_thread_stack_size)
|
|
.build_global()
|
|
.expect("set rayon stack size");
|
|
|
|
let is_serve = cli.command.is_serve();
|
|
let is_info = cli.command.is_informational();
|
|
let quiet = config.output.quiet || cli.command.is_structured_output(&config);
|
|
|
|
// Print config note before scanning (human-readable mode only). Pure
|
|
// informational commands suppress it too, their output is usually
|
|
// piped or grepped and the preamble is noise.
|
|
if let Some(note) = config_note.filter(|_| !quiet && !is_info) {
|
|
eprint!("{note}");
|
|
}
|
|
|
|
commands::handle_command(cli.command, database_dir, config_dir, &mut config)?;
|
|
|
|
// "Finished in" is useful for long scans but pure noise on fast paths
|
|
// (small repos, `index status`, `clean` etc.). Suppress it under a
|
|
// second; users who care about precise timings can use `time`/`hyperfine`.
|
|
let elapsed = now.elapsed();
|
|
if !quiet && !is_serve && !is_info && elapsed.as_secs_f32() >= 1.0 {
|
|
eprintln!(
|
|
"{} in {:.3}s.",
|
|
style("Finished").green().bold(),
|
|
elapsed.as_secs_f32()
|
|
);
|
|
}
|
|
Ok(())
|
|
}
|