feat: implement quite argument to use print less information while …

…upscaling and prepare to release.

Cargo.toml

@@ -6,7 +6,7 @@ keywords = ["realesrgan", "upscale"]
 readme = "README.md"
 repository = "https://github.com/kostya-zero/resup"
 license = "MIT"
-version = "0.2.0-dev"
+version = "0.2.0"
 edition = "2021"
 
 [dependencies]

src/args.rs

@@ -25,6 +25,12 @@ pub fn app() -> Command {
                     Arg::new("overwrite")
                         .help("Overwrite file content if file with same name exists.")
                         .long("overwrite")
+                        .short('o')
+                        .action(ArgAction::SetTrue),
+                    Arg::new("quite")
+                        .help("Run upscale without configuration info and Real-ESRGAN output.")
+                        .short('q')
+                        .long("quite")
                         .action(ArgAction::SetTrue),
                 ]),
             Command::new("list").about("List available models."),

src/main.rs

@@ -1,6 +1,5 @@
 use std::{
     fs,
-    mem::align_of_val,
     path::{Path, PathBuf},
     process::exit,
 };
@@ -25,6 +24,7 @@ fn main() {
             let input = sub.get_one::<String>("input").unwrap().to_string();
             let mut output: String = sub.get_one::<String>("output").unwrap().to_string();
             let overwrite: bool = sub.get_flag("overwrite");
+            let quite: bool = sub.get_flag("quite");
             if output.is_empty() {
                 let file_name = Path::new(&input)
                     .file_stem()
@@ -47,15 +47,14 @@ fn main() {
             }
 
             let config = Manager::load();
-            let quite: bool = false;
             Term::message("Preparing to upscale...");
-            if quite {
+            if !quite {
                 Term::display_data("Model", &config.get_model());
                 Term::display_data("Executable", &config.get_executable_path());
                 Term::display_data("Input file", &input);
                 Term::display_data("Output file", &output);
-                Term::message("Starting...");
             }
+            Term::message("Starting...");
             let upscale_result: Result<(), UpscaleError> =
                 run_upscale(config, &input, &output, quite);
             match upscale_result {

src/proc.rs

@@ -8,10 +8,15 @@ pub enum UpscaleError {
     ModelsDirectoryNotFound,
     ModelParamNotFound,
     ModelBinNotFound,
-    UnknownError
+    UnknownError,
 }
 
-pub fn run_upscale(config: Config, input: &str, output: &str, quite: bool) -> Result<(), UpscaleError> {
+pub fn run_upscale(
+    config: Config,
+    input: &str,
+    output: &str,
+    quite: bool,
+) -> Result<(), UpscaleError> {
     let mut proc: Command = Command::new(config.get_executable_path());
     proc.args(vec![
         "-i",
@@ -31,6 +36,10 @@ pub fn run_upscale(config: Config, input: &str, output: &str, quite: bool) -> Re
         proc.stdout(Stdio::inherit());
         proc.stdin(Stdio::inherit());
         proc.stderr(Stdio::inherit());
+    } else {
+        proc.stdout(Stdio::piped());
+        proc.stdin(Stdio::piped());
+        proc.stderr(Stdio::piped());
     }
     if !config.check_executable_exists() {
         return Err(UpscaleError::ExecutableNotFound);
@@ -50,7 +59,7 @@ pub fn run_upscale(config: Config, input: &str, output: &str, quite: bool) -> Re
         Err(e) => match e.kind() {
             std::io::ErrorKind::NotFound => Err(UpscaleError::ExecutableNotFound),
             std::io::ErrorKind::Interrupted => Err(UpscaleError::ProcessInterrupted),
-            _ => Err(UpscaleError::UnknownError)
-        }
+            _ => Err(UpscaleError::UnknownError),
+        },
     }
 }