get_urls_from_pdf: Take PDF path as an argument

1 files changed, 4 insertions, 2 deletions

diff --git a/src/lib.rs b/src/lib.rs
index fbe8b8a..82e58fd 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,9 +1,11 @@
 extern crate lopdf;
 
+use std::path::Path;
+
 use lopdf::{Document, Object};
 
-pub fn get_urls_from_pdf() {
-    let doc = Document::load("example.pdf").unwrap();
+pub fn get_urls_from_pdf(path: &Path) {
+    let doc = Document::load(path).unwrap();
 
     for (_, obj) in doc.objects {
         match obj {