feat(pastes): allow piping in from stdin

Signed-off-by: Amogh Lele <amolele@gmail.com>

.gitignore

@@ -1,2 +1,3 @@
 /target
-node_modules
+node_modules
+.vscode

Cargo.toml

@@ -12,6 +12,7 @@ include = ["src/**/*", "../LICENSE-*", "./README.md"]
 
 [dependencies]
 anyhow = "1.0.57"
+atty = "0.2.14"
 clap = { version = "3.1.16", features = ["derive"] }
 colored = "2.0.0"
 reqwest = { version = "0.11", features = ["blocking", "json"] }

src/main.rs

@@ -1,3 +1,9 @@
+use std::{
+    env,
+    ffi::OsString,
+    io::{self, Read},
+};
+
 use clap::{Parser, Subcommand};
 use paste::{Paste, PasteCommands};
 
@@ -19,9 +25,27 @@ enum Commands {
     Paste(Paste),
 }
 
+fn stdin_to_args() -> Vec<OsString> {
+    let mut args: Vec<OsString> = env::args_os().collect();
+
+    let mut body = String::from("");
+    io::stdin().read_to_string(&mut body).unwrap();
+    args.push(body.into());
+
+    args
+}
+
 fn main() {
     // parse CLI arguments
-    let args = Cli::parse();
+    let args: Cli;
+
+    // no input from stdin
+    if atty::is(atty::Stream::Stdin) {
+        args = Cli::parse();
+    } else {
+        // parse input from stdin
+        args = Cli::parse_from(stdin_to_args().iter());
+    }
 
     // match commands
     match args.command {