mps-rs 1.6.0

MPS — plain-text personal productivity CLI (Rust)
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
59
60
61
62
63
64
65
66
67

use std::collections::HashMap;
use anyhow::{bail, Result};
use chrono::{Local, NaiveDate};
use colored::Colorize;
use crate::config::Config;
use crate::date_parse;
use crate::store::Store;

/// Update an element's typed attributes in-place.
pub fn run(
    config:     &Config,
    ref_path:   &str,
    status:     Option<String>,
    start_time: Option<String>,
    end_time:   Option<String>,
    at:         Option<String>,
    date:       Option<String>,
) -> Result<()> {
    let mut new_attrs: HashMap<String, String> = HashMap::new();
    if let Some(s) = status     { new_attrs.insert("status".into(), s); }
    if let Some(s) = start_time { new_attrs.insert("start".into(),  s); }
    if let Some(e) = end_time   { new_attrs.insert("end".into(),    e); }
    if let Some(a) = at         { new_attrs.insert("at".into(),     a); }

    if new_attrs.is_empty() {
        bail!("No attributes specified. Use --status, --start-time, --end-time, or --at.");
    }

    let d = resolve_date(date)?;
    let store = Store::new(&config.storage_dir);

    match store.rewrite_element(ref_path, &new_attrs, d)? {
        true  => println!("  {} {}", "updated".green(), ref_path),
        false => bail!("Could not update '{}'. Check the ref is correct.", ref_path),
    }
    Ok(())
}

/// Mark a task as done — shorthand for `update REFPATH --status done`.
pub fn run_done(
    config:   &Config,
    ref_path: &str,
    date:     Option<String>,
) -> Result<()> {
    let mut new_attrs = HashMap::new();
    new_attrs.insert("status".into(), "done".into());

    let d = store_date(date)?;
    let store = Store::new(&config.storage_dir);

    match store.rewrite_element(ref_path, &new_attrs, d)? {
        true  => println!("  {} {}", "done".green(), ref_path),
        false => bail!("Could not mark '{}' as done. Check the ref is correct.", ref_path),
    }
    Ok(())
}

fn resolve_date(date: Option<String>) -> Result<NaiveDate> {
    match date {
        Some(s) => Ok(date_parse::parse_date(&s)?),
        None    => Ok(Local::now().date_naive()),
    }
}

fn store_date(date: Option<String>) -> Result<NaiveDate> {
    resolve_date(date)
}