1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
use crate::sdk::std::net::ftp::{validate_and_run_with_connection, Options}; use crate::utils::io::create_empty_file; use crate::utils::pckg; use duckscript::types::command::{Command, CommandResult}; use ftp::{FtpError, FtpStream}; use std::fs::OpenOptions; use std::io::{BufWriter, Error, Read, Write}; #[cfg(test)] #[path = "./mod_test.rs"] mod mod_test; fn write_file(reader: &mut dyn Read, target_file: &str) -> Result<(), Error> { let mut file = OpenOptions::new().append(true).open(target_file)?; { let mut writer = BufWriter::new(&mut file); let mut buffer = [0; 10240]; loop { let read_size = reader.read(&mut buffer)?; if read_size > 0 { writer.write_all(&buffer[0..read_size])?; } else { break; } } writer.flush()?; } file.sync_all()?; Ok(()) } #[derive(Clone)] pub(crate) struct CommandImpl { package: String, } impl Command for CommandImpl { fn name(&self) -> String { pckg::concat(&self.package, "Get") } fn aliases(&self) -> Vec<String> { vec!["ftp_get".to_string()] } fn help(&self) -> String { include_str!("help.md").to_string() } fn clone_and_box(&self) -> Box<dyn Command> { Box::new((*self).clone()) } fn run(&self, arguments: Vec<String>) -> CommandResult { validate_and_run_with_connection( &arguments, &|options: &Options| -> Result<(), String> { if options.remote_file.is_none() { Err("Missing remote file name".to_string()) } else if options.local_file.is_none() { Err("Missing local file name.".to_string()) } else { Ok(()) } }, &mut |options: &Options, ftp_stream: &mut FtpStream| -> CommandResult { let options_clone = options.clone(); let remote_file = options_clone.remote_file.unwrap(); let local_file = options_clone.local_file.unwrap(); match create_empty_file(&local_file) { Ok(_) => { match ftp_stream.retr(&remote_file, |reader| { match write_file(reader, &local_file) { Ok(_) => Ok(()), Err(error) => Err(FtpError::InvalidResponse(error.to_string())), } }) { Ok(_) => CommandResult::Continue(Some(true.to_string())), Err(error) => CommandResult::Error(error.to_string()), } } Err(error) => CommandResult::Error(error), } }, ) } } pub(crate) fn create(package: &str) -> Box<dyn Command> { Box::new(CommandImpl { package: package.to_string(), }) }