João Marcos
6f467281e0
After the user confirmation, only the current instance of the completions provider had the answer stored. In this PR, the changes are propagated by having each provider have an `Entity<choice>`, and having a lookup map with one `Entity<choice>` for each worktree that `Zeta` has seen. Release Notes: - N/A
49 lines
1.3 KiB
Rust
49 lines
1.3 KiB
Rust
use std::path::{Path, PathBuf};
|
|
use workspace::WorkspaceDb;
|
|
|
|
use db::sqlez_macros::sql;
|
|
use db::{define_connection, query};
|
|
|
|
define_connection!(
|
|
pub static ref DB: ZetaDb<WorkspaceDb> = &[
|
|
sql! (
|
|
CREATE TABLE zeta_preferences(
|
|
worktree_path BLOB NOT NULL PRIMARY KEY,
|
|
accepted_data_collection INTEGER
|
|
) STRICT;
|
|
),
|
|
];
|
|
);
|
|
|
|
impl ZetaDb {
|
|
query! {
|
|
pub fn get_all_data_collection_preferences() -> Result<Vec<(PathBuf, bool)>> {
|
|
SELECT worktree_path, accepted_data_collection FROM zeta_preferences
|
|
}
|
|
}
|
|
|
|
query! {
|
|
pub fn get_accepted_data_collection(worktree_path: &Path) -> Result<Option<bool>> {
|
|
SELECT accepted_data_collection FROM zeta_preferences
|
|
WHERE worktree_path = ?
|
|
}
|
|
}
|
|
|
|
query! {
|
|
pub async fn save_data_collection_choice(worktree_path: PathBuf, accepted_data_collection: bool) -> Result<()> {
|
|
INSERT INTO zeta_preferences
|
|
(worktree_path, accepted_data_collection)
|
|
VALUES
|
|
(?1, ?2)
|
|
ON CONFLICT (worktree_path) DO UPDATE SET
|
|
accepted_data_collection = ?2
|
|
}
|
|
}
|
|
|
|
query! {
|
|
pub async fn clear_all_zeta_preferences() -> Result<()> {
|
|
DELETE FROM zeta_preferences
|
|
}
|
|
}
|
|
}
|