main(): Accept input YAML from standard input

Either with a "-" argument or no file argument.

1 files changed, 14 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 5e06bb2..77092fa 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -41,9 +41,22 @@ fn main() {
             table_name,
             input_file,
         } => {
+            let input_file = match &input_file {
+                Some(f) => f,
+                None => "-",
+            };
+
             let mut dbconn = rusqlite::Connection::open(database).unwrap();
 
-            let text_data = std::fs::read_to_string(input_file.unwrap()).unwrap();
+            let mut text_data;
+            if input_file == "-" {
+                use std::io::Read;
+
+                text_data = String::new();
+                std::io::stdin().read_to_string(&mut text_data).unwrap();
+            } else {
+                text_data = std::fs::read_to_string(input_file).unwrap();
+            }
 
             let mut yaml_data = yaml::YamlLoader::load_from_str(&text_data).unwrap();