tensamin/omega
Watch
6
0
Fork
You've already forked omega
 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

User Auth / iota auth

Browse source
This commit is contained in:
Alex Emmet 2026-01-14 22:21:05 +01:00
commit a00f3dd239
1 changed files with 23 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

23
src/sql/sql.rs
View file

@ -445,18 +445,18 @@ pub async fn register_complete_user(
id: i64, id: i64,
username: String, username: String,
public_key: String, public_key: String,
private_key_hash: String,
iota_id: i64, iota_id: i64,
token: String, token: String,
) -> Result<(), sqlx::Error> { ) -> Result<(), sqlx::Error> {
let db_lock = SQL_DB.read().await; let db_lock = SQL_DB.read().await;
let pool = db_lock.as_ref().expect("Database pool is not initialized"); let pool = db_lock.as_ref().expect("Database pool is not initialized");
sqlx::query("INSERT INTO users (id, username, public_key, private_key_hash, iota_id, token) VALUES (?, ?, ?, ?, ?, ?)") sqlx::query(
"INSERT INTO users (id, username, public_key, iota_id, token) VALUES (?, ?, ?, ?, ?)",
)
.bind(id) .bind(id)
.bind(username) .bind(username)
.bind(public_key) .bind(public_key)
.bind(private_key_hash)
.bind(iota_id) .bind(iota_id)
.bind(token) .bind(token)
.execute(pool) .execute(pool)
@ -541,14 +541,23 @@ pub async fn get_iota_by_id(id: i64) -> Result<(i64, String), sqlx::Error> {
let db_lock = SQL_DB.read().await; let db_lock = SQL_DB.read().await;
let pool = db_lock.as_ref().expect("Database pool is not initialized"); let pool = db_lock.as_ref().expect("Database pool is not initialized");
sqlx::query_as::<_, (i64, Vec<u8>)>( let result = sqlx::query_as::<_, (u64, Vec<u8>)>(
"SELECT id, public_key FROM iotas WHERE id = CAST(? AS UNSIGNED)", "SELECT id, public_key FROM iotas WHERE id = CAST(? AS UNSIGNED)",
) )
.bind(id) .bind(id)
.fetch_optional(pool) .fetch_optional(pool)
.await? .await;
.map(|(id, public_key)| (id, String::from_utf8_lossy(&public_key).to_string()))
.ok_or_else(|| sqlx::Error::RowNotFound) match result {
Ok(optional_row) => match optional_row {
Some((id_u64, public_key)) => Ok((
id_u64 as i64,
String::from_utf8_lossy(&public_key).to_string(),
)),
None => Err(sqlx::Error::RowNotFound),
},
Err(e) => Err(e),
}
} }
pub async fn change_iota_key(id: i64, new_key: String) -> Result<(), sqlx::Error> { pub async fn change_iota_key(id: i64, new_key: String) -> Result<(), sqlx::Error> {