diff --git a/src/logging.rs b/src/logging.rs new file mode 100644 index 0000000..9ae098e --- /dev/null +++ b/src/logging.rs @@ -0,0 +1,12 @@ +use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt}; + +pub fn init_tracing() { + // Set up logging + tracing_subscriber::registry() + .with( + tracing_subscriber::EnvFilter::try_from_default_env() + .unwrap_or_else(|_| "nyazoom=debug,tower_http=debug".into()), + ) + .with(tracing_subscriber::fmt::layer()) + .init(); +} diff --git a/src/main.rs b/src/main.rs index b501ef8..f7a7bf6 100644 --- a/src/main.rs +++ b/src/main.rs @@ -26,9 +26,8 @@ use tokio_util::{ use tower_http::{limit::RequestBodyLimitLayer, services::ServeDir, trace::TraceLayer}; -use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt}; - mod cache; +mod logging; mod nyazoom_headers; pub mod ssr; mod state; @@ -51,14 +50,7 @@ pub mod error { #[tokio::main] async fn main() -> io::Result<()> { - // Set up logging - tracing_subscriber::registry() - .with( - tracing_subscriber::EnvFilter::try_from_default_env() - .unwrap_or_else(|_| "nyazoom=debug,tower_http=debug".into()), - ) - .with(tracing_subscriber::fmt::layer()) - .init(); + logging::init_tracing(); // uses create_dir_all to create both .cache and serve inside it in one go util::make_dir(".cache/serve").await?;