wynd
/
recipe-sync-server-rs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added support for symlink exports

master
Wynd 2026-02-28 15:14:02 +02:00
parent 2738f0e01e
commit a4c4b40d99
1 changed files with 9 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/main.rs
View File

@ -2,7 +2,7 @@ use std::{
fs, fs,
io::Write, io::Write,
net::{SocketAddr, TcpListener, TcpStream}, net::{SocketAddr, TcpListener, TcpStream},
path::Path, path::{Path, PathBuf},
result, result,
str::FromStr, str::FromStr,
}; };
@ -23,7 +23,7 @@ fn main() -> Result<()> {
let ip = args.ip.unwrap_or_default(); let ip = args.ip.unwrap_or_default();
let port = args.port.unwrap_or_default(); let port = args.port.unwrap_or_default();
let export = args.export; let export = fs::canonicalize(args.export).expect("failed to get absolute export path");
let allowed = args.allowed_devices; let allowed = args.allowed_devices;
let socket = SocketAddr::from_str(&format!("{ip}:{port}")).unwrap(); let socket = SocketAddr::from_str(&format!("{ip}:{port}")).unwrap();
@ -41,7 +41,7 @@ fn main() -> Result<()> {
fn handle_connection( fn handle_connection(
mut conn: TcpStream, mut conn: TcpStream,
export: String, export: PathBuf,
allowed_devices: Option<Vec<String>>, allowed_devices: Option<Vec<String>>,
) -> Result<()> { ) -> Result<()> {
let mut paths = vec![]; let mut paths = vec![];
@ -66,6 +66,11 @@ fn handle_connection(
walk_dir(&export, &mut paths); walk_dir(&export, &mut paths);
let export_path = export
.as_path()
.to_str()
.expect("invalid export path, aborting");
let files_sent = paths.len(); let files_sent = paths.len();
log::info!("Sending {files_sent} files to {remote_ip}"); log::info!("Sending {files_sent} files to {remote_ip}");
@ -73,7 +78,7 @@ fn handle_connection(
buffer.write_usize(files_sent); buffer.write_usize(files_sent);
for file in paths { for file in paths {
let mut path = file.replace(&export, ""); let mut path = file.replace(export_path, "");
if path.starts_with("/") { if path.starts_with("/") {
path.remove(0); path.remove(0);
} }