aboutsummaryrefslogtreecommitdiff
path: root/src/state.rs
blob: 475302b1401f444c21b7877dcdaa59f10a1e1e2d (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
use std::fs::read_dir;
use std::path::{Path, PathBuf};

use eyre::Result;
use git2::Repository;

pub struct State {
    pub root: PathBuf,
    pub data: Vec<Repository>,
}

// TODO figure out how to make this return impl Stream without living in actual hell
// Heartbreaking: E0733
fn find_repos_in(dir: impl AsRef<Path>) -> Result<Vec<Repository>> {
    let dir = dir.as_ref();
    let dir = read_dir(dir)?;
    let mut result = vec![];
    for subdir in dir {
        match subdir {
            Ok(subdir) => {
                match Repository::open_bare(subdir.path()) {
                    Ok(repo) => result.push(repo),
                    Err(err) if err.class() == git2::ErrorClass::Repository && err.code() == git2::ErrorCode::NotFound => {
                        result.extend(find_repos_in(subdir.path())?)
                    }
                    // TODO handle in a non-god-awful way
                    Err(err) => panic!("{}", err),
                }
            }
            Err(_) => {},
        }
    }
    Ok(result)
}

impl State {
    pub async fn discover(root: impl AsRef<Path>) -> Result<Self> {
        let root = root.as_ref();
        let data = find_repos_in(root)?;
        Ok(Self { root: root.to_owned(), data })
    }

    pub fn relative_path<'a>(&'a self, subdir: &'a Path) -> &'a Path {
        subdir.strip_prefix(&self.root).unwrap_or_else(|_| subdir.as_ref())
    }
}