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
|
mod config;
use std::{
env,
fs::File,
io::Read,
path::{Path, PathBuf},
};
use clap::*;
use git2::{build::CheckoutBuilder, AutotagOption, Direction, FetchOptions, 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),
));
path.push(project.name);
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),
));
path.push(project.name);
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 fo = FetchOptions::new();
fo.download_tags(AutotagOption::All);
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], Some(&mut fo), 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();
}
}
|
