jap_learn/src/data_provider/sqlite.rs

use crate::fs::*;
use rusqlite::Connection;

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");
}

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/card_set.sql");
    conn.execute(query, ())?;
    Ok(())
}

fn make_word_group(conn: &Connection) -> Result<(), rusqlite::Error> {
    let query = include_str!("../../sql/word_group.sql");
    conn.execute(query, ())?;
    conn.execute("insert into word_group (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/card_stats.sql");
    conn.execute(query, ())?;
    Ok(())
}

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