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
|
use crate::{
parser::{GitReceivePackArgs, GitUploadPackArgs},
ShackleError,
};
use git2::{Repository, RepositoryInitOptions};
use std::{
path::{Path, PathBuf},
process::Command,
};
use user_info::get_username;
pub struct GitInitResult {
pub path: PathBuf,
}
fn personal_git_dir() -> Result<PathBuf, ShackleError> {
let username = get_username().ok_or(ShackleError::UserReadError)?;
Ok(Path::new("git").join(username))
}
fn is_valid_personal_git_repo(path: &PathBuf) -> Result<bool, ShackleError> {
let valid_dir = personal_git_dir()?.canonicalize()?;
let canonical_path = path.canonicalize()?;
let relative_path = match canonical_path.strip_prefix(&valid_dir) {
Ok(relative_path) => relative_path,
Err(_) => {
return Ok(false);
}
};
if relative_path.parent() != Some(Path::new(""))
|| relative_path.extension().map(|ext| ext == "git") != Some(true)
{
return Ok(false);
}
Ok(true)
}
pub fn init(repo_name: &str) -> Result<GitInitResult, ShackleError> {
let path = personal_git_dir()?.join(repo_name).with_extension("git");
Repository::init_opts(
&path,
&RepositoryInitOptions::new()
.bare(true)
.mkdir(true)
.no_reinit(true),
)?;
Ok(GitInitResult { path })
}
pub fn upload_pack(upload_pack_args: &GitUploadPackArgs) -> Result<(), ShackleError> {
let mut command = Command::new("git-upload-pack");
if !is_valid_personal_git_repo(&upload_pack_args.directory)? {
return Err(ShackleError::InvalidDirectory);
}
if upload_pack_args.strict {
command.arg("strict");
}
if upload_pack_args.no_strict {
command.arg("no-strict");
}
if let Some(timeout) = upload_pack_args.timeout {
command.args(["timeout", &timeout.to_string()]);
}
if upload_pack_args.stateless_rpc {
command.arg("stateless-rpc");
}
if upload_pack_args.advertise_refs {
command.arg("advertise-refs");
}
command.arg(&upload_pack_args.directory);
command.spawn()?.wait()?;
Ok(())
}
pub fn receive_pack(receive_pack_args: &GitReceivePackArgs) -> Result<(), ShackleError> {
let mut command = Command::new("git-receive-pack");
if receive_pack_args.http_backend_info_refs {
command.arg("--http-backend-info-refs");
}
command.arg(&receive_pack_args.directory);
command.spawn()?.wait()?;
Ok(())
}
|
