rust-tcp-file-transfer/src/main.rs

use std::fs::{File, Metadata, self};
use std::time::Instant;
use std::net::{TcpListener, TcpStream};
use std::io::{Read, Write, self, BufWriter, BufReader, BufRead};
use std::env::{self};


const BUFFER_SIZE:usize = 512;

struct FileInfo
    {
        file:Option<File>,
        location:String,
        size_current:usize,
        metadata:Option<Metadata>,
    }
impl FileInfo 
    {
        fn reading_operations(&mut self, stream:&mut TcpStream)
            {
                self.read_metadata();
                match self.metadata
                    {
                        Some(ref mut metadata) =>
                            {
                                if Metadata::is_file(metadata)
                                    {
                                        self.read_file();
                                        self.file_to_byte(stream);
                                    }
                                else if Metadata::is_symlink(metadata)
                                    {
                                        self.read_file();
                                        self.file_to_byte(stream);
                                    }
                                else 
                                    {
                                        //path recognition and creation on the other side
                                        //std:path
                                        panic!("\n\tError: Folder Transfers've not Supported yet\n")
                                    }
                            }
                        None =>
                            {
                                println!("Error: Read Metadata -> {}", self.location);
                            }
                    }
            }
        fn writing_operations(&mut self, stream:&mut TcpStream)
            {
                self.write_file(stream);
            }
        fn read_metadata(&mut self)
            {
                //use match, there is a chance to fail creation. don't pass with just some
                self.metadata = Some(fs::metadata(&self.location).expect("Error: Read Metadata"));
            }
        fn read_file(&mut self)
            {
                match File::open(&self.location)
                    {
                        Ok(file) =>
                            {
                                self.file = Some(file);
                            }
                        Err(err_val) =>
                            {
                                println!("Error: Open File -> {} | Error: {}", self.location, err_val);
                                return;
                            }
                    }
                
            }
        fn file_to_byte(&mut self, stream:&mut TcpStream)
            {       
                let mut iteration = (self.metadata.as_ref().unwrap().len()/BUFFER_SIZE as u64)+1;
                let mut stream_writer = BufWriter::new(stream.try_clone().unwrap());
                let mut stream_reader = BufReader::new(stream.try_clone().unwrap());
                match stream_writer.write_all(self.metadata.as_ref().unwrap().len().to_string().as_bytes())
                    {
                        Ok(_) =>
                            {
                                match stream_writer.write_all(String::from("\n").as_bytes())
                                            {
                                                Ok(_) =>
                                                    {
                                                        println!("Done: Send Terminator -> {}", self.location);
                                                    }
                                                Err(err_val) =>
                                                    {
                                                        println!("Error: Send Terminator -> {} | Error: {}", self.location, err_val);
                                                        return;
                                                    }
                                            }
                                match stream_writer.flush()
                                    {
                                        Ok(_) =>
                                            {
                                                println!("Done: Flush Handshake -> {}", self.location);
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: Flush Handshake -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                                let mut handshake_callback:Vec<u8> = Vec::new();
                                match stream_reader.read_until( b'\n',&mut handshake_callback)
                                    {
                                        Ok(_) =>
                                            {
                                                println!("Done: Handshake Callback -> {}", self.location);
                                                handshake_callback.pop();
                                                if handshake_callback.as_slice() == self.metadata.as_ref().unwrap().len().to_string().as_bytes()
                                                    {
                                                        println!("Done: Handshake Correct -> {}", self.location);
                                                    }
                                                else 
                                                    {
                                                        println!("Error: Handshake Incorrect -> {}", self.location);
                                                        return;
                                                    }
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: Handshake Recv -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                            }
                        Err(err_val) =>
                            {
                                println!("Error: Handshake Send -> {} | Error: {}", self.location, err_val);
                                return;
                            }
                    }
                println!("Size = {}", self.metadata.as_ref().unwrap().len());
                println!("Iteration = {}", iteration);
                while iteration != 0
                    {
                        iteration -= 1;
                        let mut buffer = [0u8;BUFFER_SIZE];
                        let mut file_reader = BufReader::new(self.file.as_ref().unwrap());
                        if iteration != 0
                            {
                                match file_reader.read_exact(&mut buffer)
                                    {
                                        Ok(_) =>
                                            {
                                                self.size_current += buffer.len();
                                                println!("Size now = {}", self.size_current);
                                                //println!("{} | {} | {:#?}", iteration,buffer.len(), buffer);
                                                
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: File to Byte -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                            }
                        else 
                            {
                                let mut last_buffer:Vec<u8> = Vec::new();
                                match file_reader.read_to_end(&mut last_buffer)
                                    {
                                        Ok(read_size) =>
                                            {
                                                self.size_current += read_size;
                                                last_buffer.append(&mut buffer[..(self.metadata.as_ref().unwrap().len()%BUFFER_SIZE as u64) as usize].to_vec());
                                                buffer.copy_from_slice(&last_buffer);
                                                println!("Size now = {}", self.size_current);
                                                //println!("{} | {} | {:#?}", iteration,buffer.len(), buffer);
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: File to Byte Last -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                            }
                        match stream_writer.write_all(&mut buffer)
                            {
                                Ok(_) =>
                                    {
                                        println!("Done: Send Bytes -> {} | Iteration = {}", self.location, iteration);
                                    }
                                Err(err_val) =>
                                    {
                                        println!("Error: Send Bytes -> {} | Error: {}", self.location, err_val);
                                        return;
                                    }
                            }
                        match stream_writer.flush()
                            {
                                Ok(_) =>
                                    {
                                        //println!("Done: Flush -> {}", self.location);
                                    }
                                Err(err_val) =>
                                    {
                                        println!("Error: Flush -> {} | Error: {}", self.location, err_val);
                                        return;
                                    }
                            }
                    
                    
                    }
            }
        fn write_file(&mut self, stream:&mut TcpStream)
            {
                //use match, there is a chance to fail creation. don't pass with just some
                self.file = Some(File::create(&self.location).expect("Error: Create File"));
                let mut file_writer = BufWriter::new(self.file.as_ref().unwrap());
                let mut stream_reader = BufReader::new(stream.try_clone().unwrap());
                let mut stream_writer = BufWriter::new(stream.try_clone().unwrap());
                let size:u64;
                let mut handshake:Vec<u8> = Vec::new();
                match stream_reader.read_until(b'\n',&mut handshake)
                    {
                        Ok(_) =>
                            {
                                //read until and take
                                handshake.pop();
                                size = String::from_utf8(handshake.clone()).unwrap().parse().unwrap();
                                println!("Done: Handshake Recv -> {}", self.location);
                                match stream_writer.write_all(&mut handshake)
                                    {
                                        Ok(_) =>
                                            {
                                                println!("Done: Handshake Send -> {}", self.location);
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: Handshake Send -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                                match stream_writer.write_all(String::from("\n").as_bytes())
                                    {
                                        Ok(_) =>
                                            {
                                                println!("Done: Send Terminator -> {}", self.location);
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: Send Terminator -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                                match stream_writer.flush()
                                    {
                                        Ok(_) =>
                                            {
                                                println!("Done: Flush -> {}", self.location);
                                            }
                                        Err(err_val) =>
                                            {
                                                println!("Error: Flush -> {} | Error: {}", self.location, err_val);
                                                return;
                                            }
                                    }
                            }
                        Err(err_val) =>
                            {
                                println!("Error: Handshake Recv -> {} | Error: {}", self.location, err_val);
                                return;
                            }
                    }
                let mut iteration:u64 = (size/BUFFER_SIZE as u64)+1;
                println!("Size = {}", size);
                println!("Iteration = {}", iteration);
                while iteration != 0
                    {
                        iteration -= 1;
                        let mut buffer = [0u8;BUFFER_SIZE];
                        match stream_reader.read_exact(&mut buffer)
                            {
                                Ok(_) =>
                                    {
                                        self.size_current += buffer.len();
                                        println!("{} | {:#?}", iteration, buffer);
                                        if iteration != 0
                                            {
                                                match file_writer.write_all(&mut buffer)
                                                    {
                                                        Ok(_) =>
                                                            {
                                                                println!("Done: Write -> {} bytes | Iteration = {}", &mut self.size_current, iteration);
                                                            }
                                                        Err(err_val) => 
                                                            {
                                                                println!("Error: Write -> {} | Error: {}", self.location,err_val);
                                                                return;
                                                            }
                                                    }
                                            }
                                        else 
                                            {
                                                match file_writer.write_all(&mut buffer[..(size%BUFFER_SIZE as u64)as usize])
                                                    {
                                                        Ok(_) =>
                                                            {
                                                                println!("Done: Write Last -> {} bytes | Iteration = {}", &mut self.size_current, iteration);
                                                            }
                                                        Err(err_val) =>
                                                            {
                                                                println!("Error: Write Last -> {} | Error: {}", self.location,err_val);
                                                                return;
                                                            }
                                                    }
                                            }
                                    }
                                Err(err_val) =>
                                    {
                                        println!("{} | {:#?}", iteration, buffer);
                                        println!("Error: Recv Bytes -> {} | Error: {}", self.location, err_val);
                                        return;
                                    }
                            }
                    }         
            
                        
            }
    }
enum Connection
    {
        Server(String, String),
        Client(String, String),
    }
impl Connection 
    {
        fn server(self, file_info:&mut FileInfo)
            {
                print!("Server -> ");
                let ip:String;
                let port:String;
                let address:String;
                match self 
                    {
                        Connection::Server(in1, in2) => 
                            {
                                ip = in1.trim_end().to_string();
                                port = in2.trim_end().to_string();
                                address = format!("{}:{}", ip, port);
                                println!("{}", address);
                            }
                        _ => return
                    }
                let socket = TcpListener::bind(address);
                for stream in socket.expect("Error: Can't Check Connections").incoming()
                    {
                        match stream
                            {
                                Ok(mut stream) =>
                                    {
                                        let mut stay = true;
                                        while stay 
                                            {
                                                let start_time = Instant::now();
                                                FileInfo::writing_operations(file_info, &mut stream);
                                                let finish_time = Instant::now();
                                                println!("Passed: Total -> {:#?}", finish_time.duration_since(start_time));
                                                stay = false;
                                            }
                                    }
                                Err(e) =>
                                    {
                                        println!("Error: Can't Visit Stream -> {}", e);
                                        return;
                                    }
                            }
                    }
            }
        fn client(self, file_info:&mut FileInfo)
            {
                print!("Client -> ");
                let ip:String;
                let port:String;
                let address:String;
                match self 
                    {
                        Connection::Client(in1, in2) => 
                            {
                                ip = in1.trim_end().to_string();
                                port = in2.trim_end().to_string();
                                address = format!("{}:{}", ip, port);
                                println!("{}", address);
                            }
                        _ => return
                    }
                match TcpStream::connect(address) 
                    {
                        Ok(mut stream) =>
                            {
                                let start_time = Instant::now();
                                println!("Connected");
                                FileInfo::reading_operations(file_info, &mut stream);
                                let finish_time = Instant::now();
                                println!("Passed: Total -> {:#?}", finish_time.duration_since(start_time));
                            }
                        Err(e) =>
                            {
                                println!("Error: Connection -> {}", e);
                            }
                    }
                
            }
    }

fn take_string(output:String) -> String
    {
        let mut input = String::new();
        println!("{}", output);
        io::stdin().read_line(&mut input).expect("Error: Failed to Read from Console");
        input
    }
fn take_arg() -> String
    {
        env::args().last().as_deref().unwrap_or("default").to_string()
    }

fn main() 
    {
        //DONT FORGET
        //First we should check folder structure and validation then make connection.
        println!("Hello, world!");

        let mut data = FileInfo
            {
                file:None,
                location:take_arg(),
                size_current:0 as usize,
                metadata:None,
            };
        match &take_string("Input: Server 's', Client 'c'".to_string())[0..1]
            {
                "s" => 
                    {
                        Connection::server
                        (Connection::Server(take_string("Input: Server Stream IP Address".to_string()),
                         take_string("Input: Server Stream Port Address".to_string())),
                          &mut data);
                    },
                "c" => 
                    {
                        Connection::client
                        (Connection::Client(take_string("Input: Server IP Address to Connect".to_string()),
                         take_string("Input: Server Port Address to Connect".to_string())),
                          &mut data);
                    }
                input => 
                    {
                        println!("Error: Give Valid Input, You Gave : {}", input);
                        return;
                    }
            }
    }