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
119
|
use std::env;
use std::process;
use getopts::{self, Options};
use git2::{self, Repository};
use thiserror::Error;
use crate::owner_repo::{self, OwnerRepo};
const GIT_CONFIG_PREFIX: &'static str = "githubSuggestion.";
#[derive(Debug, Error)]
pub enum Error {
#[error("Unable to parse arguments: {0}")]
Opts(#[from] getopts::Fail),
#[error("Error getting environment variable '{var}'")]
EnvVar {
source: env::VarError,
var: String,
},
#[error(transparent)]
OwnerRepo(#[from] owner_repo::Error),
#[error(transparent)]
Git(#[from] git2::Error),
}
pub struct Config<'a> {
pub github_token: String,
pub o_r: Result<OwnerRepo, owner_repo::Error>,
pub suggestions: Vec<String>,
opts: Options,
usage_brief: &'a str,
}
impl<'a> Config<'a> {
pub fn get(args: &[String], usage_brief: &'a str) -> Result<Self, Error> {
let mut opts = Options::new();
opts.optopt("", "github-token", "", "TOKEN");
opts.optopt(
"",
"remote",
"remote name, defaults to 'origin'",
"REMOTE",
);
opts.optflag("h", "help", "print this help menu");
let opt_matches = opts.parse(&args[1..])?;
if opt_matches.opt_present("h") {
print!("{}", opts.usage(&usage_brief));
process::exit(exitcode::USAGE);
}
let git_config = Repository::open(".")?.config()?;
let o_r = OwnerRepo::from_remote(
Self::remote(&opt_matches, &git_config)?.as_deref(),
);
Ok(Config {
github_token: Self::github_token(&opt_matches, &git_config)?,
o_r: o_r,
suggestions: opt_matches.free,
opts: opts,
usage_brief,
})
}
pub fn print_usage(&self) {
print!("{}", self.opts.usage(&self.usage_brief))
}
fn github_token(
opt_matches: &getopts::Matches,
git_config: &git2::Config,
) -> Result<String, Error> {
match opt_matches.opt_str("github-token") {
Some(t) => Ok(t),
None =>
match git_config.get_string(&git_config_key("githubToken")) {
Err(e) if e.code() == git2::ErrorCode::NotFound => {
let key = "GITHUB_TOKEN";
env::var(key)
.map_err(|e| Error::EnvVar {
source: e,
var: key.to_owned(),
})
},
r => r.map_err(|e| Error::Git(e)),
},
}
}
fn remote(
opt_matches: &getopts::Matches,
git_config: &git2::Config,
) -> Result<Option<String>, git2::Error> {
match opt_matches.opt_str("remote") {
Some(r) => Ok(Some(r)),
None => match git_config.get_string(&git_config_key("remote")) {
Err(e) if e.code() == git2::ErrorCode::NotFound => Ok(None),
r => r.map(|r| Some(r)),
},
}
}
}
fn git_config_key(key: &str) -> String {
format!("{}.{}", GIT_CONFIG_PREFIX, key)
}
|
