the-way 0.7.0

A code snippets manager for your terminal
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

//! Code related to filtering search, list, and export results
use std::collections::HashSet;

use chrono::{Date, Utc};
use structopt::StructOpt;

use crate::the_way::{snippet::Snippet, TheWay};
use crate::utils;

#[derive(StructOpt, Debug)]
pub struct Filters {
    /// Snippets written in <language> (multiple with 'lang1 lang2')
    #[structopt(short, long)]
    pub(crate) languages: Option<Vec<String>>,
    /// Snippets with <tag> (multiple with 'tag1 tag2')
    #[structopt(short, long)]
    pub(crate) tags: Option<Vec<String>>,
    /// Snippets from <date> ("last friday" works too!)
    #[structopt(long, parse(try_from_str = utils::parse_date))]
    pub(crate) from: Option<Date<Utc>>,
    /// Snippets before <date>
    #[structopt(long, parse(try_from_str = utils::parse_date))]
    pub(crate) to: Option<Date<Utc>>,
}

impl TheWay {
    /// Filters a list of snippets by given language/tag/date
    pub(crate) fn filter_snippets(&self, filters: &Filters) -> color_eyre::Result<Vec<Snippet>> {
        let from_date = utils::date_start(filters.from);
        let to_date = utils::date_end(filters.to);
        let snippets: Option<Vec<_>> = match &filters.languages {
            Some(languages) => Some(
                self.get_snippets(
                    &languages
                        .iter()
                        .flat_map(|language| {
                            self.get_language_snippets(language).unwrap_or_default()
                        })
                        .collect::<Vec<_>>(),
                )?,
            ),
            None => None,
        };
        match (filters.tags.clone(), snippets) {
            (Some(tags), Some(snippets)) => Ok(snippets
                .into_iter()
                .filter(|snippet| {
                    snippet.in_date_range(from_date, to_date)
                        && tags.iter().any(|tag| snippet.has_tag(tag))
                })
                .collect()),
            (Some(tags), None) => {
                let indices = tags
                    .iter()
                    .flat_map(|tag| self.get_tag_snippets(tag).unwrap_or_default())
                    .collect::<HashSet<_>>()
                    .into_iter()
                    .collect::<Vec<_>>();
                Snippet::filter_in_date_range(self.get_snippets(&indices)?, from_date, to_date)
            }
            (None, Some(snippets)) => Snippet::filter_in_date_range(snippets, from_date, to_date),
            (None, None) => self.list_snippets_in_date_range(from_date, to_date),
        }
    }
}