From 92f8f57b76b32c3d3e52d4b61dcdf25969f47ab7 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Sat, 2 Nov 2019 00:35:49 +0100
Subject: get_urls_from_pdf: Remove `unwrap`s and replace with an error type

Create a custom error type to use instead of the `unwrap`s.
---
 src/errors.rs | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 src/errors.rs

(limited to 'src/errors.rs')

diff --git a/src/errors.rs b/src/errors.rs
new file mode 100644
index 0000000..6ec2ee1
--- /dev/null
+++ b/src/errors.rs
@@ -0,0 +1,43 @@
+use std::fmt;
+use std::error;
+use std::str;
+
+use lopdf;
+
+pub type Result<T> = ::std::result::Result<T, Error>;
+
+#[derive(Debug)]
+pub enum Error {
+    Lopdf(lopdf::Error),
+    Utf8(str::Utf8Error),
+}
+
+impl fmt::Display for Error {
+    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
+        match *self {
+            Error::Lopdf(ref err) => write!(f, "pdf error: {}", err),
+            Error::Utf8(ref err) => write!(f, "utf8 error: {}", err),
+        }
+    }
+}
+
+impl error::Error for Error {
+    fn source(&self) -> Option<&(dyn error::Error + 'static)> {
+        match *self {
+            Error::Lopdf(ref err) => Some(err),
+            Error::Utf8(ref err) => Some(err),
+        }
+    }
+}
+
+impl From<lopdf::Error> for Error {
+    fn from(err: lopdf::Error) -> Error {
+        Error::Lopdf(err)
+    }
+}
+
+impl From<str::Utf8Error> for Error {
+    fn from(err: str::Utf8Error) -> Error {
+        Error::Utf8(err)
+    }
+}
-- 
cgit v1.2.3