use std::sync::atomic::AtomicBool;
use std::time::{Duration, Instant};
use girt::push::{ForcePolicy, PreparedPush, PushCommand, PushLimits, send_local};
use girt::refs::RefName;
use girt::transport::TransportControl;
use girt::{
Commit, CommitFields, EntryMode, ObjectKind, Repository, Signature, Tag, TagFields, Tree,
TreeEntry,
};
fn repository() -> Result<(tempfile::TempDir, Repository), Box<dyn std::error::Error>> {
let root = tempfile::tempdir()?;
std::fs::create_dir(root.path().join("objects"))?;
std::fs::create_dir(root.path().join("refs"))?;
std::fs::write(root.path().join("HEAD"), b"ref: refs/heads/main\n")?;
std::fs::write(root.path().join("config"), b"[core]\nbare=true\n")?;
let repo = Repository::open(root.path())?;
Ok((root, repo))
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let (_source_root, source) = repository()?;
let (_destination_root, destination) = repository()?;
let loose = source.loose_objects();
let blob = loose.write_blob(b"Published by girt through native local storage\n")?;
let tree = loose.write_tree(&Tree::new(
girt::ObjectFormat::Sha1,
vec![TreeEntry {
mode: EntryMode::Blob,
name: b"hello.txt".to_vec(),
id: blob,
}],
)?)?;
let author = Signature::now(b"Push Example", b"push@example.com")?;
let commit = loose.write_commit(&Commit::new(CommitFields {
tree,
parents: vec![],
author: author.clone(),
committer: author.clone(),
extra_headers: vec![],
message: b"Local push fixture\n".to_vec(),
})?)?;
let tag = loose.write_tag(&Tag::new(TagFields {
target: commit,
target_kind: ObjectKind::Commit,
name: b"v1".to_vec(),
tagger: Some(author),
extra_headers: vec![],
message: b"First publication\n".to_vec(),
})?)?;
let cancel = AtomicBool::new(false);
let objects = source.objects()?;
let commands = vec![
PushCommand::new(RefName::new("refs/heads/main")?, None, commit)
.with_force(ForcePolicy::FastForwardOnly),
PushCommand::new(RefName::new("refs/tags/v1")?, None, tag)
.with_force(ForcePolicy::FastForwardOnly),
];
let prepared = PreparedPush::new(&objects, commands, &[], PushLimits::default(), &cancel)?;
let report = send_local(
destination.git_dir(),
girt::push::LocalPushRequest::new(&prepared),
TransportControl::new(&cancel)
.with_deadline(Some(Instant::now() + Duration::from_secs(30))),
|_| {},
)?;
if !report.all_succeeded() {
return Err(format!("native local push rejected updates: {report:?}").into());
}
let restored = destination
.objects()?
.read(blob)?
.ok_or("missing pushed blob")?;
assert_eq!(
restored.data(),
b"Published by girt through native local storage\n"
);
println!(
"Published branch and annotated tag: {} objects, {} pack bytes",
prepared.object_count(),
prepared.pack_bytes()
);
let repeated = PreparedPush::new(
&objects,
vec![
PushCommand::new(RefName::new("refs/heads/main")?, Some(commit), commit)
.with_force(ForcePolicy::FastForwardOnly),
],
&[commit],
PushLimits::default(),
&cancel,
)?;
let report = send_local(
destination.git_dir(),
girt::push::LocalPushRequest::new(&repeated),
TransportControl::new(&cancel)
.with_deadline(Some(Instant::now() + Duration::from_secs(30))),
|_| {},
)?;
if !report.all_succeeded() {
return Err(format!("repeat rejected: {report:?}").into());
}
assert_eq!(repeated.object_count(), 0);
println!(
"Repeated push: {} objects, {} pack bytes",
repeated.object_count(),
repeated.pack_bytes()
);
Ok(())
}