From f8fec23264acd576860509d6cf68d49b23330fc3 Mon Sep 17 00:00:00 2001
From: Teddy Wing
Date: Thu, 10 Mar 2022 00:02:40 +0100
Subject: 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.
---
 src/sqlite.rs | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'src/sqlite.rs')

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
 }
-- 
cgit v1.2.3