Start encoding SQLite affinity rules

Need to figure out what to do for NUMERIC affinity since Rusqlite
doesn't expose a variant for that pseudo-type.

1 files changed, 25 insertions, 0 deletions

diff --git a/src/sqlite.rs b/src/sqlite.rs
index 10dbb61..8094398 100644
--- a/src/sqlite.rs
+++ b/src/sqlite.rs
@@ -6,4 +6,29 @@ use rusqlite;
 /// Use the SQLite rules for type affinity described in:
 /// https://sqlite.org/datatype3.html#determination_of_column_affinity
 pub fn affinity(type_name: &str) -> rusqlite::types::Type {
+    use rusqlite::types::Type;
+
+    let type_name = type_name.to_uppercase();
+
+    if type_name.contains("INT") {
+        return Type::Integer;
+    } else if type_name.contains("CHAR")
+        || type_name.contains("CLOB")
+        || type_name.contains("TEXT")
+    {
+        return Type::Text;
+    } else if type_name.contains("BLOB")
+        || type_name.is_empty()
+    {
+        return Type::Blob;
+    } else if type_name.contains("REAL")
+        || type_name.contains("FLOA")
+        || type_name.contains("DOUB")
+    {
+        return Type::Real;
+    }
+
+    // TODO: Numeric affinity
+
+    Type::Text
 }