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
|
mod config;
mod utils;
use std::{
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!())
.subcommand(SubCommand::with_name("init"))
.subcommand(SubCommand::with_name("update"))
.get_matches();
if matches.subcommand_matches("init").is_some() {
init();
} else if matches.subcommand_matches("update").is_some() {
update();
}
}
fn get_config() -> Config {
let config_path = match env::var("GIT_MIRROR_CONFIG") {
Ok(c) => c,
_ => panic!("GIT_MIRROR_CONFIG isn't set."),
};
let mut config_file = match File::open(Path::new(&config_path)) {
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.")
}
let config: Config = match toml::from_str(&buf) {
Ok(c) => c,
Err(e) => panic!("Error parsing config file. {}", e),
};
config
}
fn init() {
let config = get_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() {
let config = get_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();
}
}
|
