get db url & token from proxy server

Author: codekeyz

src/lib.rs

@@ -13,11 +13,14 @@ use sqlite::{
     SQLITE_DONE, SQLITE_ERROR, SQLITE_FLOAT, SQLITE_INTEGER, SQLITE_MISUSE, SQLITE_NULL, SQLITE_OK,
     SQLITE_RANGE, SQLITE_TEXT,
 };
-use utils::{execute_async_task, read_turso_config};
-
-use crate::utils::{
-    count_parameters, extract_column_names, sql_is_begin_transaction, sql_is_commit, sql_is_pragma,
-    sql_is_rollback, TursoConfig,
+use utils::execute_async_task;
+
+use crate::{
+    proxy::get_turso_db,
+    utils::{
+        count_parameters, extract_column_names, get_tokio, sql_is_begin_transaction, sql_is_commit,
+        sql_is_pragma, sql_is_rollback,
+    },
 };
 
 mod proxy;
@@ -63,19 +66,18 @@ pub unsafe extern "C" fn sqlite3_open_v2(
         exit(1);
     }
 
-    let turso_config = read_turso_config().unwrap_or_else(|_| TursoConfig {
-        db_url: format!(
-            "https://{}.aws-us-west-2.turso.io",
-            filename
-        ),
-        db_token: String::from("eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJhIjoicnciLCJnaWQiOiIyMzBiZDc4Ni1iN2I3LTRlYjgtYjkyMy00ZjM5MDRjYTVkMzciLCJpYXQiOjE3NTEwNTc3MzYsInJpZCI6ImQ1N2NjZTQzLWVhNWItNDFmMy1hNWZlLTE2ZWI4MjIxZTkwOCJ9.ItDyuzwvUqeXwc6KsQkjf6dUVAoQ5BkhvlxFD7nDRCl6thxopIKckJ-w7boX-2ms_-jjgVQuhj9PqYAsaycFAg"),
-    });
-
     let reqwest_client = reqwest::Client::builder()
         .user_agent("libsqlite3_turso/1.0.0")
         .build()
         .unwrap();
 
+    let turso_config = get_tokio()
+        .block_on(get_turso_db(&reqwest_client, filename))
+        .unwrap_or_else(|err| {
+            eprintln!("LibSqlite3_Turso Error: {}", err);
+            exit(1);
+        });
+
     let mock_db = Box::into_raw(Box::new(SQLite3 {
         client: reqwest_client,
         error_stack: Mutex::new(vec![]),
@@ -87,10 +89,6 @@ pub unsafe extern "C" fn sqlite3_open_v2(
         turso_config: turso_config,
     }));
 
-    if let Ok(config) = read_turso_config() {
-        (*mock_db).turso_config = config;
-    }
-
     *db = mock_db;
 
     SQLITE_OK

src/proxy.rs

@@ -227,3 +227,26 @@ pub fn get_execution_result<'a>(
 
     Ok(first_execution_result)
 }
+
+pub async fn get_turso_db(client: &Client, db_name: &str) -> Result<TursoConfig, Box<dyn Error>> {
+    let globe_auth_api = std::env::var("GLOBE_DS_API")?;
+
+    let request_body = serde_json::json!({
+        "db_name": db_name,
+    });
+
+    let response = client
+        .post(format!("{}/db/auth", globe_auth_api))
+        .body(request_body.to_string())
+        .send()
+        .await?;
+
+    if !response.status().is_success() {
+        return Err(format!("Failed to get Auth Token: {}", response.status()).into());
+    }
+
+    let token: String = response.text().await?;
+    let db_info = serde_json::from_str(&token)?;
+
+    Ok(db_info)
+}

src/utils.rs

@@ -1,6 +1,7 @@
-use std::{env, ffi::c_int, sync::OnceLock};
+use std::{ffi::c_int, sync::OnceLock};
 
 use regex::Regex;
+use serde::Deserialize;
 use tokio::runtime::{self, Runtime};
 
 use crate::sqlite::{push_error, SQLite3, SQLITE_ERROR};
@@ -66,21 +67,12 @@ where
     }
 }
 
-#[derive(Debug)]
+#[derive(Debug, Deserialize)]
 pub struct TursoConfig {
     pub db_url: String,
     pub db_token: String,
 }
 
-pub fn read_turso_config() -> Result<TursoConfig, String> {
-    let db_url = env::var("TURSO_DB_URL")
-        .map_err(|_| "Missing environment variable: TURSO_DB_URL".to_string())?;
-    let db_token = env::var("TURSO_DB_TOKEN")
-        .map_err(|_| "Missing environment variable: TURSO_DB_TOKEN".to_string())?;
-
-    Ok(TursoConfig { db_url, db_token })
-}
-
 #[inline]
 pub fn sql_is_begin_transaction(sql: &String) -> bool {
     sql.starts_with("BEGIN")