Add rusqlite ToSql/FromSql implementations for CowStr if the correspo…

…nding feature is enabled

Justfile

@@ -2,7 +2,7 @@ check:
     #!/bin/bash -eux
     cargo check --example simple --no-default-features --features=json
     cargo run --example simple --features=core,json
-    cargo hack --feature-powerset --exclude-features=default,full check
+    cargo hack --feature-powerset --exclude-features=default,full check -p merde
     cargo test -F full
 
     pushd zerodeps-example

merde/Cargo.toml

@@ -44,11 +44,14 @@ merde_time = { version = "4.0.4", path = "../merde_time", optional = true, featu
 
 [features]
 default = ["core", "deserialize"]
-full = ["core", "deserialize", "json", "time"]
-deserialize = ["core", "merde_time/deserialize"]
+full = ["core", "deserialize", "json", "time", "rusqlite", "compact_str"]
 core = ["dep:merde_core"]
+deserialize = ["core", "merde_time/deserialize"]
+# merde_core re-exports
 compact_str = ["merde_core/compact_str"]
 serde = ["merde_core/serde"]
+rusqlite = ["merde_core/rusqlite"]
 
+# non-core crates
 json = ["dep:merde_json", "merde_time/json"]
 time = ["dep:merde_time"]

merde_core/Cargo.toml

@@ -13,10 +13,12 @@ categories = ["encoding", "parser-implementations"]
 [dependencies]
 compact_str = { version = "0.8.0", optional = true }
 rubicon = "3.4.2"
+rusqlite = { version = "0.32.1", optional = true }
 serde = { version = "1", optional = true }
 
 [features]
 default = []
-full = ["compact_str", "serde"]
+full = ["compact_str", "serde", "rusqlite"]
 compact_str = ["dep:compact_str"]
 serde = ["dep:serde", "compact_str/serde"]
+rusqlite = ["dep:rusqlite"]

merde_core/src/cowstr.rs

@@ -29,6 +29,17 @@ impl<'s> CowStr<'s> {
         Ok(Self::Borrowed(std::str::from_utf8(s)?))
     }
 
+    pub fn from_utf8_owned(s: Vec<u8>) -> Result<Self, std::str::Utf8Error> {
+        #[cfg(feature = "compact_str")]
+        {
+            Ok(Self::Owned(CompactString::from_utf8(s)?))
+        }
+        #[cfg(not(feature = "compact_str"))]
+        {
+            Ok(String::from_utf8(s).map_err(|e| e.utf8_error())?.into())
+        }
+    }
+
     pub fn from_utf8_lossy(s: &'s [u8]) -> Self {
         #[cfg(feature = "compact_str")]
         {
@@ -221,6 +232,29 @@ mod serde_impls {
     }
 }
 
+#[cfg(feature = "rusqlite")]
+mod rusqlite_impls {
+    use super::*;
+    use rusqlite::{types::FromSql, types::FromSqlError, types::ToSql, Result as RusqliteResult};
+
+    impl ToSql for CowStr<'_> {
+        fn to_sql(&self) -> RusqliteResult<rusqlite::types::ToSqlOutput<'_>> {
+            Ok(rusqlite::types::ToSqlOutput::Borrowed(self.as_ref().into()))
+        }
+    }
+
+    impl FromSql for CowStr<'_> {
+        fn column_result(value: rusqlite::types::ValueRef<'_>) -> Result<Self, FromSqlError> {
+            match value {
+                rusqlite::types::ValueRef::Text(s) => Ok(CowStr::from_utf8(s)
+                    .map_err(|e| FromSqlError::Other(Box::new(e)))?
+                    .into_static()),
+                _ => Err(FromSqlError::InvalidType),
+            }
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
@@ -238,4 +272,36 @@ mod tests {
         assert_ne!("world", cow_str1);
         assert_ne!(cow_str1, cow_str3);
     }
+
+    #[cfg(feature = "rusqlite")]
+    #[test]
+    fn test_rusqlite_integration() -> Result<(), Box<dyn std::error::Error>> {
+        use rusqlite::Connection;
+
+        // Create an in-memory database
+        let conn = Connection::open_in_memory()?;
+
+        // Create a table
+        conn.execute(
+            "CREATE TABLE test_table (id INTEGER PRIMARY KEY, value TEXT)",
+            [],
+        )?;
+
+        // Insert a CowStr value
+        let cow_str = CowStr::from("Hello, Rusqlite!");
+        conn.execute("INSERT INTO test_table (value) VALUES (?1)", [&cow_str])?;
+
+        // Retrieve the value
+        let mut stmt = conn.prepare("SELECT value FROM test_table")?;
+        let mut rows = stmt.query([])?;
+
+        if let Some(row) = rows.next()? {
+            let retrieved: CowStr = row.get(0)?;
+            assert_eq!(retrieved, "Hello, Rusqlite!");
+        } else {
+            panic!("No rows returned");
+        }
+
+        Ok(())
+    }
 }