jap_learn/src/data_provider/card_states.rs

use crate::data::entities::{CardState, DeckSettings};
use crate::data_provider::{SqliteDbLayer, TableAccessQuerySet};
use crate::notification::NotificationData::{DbLoadError, DbSaveError};
use crate::notification::{MaybeError, NotificationData};
use chrono::NaiveDate;
use fsrs::MemoryState;
use log::info;
use rusqlite::{Connection, Error, Params, Row, params};
use std::time::Instant;
use crate::data::repositories::DataRepository;

pub type CardStateRepository = DataRepository<CardState, SqliteDbLayer<CardState, CardStatesQuerySet>>;
#[derive(Clone)]
pub struct CardStatesQuerySet;
impl TableAccessQuerySet<CardState> for CardStatesQuerySet {
    fn load() -> &'static str {
        "SELECT id, word_id, set_id, score, last_opened, fsrs_opened, fsrs_memory, fsrs_open_day FROM states"
    }

    fn save() -> &'static str {
        "UPDATE states SET score = ?1, last_opened = ?2 WHERE id = ?3"
    }

    fn add() -> &'static str {
        "INSERT INTO states (word_id, set_id, score, last_opened, fsrs_opened, fsrs_memory, fsrs_open_day) VALUES (?1, ?2, ?3, ?4)"
    }

    fn name() -> &'static str {
        "states"
    }

    fn from_sql(row: &Row) -> Result<CardState, Error> {
        Ok(CardState {
            id: row.get(0)?,
            word_id: row.get(1)?,
            set_id: row.get(2)?,
            score: row.get(3)?,
            fsrs_opened: NaiveDate::from_epoch_days(row.get::<usize, i32>(5)?)
                .expect("Incorrect date encoding"),
            fsrs_memory: get_memory(row.get(6)?),
            last_open: row.get(4)?,
            fsrs_open_day: NaiveDate::from_epoch_days(row.get::<usize, i32>(7)?)
                .expect("Incorrect date encoding"),
        })
    }

    fn insert_params(entity: &CardState) -> impl Params {
        (
            entity.word_id,
            entity.set_id,
            entity.score,
            entity.last_open.timestamp(),
            entity.fsrs_opened.to_epoch_days(),
            memory_to_i64(&entity.fsrs_memory),
            entity.fsrs_open_day.to_epoch_days(),
        )
    }

    fn update_params(entity: &CardState) -> impl Params {
        (entity.score, entity.last_open.timestamp(), entity.id)
    }
}
pub fn load_stats_of_deck(
    set: &DeckSettings,
    connection: &Connection,
) -> Result<Vec<CardState>, NotificationData> {
    let mut stmt = connection
        .prepare("SELECT id, word_id, score, last_opened, fsrs_opened, fsrs_memory, fsrs_open_day FROM states WHERE set_id = ?1")
        .expect("Failed to prepare state query");
    let iter = stmt
        .query_map((set.id,), |row| {
            Ok(CardState {
                id: row.get(0)?,
                word_id: row.get(1)?,
                set_id: set.id,
                score: row.get(2)?,
                fsrs_opened: NaiveDate::from_epoch_days(row.get::<usize, i32>(4)?)
                    .expect("Incorrect date encoding"),
                fsrs_memory: get_memory(row.get(5)?),
                last_open: row.get(3)?,
                fsrs_open_day: NaiveDate::from_epoch_days(row.get::<usize, i32>(6)?)
                    .expect("Incorrect date encoding"),
            })
        })
        .map_err(|_| DbLoadError)?;

    let mut buffer = Vec::with_capacity(512);
    iter.filter_map(|res| res.ok()).for_each(|i| buffer.push(i));

    Ok(buffer)
}

fn get_memory(value: i64) -> Option<MemoryState> {
    if value == 0 {
        return None;
    }
    Some(MemoryState {
        stability: f32::from_bits(value as u32),
        difficulty: f32::from_bits((value >> 32) as u32),
    })
}

fn memory_to_i64(value: &Option<MemoryState>) -> i64 {
    let Some(value) = &value else {
        return 0;
    };

    value.stability.to_bits() as i64 | ((value.difficulty.to_bits() as i64) << 32)
}

pub fn add_stat_list(stats: &mut [CardState], connection: &mut Connection) -> MaybeError {
    let time = Instant::now();
    let tx = connection
        .transaction()
        .expect("Failed to create transaction");
    let count = stats.len();

    {
        let mut stmt = tx
            .prepare(
                "INSERT INTO states (word_id, set_id, score, last_opened, fsrs_opened, fsrs_memory, fsrs_open_day) VALUES (?1, ?2, ?3, ?4)",
            )
            .expect("Failed to prepare state query");

        for stat in stats.iter() {
            stmt.execute(params![
                stat.word_id,
                stat.set_id,
                stat.score,
                stat.last_open.timestamp(),
                stat.fsrs_opened.to_epoch_days(),
                memory_to_i64(&stat.fsrs_memory),
                stat.fsrs_open_day.to_epoch_days()
            ])
            .map_err(|_| DbSaveError)?;
        }
    }

    let last_index: u32 = tx.last_insert_rowid() as u32;
    tx.commit().expect("Failed to commit");

    let start_index = last_index - (count as u32) + 1;

    for (index, id) in (start_index..=last_index).enumerate() {
        stats[index].id = id.into();
    }

    info!("Added {} cards for {:?}", stats.len(), time.elapsed());
    Ok(())
}

pub fn update_stat_score(stat: &CardState, connection: &Connection) -> MaybeError {
    let time = Instant::now();

    connection
        .execute(
            "UPDATE states SET score = ?1, last_opened = ?2 WHERE id = ?3",
            (&stat.score, &stat.last_open.timestamp(), &stat.id),
        )
        .map_err(|_| DbSaveError)?;
    info!("Updated stat: {}", time.elapsed().as_millis());
    Ok(())
}

pub fn delete_stats(stat: &[CardState], connection: &Connection) -> MaybeError {
    if stat.is_empty() {
        return Ok(());
    }
    let ids = stat
        .iter()
        .map(|s| s.id.into())
        .collect::<Vec<String>>()
        .join(", ");
    connection
        .execute(
            format!("DELETE FROM states WHERE id in ({ids})").as_str(),
            (),
        )
        .map_err(|_| DbSaveError)?;
    Ok(())
}