jap_learn/src/data_provider/sqlite.rs

use crate::fs::*;
use log::error;
use rusqlite::Connection;
use std::ops::{Deref, DerefMut};

pub fn create_db() {
    let db_file = db_path();
    if !db_file.exists() {
        std::fs::File::create(&db_file).expect("Database file creation failed");
        let connection = Connection::open(&db_file).expect("Connection open failed");
        connection
            .execute("PRAGMA foreign_keys = ON;", [])
            .expect("Failed to set PRAGMA");

        create_tables(&connection);
    }
}

fn create_tables(conn: &Connection) {
    make_card_set(conn).expect("Failed to init table");
    make_settings(conn).expect("Failed to init table");
    make_word_group(conn).expect("Failed to init table");
    make_words(conn).expect("Failed to init table");
    make_card_stats(conn).expect("Failed to init table");
    make_history(conn).expect("Failed to init table");
    make_custom_fields(conn).expect("Failed to init table");
}

fn make_settings(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/settings.sql");
    conn.execute(query, ())?;
    Ok(())
}

fn make_card_set(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/decks.sql");
    conn.execute(query, ())?;
    Ok(())
}

fn make_word_group(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/word_groups.sql");
    conn.execute(query, ())?;
    conn.execute("insert into word_groups (name) values (\"Слова\");", ())?;
    Ok(())
}
fn make_words(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/words.sql");
    conn.execute(query, ())?;
    Ok(())
}
fn make_card_stats(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/states.sql");
    conn.execute(query, ())?;
    Ok(())
}

fn make_history(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/history.sql");
    conn.execute(query, ())?;
    Ok(())
}

fn make_custom_fields(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/custom_fields.sql");
    conn.execute(query, ())?;
    conn.execute("insert into custom_fields (name, identifier) values (\"Чтение\", \"reading\"), (\"Описание\", \"description\"), (\"Пример использования\", \"context\")", ())?;
    Ok(())
}

pub fn default_connection() -> Connection {
    let db_file = db_path();
    let connection = Connection::open(db_file).expect("Failed to open database");
    connection
        .execute("PRAGMA foreign_keys = ON;", [])
        .expect("Failed to set PRAGMA");
    connection
        .query_one("PRAGMA journal_mode = WAL;", [], |row| {
            println!("{:?}", row);
            Ok(())
        })
        .unwrap_or_else(|e| {
            println!("{}", e);
        });
    connection
        .execute("PRAGMA synchronous = NORMAL;", [])
        .expect("Failed to set PRAGMA");
    connection
}

pub struct SqliteConnection(Option<Connection>);
impl Deref for SqliteConnection {
    type Target = Connection;
    fn deref(&self) -> &Self::Target {
        self.0.as_ref().unwrap()
    }
}

impl DerefMut for SqliteConnection {
    fn deref_mut(&mut self) -> &mut Self::Target {
        self.0.as_mut().unwrap()
    }
}

impl Default for SqliteConnection {
    fn default() -> Self {
        SqliteConnection(Some(default_connection()))
    }
}

impl SqliteConnection {
    pub fn finish(&mut self) {
        let connection = self.0.take().unwrap();
        connection
            .query_row("PRAGMA wal_checkpoint(TRUNCATE)", [], |_| Ok(()))
            .unwrap_or_else(|e| {
                error!("Ошибка при checkpoint WAL: {e}");
            });
        connection
            .execute("PRAGMA optimize", [])
            .unwrap_or_else(|e| {
                error!("Ошибка при  оптимизации БД: {e}");
                0
            });

        self.0 = None;
    }
    pub fn restart(&mut self) {
        self.0 = Some(default_connection());
    }
}

impl Drop for SqliteConnection {
    fn drop(&mut self) {
        self.finish();
    }
}