From 568e61810601768d4eee4f00b71262d43b5804c2 Mon Sep 17 00:00:00 2001
From: Ade Attwood <adrian.attwood@marketdojo.com>
Date: Sat, 11 May 2024 21:35:56 +0100
Subject: [PATCH] feat: add input argument to read input from text or stdin

You can now provide input as the content you want to search and replace in. If
you don't provide input, the program will read from stdin.
---
 src/main.rs | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/src/main.rs b/src/main.rs
index 1eaafda..65df5bd 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,17 +11,26 @@ struct Args {
     /// The replacement pattern.
     #[arg(short, long)]
     replace: String,
+
+    /// The input content to search and replace. If not provided, input will be read from stdin.
+    #[arg(short, long)]
+    input: Option<String>,
 }
 
 fn main() {
     let args = Args::parse();
 
-    let mut input = String::new();
-    match std::io::stdin().read_to_string(&mut input) {
-        Ok(_) => (),
-        Err(err) => {
-            eprintln!("Error reading from stdin: {}", err);
-            return;
+    let input = match args.input {
+        Some(input) => input,
+        None => {
+            let mut input = String::new();
+            match std::io::stdin().read_to_string(&mut input) {
+                Ok(_) => input,
+                Err(err) => {
+                    eprintln!("Error reading from stdin: {}", err);
+                    return;
+                }
+            }
         }
     };