feat(api): extend commit and diff services with new functionality
- Add FindCommit, ListCommitsByOid, CommitIsAncestor RPCs to CommitService - Add CheckObjectsExist, CommitsByMessage, GetCommitStats RPCs to CommitService - Add LastCommitForPath, CountCommits, CountDivergingCommits RPCs to CommitService - Add RawDiff, RawPatch, FindChangedPaths RPCs to DiffService - Add FindMergeBase, WriteRef, SearchFilesByContent RPCs to RepositoryService - Add SearchFilesByName, ObjectsSize, RepositorySize RPCs to RepositoryService - Add FindLicense, OptimizeRepository, GetRawChanges RPCs to RepositoryService - Add FetchRemote, CreateRepositoryFromURL RPCs to RepositoryService - Implement server handlers for all new RPC methods - Add new modules for commit counting, finding, and querying features - Add new modules for diff changed paths and raw operations - Add new modules for refs and remote operations - Remove unnecessary comments from various source files - Update proto definitions with new message types and service methods
This commit is contained in:
zhenyi
@@ -0,0 +1,155 @@
|
||||
use crate::bare::GitBare;
|
||||
use crate::error::GitResult;
|
||||
use crate::paginate;
|
||||
use crate::pb::*;
|
||||
|
||||
impl GitBare {
|
||||
/// Find all refs pointing to a given OID.
|
||||
pub fn find_refs_by_oid(&self, request: FindRefsByOidRequest) -> GitResult<FindRefsByOidResponse> {
|
||||
crate::sanitize::validate_revision(&request.oid)?;
|
||||
|
||||
let mut args = vec![
|
||||
"--git-dir".to_string(),
|
||||
self.bare_dir.to_string_lossy().into_owned(),
|
||||
"for-each-ref".to_string(),
|
||||
"--format=%(refname)%00%(objectname)%00%(symref)".to_string(),
|
||||
format!("--points-at={}", request.oid),
|
||||
];
|
||||
|
||||
if let Some(ref filter) = request.filter {
|
||||
for prefix in &filter.prefixes {
|
||||
args.push(prefix.clone());
|
||||
}
|
||||
if filter.limit > 0 {
|
||||
args.push(format!("--count={}", filter.limit));
|
||||
}
|
||||
}
|
||||
|
||||
let output = std::process::Command::new("git")
|
||||
.args(&args)
|
||||
.output()
|
||||
.map_err(|e| crate::error::GitError::CommandFailed {
|
||||
status_code: None,
|
||||
stderr: e.to_string(),
|
||||
})?;
|
||||
|
||||
let stdout = String::from_utf8_lossy(&output.stdout);
|
||||
let mut refs = Vec::new();
|
||||
|
||||
for line in stdout.lines() {
|
||||
let parts: Vec<&str> = line.split('\0').collect();
|
||||
if parts.len() >= 2 {
|
||||
let ref_name = parts[0].to_string();
|
||||
let oid = parts[1].to_string();
|
||||
let symref = parts.get(2).map(|s| s.to_string()).unwrap_or_default();
|
||||
refs.push(FoundRef {
|
||||
ref_name,
|
||||
target_oid: oid,
|
||||
symbolic: !symref.is_empty(),
|
||||
symbolic_target: symref,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
Ok(FindRefsByOidResponse { refs })
|
||||
}
|
||||
|
||||
/// List refs with optional prefix/pagination/sorting.
|
||||
pub fn list_all_refs(&self, request: ListRefsRequest) -> GitResult<ListRefsResponse> {
|
||||
let mut args = vec![
|
||||
"--git-dir".to_string(),
|
||||
self.bare_dir.to_string_lossy().into_owned(),
|
||||
"for-each-ref".to_string(),
|
||||
"--format=%(refname)%00%(objectname)%00%(symref)".to_string(),
|
||||
];
|
||||
|
||||
// Sort direction
|
||||
let sort_prefix = match SortDirection::try_from(request.sort_direction) {
|
||||
Ok(SortDirection::Asc) => "",
|
||||
Ok(SortDirection::Desc) | _ => "-",
|
||||
};
|
||||
args.push(format!("--sort={sort_prefix}refname"));
|
||||
|
||||
// Containing OIDs filter
|
||||
if let Some(first_oid) = request.containing_oids.first() {
|
||||
args.push(format!("--points-at={first_oid}"));
|
||||
}
|
||||
|
||||
// Prefix or pattern
|
||||
if !request.prefixes.is_empty() {
|
||||
for prefix in &request.prefixes {
|
||||
args.push(prefix.clone());
|
||||
}
|
||||
}
|
||||
|
||||
let output = std::process::Command::new("git")
|
||||
.args(&args)
|
||||
.output()
|
||||
.map_err(|e| crate::error::GitError::CommandFailed {
|
||||
status_code: None,
|
||||
stderr: e.to_string(),
|
||||
})?;
|
||||
|
||||
let stdout = String::from_utf8_lossy(&output.stdout);
|
||||
let mut all_refs: Vec<FoundRef> = Vec::new();
|
||||
|
||||
for line in stdout.lines() {
|
||||
let parts: Vec<&str> = line.split('\0').collect();
|
||||
if parts.len() >= 2 {
|
||||
let ref_name = parts[0].to_string();
|
||||
let oid = parts[1].to_string();
|
||||
let symref = parts.get(2).map(|s| s.to_string()).unwrap_or_default();
|
||||
|
||||
// Apply glob pattern filter if set
|
||||
if !request.pattern.is_empty() && !simple_glob_match(&request.pattern, &ref_name) {
|
||||
continue;
|
||||
}
|
||||
|
||||
all_refs.push(FoundRef {
|
||||
ref_name,
|
||||
target_oid: oid,
|
||||
symbolic: !symref.is_empty(),
|
||||
symbolic_target: symref,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
let _total = all_refs.len() as u64;
|
||||
let (paged, page_info) = paginate::paginate(&all_refs, request.pagination.as_ref());
|
||||
|
||||
Ok(ListRefsResponse {
|
||||
refs: paged,
|
||||
page_info: Some(page_info),
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
/// Simple glob match. Supports `*` (any chars) and `?` (single char).
|
||||
fn simple_glob_match(pattern: &str, name: &str) -> bool {
|
||||
let pat_bytes = pattern.as_bytes();
|
||||
let name_bytes = name.as_bytes();
|
||||
let mut pi = 0;
|
||||
let mut ni = 0;
|
||||
let mut star_pi = None;
|
||||
let mut star_ni = 0;
|
||||
|
||||
while ni < name_bytes.len() || pi < pat_bytes.len() {
|
||||
if pi < pat_bytes.len() && pat_bytes[pi] == b'*' {
|
||||
star_pi = Some(pi);
|
||||
star_ni = ni;
|
||||
pi += 1;
|
||||
} else if pi < pat_bytes.len() && ni < name_bytes.len()
|
||||
&& (pat_bytes[pi] == b'?' || pat_bytes[pi] == name_bytes[ni])
|
||||
{
|
||||
pi += 1;
|
||||
ni += 1;
|
||||
} else if let Some(sp) = star_pi {
|
||||
pi = sp + 1;
|
||||
star_ni += 1;
|
||||
ni = star_ni;
|
||||
} else {
|
||||
return false;
|
||||
}
|
||||
}
|
||||
true
|
||||
}
|
||||
@@ -1 +1,3 @@
|
||||
pub mod find_refs;
|
||||
pub mod list_refs;
|
||||
pub mod update_refs;
|
||||
|
||||
@@ -0,0 +1,149 @@
|
||||
use crate::bare::GitBare;
|
||||
use crate::error::GitResult;
|
||||
use crate::pb::*;
|
||||
|
||||
impl GitBare {
|
||||
/// Update multiple refs atomically using `git update-ref --stdin`.
|
||||
pub fn update_references(&self, request: UpdateReferencesRequest) -> GitResult<UpdateReferencesResponse> {
|
||||
let mut stdin_input = String::new();
|
||||
for update in &request.updates {
|
||||
crate::sanitize::validate_ref_name(&update.ref_name)?;
|
||||
crate::sanitize::validate_revision(&update.new_oid)?;
|
||||
if !update.old_oid.is_empty() {
|
||||
crate::sanitize::validate_revision(&update.old_oid)?;
|
||||
stdin_input.push_str(&format!(
|
||||
"update {} {}\0{}\n",
|
||||
update.ref_name, update.new_oid, update.old_oid
|
||||
));
|
||||
} else {
|
||||
stdin_input.push_str(&format!(
|
||||
"update {} {}\n",
|
||||
update.ref_name, update.new_oid
|
||||
));
|
||||
}
|
||||
}
|
||||
if stdin_input.is_empty() {
|
||||
return Ok(UpdateReferencesResponse::default());
|
||||
}
|
||||
|
||||
let output = std::process::Command::new("git")
|
||||
.args(["--git-dir", &self.bare_dir.to_string_lossy(), "update-ref", "--stdin", "-z"])
|
||||
.stdin(std::process::Stdio::piped())
|
||||
.stdout(std::process::Stdio::piped())
|
||||
.stderr(std::process::Stdio::piped())
|
||||
.output()
|
||||
.map_err(|e| crate::error::GitError::CommandFailed {
|
||||
status_code: None,
|
||||
stderr: e.to_string(),
|
||||
})?;
|
||||
|
||||
let stderr = String::from_utf8_lossy(&output.stderr).into_owned();
|
||||
if !output.status.success() {
|
||||
return Ok(UpdateReferencesResponse {
|
||||
failed_refs: request.updates.iter().map(|u| u.ref_name.clone()).collect(),
|
||||
error: stderr.trim().to_string(),
|
||||
});
|
||||
}
|
||||
Ok(UpdateReferencesResponse::default())
|
||||
}
|
||||
|
||||
/// Delete refs in bulk.
|
||||
pub fn delete_refs(&self, request: DeleteRefsRequest) -> GitResult<DeleteRefsResponse> {
|
||||
let mut failed = Vec::new();
|
||||
let mut error_msg = String::new();
|
||||
|
||||
for ref_name in &request.ref_names {
|
||||
crate::sanitize::validate_ref_name(ref_name)?;
|
||||
let output = std::process::Command::new("git")
|
||||
.args([
|
||||
"--git-dir",
|
||||
&self.bare_dir.to_string_lossy(),
|
||||
"update-ref",
|
||||
"-d",
|
||||
ref_name,
|
||||
])
|
||||
.stdout(std::process::Stdio::piped())
|
||||
.stderr(std::process::Stdio::piped())
|
||||
.output()
|
||||
.map_err(|e| crate::error::GitError::CommandFailed {
|
||||
status_code: None,
|
||||
stderr: e.to_string(),
|
||||
})?;
|
||||
|
||||
if !output.status.success() {
|
||||
failed.push(ref_name.clone());
|
||||
if error_msg.is_empty() {
|
||||
error_msg = String::from_utf8_lossy(&output.stderr).trim().to_string();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(DeleteRefsResponse {
|
||||
failed_refs: failed,
|
||||
error: error_msg,
|
||||
})
|
||||
}
|
||||
|
||||
/// Write a single ref with optional expected-old-oid check.
|
||||
pub fn write_ref(&self, request: WriteRefRequest) -> GitResult<WriteRefResponse> {
|
||||
crate::sanitize::validate_ref_name(&request.ref_name)?;
|
||||
crate::sanitize::validate_revision(&request.new_oid)?;
|
||||
|
||||
let mut args = vec![
|
||||
"--git-dir".to_string(),
|
||||
self.bare_dir.to_string_lossy().into_owned(),
|
||||
"update-ref".to_string(),
|
||||
request.ref_name.clone(),
|
||||
request.new_oid.clone(),
|
||||
];
|
||||
|
||||
if !request.old_oid.is_empty() {
|
||||
crate::sanitize::validate_revision(&request.old_oid)?;
|
||||
args.push(request.old_oid.clone());
|
||||
}
|
||||
|
||||
let output = std::process::Command::new("git")
|
||||
.args(&args)
|
||||
.stdout(std::process::Stdio::piped())
|
||||
.stderr(std::process::Stdio::piped())
|
||||
.output()
|
||||
.map_err(|e| crate::error::GitError::CommandFailed {
|
||||
status_code: None,
|
||||
stderr: e.to_string(),
|
||||
})?;
|
||||
|
||||
if !output.status.success() {
|
||||
return Ok(WriteRefResponse {
|
||||
ok: false,
|
||||
error: String::from_utf8_lossy(&output.stderr).trim().to_string(),
|
||||
});
|
||||
}
|
||||
|
||||
Ok(WriteRefResponse { ok: true, error: String::new() })
|
||||
}
|
||||
|
||||
/// Check if a ref exists.
|
||||
pub fn ref_exists(&self, request: RefExistsRequest) -> GitResult<RefExistsResponse> {
|
||||
crate::sanitize::validate_ref_name(&request.ref_name)?;
|
||||
let repo = self.gix_repo()?;
|
||||
let exists = repo.try_find_reference(&request.ref_name).ok().flatten().is_some();
|
||||
Ok(RefExistsResponse { exists })
|
||||
}
|
||||
|
||||
/// Find the default branch name.
|
||||
pub fn find_default_branch_name(&self) -> GitResult<FindDefaultBranchNameResponse> {
|
||||
let result = std::process::Command::new("git")
|
||||
.args(["--git-dir", &self.bare_dir.to_string_lossy(), "symbolic-ref", "HEAD"])
|
||||
.output()
|
||||
.map_err(|e| crate::error::GitError::CommandFailed {
|
||||
status_code: None,
|
||||
stderr: e.to_string(),
|
||||
})?;
|
||||
let name = String::from_utf8_lossy(&result.stdout)
|
||||
.trim()
|
||||
.strip_prefix("refs/heads/")
|
||||
.map(|b| b.to_string())
|
||||
.unwrap_or_default();
|
||||
Ok(FindDefaultBranchNameResponse { name })
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user