main: Accept a file path as a command line argument

1 files changed, 18 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 049fcd2..3f77e96 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,7 +1,24 @@
+extern crate exitcode;
+
+use std::env;
+use std::process;
+
 use pdf_urls::get_urls_from_pdf;
 
+fn print_usage() {
+    println!("usage: pdf-urls FILE");
+}
+
 fn main() {
-    match get_urls_from_pdf("example.pdf") {
+    let args: Vec<String> = env::args().collect();
+
+    if args.len() != 2 {
+        print_usage();
+
+        process::exit(exitcode::USAGE);
+    }
+
+    match get_urls_from_pdf(&args[1]) {
         Ok(urls) => {
             for url in urls {
                 println!("{}", url);