commit
5082b205d9
GPG Key ID:
8F93B68BD37255B8
4 changed files with 748 additions and 0 deletions
@ -0,0 +1,430 @@
@@ -0,0 +1,430 @@
|
||||
# This file is automatically @generated by Cargo. |
||||
# It is not intended for manual editing. |
||||
version = 3 |
||||
|
||||
[[package]] |
||||
name = "android-tzdata" |
||||
version = "0.1.1" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "e999941b234f3131b00bc13c22d06e8c5ff726d1b6318ac7eb276997bbb4fef0" |
||||
|
||||
[[package]] |
||||
name = "android_system_properties" |
||||
version = "0.1.5" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "819e7219dbd41043ac279b19830f2efc897156490d7fd6ea916720117ee66311" |
||||
dependencies = [ |
||||
"libc", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "autocfg" |
||||
version = "1.2.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "f1fdabc7756949593fe60f30ec81974b613357de856987752631dea1e3394c80" |
||||
|
||||
[[package]] |
||||
name = "bumpalo" |
||||
version = "3.16.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c" |
||||
|
||||
[[package]] |
||||
name = "cc" |
||||
version = "1.0.92" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "2678b2e3449475e95b0aa6f9b506a28e61b3dc8996592b983695e8ebb58a8b41" |
||||
|
||||
[[package]] |
||||
name = "cfg-if" |
||||
version = "1.0.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd" |
||||
|
||||
[[package]] |
||||
name = "chrono" |
||||
version = "0.4.37" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "8a0d04d43504c61aa6c7531f1871dd0d418d91130162063b789da00fd7057a5e" |
||||
dependencies = [ |
||||
"android-tzdata", |
||||
"iana-time-zone", |
||||
"js-sys", |
||||
"num-traits", |
||||
"wasm-bindgen", |
||||
"windows-targets", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "core-foundation-sys" |
||||
version = "0.8.6" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "06ea2b9bc92be3c2baa9334a323ebca2d6f074ff852cd1d7b11064035cd3868f" |
||||
|
||||
[[package]] |
||||
name = "gay_tracing_formatter" |
||||
version = "0.1.0" |
||||
dependencies = [ |
||||
"chrono", |
||||
"owo-colors", |
||||
"tracing", |
||||
"tracing-subscriber", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "iana-time-zone" |
||||
version = "0.1.60" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "e7ffbb5a1b541ea2561f8c41c087286cc091e21e556a4f09a8f6cbf17b69b141" |
||||
dependencies = [ |
||||
"android_system_properties", |
||||
"core-foundation-sys", |
||||
"iana-time-zone-haiku", |
||||
"js-sys", |
||||
"wasm-bindgen", |
||||
"windows-core", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "iana-time-zone-haiku" |
||||
version = "0.1.2" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "f31827a206f56af32e590ba56d5d2d085f558508192593743f16b2306495269f" |
||||
dependencies = [ |
||||
"cc", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "js-sys" |
||||
version = "0.3.69" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "29c15563dc2726973df627357ce0c9ddddbea194836909d655df6a75d2cf296d" |
||||
dependencies = [ |
||||
"wasm-bindgen", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "lazy_static" |
||||
version = "1.4.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646" |
||||
|
||||
[[package]] |
||||
name = "libc" |
||||
version = "0.2.153" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "9c198f91728a82281a64e1f4f9eeb25d82cb32a5de251c6bd1b5154d63a8e7bd" |
||||
|
||||
[[package]] |
||||
name = "log" |
||||
version = "0.4.21" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "90ed8c1e510134f979dbc4f070f87d4313098b704861a105fe34231c70a3901c" |
||||
|
||||
[[package]] |
||||
name = "nu-ansi-term" |
||||
version = "0.46.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "77a8165726e8236064dbb45459242600304b42a5ea24ee2948e18e023bf7ba84" |
||||
dependencies = [ |
||||
"overload", |
||||
"winapi", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "num-traits" |
||||
version = "0.2.18" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "da0df0e5185db44f69b44f26786fe401b6c293d1907744beaa7fa62b2e5a517a" |
||||
dependencies = [ |
||||
"autocfg", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "once_cell" |
||||
version = "1.19.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92" |
||||
|
||||
[[package]] |
||||
name = "overload" |
||||
version = "0.1.1" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "b15813163c1d831bf4a13c3610c05c0d03b39feb07f7e09fa234dac9b15aaf39" |
||||
|
||||
[[package]] |
||||
name = "owo-colors" |
||||
version = "4.0.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "caff54706df99d2a78a5a4e3455ff45448d81ef1bb63c22cd14052ca0e993a3f" |
||||
|
||||
[[package]] |
||||
name = "pin-project-lite" |
||||
version = "0.2.14" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "bda66fc9667c18cb2758a2ac84d1167245054bcf85d5d1aaa6923f45801bdd02" |
||||
|
||||
[[package]] |
||||
name = "proc-macro2" |
||||
version = "1.0.79" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "e835ff2298f5721608eb1a980ecaee1aef2c132bf95ecc026a11b7bf3c01c02e" |
||||
dependencies = [ |
||||
"unicode-ident", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "quote" |
||||
version = "1.0.36" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "0fa76aaf39101c457836aec0ce2316dbdc3ab723cdda1c6bd4e6ad4208acaca7" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "sharded-slab" |
||||
version = "0.1.7" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "f40ca3c46823713e0d4209592e8d6e826aa57e928f09752619fc696c499637f6" |
||||
dependencies = [ |
||||
"lazy_static", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "smallvec" |
||||
version = "1.13.2" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "3c5e1a9a646d36c3599cd173a41282daf47c44583ad367b8e6837255952e5c67" |
||||
|
||||
[[package]] |
||||
name = "syn" |
||||
version = "2.0.58" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "44cfb93f38070beee36b3fef7d4f5a16f27751d94b187b666a5cc5e9b0d30687" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
"quote", |
||||
"unicode-ident", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "thread_local" |
||||
version = "1.1.8" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "8b9ef9bad013ada3808854ceac7b46812a6465ba368859a37e2100283d2d719c" |
||||
dependencies = [ |
||||
"cfg-if", |
||||
"once_cell", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "tracing" |
||||
version = "0.1.40" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "c3523ab5a71916ccf420eebdf5521fcef02141234bbc0b8a49f2fdc4544364ef" |
||||
dependencies = [ |
||||
"pin-project-lite", |
||||
"tracing-attributes", |
||||
"tracing-core", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "tracing-attributes" |
||||
version = "0.1.27" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "34704c8d6ebcbc939824180af020566b01a7c01f80641264eba0999f6c2b6be7" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
"quote", |
||||
"syn", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "tracing-core" |
||||
version = "0.1.32" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "c06d3da6113f116aaee68e4d601191614c9053067f9ab7f6edbcb161237daa54" |
||||
dependencies = [ |
||||
"once_cell", |
||||
"valuable", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "tracing-log" |
||||
version = "0.2.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "ee855f1f400bd0e5c02d150ae5de3840039a3f54b025156404e34c23c03f47c3" |
||||
dependencies = [ |
||||
"log", |
||||
"once_cell", |
||||
"tracing-core", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "tracing-subscriber" |
||||
version = "0.3.18" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "ad0f048c97dbd9faa9b7df56362b8ebcaa52adb06b498c050d2f4e32f90a7a8b" |
||||
dependencies = [ |
||||
"nu-ansi-term", |
||||
"sharded-slab", |
||||
"smallvec", |
||||
"thread_local", |
||||
"tracing-core", |
||||
"tracing-log", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "unicode-ident" |
||||
version = "1.0.12" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "3354b9ac3fae1ff6755cb6db53683adb661634f67557942dea4facebec0fee4b" |
||||
|
||||
[[package]] |
||||
name = "valuable" |
||||
version = "0.1.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "830b7e5d4d90034032940e4ace0d9a9a057e7a45cd94e6c007832e39edb82f6d" |
||||
|
||||
[[package]] |
||||
name = "wasm-bindgen" |
||||
version = "0.2.92" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "4be2531df63900aeb2bca0daaaddec08491ee64ceecbee5076636a3b026795a8" |
||||
dependencies = [ |
||||
"cfg-if", |
||||
"wasm-bindgen-macro", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "wasm-bindgen-backend" |
||||
version = "0.2.92" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "614d787b966d3989fa7bb98a654e369c762374fd3213d212cfc0251257e747da" |
||||
dependencies = [ |
||||
"bumpalo", |
||||
"log", |
||||
"once_cell", |
||||
"proc-macro2", |
||||
"quote", |
||||
"syn", |
||||
"wasm-bindgen-shared", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "wasm-bindgen-macro" |
||||
version = "0.2.92" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "a1f8823de937b71b9460c0c34e25f3da88250760bec0ebac694b49997550d726" |
||||
dependencies = [ |
||||
"quote", |
||||
"wasm-bindgen-macro-support", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "wasm-bindgen-macro-support" |
||||
version = "0.2.92" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "e94f17b526d0a461a191c78ea52bbce64071ed5c04c9ffe424dcb38f74171bb7" |
||||
dependencies = [ |
||||
"proc-macro2", |
||||
"quote", |
||||
"syn", |
||||
"wasm-bindgen-backend", |
||||
"wasm-bindgen-shared", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "wasm-bindgen-shared" |
||||
version = "0.2.92" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "af190c94f2773fdb3729c55b007a722abb5384da03bc0986df4c289bf5567e96" |
||||
|
||||
[[package]] |
||||
name = "winapi" |
||||
version = "0.3.9" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419" |
||||
dependencies = [ |
||||
"winapi-i686-pc-windows-gnu", |
||||
"winapi-x86_64-pc-windows-gnu", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "winapi-i686-pc-windows-gnu" |
||||
version = "0.4.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6" |
||||
|
||||
[[package]] |
||||
name = "winapi-x86_64-pc-windows-gnu" |
||||
version = "0.4.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f" |
||||
|
||||
[[package]] |
||||
name = "windows-core" |
||||
version = "0.52.0" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "33ab640c8d7e35bf8ba19b884ba838ceb4fba93a4e8c65a9059d08afcfc683d9" |
||||
dependencies = [ |
||||
"windows-targets", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows-targets" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "7dd37b7e5ab9018759f893a1952c9420d060016fc19a472b4bb20d1bdd694d1b" |
||||
dependencies = [ |
||||
"windows_aarch64_gnullvm", |
||||
"windows_aarch64_msvc", |
||||
"windows_i686_gnu", |
||||
"windows_i686_msvc", |
||||
"windows_x86_64_gnu", |
||||
"windows_x86_64_gnullvm", |
||||
"windows_x86_64_msvc", |
||||
] |
||||
|
||||
[[package]] |
||||
name = "windows_aarch64_gnullvm" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "bcf46cf4c365c6f2d1cc93ce535f2c8b244591df96ceee75d8e83deb70a9cac9" |
||||
|
||||
[[package]] |
||||
name = "windows_aarch64_msvc" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "da9f259dd3bcf6990b55bffd094c4f7235817ba4ceebde8e6d11cd0c5633b675" |
||||
|
||||
[[package]] |
||||
name = "windows_i686_gnu" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "b474d8268f99e0995f25b9f095bc7434632601028cf86590aea5c8a5cb7801d3" |
||||
|
||||
[[package]] |
||||
name = "windows_i686_msvc" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "1515e9a29e5bed743cb4415a9ecf5dfca648ce85ee42e15873c3cd8610ff8e02" |
||||
|
||||
[[package]] |
||||
name = "windows_x86_64_gnu" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "5eee091590e89cc02ad514ffe3ead9eb6b660aedca2183455434b93546371a03" |
||||
|
||||
[[package]] |
||||
name = "windows_x86_64_gnullvm" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "77ca79f2451b49fa9e2af39f0747fe999fcda4f5e241b2898624dca97a1f2177" |
||||
|
||||
[[package]] |
||||
name = "windows_x86_64_msvc" |
||||
version = "0.52.4" |
||||
source = "registry+https://github.com/rust-lang/crates.io-index" |
||||
checksum = "32b752e52a2da0ddfbdbcc6fceadfeede4c939ed16d13e648833a61dfb611ed8" |
||||
@ -0,0 +1,10 @@
@@ -0,0 +1,10 @@
|
||||
[package] |
||||
name = "gay_tracing_formatter" |
||||
version = "0.1.0" |
||||
edition = "2021" |
||||
|
||||
[dependencies] |
||||
chrono = "0.4.37" |
||||
owo-colors = "4.0.0" |
||||
tracing = "0.1.40" |
||||
tracing-subscriber = "0.3.18" |
||||
@ -0,0 +1,307 @@
@@ -0,0 +1,307 @@
|
||||
use std::{ |
||||
cell::RefCell, |
||||
env, |
||||
fmt::{self, Display}, |
||||
io::{stdout, IsTerminal, Write}, |
||||
marker::PhantomData, |
||||
}; |
||||
|
||||
use chrono::{DateTime, Local, TimeZone}; |
||||
use owo_colors::{AnsiColors, OwoColorize}; |
||||
use tracing::{ |
||||
field::{Field, Visit}, |
||||
Event, Level, Metadata, Subscriber, |
||||
}; |
||||
use tracing_subscriber::{ |
||||
fmt::{format::Writer, MakeWriter}, |
||||
Layer, |
||||
}; |
||||
|
||||
pub struct GayFormattingLayer< |
||||
S, |
||||
W: for<'w> MakeWriter<'w> + 'static = fn() -> std::io::Stdout, |
||||
Tz: TimeZone = Local, |
||||
> { |
||||
timer: DateTime<Tz>, |
||||
with_ansi: bool, |
||||
colors: GayColors, |
||||
make_writer: W, |
||||
_inner: PhantomData<fn(S)>, |
||||
} |
||||
|
||||
struct GayVisitor<'w> { |
||||
writer: Writer<'w>, |
||||
indent: usize, |
||||
injected_message: Option<String>, |
||||
} |
||||
|
||||
pub struct GayColors { |
||||
pub crate_name: AnsiColors, |
||||
pub trace: AnsiColors, |
||||
pub debug: AnsiColors, |
||||
pub info: AnsiColors, |
||||
pub warn: AnsiColors, |
||||
pub error: AnsiColors, |
||||
pub timer: AnsiColors, |
||||
} |
||||
|
||||
impl<S> Default for GayFormattingLayer<S> { |
||||
fn default() -> Self { |
||||
Self::new() |
||||
} |
||||
} |
||||
|
||||
impl<S, W> Layer<S> for GayFormattingLayer<S, W> |
||||
where |
||||
S: Subscriber, |
||||
W: for<'w> MakeWriter<'w> + 'static, |
||||
{ |
||||
fn on_event(&self, event: &Event<'_>, _ctx: tracing_subscriber::layer::Context<'_, S>) { |
||||
thread_local! { |
||||
static BUF: RefCell<String> = RefCell::new(String::new()); |
||||
} |
||||
|
||||
BUF.with(|buf: &RefCell<String>| { |
||||
let borrow = buf.try_borrow_mut(); |
||||
let mut a; |
||||
let mut b; |
||||
let buf = match borrow { |
||||
Ok(buf) => { |
||||
a = buf; |
||||
&mut *a |
||||
} |
||||
_ => { |
||||
b = String::new(); |
||||
&mut b |
||||
} |
||||
}; |
||||
|
||||
let res = self.format_event(event, tracing_subscriber::fmt::format::Writer::new(buf)); |
||||
|
||||
if res.is_ok() { |
||||
let mut writer = self.make_writer.make_writer(); |
||||
if let Err(fuck) = writer.write_all(buf.as_bytes()) { |
||||
eprintln!("[gay tracing formatter crate] unable to write event: {fuck}\n") |
||||
} |
||||
} else { |
||||
eprintln!( |
||||
"[gay tracing formatter crate] unable to format event; name: {}; fields: {:?}", |
||||
event.metadata().name(), |
||||
event.fields() |
||||
); |
||||
} |
||||
}); |
||||
} |
||||
} |
||||
|
||||
impl<'w> Visit for GayVisitor<'w> { |
||||
fn record_debug(&mut self, field: &Field, value: &dyn fmt::Debug) { |
||||
match field.name() { |
||||
"message" => { |
||||
write!(self.writer, " {:?}", value).ok(); |
||||
if let Some(msg) = self.injected_message.take() { |
||||
write!(self.writer, "\n{}", msg).ok(); |
||||
} |
||||
} |
||||
_ => { |
||||
if let Some(msg) = self.injected_message.take() { |
||||
write!(self.writer, "\n{}", msg).ok(); |
||||
} |
||||
write!( |
||||
self.writer, |
||||
"\n{:>indent$}: {:?}", |
||||
field.name(), |
||||
value, |
||||
indent = self.indent - 1 |
||||
) |
||||
.ok(); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl Default for GayColors { |
||||
fn default() -> Self { |
||||
Self { |
||||
crate_name: AnsiColors::Green, |
||||
trace: AnsiColors::Magenta, |
||||
debug: AnsiColors::Blue, |
||||
info: AnsiColors::White, |
||||
warn: AnsiColors::BrightYellow, |
||||
error: AnsiColors::Red, |
||||
timer: AnsiColors::BrightBlack, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl<S> GayFormattingLayer<S> { |
||||
pub fn new() -> Self { |
||||
Self { |
||||
timer: Local::now(), |
||||
with_ansi: stdout().is_terminal(), |
||||
colors: GayColors::default(), |
||||
make_writer: std::io::stdout, |
||||
_inner: PhantomData, |
||||
} |
||||
} |
||||
} |
||||
|
||||
impl<S, W, Tz> GayFormattingLayer<S, W, Tz> |
||||
where |
||||
W: for<'w> MakeWriter<'w> + 'static, |
||||
Tz: TimeZone, |
||||
Tz::Offset: Display, |
||||
{ |
||||
pub fn timer(self, timer: DateTime<Tz>) -> Self { |
||||
let Self { |
||||
with_ansi, |
||||
colors, |
||||
make_writer, |
||||
_inner, |
||||
.. |
||||
} = self; |
||||
Self { |
||||
timer, |
||||
with_ansi, |
||||
colors, |
||||
make_writer, |
||||
_inner, |
||||
} |
||||
} |
||||
|
||||
pub fn with_ansi(self, with_ansi: bool) -> Self { |
||||
let Self { |
||||
timer, |
||||
colors, |
||||
make_writer, |
||||
_inner, |
||||
.. |
||||
} = self; |
||||
Self { |
||||
timer, |
||||
with_ansi, |
||||
colors, |
||||
make_writer, |
||||
_inner, |
||||
} |
||||
} |
||||
|
||||
pub fn colors(self, colors: GayColors) -> Self { |
||||
let Self { |
||||
timer, |
||||
with_ansi, |
||||
colors: _, |
||||
make_writer, |
||||
_inner, |
||||
} = self; |
||||
Self { |
||||
timer, |
||||
with_ansi, |
||||
colors, |
||||
make_writer, |
||||
_inner, |
||||
} |
||||
} |
||||
|
||||
pub fn make_writer(self, make_writer: W) -> Self { |
||||
let Self { |
||||
timer, |
||||
with_ansi, |
||||
colors, |
||||
_inner, |
||||
.. |
||||
} = self; |
||||
Self { |
||||
timer, |
||||
with_ansi, |
||||
colors, |
||||
make_writer, |
||||
_inner, |
||||
} |
||||
} |
||||
|
||||
const TIME_LEN: usize = "YYYY-MM-DD HH-MM-SS".len(); |
||||
|
||||
fn make_from_crate_string(&self, meta: &Metadata<'static>) -> Option<String> { |
||||
static CURRENT_CRATE_NAME: &'static str = env!("CARGO_PKG_NAME"); |
||||
match meta.module_path() { |
||||
Some(path) => path |
||||
.split_once("::") |
||||
.map(|(crate_name, _)| crate_name) |
||||
.filter(|name| *name != CURRENT_CRATE_NAME) |
||||
.map(|name| { |
||||
name.to_owned() |
||||
.colorize_if_true(self.with_ansi, self.colors.crate_name) |
||||
}) |
||||
.map(|name| format!("{:>indent$} {name}", "from crate", indent = Self::TIME_LEN)), |
||||
None => Some(format!( |
||||
"{:>indent$}", |
||||
"from unknown crate", |
||||
indent = Self::TIME_LEN, |
||||
)), |
||||
} |
||||
} |
||||
|
||||
fn format_event( |
||||
&self, |
||||
event: &Event, |
||||
mut writer: tracing_subscriber::fmt::format::Writer, |
||||
) -> std::fmt::Result { |
||||
let meta = event.metadata(); |
||||
write!(writer, "{} {}", self.datetime(), self.level(meta.level()))?; |
||||
|
||||
let mut v = GayVisitor::new( |
||||
writer.by_ref(), |
||||
Self::TIME_LEN, |
||||
self.make_from_crate_string(meta), |
||||
); |
||||
|
||||
event.record(&mut v); |
||||
|
||||
write!(writer, "\n")?; |
||||
|
||||
Ok(()) |
||||
} |
||||
|
||||
fn datetime(&self) -> String { |
||||
self.timer |
||||
.format("%F %T") |
||||
.to_string() |
||||
.colorize_if_true(self.with_ansi, self.colors.timer) |
||||
} |
||||
|
||||
fn level(&self, level: &Level) -> String { |
||||
let (icon, color) = match *level { |
||||
Level::TRACE => ("TRACE", self.colors.trace), |
||||
Level::DEBUG => ("DEBUG", self.colors.debug), |
||||
Level::INFO => ("INFO ", self.colors.info), |
||||
Level::WARN => ("WARN ", self.colors.warn), |
||||
Level::ERROR => ("ERROR", self.colors.error), |
||||
}; |
||||
icon.to_string().colorize_if_true(self.with_ansi, color) |
||||
} |
||||
} |
||||
|
||||
impl<'w> GayVisitor<'w> { |
||||
pub fn new(writer: Writer<'w>, indent: usize, message: Option<String>) -> Self { |
||||
GayVisitor { |
||||
writer, |
||||
indent, |
||||
injected_message: message, |
||||
} |
||||
} |
||||
} |
||||
|
||||
trait ColorizeIfTrue { |
||||
fn colorize_if_true(self, cond: bool, color: AnsiColors) -> Self; |
||||
} |
||||
|
||||
impl ColorizeIfTrue for String { |
||||
fn colorize_if_true(self, cond: bool, color: AnsiColors) -> Self { |
||||
if cond { |
||||
self.color(color).to_string() |
||||
} else { |
||||
self |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue