mirror of
https://github.com/ankitects/anki.git
synced 2025-09-21 07:22:23 -04:00
c521753057
* Add crate snafu * Replace all inline structs in AnkiError * Derive Snafu on AnkiError * Use snafu for card type errors * Use snafu whatever error for InvalidInput * Use snafu for NotFoundError and improve message * Use snafu for FileIoError to attach context Remove IoError. Add some context-attaching helpers to replace code returning bare io::Errors. * Add more context-attaching io helpers * Add message, context and backtrace to new snafus * Utilize error context and backtrace on frontend * Rename LocalizedError -> BackendError. * Remove DocumentedError. * Have all backend exceptions inherit BackendError. * Rename localized(_description) -> message * Remove accidentally committed experimental trait * invalid_input_context -> ok_or_invalid * ensure_valid_input! -> require! * Always return `Err` from `invalid_input!` Instead of a Result to unwrap, the macro accepts a source error now. * new_tempfile_in_parent -> new_tempfile_in_parent_of * ok_or_not_found -> or_not_found * ok_or_invalid -> or_invalid * Add crate convert_case * Use unqualified lowercase type name * Remove uses of snafu::ensure * Allow public construction of InvalidInputErrors (dae) Needed to port the AnkiDroid changes. * Make into_protobuf() public (dae) Also required for AnkiDroid. Not sure why it worked previously - possible bug in older Rust version?
80 lines
2.4 KiB
Rust
80 lines
2.4 KiB
Rust
// Copyright: Ankitects Pty Ltd and contributors
|
|
// License: GNU AGPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
|
|
|
|
use std::path::Path;
|
|
|
|
use rusqlite::{params, types::FromSql, Connection, ToSql};
|
|
|
|
use super::*;
|
|
use crate::prelude::*;
|
|
|
|
impl SqliteStorage {
|
|
pub(crate) fn usn(&self, server: bool) -> Result<Usn> {
|
|
if server {
|
|
Ok(Usn(self
|
|
.db
|
|
.prepare_cached("select usn from col")?
|
|
.query_row([], |row| row.get(0))?))
|
|
} else {
|
|
Ok(Usn(-1))
|
|
}
|
|
}
|
|
|
|
pub(crate) fn set_usn(&self, usn: Usn) -> Result<()> {
|
|
self.db
|
|
.prepare_cached("update col set usn = ?")?
|
|
.execute([usn])?;
|
|
Ok(())
|
|
}
|
|
|
|
pub(crate) fn increment_usn(&self) -> Result<()> {
|
|
self.db
|
|
.prepare_cached("update col set usn = usn + 1")?
|
|
.execute([])?;
|
|
Ok(())
|
|
}
|
|
|
|
pub(crate) fn objects_pending_sync<T: FromSql>(&self, table: &str, usn: Usn) -> Result<Vec<T>> {
|
|
self.db
|
|
.prepare_cached(&format!(
|
|
"select id from {} where {}",
|
|
table,
|
|
usn.pending_object_clause()
|
|
))?
|
|
.query_and_then([usn], |r| r.get(0).map_err(Into::into))?
|
|
.collect()
|
|
}
|
|
|
|
pub(crate) fn maybe_update_object_usns<I: ToSql>(
|
|
&self,
|
|
table: &str,
|
|
ids: &[I],
|
|
new_usn: Option<Usn>,
|
|
) -> Result<()> {
|
|
if let Some(new_usn) = new_usn {
|
|
let mut stmt = self
|
|
.db
|
|
.prepare_cached(&format!("update {} set usn=? where id=?", table))?;
|
|
for id in ids {
|
|
stmt.execute(params![new_usn, id])?;
|
|
}
|
|
}
|
|
Ok(())
|
|
}
|
|
}
|
|
|
|
/// Return error if file is unreadable, fails the sqlite
|
|
/// integrity check, or is not in the 'delete' journal mode.
|
|
/// On success, returns the opened DB.
|
|
pub(crate) fn open_and_check_sqlite_file(path: &Path) -> Result<Connection> {
|
|
let db = Connection::open(path)?;
|
|
match db.pragma_query_value(None, "integrity_check", |row| row.get::<_, String>(0)) {
|
|
Ok(s) => require!(s == "ok", "corrupt: {s}"),
|
|
Err(e) => return Err(e.into()),
|
|
};
|
|
match db.pragma_query_value(None, "journal_mode", |row| row.get::<_, String>(0)) {
|
|
Ok(s) if s == "delete" => Ok(db),
|
|
Ok(s) => invalid_input!("corrupt: {s}"),
|
|
Err(e) => Err(e.into()),
|
|
}
|
|
}
|