diff --git a/cli/src/main.rs b/cli/src/main.rs
index 4c29222..e02972d 100644
--- a/cli/src/main.rs
+++ b/cli/src/main.rs
@@ -17,7 +17,7 @@
// You should have received a copy of the GNU General Public License
// along with this program. If not, see .
-use std::io::Write;
+use std::io::{Read, Write};
use std::path::PathBuf;
use anyhow::{anyhow, bail, Context, Result};
@@ -50,7 +50,9 @@ enum Action {
password: bool,
#[clap(short, long)]
duration: Option,
- path: PathBuf,
+ /// The path to the file to upload. If none is provided, then reads
+ /// stdin instead.
+ path: Option,
},
Download {
/// The paste to download.
@@ -78,12 +80,27 @@ fn handle_upload(
mut url: Url,
password: bool,
duration: Option,
- path: PathBuf,
+ path: Option,
) -> Result<()> {
url.set_fragment(None);
+ if password && path.is_none() {
+ bail!("Reading data from stdin is incompatible with a password. Provide a path to a file to upload.");
+ }
+
let (data, key) = {
- let mut container = std::fs::read(path)?;
+ let mut container = if let Some(path) = path {
+ std::fs::read(path)?
+ } else {
+ let mut container = vec![];
+ std::io::stdin().lock().read_to_end(&mut container)?;
+ container
+ };
+
+ if container.is_empty() {
+ bail!("Nothing to upload.");
+ }
+
let password = if password {
let maybe_password =
prompt_password_stderr("Please set the password for this paste: ")?;