activity-tracker 1.13.0

Captura atividades do sistema e resume com Ollama
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use super::{
    CF_ADD_BLOCK, CF_MACHINE, CF_MODEL, CF_NOTION_PAGE, CF_NOTION_TOKEN, CF_PROMPT, CF_SLACK,
    CF_URL_OR_KEY,
};
use crate::config::Config;
use crate::summarizer;

pub(super) fn cfg_cycle(current: &mut String, options: &[&str], dir: i32) {
    let pos = options
        .iter()
        .position(|&o| o == current.as_str())
        .unwrap_or(0);
    let n = options.len();
    let new_pos = ((pos as i32 + dir).rem_euclid(n as i32)) as usize;
    *current = options[new_pos].to_string();
}

pub(super) fn cfg_initial_value(
    cfg: &Config,
    cursor: usize,
    cf_add_git_path: usize,
) -> Option<String> {
    match cursor {
        CF_MODEL => Some(cfg.model.clone()),
        CF_URL_OR_KEY => Some(if cfg.provider == "ollama" {
            cfg.ollama_url.clone()
        } else {
            cfg.api_key.clone().unwrap_or_default()
        }),
        CF_PROMPT => Some(
            cfg.custom_prompt
                .clone()
                .unwrap_or_else(|| summarizer::DEFAULT_PROMPT_TEMPLATE.to_string()),
        ),
        CF_MACHINE => Some(cfg.machine_name.clone().unwrap_or_default()),
        CF_NOTION_TOKEN => Some(cfg.notion_token.clone().unwrap_or_default()),
        CF_NOTION_PAGE => Some(cfg.notion_page_id.clone().unwrap_or_default()),
        CF_SLACK => Some(cfg.slack_webhook.clone().unwrap_or_default()),
        CF_ADD_BLOCK => Some(String::new()),
        c if c > CF_ADD_BLOCK && c < cf_add_git_path => {
            cfg.blocked_patterns.get(c - CF_ADD_BLOCK - 1).cloned()
        }
        c if c == cf_add_git_path => Some(String::new()),
        c if c > cf_add_git_path => cfg.ignored_git_paths.get(c - cf_add_git_path - 1).cloned(),
        _ => None,
    }
}

pub(super) fn cfg_apply(cfg: &mut Config, cursor: usize, value: String, cf_add_git_path: usize) {
    let v = value.trim().to_string();
    match cursor {
        CF_MODEL => cfg.model = v,
        CF_URL_OR_KEY => {
            if cfg.provider == "ollama" {
                cfg.ollama_url = v;
            } else {
                cfg.api_key = if v.is_empty() { None } else { Some(v) };
            }
        }
        CF_PROMPT => cfg.custom_prompt = if v.is_empty() { None } else { Some(v) },
        CF_MACHINE => cfg.machine_name = if v.is_empty() { None } else { Some(v) },
        CF_NOTION_TOKEN => cfg.notion_token = if v.is_empty() { None } else { Some(v) },
        CF_NOTION_PAGE => cfg.notion_page_id = if v.is_empty() { None } else { Some(v) },
        CF_SLACK => cfg.slack_webhook = if v.is_empty() { None } else { Some(v) },
        CF_ADD_BLOCK => {
            if !v.is_empty() {
                cfg.blocked_patterns.push(v);
            }
        }
        c if c > CF_ADD_BLOCK && c < cf_add_git_path => {
            let idx = c - CF_ADD_BLOCK - 1;
            if idx < cfg.blocked_patterns.len() {
                if v.is_empty() {
                    cfg.blocked_patterns.remove(idx);
                } else {
                    cfg.blocked_patterns[idx] = v;
                }
            }
        }
        c if c == cf_add_git_path => {
            if !v.is_empty() {
                cfg.ignored_git_paths.push(v);
            }
        }
        c if c > cf_add_git_path => {
            let idx = c - cf_add_git_path - 1;
            if idx < cfg.ignored_git_paths.len() {
                if v.is_empty() {
                    cfg.ignored_git_paths.remove(idx);
                } else {
                    cfg.ignored_git_paths[idx] = v;
                }
            }
        }
        _ => {}
    }
}