changed file section

database/src/db/char/char_awake_info.rs

@@ -0,0 +1,118 @@
+use crate::models::game::char_awake_info::CharAwakeInfo;
+use sqlx::SqlitePool;
+
+/// Add a single CharAwakeInfo record from a Rust struct.
+pub async fn add_char_awake_info(pool: &SqlitePool, data: &CharAwakeInfo) -> sqlx::Result<()> {
+    sqlx::query(
+        r#"
+INSERT INTO CharAwakeInfo (
+    Uid,
+    UniqueCharId,
+    IsAwake,
+    ImprintSlot1Level,
+    ImprintSlot2Level,
+    ImprintSlot3Level
+) VALUES (
+    ?,
+    ?,
+    ?,
+    ?,
+    ?,
+    ?
+)
+"#,
+    )
+    .bind(&data.uid)
+    .bind(&data.unique_char_id)
+    .bind(&data.is_awake)
+    .bind(&data.imprint_slot_1_level)
+    .bind(&data.imprint_slot_2_level)
+    .bind(&data.imprint_slot_3_level)
+    .execute(pool)
+    .await?;
+
+    Ok(())
+}
+
+/// Fetch all records for a given UID.
+pub async fn get_char_awake_info(pool: &SqlitePool, uid: i64) -> sqlx::Result<Vec<CharAwakeInfo>> {
+    sqlx::query_as::<_, CharAwakeInfo>("SELECT * FROM CharAwakeInfo WHERE Uid = ?")
+        .bind(uid)
+        .fetch_all(pool)
+        .await
+}
+
+/// Delete all CharAwakeInfo rows for a UID.
+pub async fn delete_char_awake_info(pool: &SqlitePool, uid: i64) -> sqlx::Result<()> {
+    sqlx::query("DELETE FROM CharAwakeInfo WHERE Uid = ?")
+        .bind(uid)
+        .execute(pool)
+        .await?;
+    Ok(())
+}
+
+/// Get a single record by Index (rowid)
+pub async fn get_by_index(pool: &SqlitePool, uid: i64, index: i64) -> sqlx::Result<CharAwakeInfo> {
+    sqlx::query_as::<_, CharAwakeInfo>("SELECT * FROM CharAwakeInfo WHERE Uid = ? AND Index = ?")
+        .bind(uid)
+        .bind(index)
+        .fetch_one(pool)
+        .await
+}
+
+/// Get multiple records by their Index (rowids)
+pub async fn get_all_by_index(
+    pool: &SqlitePool,
+    uid: i64,
+    index: &[i64],
+) -> sqlx::Result<Vec<CharAwakeInfo>> {
+    if index.is_empty() {
+        return Ok(vec![]);
+    }
+
+    let placeholders = index.iter().map(|_| "?").collect::<Vec<_>>().join(",");
+    let query = format!(
+        "SELECT * FROM CharAwakeInfo WHERE Uid = ? AND Index IN ({})",
+        placeholders
+    );
+
+    let mut query = sqlx::query_as::<_, CharAwakeInfo>(&query).bind(uid);
+    for idx in index {
+        query = query.bind(idx);
+    }
+
+    query.fetch_all(pool).await
+}
+
+/// Insert and return the rowid (Index)
+pub async fn insert(pool: &SqlitePool, data: &CharAwakeInfo) -> sqlx::Result<i64> {
+    let result = sqlx::query(
+        r#"
+INSERT INTO CharAwakeInfo (
+    Uid,
+    UniqueCharId,
+    IsAwake,
+    ImprintSlot1Level,
+    ImprintSlot2Level,
+    ImprintSlot3Level
+) VALUES (
+    ?,
+    ?,
+    ?,
+    ?,
+    ?,
+    ?
+)
+"#,
+    )
+    .bind(&data.uid)
+    .bind(&data.unique_char_id)
+    .bind(&data.is_awake)
+    .bind(&data.imprint_slot_1_level)
+    .bind(&data.imprint_slot_2_level)
+    .bind(&data.imprint_slot_3_level)
+    .execute(pool)
+    .await?;
+
+    Ok(result.last_insert_rowid())
+}