genee 0.9.3

A habit tracker app with command-line interface
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

use anyhow::Result;
use chrono::{Duration, Local, NaiveDate};
use clap::Parser;
use genee::datafile;
use rand::prelude::*;
use std::char;
use std::path::PathBuf;

const A_IDX: u32 = b'A' as u32;
const Z_IDX: u32 = b'Z' as u32;

#[derive(Parser)]
struct Opt {
    #[arg(short, long)]
    file: PathBuf,

    #[arg(short, long)]
    rows: usize,

    #[arg(short, long)]
    cols: usize,
}

fn main() -> Result<()> {
    let opt = Opt::parse();
    let headers = generate_header(opt.cols);
    datafile::create_new_datafile(&opt.file, &headers)?;
    let mut data = datafile::open_datafile(&opt.file)?;
    data.update_data_batch(&generate_data(opt.cols, opt.rows))?;
    Ok(())
}

fn generate_header(cols: usize) -> Vec<String> {
    let mut rng = rand::rng();
    let mut header = vec![];
    for _col in 0..cols {
        let rand_char = A_IDX + rng.next_u32() % (Z_IDX - A_IDX);
        header.push(String::from(char::from_u32(rand_char).unwrap()));
    }
    header
}

fn generate_data(cols: usize, rows: usize) -> Vec<(NaiveDate, Vec<usize>)> {
    let mut rng = rand::rng();
    let mut data = vec![];
    for row in 0..rows {
        let mut row_data = vec![];
        for col in 1..cols + 1 {
            if rng.random_bool(0.5) {
                row_data.push(col)
            }
        }
        let date =
            Local::now().naive_local() + Duration::try_days(1 + row as i64 - rows as i64).unwrap();
        data.push((date.date(), row_data));
    }
    data
}