87 lines
1.8 KiB
Rust
87 lines
1.8 KiB
Rust
use std::{
|
|
fs,
|
|
io::{BufWriter, Write},
|
|
sync::Mutex,
|
|
};
|
|
|
|
use chrono::Utc;
|
|
use lazy_static::lazy_static;
|
|
use serde::Serialize;
|
|
|
|
lazy_static! {
|
|
static ref WRITER: Mutex<BufWriter<fs::File>> = {
|
|
let events_dir = crate::path::events_dir();
|
|
std::fs::create_dir_all(events_dir.as_path()).ok();
|
|
|
|
let now = Utc::now();
|
|
let fname = now.format("%Y-%m-%d.json");
|
|
let file = fs::OpenOptions::new()
|
|
.create(true)
|
|
.append(true)
|
|
.write(true)
|
|
.open(events_dir.join(fname.to_string()))
|
|
.ok()
|
|
.unwrap();
|
|
|
|
Mutex::new(BufWriter::new(file))
|
|
};
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
pub struct Choice<'a> {
|
|
pub index: u32,
|
|
pub text: &'a str,
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
#[serde(rename_all = "snake_case")]
|
|
pub enum Event<'a> {
|
|
View {
|
|
completion_id: &'a str,
|
|
choice_index: u32,
|
|
},
|
|
Select {
|
|
completion_id: &'a str,
|
|
choice_index: u32,
|
|
},
|
|
Completion {
|
|
completion_id: &'a str,
|
|
language: &'a str,
|
|
prompt: &'a str,
|
|
choices: Vec<Choice<'a>>,
|
|
user: Option<&'a str>,
|
|
},
|
|
}
|
|
|
|
#[derive(Serialize)]
|
|
struct Log<'a> {
|
|
ts: u128,
|
|
event: &'a Event<'a>,
|
|
}
|
|
|
|
impl Event<'_> {
|
|
pub fn log(&self) {
|
|
let mut writer = WRITER.lock().unwrap();
|
|
|
|
serdeconv::to_json_writer(
|
|
&Log {
|
|
ts: timestamp(),
|
|
event: self,
|
|
},
|
|
writer.by_ref(),
|
|
)
|
|
.unwrap();
|
|
writeln!(writer).unwrap();
|
|
writer.flush().unwrap();
|
|
}
|
|
}
|
|
|
|
fn timestamp() -> u128 {
|
|
use std::time::{SystemTime, UNIX_EPOCH};
|
|
let start = SystemTime::now();
|
|
start
|
|
.duration_since(UNIX_EPOCH)
|
|
.expect("Time went backwards")
|
|
.as_millis()
|
|
}
|