Compare commits
8 Commits
v0.0.1
...
09e1c57f32
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
09e1c57f32 | ||
|
|
fcb144b437 | ||
|
|
e6ceef2ead | ||
|
|
7463ca2069 | ||
|
|
1235b38636 | ||
|
|
cf47c63bd7 | ||
|
|
0f9711aaa1 | ||
|
|
326b79a3b1 |
62
src/main.rs
62
src/main.rs
@@ -4,7 +4,7 @@ use std::fs::File;
|
||||
use std::time::{SystemTime, UNIX_EPOCH, Duration};
|
||||
use std::ops::{Add, Sub};
|
||||
use clap::Parser;
|
||||
use chrono::DateTime;
|
||||
use chrono::{TimeZone, Local, Timelike};
|
||||
|
||||
#[derive(Debug, Parser)]
|
||||
struct Flags {
|
||||
@@ -22,15 +22,25 @@ struct Flags {
|
||||
|
||||
#[arg(short = 't', long = "tag")]
|
||||
tag: Option<String>,
|
||||
|
||||
#[arg(short = 'c', long = "clock")]
|
||||
clock: bool,
|
||||
}
|
||||
|
||||
fn main() {
|
||||
let flags = Flags::parse();
|
||||
let mut flags = Flags::parse();
|
||||
flags.log = flags.log || flags.since.is_some();
|
||||
|
||||
clock(&flags.f, &flags.clock).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 clock(f: &String, clock: &bool) -> Result<(), String> {
|
||||
match clock {
|
||||
true => add(&f, &Some("".to_string()), &None),
|
||||
false => Ok(()),
|
||||
}
|
||||
}
|
||||
|
||||
fn add(f: &String, x: &Option<String>, tag: &Option<String>) -> Result<(), String> {
|
||||
@@ -51,23 +61,26 @@ fn add(f: &String, x: &Option<String>, tag: &Option<String>) -> Result<(), Strin
|
||||
#[derive(Debug, Serialize, Clone)]
|
||||
struct Log {
|
||||
t: String,
|
||||
d: u8,
|
||||
d: f32,
|
||||
xs: Vec<LogX>,
|
||||
}
|
||||
#[derive(Debug, Serialize, Clone)]
|
||||
struct LogX {
|
||||
d: u8,
|
||||
d: f32,
|
||||
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 tsheet = load(&f)?;
|
||||
let tsheet = tsheet.since(since);
|
||||
let tsheet = tsheet.sorted();
|
||||
|
||||
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() {
|
||||
let x = &tsheet.xs[i];
|
||||
if curr.t != x.timestamp() {
|
||||
@@ -76,15 +89,20 @@ fn log(f: &String, since: &Option<String>) -> Result<(), String> {
|
||||
}
|
||||
curr.xs.truncate(0);
|
||||
curr.t = x.timestamp();
|
||||
curr.d = 0;
|
||||
curr.d = 0.0;
|
||||
}
|
||||
let d = match curr.xs.len() {
|
||||
0 if x.x.len() == 0 => 0,
|
||||
0 => 1,
|
||||
_ => ((tsheet.xs[i].t - tsheet.xs[i-1].t + 60*30) / (60*60)) as u8,
|
||||
0 if x.x.len() == 0 => 0.0,
|
||||
0 => 1.0,
|
||||
_ => ((tsheet.xs[i].t - tsheet.xs[i-1].t) as f32 / (60.0*60.0)) as f32,
|
||||
};
|
||||
match x.x.len() {
|
||||
0 => {},
|
||||
_ => {
|
||||
curr.t = x.timestamp();
|
||||
curr.xs.push(LogX{d: d, x: x.x.clone()});
|
||||
},
|
||||
};
|
||||
curr.t = x.timestamp();
|
||||
curr.xs.push(LogX{d: d, x: x.x.clone()});
|
||||
}
|
||||
if curr.xs.len() > 0 {
|
||||
result.push(curr.clone());
|
||||
@@ -96,7 +114,7 @@ fn log(f: &String, since: &Option<String>) -> Result<(), String> {
|
||||
for log in result {
|
||||
for x in log.xs {
|
||||
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 +125,17 @@ fn parse_time(since: &Option<String>) -> Result<SystemTime, String> {
|
||||
match since {
|
||||
Some(since) => {
|
||||
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(
|
||||
Duration::from_secs(
|
||||
dt.and_hms_opt(1, 1, 1)
|
||||
.unwrap()
|
||||
.timestamp() as u64
|
||||
)
|
||||
Duration::from_secs(dt.timestamp() as u64)
|
||||
))
|
||||
},
|
||||
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 +237,7 @@ impl X {
|
||||
}
|
||||
|
||||
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()
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user