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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
|
mod config;
mod utils;
use std::{
borrow::Cow,
env,
fs::File,
io::Read,
path::{Path, PathBuf},
};
use clap::*;
use git2::{build::CheckoutBuilder, Direction, Repository};
use config::*;
fn main() {
let matches = App::new(crate_name!())
.version(crate_version!())
.author(crate_authors!())
.about(crate_description!())
.arg(Arg::with_name("config")
.short("c")
.long("config")
.value_name("FILE")
.help("Sets a custom config file")
.takes_value(true))
.subcommand(SubCommand::with_name("init"))
.subcommand(SubCommand::with_name("update"))
.get_matches();
let config = get_config(matches.value_of("config"));
if matches.subcommand_matches("init").is_some() {
init(&config);
} else if matches.subcommand_matches("update").is_some() {
update(&config);
}
}
fn get_config(config: Option<&str>) -> Config {
let config_path = match config
{
Some(c) => Cow::Borrowed(c),
None => match env::var("GIT_MIRROR_CONFIG") {
Ok(c) => Cow::Owned(c),
_ => panic!("GIT_MIRROR_CONFIG isn't set."),
}
};
let s: &str = &config_path;
let mut config_file = match File::open(Path::new(s)) {
Ok(c) => c,
_ => panic!("Error opening config file."),
};
let mut buf = String::new();
if config_file.read_to_string(&mut buf).is_err() {
panic!("Error reading config file.")
}
match toml::from_str(&buf) {
Ok(c) => c,
Err(e) => panic!("Error parsing config file. {}", e),
}
}
fn init(config: &Config) {
let projects = match &config.projects {
Some(p) => p,
_ => return,
};
for project in projects {
let mut path = PathBuf::new();
path.push(Path::new(
project.path.as_ref().unwrap_or(&config.default_path),
));
match &project.name {
Some(name) => path.push(name),
None => path.push(utils::get_repo_name(&project.url))
};
if path.exists() {
println!("{:#?} already exists, skipping.", path);
continue;
}
let _repo = match Repository::clone(&project.url, path) {
Ok(r) => r,
Err(e) => {
eprintln!("{}", e);
continue;
}
};
}
}
fn update(config: &Config) {
let projects = match &config.projects {
Some(p) => p,
_ => return,
};
for project in projects {
let mut path = PathBuf::new();
path.push(Path::new(
project.path.as_ref().unwrap_or(&config.default_path),
));
match &project.name {
Some(name) => path.push(name),
None => path.push(utils::get_repo_name(&project.url))
};
if !path.exists() {
println!("{:#?} doesn't exist, skipping.", &path);
continue;
}
let repo = match Repository::open(&path) {
Ok(r) => r,
Err(e) => {
eprintln!("{}", e);
continue;
}
};
let mut remote = repo.find_remote("origin").unwrap();
remote.connect(Direction::Fetch).unwrap();
let branch_buf = remote.default_branch().unwrap();
let branch = branch_buf.as_str().unwrap();
if remote.fetch(&[branch], None, None).is_err() {
eprintln!("Failed fetching branch {} for {:#?}", branch, &path);
continue;
}
let fetch_head = repo.find_reference("FETCH_HEAD").unwrap();
let com = repo.reference_to_annotated_commit(&fetch_head).unwrap();
let mut branch_ref = repo.find_reference(&branch).unwrap();
branch_ref.set_target(com.id(), "IDK").unwrap();
repo.set_head(&branch).unwrap();
repo.checkout_head(Some(CheckoutBuilder::default().force()))
.unwrap();
}
}
|
