diff --git a/src/main.rs b/src/main.rs index a19e646..de4bbbb 100644 --- a/src/main.rs +++ b/src/main.rs @@ -5,7 +5,7 @@ use crate::params::{parse_params, Format}; use std::env; - +use std::ffi::OsString; use std::fs; use std::io::{self, Write}; use std::process::{exit, ExitCode}; @@ -38,19 +38,25 @@ fn main() -> ExitCode { ); } }; - if same_file::is_same_file(¶ms.from, ¶ms.to).unwrap_or(false) { + if params.from == "-" && params.to == "-" + || same_file::is_same_file(¶ms.from, ¶ms.to).unwrap_or(false) + { maybe_report_identical_files(); return ExitCode::SUCCESS; } // read files - let from_content = match fs::read(¶ms.from) { + fn read_file_contents(filepath: &OsString) -> io::Result> { + let stdin = OsString::from("/dev/stdin"); + fs::read(if filepath == "-" { &stdin } else { filepath }) + } + let from_content = match read_file_contents(¶ms.from) { Ok(from_content) => from_content, Err(e) => { eprintln!("Failed to read from-file: {e}"); return ExitCode::from(2); } }; - let to_content = match fs::read(¶ms.to) { + let to_content = match read_file_contents(¶ms.to) { Ok(to_content) => to_content, Err(e) => { eprintln!("Failed to read to-file: {e}"); diff --git a/src/params.rs b/src/params.rs index 74e8d15..38d8238 100644 --- a/src/params.rs +++ b/src/params.rs @@ -67,9 +67,9 @@ pub fn parse_params>(opts: I) -> Result ", exe.to_string_lossy())); } @@ -461,14 +461,14 @@ mod tests { assert_eq!( Ok(Params { from: os("foo"), - to: os("/dev/stdin"), + to: os("-"), ..Default::default() }), parse_params([os("diff"), os("foo"), os("-")].iter().cloned()) ); assert_eq!( Ok(Params { - from: os("/dev/stdin"), + from: os("-"), to: os("bar"), ..Default::default() }), @@ -476,8 +476,8 @@ mod tests { ); assert_eq!( Ok(Params { - from: os("/dev/stdin"), - to: os("/dev/stdin"), + from: os("-"), + to: os("-"), ..Default::default() }), parse_params([os("diff"), os("-"), os("-")].iter().cloned()) diff --git a/tests/integration.rs b/tests/integration.rs index 5ad624e..240d60b 100644 --- a/tests/integration.rs +++ b/tests/integration.rs @@ -22,25 +22,30 @@ fn unknown_param() -> Result<(), Box> { } #[test] -fn cannot_read_from_file() -> Result<(), Box> { +fn cannot_read_files() -> Result<(), Box> { + let file = NamedTempFile::new()?; + let mut cmd = Command::cargo_bin("diffutils")?; - cmd.arg("foo.txt").arg("bar.txt"); + cmd.arg("foo.txt").arg(file.path()); cmd.assert() .code(predicate::eq(2)) .failure() .stderr(predicate::str::starts_with("Failed to read from-file")); - Ok(()) -} -#[test] -fn cannot_read_to_file() -> Result<(), Box> { - let file = NamedTempFile::new()?; let mut cmd = Command::cargo_bin("diffutils")?; - cmd.arg(file.path()).arg("bar.txt"); + cmd.arg(file.path()).arg("foo.txt"); cmd.assert() .code(predicate::eq(2)) .failure() .stderr(predicate::str::starts_with("Failed to read to-file")); + + let mut cmd = Command::cargo_bin("diffutils")?; + cmd.arg("foo.txt").arg("foo.txt"); + cmd.assert() + .code(predicate::eq(2)) + .failure() + .stderr(predicate::str::starts_with("Failed to read from-file")); + Ok(()) } @@ -177,7 +182,7 @@ fn read_from_stdin() -> Result<(), Box> { .code(predicate::eq(1)) .failure() .stdout(predicate::eq(format!( - "--- {}\t\n+++ /dev/stdin\t\n@@ -1 +1 @@\n-foo\n+bar\n", + "--- {}\t\n+++ -\t\n@@ -1 +1 @@\n-foo\n+bar\n", file1.path().to_string_lossy() ))); @@ -190,12 +195,12 @@ fn read_from_stdin() -> Result<(), Box> { .code(predicate::eq(1)) .failure() .stdout(predicate::eq(format!( - "--- /dev/stdin\t\n+++ {}\t\n@@ -1 +1 @@\n-foo\n+bar\n", + "--- -\t\n+++ {}\t\n@@ -1 +1 @@\n-foo\n+bar\n", file2.path().to_string_lossy() ))); let mut cmd = Command::cargo_bin("diffutils")?; - cmd.arg("-u").arg("-").arg("-").write_stdin("foo\n"); + cmd.arg("-u").arg("-").arg("-"); cmd.assert() .code(predicate::eq(0)) .success()