From 2a6496c631faba7d5f8c69c514e8db9d5f0c0b35 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Fri, 1 Nov 2019 21:47:46 +0100
Subject: main: Move URL extraction code into lib.rs

---
 src/main.rs | 37 ++-----------------------------------
 1 file changed, 2 insertions(+), 35 deletions(-)

(limited to 'src/main.rs')

diff --git a/src/main.rs b/src/main.rs
index 4b2d541..3964551 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,38 +1,5 @@
-extern crate lopdf;
-
-use lopdf::{Document, Object};
+use pdf_urls::get_urls_from_pdf;
 
 fn main() {
-    let doc = Document::load("example.pdf").unwrap();
-
-    for (_, obj) in doc.objects {
-        match obj {
-            Object::Dictionary(d) => {
-                for (k, v) in d.iter() {
-                    let key = ::std::str::from_utf8(&k).unwrap();
-
-                    if key == "A" {
-                        for (k, v) in v.as_dict().unwrap() {
-                            let key = ::std::str::from_utf8(&k).unwrap();
-
-                            if key == "URI" {
-
-                                match v {
-                                    Object::String(s, _) => {
-                                        println!("{}", ::std::str::from_utf8(s).unwrap());
-
-                                        ()
-                                    },
-                                    _ => (),
-                                }
-                            }
-                        }
-                    }
-                }
-
-                ()
-            },
-            _ => (),
-        }
-    }
+    get_urls_from_pdf();
 }
-- 
cgit v1.2.3