Add powersync_client_id() function.

Author: rkistner

crates/core/src/client_id.rs

@@ -0,0 +1,58 @@
+extern crate alloc;
+
+use alloc::format;
+use alloc::string::{String, ToString};
+use alloc::vec::Vec;
+use core::ffi::c_int;
+use core::slice;
+
+use serde::{Deserialize, Serialize};
+use serde_json as json;
+use sqlite::ResultCode;
+use sqlite_nostd as sqlite;
+use sqlite_nostd::{Connection, Context, Value};
+
+use crate::create_sqlite_text_fn;
+use crate::error::SQLiteError;
+use crate::sync_types::Checkpoint;
+
+fn powersync_client_id_impl(
+    ctx: *mut sqlite::context,
+    args: &[*mut sqlite::value],
+) -> Result<String, SQLiteError> {
+    let db = ctx.db_handle();
+
+    // language=SQLite
+    let statement = db.prepare_v2("select value from ps_kv where key = 'client_id'")?;
+
+    if statement.step()? == ResultCode::ROW {
+        let client_id = statement.column_text(0)?;
+        return Ok(client_id.to_string());
+    } else {
+        return Err(SQLiteError(
+            ResultCode::ABORT,
+            Some(format!("No client_id found in ps_kv")),
+        ));
+    }
+}
+
+create_sqlite_text_fn!(
+    powersync_client_id,
+    powersync_client_id_impl,
+    "powersync_client_id"
+);
+
+pub fn register(db: *mut sqlite::sqlite3) -> Result<(), ResultCode> {
+    db.create_function_v2(
+        "powersync_client_id",
+        0,
+        sqlite::UTF8 | sqlite::DETERMINISTIC,
+        None,
+        Some(powersync_client_id),
+        None,
+        None,
+        None,
+    )?;
+
+    Ok(())
+}

crates/core/src/lib.rs

@@ -11,23 +11,24 @@ use core::ffi::{c_char, c_int};
 use sqlite::ResultCode;
 use sqlite_nostd as sqlite;
 
-mod util;
-mod uuid;
-mod views;
-mod view_admin;
-mod macros;
+mod checkpoint;
+mod client_id;
+mod crud_vtab;
 mod diff;
-mod schema_management;
-mod operations_vtab;
-mod operations;
-mod ext;
 mod error;
-mod crud_vtab;
-mod vtab_util;
+mod ext;
+mod macros;
+mod operations;
+mod operations_vtab;
+mod schema_management;
 mod sync_local;
-mod checkpoint;
-mod version;
 mod sync_types;
+mod util;
+mod uuid;
+mod version;
+mod view_admin;
+mod views;
+mod vtab_util;
 
 #[no_mangle]
 pub extern "C" fn sqlite3_powersync_init(
@@ -43,7 +44,7 @@ pub extern "C" fn sqlite3_powersync_init(
         code as c_int
     } else {
         ResultCode::OK as c_int
-    }
+    };
 }
 
 fn init_extension(db: *mut sqlite::sqlite3) -> Result<(), ResultCode> {
@@ -53,6 +54,7 @@ fn init_extension(db: *mut sqlite::sqlite3) -> Result<(), ResultCode> {
     crate::diff::register(db)?;
     crate::view_admin::register(db)?;
     crate::checkpoint::register(db)?;
+    crate::client_id::register(db)?;
 
     crate::schema_management::register(db)?;
     crate::operations_vtab::register(db)?;
@@ -61,12 +63,17 @@ fn init_extension(db: *mut sqlite::sqlite3) -> Result<(), ResultCode> {
     Ok(())
 }
 
-
 extern "C" {
     #[cfg(feature = "static")]
     #[allow(non_snake_case)]
     pub fn sqlite3_auto_extension(
-        xEntryPoint: Option<extern "C" fn(*mut sqlite::sqlite3, *mut *mut c_char, *mut sqlite::api_routines) -> c_int>,
+        xEntryPoint: Option<
+            extern "C" fn(
+                *mut sqlite::sqlite3,
+                *mut *mut c_char,
+                *mut sqlite::api_routines,
+            ) -> c_int,
+        >,
     ) -> ::core::ffi::c_int;
 }