beet_core 0.0.8

Core utilities and types for other beet crates
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73

use crate::prelude::*;
use bevy::log::tracing;
use bevy::log::tracing_subscriber;
use std::str::FromStr;
use tracing::level_filters::LevelFilter;

/// Opinionated high level tracing initialization
pub struct PrettyTracing {
	default_level: tracing::Level,
}

impl Default for PrettyTracing {
	fn default() -> Self {
		#[cfg(test)]
		let default_level = tracing::Level::WARN;
		#[cfg(not(test))]
		let default_level = tracing::Level::DEBUG;
		Self { default_level }
	}
}

impl PrettyTracing {
	/// Opinionated tracing defaults for bevy, using the provided level
	/// if none in environment variables.
	/// if already initialized, this will do nothing
	///
	/// ## AWS Lambda
	/// This also considers the AWS Lambda tracing environment variables,
	/// as defined in [`lambda_http::tracing::init_default_subscriber_with_writer`]
	///
	pub fn init(&self) {
		let log_level = env_ext::var("AWS_LAMBDA_LOG_LEVEL")
			.or_else(|_| env_ext::var("RUST_LOG"))
			.map(|val| LevelFilter::from_str(&val).ok())
			.ok()
			.flatten()
			.unwrap_or(self.default_level.into());

		let sub = tracing_subscriber::fmt()
			.compact()
			.with_level(true)
			.with_target(false)
			.with_thread_ids(false)
			.with_thread_names(false)
			.with_file(true)
			.without_time()
			.with_line_number(true)
			.with_env_filter(
				tracing_subscriber::EnvFilter::from_default_env()
					.add_directive("wgpu=error".parse().unwrap())
					.add_directive("naga=warn".parse().unwrap())
					.add_directive("bevy_app=warn".parse().unwrap())
					.add_directive("walrus=warn".parse().unwrap())
					.add_directive("aws=warn".parse().unwrap())
					.add_directive("hyper-util=warn".parse().unwrap())
					.add_directive(log_level.into()),
			)
			.with_writer(std::io::stdout);
		// #[cfg(debug_assertions)]
		// // remove timestamps from the output in debug mode
		// let sub = sub.without_time();


		// if self.aws_lambda && std::env::var("AWS_LAMBDA_LOG_FORMAT")
		// 	.unwrap_or_default()
		// 	.eq_ignore_ascii_case("json") {
		// 	sub.json().try_init()
		// }else{
		// 	sub.pretty().try_init()
		// }.ok();
		sub.pretty().try_init().ok();
	}
}