small folder support changes

src/main.rs

@@ -258,6 +258,10 @@ impl FileLocation {
  self.host.is_none() || (self.host.is_some() && self.file_path.exists())
  }
 
+ fn is_dir(&self) -> bool {
+ self.file_path.is_dir()
+ }
+
  async fn file_size(&self) -> io::Result<u64> {
  let metadata = tokio::fs::metadata(&self.file_path).await?;
  Ok(metadata.len())

src/receiver/mod.rs

@@ -37,6 +37,7 @@ pub(crate) async fn main(
  mut control_stream: TcpStream,
  remote_addr: IpAddr,
 ) -> Result<()> {
+ // TODO what if the source is a directory?
  if options.destination.file_path.is_dir() {
  info!("destination is a folder, reformatting path with target file");
 

src/sender/mod.rs

@@ -36,6 +36,17 @@ pub(crate) async fn main(
 ) -> Result<()> {
  info!("sending {} -> {}", options.source, options.destination);
 
+ if options.source.is_dir() {
+ for entry in options.source.file_path.read_dir()? {
+ if let Ok(entry) = entry {
+ println!("{:?}", entry);
+ entry.path().is_dir(); // lol
+ }
+ }
+
+ return Ok(());
+ }
+
  let file_size = options.source.file_size().await?;
  stats.total_data.store(file_size as usize, Relaxed);
  // send the file size to the remote client