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
use cmd_lib::run_cmd;
use std::path::{Path, PathBuf};
pub fn repo(path: impl Into<String>) -> Downloader {
Downloader::new(path)
}
struct CopyRequest {
from: PathBuf,
to: PathBuf,
}
pub struct Downloader {
repo_path: String,
branch_name: String,
out_dir: PathBuf,
copy_requests: Vec<CopyRequest>,
}
impl Downloader {
fn new(repo: impl Into<String>) -> Self {
let cur_dir = std::env::current_dir().unwrap();
Self {
repo_path: repo.into(),
branch_name: "master".to_owned(),
out_dir: cur_dir,
copy_requests: vec![],
}
}
pub fn out_dir(mut self, path: impl AsRef<Path>) -> Self {
self.out_dir = path.as_ref().to_owned();
self
}
pub fn branch_name(mut self, name: impl Into<String>) -> Self {
self.branch_name = name.into();
self
}
pub fn add_file(mut self, src: impl AsRef<Path>, dst: impl AsRef<Path>) -> Self {
let from = src.as_ref().to_owned();
let to = dst.as_ref().to_owned();
let req = CopyRequest { from, to };
self.copy_requests.push(req);
self
}
pub fn exec(self) -> anyhow::Result<()> {
let old_pwd = std::env::current_dir()?;
let dir = tempfile::tempdir()?;
let dir_path = dir.path();
std::env::set_current_dir(dir_path)?;
let repo = &self.repo_path;
run_cmd! {
git init .;
git config core.sparsecheckout true;
git remote add origin $repo;
}?;
for req in &self.copy_requests {
let from = &req.from;
run_cmd! {
echo $from >> .git/info/sparse-checkout;
}?;
}
let branch_name = &self.branch_name;
run_cmd! {
git pull origin $branch_name;
}?;
for req in &self.copy_requests {
let from = &req.from;
let to = &req.to;
let to = self.out_dir.join(to);
let to_dir = to.parent().unwrap();
if !to_dir.exists() {
std::fs::create_dir_all(to_dir)?;
}
run_cmd! {
mv $from $to;
}?;
}
std::env::set_current_dir(old_pwd)?;
Ok(())
}
}