From 523170630fd8e9ba695f4167d132a8812d814a1e Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Sat, 2 Nov 2019 04:28:56 +0100
Subject: main: Accept a file path as a command line argument

---
 src/main.rs | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

(limited to 'src')

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);
-- 
cgit v1.2.3