Skip to content
Merged
Show file tree
Hide file tree
Changes from 3 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
92 changes: 92 additions & 0 deletions examples/transfer.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,92 @@
use std::path::PathBuf;

use iroh::{protocol::Router, Endpoint};
use iroh_blobs::{net_protocol::Blobs, store::mem::MemStore, ticket::BlobTicket};

#[tokio::main]
async fn main() -> anyhow::Result<()> {
// Create an endpoint, it allows creating and accepting
// connections in the iroh p2p world
let endpoint = Endpoint::builder().discovery_n0().bind().await?;

// We initialize an in-memory backing store for iroh-blobs
let store = MemStore::new();
// Then we initialize a struct that can accept blobs requests over iroh connections
let blobs = Blobs::new(&store, endpoint.clone(), None);

// Grab all passed in arguments, the first one is the binary itself, so we skip it.
let args: Vec<String> = std::env::args().skip(1).collect();
// Convert to &str, so we can pattern-match easily:
let arg_refs: Vec<&str> = args.iter().map(String::as_str).collect();

match arg_refs.as_slice() {
["send", filename] => {
let filename: PathBuf = filename.parse()?;
let abs_path = std::path::absolute(&filename)?;

println!("Hashing file.");

// When we import a blob, we get back a "tag" that refers to said blob in the store
// and allows us to control when/if it gets garbage-collected
let tag = store.blobs().add_path(abs_path).await?;

let node_id = endpoint.node_id();
let ticket = BlobTicket::new(node_id.into(), tag.hash, tag.format);

println!("File hashed. Fetch this file by running:");
println!(
"cargo run --example transfer -- receive {ticket} {}",
filename.display()
);

// For sending files we build a router that accepts blobs connections & routes them
// to the blobs protocol.
let router = Router::builder(endpoint)
.accept(iroh_blobs::ALPN, blobs)
.spawn();

tokio::signal::ctrl_c().await?;

// Gracefully shut down the node
println!("Shutting down.");
router.shutdown().await?;
}
["receive", ticket, filename] => {
let filename: PathBuf = filename.parse()?;
let abs_path = std::path::absolute(filename)?;
let ticket: BlobTicket = ticket.parse()?;

println!("Starting download.");

// For receiving files, we create a "downloader" that allows us to fetch files
// from other nodes via iroh connections
store
.downloader(&endpoint)
.download(ticket.hash(), Some(ticket.node_addr().node_id))
.await?;

println!("Finished download.");
println!("Copying to destination.");

store.blobs().export(ticket.hash(), abs_path).await?;

println!("Finished copying.");

// Gracefully shut down the node
println!("Shutting down.");
endpoint.close().await;
}
_ => {
println!("Couldn't parse command line arguments: {args:?}");
println!("Usage:");
println!(" # to send:");
println!(" cargo run --example transfer -- send [FILE]");
println!(" # this will print a ticket.");
println!();
println!(" # to receive:");
println!(" cargo run --example transfer -- receive [TICKET] [FILE]");
}
}

Ok(())
}
2 changes: 1 addition & 1 deletion src/api/downloader.rs
Original file line number Diff line number Diff line change
Expand Up @@ -364,7 +364,7 @@ impl Downloader {
self.download_with_opts(DownloadOptions {
request,
providers,
strategy: SplitStrategy::Split,
strategy: SplitStrategy::None,
})
}

Expand Down
Loading