Make connect/create return a tuple

src/vtab/array.rs

@@ -4,6 +4,7 @@ use std::default::Default;
 use std::os::raw::{c_char, c_int, c_void};
 use std::rc::Rc;
 
+use error::error_from_sqlite_code;
 use ffi;
 use types::{ToSql, ToSqlOutput, Value};
 use vtab::{self, Context, IndexInfo, Module, VTab, VTabCursor, Values};
@@ -62,15 +63,15 @@ impl Module for ArrayModule {
         self.0
     }
 
-    fn connect(db: &mut ffi::sqlite3, _aux: Option<&()>, _args: &[&[u8]]) -> Result<ArrayTab> {
+    fn connect(
+        _: &mut ffi::sqlite3,
+        _aux: Option<&()>,
+        _args: &[&[u8]],
+    ) -> Result<(String, ArrayTab)> {
         let vtab = ArrayTab {
             base: Default::default(),
         };
-        try!(ArrayModule::declare_vtab(
-            db,
-            "CREATE TABLE x(value,pointer hidden)"
-        ));
-        Ok(vtab)
+        Ok(("CREATE TABLE x(value,pointer hidden)".to_owned(), vtab))
     }
 }
 

src/vtab/csvtab.rs

@@ -7,6 +7,7 @@ use std::path::Path;
 use std::result;
 use std::str;
 
+use error::error_from_sqlite_code;
 use ffi;
 use types::Null;
 use vtab::{
@@ -84,7 +85,11 @@ impl Module for CSVModule {
         self.0
     }
 
-    fn connect(db: &mut ffi::sqlite3, _aux: Option<&()>, args: &[&[u8]]) -> Result<CSVTab> {
+    fn connect(
+        _: &mut ffi::sqlite3,
+        _aux: Option<&()>,
+        args: &[&[u8]],
+    ) -> Result<(String, CSVTab)> {
         if args.len() < 4 {
             return Err(Error::ModuleError("no CSV file specified".to_owned()));
         }
@@ -230,8 +235,7 @@ impl Module for CSVModule {
             schema = Some(sql);
         }
 
-        try!(CSVModule::declare_vtab(db, &schema.unwrap()));
-        Ok(vtab)
+        Ok((schema.unwrap().to_owned(), vtab))
     }
 }
 

src/vtab/mod.rs

@@ -7,7 +7,6 @@ use std::ptr;
 use std::slice;
 
 use context::set_result;
-use error::error_from_sqlite_code;
 use ffi;
 use types::{FromSql, FromSqlError, ToSql, ValueRef};
 use {str_to_cstring, Connection, Error, InnerConnection, Result};
@@ -54,7 +53,7 @@ pub trait Module {
         db: &mut ffi::sqlite3,
         aux: Option<&Self::Aux>,
         args: &[&[u8]],
-    ) -> Result<Self::Table> {
+    ) -> Result<(String, Self::Table)> {
         Self::connect(db, aux, args)
     }
     /// Similar to `create`. The difference is that `connect` is called to establish a new connection
@@ -63,18 +62,7 @@ pub trait Module {
         db: &mut ffi::sqlite3,
         aux: Option<&Self::Aux>,
         args: &[&[u8]],
-    ) -> Result<Self::Table>;
-
-    /// Declare the schema of a virtual table.
-    fn declare_vtab(db: &mut ffi::sqlite3, sql: &str) -> Result<()> {
-        let c_sql = try!(CString::new(sql));
-        let rc = unsafe { ffi::sqlite3_declare_vtab(db, c_sql.as_ptr()) };
-        if rc == ffi::SQLITE_OK {
-            Ok(())
-        } else {
-            Err(error_from_sqlite_code(rc, None))
-        }
-    }
+    ) -> Result<(String, Self::Table)>;
 }
 
 /// Virtual table instance trait.
@@ -573,10 +561,25 @@ macro_rules! create_or_connect {
                 .map(|&cs| CStr::from_ptr(cs).to_bytes()) // FIXME .to_str() -> Result<&str, Utf8Error>
                 .collect::<Vec<_>>();
             match $module::$module_func(db.as_mut().expect("non null db pointer"), aux.as_ref(), &vec[..]) {
-                Ok(vtab) => {
+                Ok((sql, vtab)) => {
+                    match ::std::ffi::CString::new(sql) {
+                        Ok(c_sql) => {
+                            let rc = ffi::sqlite3_declare_vtab(db, c_sql.as_ptr());
+                            if rc == ffi::SQLITE_OK {
                                         let boxed_vtab: *mut $vtab = Box::into_raw(Box::new(vtab));
                                         *pp_vtab = boxed_vtab as *mut ffi::sqlite3_vtab;
                                         ffi::SQLITE_OK
+                            } else {
+                                let err = error_from_sqlite_code(rc, None);
+                                *err_msg = mprintf(err.description());
+                                rc
+                            }
+                        }
+                        Err(err) => {
+                            *err_msg = mprintf(err.description());
+                            ffi::SQLITE_ERROR
+                        }
+                    }
                 }
                 Err(Error::SqliteFailure(err, s)) => {
                     if let Some(s) = s {

src/vtab/series.rs

@@ -3,6 +3,7 @@
 use std::default::Default;
 use std::os::raw::{c_char, c_int, c_void};
 
+use error::error_from_sqlite_code;
 use ffi;
 use types::Type;
 use vtab::{self, Context, IndexInfo, Module, VTab, VTabCursor, Values};
@@ -45,15 +46,18 @@ impl Module for Series {
         self.0
     }
 
-    fn connect(db: &mut ffi::sqlite3, _aux: Option<&()>, _args: &[&[u8]]) -> Result<SeriesTab> {
+    fn connect(
+        _: &mut ffi::sqlite3,
+        _aux: Option<&()>,
+        _args: &[&[u8]],
+    ) -> Result<(String, SeriesTab)> {
         let vtab = SeriesTab {
             base: Default::default(),
         };
-        try!(Series::declare_vtab(
-            db,
-            "CREATE TABLE x(value,start hidden,stop hidden,step hidden)"
-        ));
-        Ok(vtab)
+        Ok((
+            "CREATE TABLE x(value,start hidden,stop hidden,step hidden)".to_owned(),
+            vtab,
+        ))
     }
 }