CardsSimulator
/
cards-simulator
3
1
Fork 0
Code Issues 7 Pull Requests Projects Releases Wiki Activity
Browse Source

Finish logger

new_protocol
ThePerkinrex 5 years ago
parent
24980707ec
commit
5a4b7f953c
No known key found for this signature in database GPG Key ID: 1F45A7C4BFB41607
5 changed files with 257 additions and 109 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 3
      .gitignore
  2. 22
      .vscode/launch.json
  3. 207
      server/src/logger.rs
  4. 36
      server/src/logger/color_message.rs
  5. 28
      server/src/main.rs

3
.gitignore
View File

@ -0,0 +1,3 @@
db.sqlite*
output.log
properties.json

22
.vscode/launch.json
View File

@ -0,0 +1,22 @@
{
// Use IntelliSense to learn about possible attributes.
// Hover to view descriptions of existing attributes.
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"type": "lldb",
"request": "launch",
"name": "Cargo launch",
"cargo": {
"args": [
"build",
"--manifest-path",
"server/Cargo.toml"
]
},
"program": "${cargo:program}",
"args": []
}
]
}

207
server/src/logger.rs
View File

@ -10,47 +10,42 @@ use crossterm::{
terminal::{Clear, ClearType},
};
use fern::colors::{ColoredLevelConfig, Color};
pub fn setup() -> Result<(Close, Stdin<String>), fern::InitError> {
let (stdout, stdin, close, join_handle) = TerminalHandler::new(|x| {
use fern::colors::{Color, ColoredLevelConfig};
mod color_message;
use color_message::print_line;
// TODO: Cleanup
pub fn setup(
properties: &crate::server_properties::ServerProperties,
) -> Result<(Close, Stdin<String>), fern::InitError> {
let (stdout, stdin, close, join_handle) = TerminalHandler::new(
|x| {
if x == "sv_cheats" {
log::info!("CHEATS ENABLED")
}
x
});
let colors_line = ColoredLevelConfig::new()
.error(Color::Red)
.warn(Color::Yellow)
// we actually don't need to specify the color for debug and info, they are white by default
.info(Color::White)
.debug(Color::BrightBlack)
// depending on the terminals color scheme, this is the same as the background color
.trace(Color::BrightBlack);
let colors_level = colors_line.clone().info(Color::Green).debug(Color::BrightBlack);
let stdout_logger = fern::Dispatch::new()
.format(move |out, message, record| {
out.finish(format_args!(
"{colors_line}{}[{}{colors_line}][{}] {}\x1B[0m",
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
colors_level.clone().color(record.level()),
record.target(),
message,
colors_line = format!("\x1B[{}m", colors_line.clone().get_color(&record.level()).to_fg_str()),
// colors_level = colors_level.clone().get_color(&record.level()).to_fg_str(),
))
})
.level(log::LevelFilter::Debug)
},
properties.use_colors,
);
// let simple_formatter = |out: fern::FormatCallback, message: _, record: &log::Record| {
// out.finish(format_args!(
// "{}[{}][{}] {}",
// chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
// record.level(),
// record.target(),
// message,
// ))
// };
let stdout_logger = if properties.use_colors {
colored_formatter(fern::Dispatch::new())
} else {
simple_formatter(fern::Dispatch::new())
}
.chain(stdout);
let file = fern::Dispatch::new().format(|out, message, record| {
out.finish(format_args!(
"{}[{}][{}] {}",
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
record.level(),
record.target(),
message,
))
})
.chain(
let file = simple_formatter(fern::Dispatch::new()).chain(
std::fs::OpenOptions::new()
.write(true)
.create(true)
@ -59,16 +54,67 @@ pub fn setup() -> Result<(Close, Stdin<String>), fern::InitError> {
);
fern::Dispatch::new()
.level(log::LevelFilter::Debug)
.level_for("h2::codec", log::LevelFilter::Info)
.level_for("tokio::codec", log::LevelFilter::Info)
.level_for("sqlx::query", log::LevelFilter::Info)
.chain(stdout_logger)
.chain(file)
.apply()?;
log::warn!("TEST1");
log::error!("TEST2");
log::info!("Saving output to output.log");
// std::process::exit(1);
Ok((Close{close, handle: join_handle}, stdin))
Ok((
Close {
close,
handle: join_handle,
},
stdin,
))
}
fn simple_formatter(d: fern::Dispatch) -> fern::Dispatch {
d.format(
|out: fern::FormatCallback, message: _, record: &log::Record| {
out.finish(format_args!(
"{}[{}][{}] {}",
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
record.level(),
record.target(),
message,
))
},
)
}
fn colored_formatter(d: fern::Dispatch) -> fern::Dispatch {
let colors_line = ColoredLevelConfig::new()
.error(Color::Red)
.warn(Color::Yellow)
// we actually don't need to specify the color for debug and info, they are white by default
.info(Color::White)
.debug(Color::BrightBlack)
// depending on the terminals color scheme, this is the same as the background color
.trace(Color::BrightBlack);
let colors_level = colors_line
.clone()
.info(Color::Green)
.debug(Color::BrightBlack);
d.format(
move |out: fern::FormatCallback, message: _, record: &log::Record| {
out.finish(format_args!(
"{colors_line}{}[{}{colors_line}][{}] {}\x1B[0m",
chrono::Local::now().format("[%Y-%m-%d][%H:%M:%S]"),
colors_level.clone().color(record.level()),
record.target(),
message,
colors_line = format!(
"\x1B[{}m",
colors_line.clone().get_color(&record.level()).to_fg_str()
),
// colors_level = colors_level.clone().get_color(&record.level()).to_fg_str(),
))
},
)
}
pub type Stdout = mpsc::Sender<String>;
@ -85,6 +131,7 @@ where
handler: F,
lines: LimitedVec<String>,
command: String,
use_colors: bool,
}
impl<T, F> TerminalHandler<T, F>
@ -94,6 +141,7 @@ where
{
fn new(
handler: F,
use_colors: bool,
) -> (
Stdout,
Stdin<T>,
@ -114,15 +162,14 @@ where
handler,
lines: LimitedVec::new(crossterm::terminal::size().unwrap().1 as usize - 1),
command: String::new(),
use_colors,
};
loop {
if s.iterate() {
break;
}
}
queue!(stdout, crossterm::event::DisableMouseCapture).unwrap();
crossterm::terminal::disable_raw_mode().ok();
println!("Closing printing thread");
cleanup();
});
(stdout_send, stdin_recv, close_send, join_handle)
}
@ -138,19 +185,16 @@ where
Event::Resize(_cols, rows) => self.lines.size = rows as usize - 1,
Event::Key(k) => match (k.code, k.modifiers) {
(KeyCode::Char('c'), KeyModifiers::CONTROL) => {
let mut stdout = std::io::stdout();
queue!(stdout, crossterm::event::DisableMouseCapture).unwrap();
crossterm::terminal::disable_raw_mode().unwrap();
stdout.flush().unwrap();
println!("Exiting server");
cleanup();
std::process::exit(0)
}
(k, _m) => {
match k {
(k, _m) => match k {
KeyCode::Char(c) => self.command.push(c),
KeyCode::Backspace => {self.command.pop();},
KeyCode::Backspace => {
self.command.pop();
}
KeyCode::Enter => {
log::info!(target: "COMMAND", "{}", self.command);
log::info!(target: "command", " >> {}", self.command);
match self.stdin.send((self.handler)(self.command.clone())) {
Ok(_) => (),
Err(e) => log::error!("{}", e),
@ -160,8 +204,7 @@ where
KeyCode::Esc => {
self.lines.scroll_to_bottom();
}
_ => ()
}
_ => (),
},
},
Event::Mouse(m) => match m {
@ -190,18 +233,44 @@ where
// println!("Got stdout");
if updated {
queue!(stdout, Clear(ClearType::All), MoveTo(0, 0)).unwrap();
for v in self.lines.iter() {
queue!(stdout, Print(v), MoveToNextLine(1)).unwrap();
let mut i = 0;
let size = crossterm::terminal::size().unwrap();
for v in self.lines.iter(self.use_colors) {
print_line(&v, self.use_colors, &mut stdout);
queue!(stdout, MoveToNextLine(1)).unwrap();
i += 1;
if i >= size.1 as usize - 1 {
break;
}
// println!("{}", v);
}
let size = crossterm::terminal::size().unwrap();
queue!(stdout, MoveTo(0, size.1), Clear(ClearType::CurrentLine), Print(&self.command)).unwrap();
queue!(
stdout,
MoveTo(0, size.1),
// Clear(ClearType::CurrentLine),
Print(&format!(" >> {}", self.command))
)
.unwrap();
stdout.flush().unwrap();
}
false
}
}
fn cleanup() {
let mut stdout = std::io::stdout();
queue!(
stdout,
Clear(ClearType::All),
MoveTo(0, 0),
crossterm::event::DisableMouseCapture
)
.unwrap();
crossterm::terminal::disable_raw_mode().unwrap();
stdout.flush().unwrap();
println!("Exiting server, output has been saved to output.log");
}
impl<T, F> Drop for TerminalHandler<T, F>
where
T: Send + 'static,
@ -257,7 +326,10 @@ impl<T> LimitedVec<T> {
// assert!(self.inner.len() <= self.size)
}
fn iter(&self) -> std::slice::Iter<T> {
fn iter<'a>(&'a self, use_colors: bool) -> Box<dyn Iterator<Item = String> + 'a>
where
T: std::fmt::Display,
{
let start = if self.inner.len() < self.size {
0
} else {
@ -271,7 +343,24 @@ impl<T> LimitedVec<T> {
if end < self.inner.len() {
end = self.inner.len()
}
self.inner[start..end].iter()
let iter = self.inner[start..end].iter();
let map = iter.enumerate().map(move |(i, x)| {
format!(
"{}{}",
if use_colors {
format!(
"\x1B[{}m{:>5}\x1B[0m ",
fern::colors::Color::BrightBlack.to_fg_str(),
start + i
)
} else {
String::new()
},
x
)
});
Box::new(map)
}
fn scroll_up(&mut self) {

36
server/src/logger/color_message.rs
View File

@ -0,0 +1,36 @@
use crossterm::{
cursor::{MoveRight, MoveToNextLine},
event::{Event, KeyCode, KeyModifiers, MouseEvent},
queue,
style::{Print, PrintStyledContent},
};
use std::io::Write;
pub fn print_line(message: &String, use_colors: bool, stdout: &mut std::io::Stdout) {
if use_colors {
let mut s = String::new();
let mut highlight = false;
for c in message.chars() {
if c == '`' {
// queue!(stdout, Print(format!("{}`", highlight))).unwrap();
if highlight {
let styled = format!("\x1B[{}m{}\x1B[0m", fern::colors::Color::Blue.to_fg_str(), s);
queue!(stdout, Print(styled)).unwrap();
s = String::new();
}else{
queue!(stdout, Print(s)).unwrap();
s = String::new();
}
highlight = !highlight;
continue;
}
s.push(c);
}
queue!(stdout, Print(&s)).unwrap();
}else {
queue!(stdout, Print(message)).unwrap();
}
}

28
server/src/main.rs
View File

@ -4,31 +4,29 @@ mod games;
mod logger;
mod server_properties;
use log::info;
#[tokio::main]
async fn main() {
server_properties::setup();
let p = server_properties::ServerProperties::load();
let (close, _stdin) = logger::setup().unwrap();
// protobuf::route_guide_grpc::RouteGuid
// for i in 0..100 {
// std::thread::sleep(std::time::Duration::from_secs_f32(0.1));
let (close, _stdin) = logger::setup(&p).unwrap();
info!(target: "setup", "Server name: `{}`", p.name);
info!(target: "setup", "Serving on address `{}`", p.addr);
// for i in 0..1000 {
// std::thread::sleep(std::time::Duration::from_secs_f32(1.));
// if i == 69 {
// log::info!("69 Nͬ͂ͧi̽̅̽҉̫̕͢ĉ͆̚҉̢͞͏eͫ̑͑");
// std::thread::sleep(std::time::Duration::from_secs_f32(0.2));
// }else{
// log::info!("{}", i);
// info!("{}\nTEXTETX\nTEXT", i)
// }
// }
// std::thread::sleep(std::time::Duration::from_secs_f32(200.));
info!(target: "setup", "Loading games from the games directory");
let games = games::load_games();
log::info!("{:?}", games);
games[0].run(4);
info!(target: "setup", "Loading database");
let pool = db::DbPool::new().await;
// let mut conn = pool.acquire().await;
// println!("{}", conn.add_user("Hi").await);
// println!("{:?}", conn.users().await);
// conn.close().await;
info!(target: "setup", "Starting server");
grpc::start(pool, games, p).await;
close.close();
}

Write Preview
Loading…
Cancel
Save