zhenyi
123 lines
4.1 KiB
Rust
123 lines
4.1 KiB
Rust
//! Copyright (c) 2022-2026 GitDataAi All rights reserved.
|
|
|
|
use crate::bare::GitBare;
|
|
use crate::error::GitResult;
|
|
use crate::pb::*;
|
|
|
|
const MAX_RAW_DIFF_OUTPUT_BYTES: usize = 64 * 1024 * 1024;
|
|
|
|
impl GitBare {
|
|
/// Stream raw diff output.
|
|
pub fn raw_diff(&self, request: RawDiffRequest) -> GitResult<Vec<RawDiffResponse>> {
|
|
let base = &request.base;
|
|
let head = &request.head;
|
|
crate::sanitize::validate_revision(base)?;
|
|
crate::sanitize::validate_revision(head)?;
|
|
|
|
let mut args = vec![
|
|
"--git-dir".to_string(),
|
|
self.bare_dir.to_string_lossy().into_owned(),
|
|
"diff".to_string(),
|
|
];
|
|
|
|
let mut pathspecs = Vec::new();
|
|
if let Some(ref opts) = request.options {
|
|
if opts.recursive {
|
|
args.push("--recursive".to_string());
|
|
}
|
|
if opts.include_binary {
|
|
args.push("--binary".to_string());
|
|
} else {
|
|
args.push("--no-binary".to_string());
|
|
}
|
|
for pathspec in &opts.pathspec {
|
|
crate::sanitize::validate_file_path(pathspec)?;
|
|
pathspecs.push(pathspec.clone());
|
|
}
|
|
}
|
|
|
|
args.push(base.clone());
|
|
args.push(head.clone());
|
|
if !pathspecs.is_empty() {
|
|
args.push("--".to_string());
|
|
args.extend(pathspecs);
|
|
}
|
|
|
|
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 Err(crate::error::GitError::CommandFailed {
|
|
status_code: output.status.code(),
|
|
stderr: String::from_utf8_lossy(&output.stderr).trim().to_string(),
|
|
});
|
|
}
|
|
|
|
if output.stdout.len() > MAX_RAW_DIFF_OUTPUT_BYTES {
|
|
return Err(crate::error::GitError::InvalidArgument(format!(
|
|
"raw diff output too large (max {MAX_RAW_DIFF_OUTPUT_BYTES} bytes)"
|
|
)));
|
|
}
|
|
|
|
const CHUNK_SIZE: usize = 32768;
|
|
let data = output.stdout;
|
|
let chunks: Vec<RawDiffResponse> = data
|
|
.chunks(CHUNK_SIZE)
|
|
.map(|c| RawDiffResponse { data: c.to_vec() })
|
|
.collect();
|
|
|
|
Ok(chunks)
|
|
}
|
|
|
|
/// Stream raw patch (format-patch) output.
|
|
pub fn raw_patch(&self, request: RawPatchRequest) -> GitResult<Vec<RawPatchResponse>> {
|
|
crate::sanitize::validate_revision(&request.base)?;
|
|
crate::sanitize::validate_revision(&request.head)?;
|
|
|
|
let range = format!("{}..{}", request.base, request.head);
|
|
|
|
let output = std::process::Command::new("git")
|
|
.args([
|
|
"--git-dir",
|
|
&self.bare_dir.to_string_lossy(),
|
|
"format-patch",
|
|
"--stdout",
|
|
&range,
|
|
])
|
|
.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 Err(crate::error::GitError::CommandFailed {
|
|
status_code: output.status.code(),
|
|
stderr: String::from_utf8_lossy(&output.stderr).trim().to_string(),
|
|
});
|
|
}
|
|
|
|
if output.stdout.len() > MAX_RAW_DIFF_OUTPUT_BYTES {
|
|
return Err(crate::error::GitError::InvalidArgument(format!(
|
|
"raw patch output too large (max {MAX_RAW_DIFF_OUTPUT_BYTES} bytes)"
|
|
)));
|
|
}
|
|
|
|
const CHUNK_SIZE: usize = 32768;
|
|
let data = output.stdout;
|
|
let chunks: Vec<RawPatchResponse> = data
|
|
.chunks(CHUNK_SIZE)
|
|
.map(|c| RawPatchResponse { data: c.to_vec() })
|
|
.collect();
|
|
|
|
Ok(chunks)
|
|
}
|
|
}
|