Compare commits
5 Commits
v0.0.1
...
7463ca2069
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
7463ca2069 | ||
|
|
1235b38636 | ||
|
|
cf47c63bd7 | ||
|
|
0f9711aaa1 | ||
|
|
326b79a3b1 |
41
src/main.rs
41
src/main.rs
@@ -4,7 +4,7 @@ use std::fs::File;
|
|||||||
use std::time::{SystemTime, UNIX_EPOCH, Duration};
|
use std::time::{SystemTime, UNIX_EPOCH, Duration};
|
||||||
use std::ops::{Add, Sub};
|
use std::ops::{Add, Sub};
|
||||||
use clap::Parser;
|
use clap::Parser;
|
||||||
use chrono::DateTime;
|
use chrono::{TimeZone, Local, Timelike};
|
||||||
|
|
||||||
#[derive(Debug, Parser)]
|
#[derive(Debug, Parser)]
|
||||||
struct Flags {
|
struct Flags {
|
||||||
@@ -28,9 +28,7 @@ fn main() {
|
|||||||
let flags = Flags::parse();
|
let flags = Flags::parse();
|
||||||
|
|
||||||
add(&flags.f, &flags.add, &flags.tag).unwrap();
|
add(&flags.f, &flags.add, &flags.tag).unwrap();
|
||||||
log(&flags.f, &flags.since).unwrap();
|
log(&flags.f, &flags.log, &flags.since).unwrap();
|
||||||
|
|
||||||
println!("{:?}", flags);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
fn add(f: &String, x: &Option<String>, tag: &Option<String>) -> Result<(), String> {
|
fn add(f: &String, x: &Option<String>, tag: &Option<String>) -> Result<(), String> {
|
||||||
@@ -51,23 +49,26 @@ fn add(f: &String, x: &Option<String>, tag: &Option<String>) -> Result<(), Strin
|
|||||||
#[derive(Debug, Serialize, Clone)]
|
#[derive(Debug, Serialize, Clone)]
|
||||||
struct Log {
|
struct Log {
|
||||||
t: String,
|
t: String,
|
||||||
d: u8,
|
d: f32,
|
||||||
xs: Vec<LogX>,
|
xs: Vec<LogX>,
|
||||||
}
|
}
|
||||||
#[derive(Debug, Serialize, Clone)]
|
#[derive(Debug, Serialize, Clone)]
|
||||||
struct LogX {
|
struct LogX {
|
||||||
d: u8,
|
d: f32,
|
||||||
x: String,
|
x: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
fn log(f: &String, since: &Option<String>) -> Result<(), String> {
|
fn log(f: &String, enabled: &bool, since: &Option<String>) -> Result<(), String> {
|
||||||
|
if !enabled {
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
let since = parse_time(since)?;
|
let since = parse_time(since)?;
|
||||||
let tsheet = load(&f)?;
|
let tsheet = load(&f)?;
|
||||||
let tsheet = tsheet.since(since);
|
let tsheet = tsheet.since(since);
|
||||||
let tsheet = tsheet.sorted();
|
let tsheet = tsheet.sorted();
|
||||||
|
|
||||||
let mut result = vec![];
|
let mut result = vec![];
|
||||||
let mut curr = Log{t: "".to_string(), d: 0, xs: vec![]};
|
let mut curr = Log{t: "".to_string(), d: 0.0, xs: vec![]};
|
||||||
for i in 0..tsheet.xs.len() {
|
for i in 0..tsheet.xs.len() {
|
||||||
let x = &tsheet.xs[i];
|
let x = &tsheet.xs[i];
|
||||||
if curr.t != x.timestamp() {
|
if curr.t != x.timestamp() {
|
||||||
@@ -76,12 +77,12 @@ fn log(f: &String, since: &Option<String>) -> Result<(), String> {
|
|||||||
}
|
}
|
||||||
curr.xs.truncate(0);
|
curr.xs.truncate(0);
|
||||||
curr.t = x.timestamp();
|
curr.t = x.timestamp();
|
||||||
curr.d = 0;
|
curr.d = 0.0;
|
||||||
}
|
}
|
||||||
let d = match curr.xs.len() {
|
let d = match curr.xs.len() {
|
||||||
0 if x.x.len() == 0 => 0,
|
0 if x.x.len() == 0 => 0.0,
|
||||||
0 => 1,
|
0 => 1.0,
|
||||||
_ => ((tsheet.xs[i].t - tsheet.xs[i-1].t + 60*30) / (60*60)) as u8,
|
_ => ((tsheet.xs[i].t - tsheet.xs[i-1].t) as f32 / (60.0*60.0)) as f32,
|
||||||
};
|
};
|
||||||
curr.t = x.timestamp();
|
curr.t = x.timestamp();
|
||||||
curr.xs.push(LogX{d: d, x: x.x.clone()});
|
curr.xs.push(LogX{d: d, x: x.x.clone()});
|
||||||
@@ -96,7 +97,7 @@ fn log(f: &String, since: &Option<String>) -> Result<(), String> {
|
|||||||
for log in result {
|
for log in result {
|
||||||
for x in log.xs {
|
for x in log.xs {
|
||||||
if x.x.len() > 0 {
|
if x.x.len() > 0 {
|
||||||
println!("{} ({}) {} ({})", log.t, log.d, x.x, x.d);
|
println!("{} ({}) {} ({:.1})", log.t, (0.5 + log.d) as i64, x.x, x.d + 0.05);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -107,19 +108,17 @@ fn parse_time(since: &Option<String>) -> Result<SystemTime, String> {
|
|||||||
match since {
|
match since {
|
||||||
Some(since) => {
|
Some(since) => {
|
||||||
match chrono::NaiveDate::parse_from_str(since, "%Y-%m-%d") {
|
match chrono::NaiveDate::parse_from_str(since, "%Y-%m-%d") {
|
||||||
Ok(dt) => {
|
Ok(nd) => {
|
||||||
|
let ndt = nd.and_hms_opt(1, 1, 1).unwrap();
|
||||||
|
let dt = Local.from_local_datetime(&ndt).unwrap();
|
||||||
Ok(UNIX_EPOCH.add(
|
Ok(UNIX_EPOCH.add(
|
||||||
Duration::from_secs(
|
Duration::from_secs(dt.timestamp() as u64)
|
||||||
dt.and_hms_opt(1, 1, 1)
|
|
||||||
.unwrap()
|
|
||||||
.timestamp() as u64
|
|
||||||
)
|
|
||||||
))
|
))
|
||||||
},
|
},
|
||||||
Err(msg) => Err(format!("failed to parse {}: {}", since, msg)),
|
Err(msg) => Err(format!("failed to parse {}: {}", since, msg)),
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
None => Ok(SystemTime::now().sub(Duration::from_secs(60*60*24*7))),
|
None => Ok(SystemTime::now().sub(Duration::from_secs(Local::now().hour() as u64*60*60))),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -221,7 +220,7 @@ impl X {
|
|||||||
}
|
}
|
||||||
|
|
||||||
fn timestamp(&self) -> String {
|
fn timestamp(&self) -> String {
|
||||||
let dt = DateTime::from_timestamp(self.t, 0).unwrap();
|
let dt = Local.timestamp_opt(self.t, 0).unwrap();
|
||||||
dt.format("%Y-%m-%d").to_string()
|
dt.format("%Y-%m-%d").to_string()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user