jap_learn/src/data_provider/sqlite.rs

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

pub fn create_db() {
    let db_file = db_path();
    if !db_file.exists() {
        std::fs::File::create(&db_file).unwrap();
        let connection = Connection::open(&db_file).unwrap();
        connection.execute("PRAGMA foreign_keys = ON;", []).unwrap();

        create_tables(&connection);
    }
}

fn create_tables(conn: &Connection) {
    make_card_set(conn).unwrap();
    make_settings(conn).unwrap();
    make_word_group(conn).unwrap();
    make_words(conn).unwrap();
    make_card_stats(conn).unwrap();
}

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(())
}

pub fn default_connection() -> Connection {
    let db_file = db_path();
    let connection = Connection::open(db_file).unwrap();
    connection.execute("PRAGMA foreign_keys = ON;", []).unwrap();
    connection
}