zhenyi
81 lines
2.7 KiB
Rust
81 lines
2.7 KiB
Rust
//! Copyright (c) 2022-2026 GitDataAi All rights reserved.
|
|
|
|
use crate::bare::GitBare;
|
|
use crate::error::GitResult;
|
|
use crate::paginate;
|
|
use crate::pb::{Branch, ListBranchesRequest, ListBranchesResponse};
|
|
|
|
impl GitBare {
|
|
pub fn list_branches(&self, request: ListBranchesRequest) -> GitResult<ListBranchesResponse> {
|
|
let repo = self.gix_repo()?;
|
|
|
|
let merged_set = if request.merged_into_head || request.not_merged_into_head {
|
|
let flag = if request.merged_into_head {
|
|
"--merged"
|
|
} else {
|
|
"--no-merged"
|
|
};
|
|
let check = duct::cmd(
|
|
"git",
|
|
[
|
|
"--git-dir",
|
|
self.bare_dir.to_string_lossy().as_ref(),
|
|
"branch",
|
|
flag,
|
|
"HEAD",
|
|
],
|
|
)
|
|
.stdout_capture()
|
|
.stderr_capture()
|
|
.unchecked()
|
|
.run();
|
|
match check {
|
|
Ok(out) => String::from_utf8_lossy(&out.stdout)
|
|
.lines()
|
|
.map(|l| l.trim().trim_start_matches("* ").to_string())
|
|
.collect::<Vec<_>>(),
|
|
Err(_) => Vec::new(),
|
|
}
|
|
} else {
|
|
Vec::new()
|
|
};
|
|
|
|
let mut branches: Vec<Branch> = Vec::new();
|
|
for r in repo.references()?.local_branches()? {
|
|
let mut r = r.map_err(|e| crate::error::GitError::Gix(e.to_string()))?;
|
|
let name = r.name().shorten().to_string();
|
|
if !request.pattern.is_empty() && !name.contains(&request.pattern) {
|
|
continue;
|
|
}
|
|
if request.merged_into_head && !merged_set.contains(&name) {
|
|
continue;
|
|
}
|
|
if request.not_merged_into_head && merged_set.contains(&name) {
|
|
continue;
|
|
}
|
|
let hex = r
|
|
.peel_to_id()
|
|
.ok()
|
|
.map(|id| id.to_string())
|
|
.unwrap_or_default();
|
|
branches.push(Branch {
|
|
name,
|
|
full_ref: r.name().to_string(),
|
|
target_oid: Some(self.oid_to_pb(hex)),
|
|
commit: None,
|
|
upstream: None,
|
|
is_default: false,
|
|
is_head: false,
|
|
is_merged: false,
|
|
is_detached: false,
|
|
});
|
|
}
|
|
paginate::apply_sort(&mut branches, request.sort_direction);
|
|
let (branches, page_info) = paginate::paginate(&branches, request.pagination.as_ref());
|
|
Ok(ListBranchesResponse {
|
|
branches,
|
|
page_info: Some(page_info),
|
|
})
|
|
}
|
|
}
|