typ2anki 0.1.0

Compile Typst flashcards into Anki decks
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

use std::collections::HashMap;

use crate::output::{OutputManager, OutputMessage};
use crate::utils::{self, hash_string};
use crate::{anki_api, config};

const CACHE_HASH_PART_LENGTH: usize = 34;

#[derive(Debug, Clone)]
pub struct CardsCacheManager {
    pub static_hash: String,
    pub old_cache: HashMap<String, String>,
    pub new_cache: HashMap<String, String>,
}

pub fn card_key(deck_name: &str, card_id: &str) -> String {
    format!("{}_{}", deck_name, card_id)
}

fn cache_concat_hashes_padding(hash1: &str, hash2: &str) -> String {
    let mut out = String::new();
    out.push_str(hash1);
    out.push_str(&"0".repeat(CACHE_HASH_PART_LENGTH.saturating_sub(hash1.len())));
    out.push_str(&"0".repeat(CACHE_HASH_PART_LENGTH.saturating_sub(hash2.len())));
    out.push_str(hash2);
    out
}

impl CardsCacheManager {
    pub fn init(ankiconf_hash: String, _output: &impl OutputManager) -> Self {
        let cfg = config::get();
        let static_hash =
            hash_string(format!("{}{}", ankiconf_hash, cfg.config_hash.as_ref().unwrap()).as_str());
        let cache: HashMap<String, String>;

        if cfg.skip_cache {
            cache = HashMap::new();
        } else {
            let s = anki_api::get_cards_cache_string().unwrap_or("{}".to_string());
            cache = serde_json::from_str(&s).unwrap_or(HashMap::new());
        }

        Self {
            static_hash,
            new_cache: HashMap::new(),
            old_cache: cache,
        }
    }

    pub fn add_card_hash(&mut self, deck_name: &str, card_id: &str, content_hash: &str) {
        self.new_cache.insert(
            card_key(deck_name, card_id),
            cache_concat_hashes_padding(&self.static_hash, content_hash),
        );
    }

    // Removes the new hash for a card (used when a card fails to compile/upload)
    pub fn remove_card_hash(&mut self, deck_name: &str, card_id: &str) {
        let key = card_key(deck_name, card_id);
        self.new_cache.remove(&key);
        self.old_cache.remove(&key);
    }

    pub fn detect_configuration_change(&mut self, output: &impl OutputManager) {
        let cfg = config::get();
        if cfg.skip_cache {
            return;
        }

        let mut config_changes = 0;
        let mut total_cards = 0;
        for (k, v) in &self.old_cache {
            total_cards += 1;
            if let Some(new_v) = self.new_cache.get(k) {
                if v[..CACHE_HASH_PART_LENGTH] != new_v[..CACHE_HASH_PART_LENGTH] {
                    config_changes += 1;
                }
            }
        }

        if cfg.dry_run {
            output.send(OutputMessage::DbgConfigChangeDetection {
                total_cards,
                config_changes,
            });
        }

        if cfg.recompile_on_config_change.read().unwrap().is_none() {
            if total_cards > 0 && (config_changes as f64) / (total_cards as f64) >= 0.2 {
                if output.ask_yes_no("A configuration or ankiconf.typ change has been detected. Do you wish to recompile all cards with this new config? (Y/n)") {
                    *cfg.recompile_on_config_change.write().unwrap() = Some(true);
                } else {
                    *cfg.recompile_on_config_change.write().unwrap() = Some(false);
                }
            }
        }
    }

    pub fn save_cache(&self, output: &impl OutputManager) {
        let cfg = config::get();
        if cfg.dry_run || cfg.skip_cache {
            return;
        }
        let push: HashMap<String, String> = self
            .old_cache
            .clone()
            .into_iter()
            .chain(self.new_cache.clone().into_iter())
            .collect();
        let s = serde_json::to_string(&push).unwrap_or("{}".to_string());
        let payload = utils::b64_encode(s);
        if let Err(e) = anki_api::upload_file(anki_api::CARDS_CACHE_FILENAME.into(), &payload) {
            output.send(OutputMessage::ErrorSavingCache(e));
        } else {
            output.send(OutputMessage::DbgSavedCache);
        }
    }
}