summaryrefslogtreecommitdiffstats
path: root/src/model/backend/worker.rs
blob: be347acaf73bf603c65fe311d822008ab41b5cd2 (plain)
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
use std::sync::Mutex;

use anyhow::anyhow;
use anyhow::Result;
use cached::Return;
use cached::TimedSizedCache;
use cached::proc_macro::cached;

use crate::model::backend::error::BackendError;
use crate::model::backend::messages::BackendRequest;
use crate::model::backend::messages::BackendResponse;
use crate::model::backend::channels::BackendRequestdReceiver;
use crate::model::backend::channels::BackendResponseSender;

pub struct BackendWorker {
    repo_name: String,
    repo: Mutex<git2::Repository>,
    receiver: BackendRequestdReceiver,
}

impl BackendWorker {
    pub fn new(
        repo_name: String,
        repo: Mutex<git2::Repository>,
        receiver: BackendRequestdReceiver,
    ) -> Self {
        BackendWorker { repo_name, repo, receiver }
    }

    pub fn run(self) {
        loop {
            if let Err(e) = self.handle() {
                log::error!("Error in backend: {}", e);
            }
        }
    }

    fn handle(&self) -> std::result::Result<(), BackendError> {
        match self.receiver.recv().map_err(|_| BackendError::ChannelError)? {
            (BackendRequest::BranchList, chnl) => self.send_branch_list(chnl),
            (BackendRequest::TagList, chnl) => self.send_tag_list(chnl),
        }
    }

    fn send_branch_list(&self, sender: BackendResponseSender) -> std::result::Result<(), BackendError> {
        let repo_lock = self.repo
            .lock()
            .map_err(|_| BackendError::LockError)?;

        let branch_names = get_branch_names(&self.repo_name, &*repo_lock)
            .map_err(|e| BackendError::Str(e.to_string()))?;

        sender.send(BackendResponse::BranchList(branch_names.value))
            .map_err(|_| BackendError::ChannelError)
    }

    fn send_tag_list(&self, sender: BackendResponseSender) -> std::result::Result<(), BackendError> {
        let repo_lock = self.repo
            .lock()
            .map_err(|_| BackendError::LockError)?;

        let tag_names = get_tag_names(&self.repo_name, &*repo_lock)
            .map_err(|e| BackendError::Str(e.to_string()))?;

        sender.send(BackendResponse::TagList(tag_names.value))
            .map_err(|_| BackendError::ChannelError)
    }
}

#[cached(
    type = "TimedSizedCache<String, Return<Vec<String>>>",
    create = "{ TimedSizedCache::with_size_and_lifespan(1, 60) }",
    convert = r#"{ _repo_name.to_owned() }"#,
    with_cached_flag = true,
    result = true,
)]
fn get_branch_names(_repo_name: &str, repo: &git2::Repository) -> Result<cached::Return<Vec<String>>> {
    repo.branches(None)?
        .map(|branch| {
            let branch = branch?.0;
            branch.name()?
                .ok_or_else(|| anyhow!("Branch name is not valid UTF8: {:?}", branch.name()))
                .map(String::from)
        })
        .collect::<Result<Vec<String>>>()
        .map(cached::Return::new)
}

#[cached(
    type = "TimedSizedCache<String, Return<Vec<String>>>",
    create = "{ TimedSizedCache::with_size_and_lifespan(1, 60) }",
    convert = r#"{ _repo_name.to_owned() }"#,
    with_cached_flag = true,
    result = true,
)]
fn get_tag_names(_repo_name: &str, repo: &git2::Repository) -> Result<cached::Return<Vec<String>>> {
    repo.tag_names(None)?
        .into_iter()
        .map(|tag| {
            tag.ok_or_else(|| anyhow!("Tag name is not valid UTF8: {:?}", tag))
                .map(String::from)
        })
        .collect::<Result<Vec<String>>>()
        .map(cached::Return::new)
}