Implement Display for Error and ResultCode

src/error.rs

@@ -1,5 +1,6 @@
 use raw;
 use std::convert::{From, Into};
+use std::fmt::{self, Display, Formatter};
 
 use {Database, ResultCode};
 
@@ -49,3 +50,12 @@ impl From<ResultCode> for Error {
         }
     }
 }
+
+impl Display for Error {
+    fn fmt(&self, formatter: &mut Formatter) -> fmt::Result {
+        match self.message {
+            Some(ref message) => Display::fmt(message, formatter),
+            None => Display::fmt(&self.code, formatter),
+        }
+    }
+}

src/result.rs

@@ -1,4 +1,5 @@
 use libc::c_int;
+use std::fmt::{self, Display, Formatter};
 
 use raw;
 use Error;
@@ -6,43 +7,75 @@ use Error;
 /// A result.
 pub type Result<T> = ::std::result::Result<T, Error>;
 
-/// A result code.
+macro_rules! declare(
-#[derive(Clone, Copy, Debug, PartialEq, Eq)]
+    ($($left:ident => $right:ident,)*) => (
-pub enum ResultCode {
+        /// A result code.
-    Abort = raw::SQLITE_ABORT as isize,
+        #[derive(Clone, Copy, Debug, PartialEq, Eq)]
-    Authorization = raw::SQLITE_AUTH as isize,
+        pub enum ResultCode {
-    Busy = raw::SQLITE_BUSY as isize,
+            $($left = raw::$right as isize,)*
-    CantOpen = raw::SQLITE_CANTOPEN as isize,
+            Unknown,
-    Constraint = raw::SQLITE_CONSTRAINT as isize,
+        }
-    Corruption = raw::SQLITE_CORRUPT as isize,
+
-    Done = raw::SQLITE_DONE as isize,
+        pub fn code_from_raw(code: c_int) -> ResultCode {
-    Empty = raw::SQLITE_EMPTY as isize,
+            match code {
-    Error = raw::SQLITE_ERROR as isize,
+                $(raw::$right => ResultCode::$left,)*
-    Format = raw::SQLITE_FORMAT as isize,
+                _ => ResultCode::Unknown,
-    Full = raw::SQLITE_FULL as isize,
+            }
-    Internal = raw::SQLITE_INTERNAL as isize,
+        }
-    Interruption = raw::SQLITE_INTERRUPT as isize,
+    );
-    IOError = raw::SQLITE_IOERR as isize,
+);
-    Locked = raw::SQLITE_LOCKED as isize,
+
-    Mismatch = raw::SQLITE_MISMATCH as isize,
+declare!(
-    Misuse = raw::SQLITE_MISUSE as isize,
+    Abort => SQLITE_ABORT,
-    NoLargeFileSupport = raw::SQLITE_NOLFS as isize,
+    Authorization => SQLITE_AUTH,
-    NoMemory = raw::SQLITE_NOMEM as isize,
+    Busy => SQLITE_BUSY,
-    NotDatabase = raw::SQLITE_NOTADB as isize,
+    CantOpen => SQLITE_CANTOPEN,
-    NotFound = raw::SQLITE_NOTFOUND as isize,
+    Constraint => SQLITE_CONSTRAINT,
-    Notice = raw::SQLITE_NOTICE as isize,
+    Corruption => SQLITE_CORRUPT,
-    OK = raw::SQLITE_OK as isize,
+    Done => SQLITE_DONE,
-    Permission = raw::SQLITE_PERM as isize,
+    Empty => SQLITE_EMPTY,
-    Protocol = raw::SQLITE_PROTOCOL as isize,
+    Error => SQLITE_ERROR,
-    Range = raw::SQLITE_RANGE as isize,
+    Format => SQLITE_FORMAT,
-    ReadOnly = raw::SQLITE_READONLY as isize,
+    Full => SQLITE_FULL,
-    Row = raw::SQLITE_ROW as isize,
+    Internal => SQLITE_INTERNAL,
-    Schema = raw::SQLITE_SCHEMA as isize,
+    Interruption => SQLITE_INTERRUPT,
-    TooBig = raw::SQLITE_TOOBIG as isize,
+    IOError => SQLITE_IOERR,
-    Warning = raw::SQLITE_WARNING as isize,
+    Locked => SQLITE_LOCKED,
+    Mismatch => SQLITE_MISMATCH,
+    Misuse => SQLITE_MISUSE,
+    NoLargeFileSupport => SQLITE_NOLFS,
+    NoMemory => SQLITE_NOMEM,
+    NotDatabase => SQLITE_NOTADB,
+    NotFound => SQLITE_NOTFOUND,
+    Notice => SQLITE_NOTICE,
+    OK => SQLITE_OK,
+    Permission => SQLITE_PERM,
+    Protocol => SQLITE_PROTOCOL,
+    Range => SQLITE_RANGE,
+    ReadOnly => SQLITE_READONLY,
+    Row => SQLITE_ROW,
+    Schema => SQLITE_SCHEMA,
+    TooBig => SQLITE_TOOBIG,
+    Warning => SQLITE_WARNING,
+);
+
+impl Display for ResultCode {
+    fn fmt(&self, formatter: &mut Formatter) -> fmt::Result {
+        match *self {
+            ResultCode::Unknown => write!(formatter, "an unknown SQLite code"),
+            _ => write!(formatter, "SQLite code {}", *self as isize),
+        }
+    }
 }
 
-#[inline]
+#[cfg(test)]
-pub fn code_from_raw(code: c_int) -> ResultCode {
+mod tests {
-    unsafe { ::std::mem::transmute(code as i8) }
+    use ResultCode;
+    use super::code_from_raw;
+
+    #[test]
+    fn fmt() {
+        assert_eq!(format!("{}", ResultCode::OK), String::from("SQLite code 0"));
+        assert_eq!(format!("{}", code_from_raw(777)), String::from("an unknown SQLite code"));
+    }
 }