zr 2.0.0

quick, simple zsh package manager
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

use std::fmt;
use std::fs::create_dir_all;
use std::io::Error;
use std::path::{Path, PathBuf};

use git2_credentials::CredentialHandler;

use crate::identifier::Identifier;
use crate::plugin::Plugin;

/// Plugins are collected into different `home`s
pub struct Plugins {
    plugins: Vec<Plugin>,
    cache: PathBuf,
}

impl Plugins {
    pub fn update(&self) -> Result<(), Error> {
        for plugin in &self.plugins {
            let dir = self.cache.join(&plugin.identifier.dir);
            let repo = git2::Repository::open(&dir).unwrap();
            let mut callbacks = git2::RemoteCallbacks::new();

            let git_config = git2::Config::open_default().unwrap();
            let mut credential_handler = CredentialHandler::new(git_config);
            callbacks.credentials(move |url, username, allowed| {
                credential_handler.try_next_credential(url, username, allowed)
            });

            callbacks.update_tips(|refspec, from, to| {
                eprintln!(
                    "{} {} {:.6} ↓ {:.6}",
                    &plugin.identifier.name,
                    refspec.replace("refs/heads/", ""),
                    from,
                    to
                );
                true
            });
            let mut options = git2::FetchOptions::new();
            options.remote_callbacks(callbacks);

            let mut remote = repo.find_remote("origin").unwrap();
            remote
                .fetch(&["refs/heads/*:refs/heads/*"], Some(&mut options), None)
                .unwrap();
        }
        Ok(())
    }

    pub fn new(cache: &Path, identifiers: Vec<Identifier>) -> Plugins {
        if !cache.exists() {
            create_dir_all(cache).expect("failed to create the cache directory");
        }

        let mut plugins: Vec<Plugin> = vec![];

        for identifier in identifiers {
            if let Some(plugin) = plugins
                .iter_mut()
                .find(|plugin| plugin.identifier == identifier)
            {
                if let Some(file) = &identifier.file {
                    plugin.files.insert(cache.join(&identifier.dir).join(file));
                }
            };

            if let Ok(plugin) = Plugin::new(cache, identifier) {
                plugins.push(plugin);
            }
        }

        Plugins {
            plugins,
            cache: cache.to_path_buf(),
        }
    }
}

impl fmt::Display for Plugins {
    fn fmt(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
        writeln!(formatter, "export _ZR=$0")?;
        for plugin in &self.plugins {
            writeln!(formatter, "{}", plugin)?;
        }
        Ok(())
    }
}