LibrePages/librepages
4
0
Fork 0
Code Issues 6 Pull requests 10 Projects Releases Packages Wiki Activity

feat: read files off of preview branches

Browse source
This commit is contained in:
Aravinth Manivannan 2022-09-08 00:58:27 +05:30
parent 53bd8c54bb
commit 313264a5e1
Signed by: realaravinth
GPG key ID: AD9F0F08E855ED88
1 changed files with 36 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
src/git.rs
View file

@ -114,6 +114,32 @@ impl ContentType {
/// in the repository. However, it will succeed if the output of [escape_spaces] is /// in the repository. However, it will succeed if the output of [escape_spaces] is
/// used in the request. /// used in the request.
pub fn read_file(repo_path: &str, path: &str) -> ServiceResult<FileInfo> { pub fn read_file(repo_path: &str, path: &str) -> ServiceResult<FileInfo> {
let repo = git2::Repository::open(repo_path).unwrap();
let head = repo.head().unwrap();
let tree = head.peel_to_tree().unwrap();
read_file_inner(&repo, path, &tree)
}
pub fn read_preview_file(
repo_path: &str,
preview_name: &str,
path: &str,
) -> ServiceResult<FileInfo> {
let repo = git2::Repository::open(repo_path).unwrap();
let branch = repo
.find_branch(preview_name, git2::BranchType::Local)
.unwrap();
// let tree = head.peel_to_tree().unwrap();
let branch = branch.into_reference();
let tree = branch.peel_to_tree().unwrap();
read_file_inner(&repo, path, &tree)
}
fn read_file_inner(
repo: &git2::Repository,
path: &str,
tree: &git2::Tree,
) -> ServiceResult<FileInfo> {
fn read_file(id: Oid, repo: &git2::Repository) -> ContentType { fn read_file(id: Oid, repo: &git2::Repository) -> ContentType {
let blob = repo.find_blob(id).unwrap(); let blob = repo.find_blob(id).unwrap();
ContentType::from_blob(&blob) ContentType::from_blob(&blob)
@ -145,9 +171,9 @@ pub fn read_file(repo_path: &str, path: &str) -> ServiceResult<FileInfo> {
content content
} }
let inner = |repo: &git2::Repository| -> ServiceResult<FileInfo> { let inner = |repo: &git2::Repository, tree: &git2::Tree| -> ServiceResult<FileInfo> {
let head = repo.head().unwrap(); // let head = repo.head().unwrap();
let tree = head.peel_to_tree().unwrap(); // let tree = head.peel_to_tree().unwrap();
let mut path = path; let mut path = path;
if path == "/" { if path == "/" {
let content = get_index_file(tree.id(), repo); let content = get_index_file(tree.id(), repo);
@ -185,8 +211,8 @@ pub fn read_file(repo_path: &str, path: &str) -> ServiceResult<FileInfo> {
} }
}; };
let repo = git2::Repository::open(repo_path).unwrap(); //let repo = git2::Repository::open(repo_path).unwrap();
inner(&repo) inner(repo, tree)
} }
#[cfg(test)] #[cfg(test)]
@ -257,5 +283,10 @@ pub mod tests {
assert_eq!(resp.filename, "README.txt"); assert_eq!(resp.filename, "README.txt");
assert_eq!(resp.content.bytes(), FILE_CONTENT.as_bytes()); assert_eq!(resp.content.bytes(), FILE_CONTENT.as_bytes());
assert_eq!(resp.mime.first().unwrap(), "text/plain"); assert_eq!(resp.mime.first().unwrap(), "text/plain");
let resp = read_preview_file(PATH, "master", "README.txt").unwrap();
assert_eq!(resp.filename, "README.txt");
assert_eq!(resp.content.bytes(), FILE_CONTENT.as_bytes());
assert_eq!(resp.mime.first().unwrap(), "text/plain");
} }
} }