163 lines
4.3 KiB
Rust
163 lines
4.3 KiB
Rust
use rusqlite::{Connection, params};
|
|
use std::collections::HashMap;
|
|
use crate::data::entities::{WordData, WordGroup};
|
|
|
|
pub fn add_word(word: &mut WordData, connection: &Connection) {
|
|
let index : u32 = connection
|
|
.query_row(
|
|
"INSERT INTO words (key, value, tags, more, group_id) VALUES (?1, ?2, ?3, ?4, ?5) RETURNING id",
|
|
(
|
|
&word.key,
|
|
&word.value,
|
|
&word.tags,
|
|
serde_json::to_string(&word.additional).unwrap(),
|
|
&word.group_id,
|
|
),
|
|
|row| row.get(0),
|
|
).unwrap();
|
|
|
|
word.id = index.into();
|
|
}
|
|
|
|
pub fn add_words(words: &mut [WordData], connection: &mut Connection) {
|
|
let tx = connection.transaction().unwrap();
|
|
let count = words.len();
|
|
|
|
{
|
|
let mut stmt = tx
|
|
.prepare(
|
|
"INSERT INTO words (key, value, tags, more, group_id) VALUES (?1, ?2, ?3, ?4, ?5)",
|
|
)
|
|
.unwrap();
|
|
|
|
for word in words.iter() {
|
|
stmt.execute(params![
|
|
word.key,
|
|
word.value,
|
|
word.tags,
|
|
serde_json::to_string(&word.additional).unwrap(),
|
|
word.group_id
|
|
])
|
|
.unwrap();
|
|
}
|
|
}
|
|
|
|
let last_index: u32 = tx.last_insert_rowid() as u32;
|
|
tx.commit().unwrap();
|
|
|
|
let start_index = last_index - (count as u32) + 1;
|
|
|
|
for (index, id) in (start_index..=last_index).enumerate() {
|
|
words[index].id = id.into();
|
|
}
|
|
}
|
|
|
|
pub fn update_word(word: &mut WordData, connection: &Connection) {
|
|
if !word.id.is_valid() {
|
|
add_word(word, connection);
|
|
} else {
|
|
connection
|
|
.execute(
|
|
"UPDATE words SET key = ?1, value = ?2, tags = ?3, more = ?4 WHERE id = ?5",
|
|
(
|
|
&word.key,
|
|
&word.value,
|
|
&word.tags,
|
|
serde_json::to_string(&word.additional).unwrap(),
|
|
&word.id,
|
|
),
|
|
)
|
|
.unwrap();
|
|
}
|
|
}
|
|
|
|
pub fn delete_word(word: &WordData, connection: &Connection) {
|
|
if !word.id.is_valid() {
|
|
return;
|
|
}
|
|
connection
|
|
.execute("DELETE FROM words WHERE id = ?1", (&word.id,))
|
|
.unwrap();
|
|
}
|
|
|
|
pub fn load_words(connection: &Connection) -> Vec<WordData> {
|
|
let mut stmt = connection
|
|
.prepare("SELECT id, key, value, tags, more, group_id FROM words")
|
|
.unwrap();
|
|
let word_iter = stmt
|
|
.query_map([], |row| {
|
|
let additional: String = row.get(4)?;
|
|
Ok(WordData {
|
|
id: row.get(0)?,
|
|
key: row.get(1)?,
|
|
value: row.get(2)?,
|
|
tags: row.get(3)?,
|
|
additional: serde_json::from_str::<HashMap<String, String>>(&additional).unwrap(),
|
|
group_id: row.get(5)?,
|
|
})
|
|
})
|
|
.unwrap();
|
|
|
|
let mut buffer = vec![];
|
|
for word in word_iter {
|
|
buffer.push(word.unwrap());
|
|
}
|
|
|
|
buffer
|
|
}
|
|
|
|
pub fn load_word_groups(connection: &Connection) -> Vec<WordGroup> {
|
|
let mut stmt = connection
|
|
.prepare("SELECT id, name FROM word_group")
|
|
.unwrap();
|
|
let group_iter = stmt
|
|
.query_map([], |row| {
|
|
Ok(WordGroup {
|
|
id: row.get(0)?,
|
|
name: row.get(1)?,
|
|
})
|
|
})
|
|
.unwrap();
|
|
|
|
let mut buffer = vec![];
|
|
for group in group_iter {
|
|
buffer.push(group.unwrap());
|
|
}
|
|
|
|
buffer
|
|
}
|
|
|
|
pub fn add_group(group: &mut WordGroup, connection: &Connection) {
|
|
let index: u32 = connection
|
|
.query_row(
|
|
"INSERT INTO word_group (name) VALUES (?1) RETURNING id",
|
|
(&group.name,),
|
|
|row| row.get(0),
|
|
)
|
|
.unwrap();
|
|
|
|
group.id = index.into();
|
|
}
|
|
|
|
pub fn update_group(group: &mut WordGroup, connection: &Connection) {
|
|
if !group.id.is_valid() {
|
|
add_group(group, connection);
|
|
} else {
|
|
connection
|
|
.execute(
|
|
"UPDATE word_group SET name = ?1 WHERE id = ?2",
|
|
(&group.name, &group.id),
|
|
)
|
|
.unwrap();
|
|
}
|
|
}
|
|
|
|
pub fn delete_group(group: &WordGroup, connection: &Connection) {
|
|
if !group.id.is_valid() {
|
|
return;
|
|
}
|
|
connection
|
|
.execute("DELETE FROM word_group WHERE id = ?1", (&group.id,))
|
|
.unwrap();
|
|
}
|