diff options
| author | Ellie Huxtable <ellie@elliehuxtable.com> | 2024-02-02 15:05:07 +0000 |
|---|---|---|
| committer | Ellie Huxtable <ellie@elliehuxtable.com> | 2024-02-02 18:01:09 +0000 |
| commit | c9a453289e2fea97b5ac17e265f99332edcdcd4c (patch) | |
| tree | e51eb1f7d2fd3b44fabf2c1260802232fa0367b2 /atuin-server | |
| parent | feat: failure to decrypt history = failure to sync (diff) | |
| download | atuin-c9a453289e2fea97b5ac17e265f99332edcdcd4c.zip | |
feat: add `store push --force`
This will
1. Wipe the remote store
2. Upload all of the local store to remote
Imagine the scenario where you end up with some mixed keys locally :(
You confirm this with
```
atuin store verify
```
You then fix it locally with
```
atuin store purge
```
Ensure that your local changes are reflected remotely with
```
atuin store push --force
```
and then (another PR, coming soon), update all other hosts with
```
atuin store pull --force
```
Diffstat (limited to '')
| -rw-r--r-- | atuin-server-database/src/lib.rs | 3 | ||||
| -rw-r--r-- | atuin-server-postgres/src/lib.rs | 13 | ||||
| -rw-r--r-- | atuin-server/src/handlers/v0/mod.rs | 1 | ||||
| -rw-r--r-- | atuin-server/src/handlers/v0/store.rs | 37 | ||||
| -rw-r--r-- | atuin-server/src/router.rs | 3 |
5 files changed, 55 insertions, 2 deletions
diff --git a/atuin-server-database/src/lib.rs b/atuin-server-database/src/lib.rs index dff1204d..d2c16b3d 100644 --- a/atuin-server-database/src/lib.rs +++ b/atuin-server-database/src/lib.rs @@ -53,15 +53,16 @@ pub trait Database: Sized + Clone + Send + Sync + 'static { async fn get_user(&self, username: &str) -> DbResult<User>; async fn get_user_session(&self, u: &User) -> DbResult<Session>; async fn add_user(&self, user: &NewUser) -> DbResult<i64>; - async fn delete_user(&self, u: &User) -> DbResult<()>; async fn update_user_password(&self, u: &User) -> DbResult<()>; async fn total_history(&self) -> DbResult<i64>; async fn count_history(&self, user: &User) -> DbResult<i64>; async fn count_history_cached(&self, user: &User) -> DbResult<i64>; + async fn delete_user(&self, u: &User) -> DbResult<()>; async fn delete_history(&self, user: &User, id: String) -> DbResult<()>; async fn deleted_history(&self, user: &User) -> DbResult<Vec<String>>; + async fn delete_store(&self, user: &User) -> DbResult<()>; async fn add_records(&self, user: &User, record: &[Record<EncryptedData>]) -> DbResult<()>; async fn next_records( diff --git a/atuin-server-postgres/src/lib.rs b/atuin-server-postgres/src/lib.rs index 1f7cf47a..0ad33076 100644 --- a/atuin-server-postgres/src/lib.rs +++ b/atuin-server-postgres/src/lib.rs @@ -133,6 +133,19 @@ impl Database for Postgres { Ok(res.0 as i64) } + async fn delete_store(&self, user: &User) -> DbResult<()> { + sqlx::query( + "delete from store + where user_id = $1", + ) + .bind(user.id) + .execute(&self.pool) + .await + .map_err(fix_error)?; + + Ok(()) + } + async fn delete_history(&self, user: &User, id: String) -> DbResult<()> { sqlx::query( "update history diff --git a/atuin-server/src/handlers/v0/mod.rs b/atuin-server/src/handlers/v0/mod.rs index 78fb47b8..2d6745cf 100644 --- a/atuin-server/src/handlers/v0/mod.rs +++ b/atuin-server/src/handlers/v0/mod.rs @@ -1 +1,2 @@ pub(crate) mod record; +pub(crate) mod store; diff --git a/atuin-server/src/handlers/v0/store.rs b/atuin-server/src/handlers/v0/store.rs new file mode 100644 index 00000000..941f2487 --- /dev/null +++ b/atuin-server/src/handlers/v0/store.rs @@ -0,0 +1,37 @@ +use axum::{extract::Query, extract::State, http::StatusCode}; +use metrics::counter; +use serde::Deserialize; +use tracing::{error, instrument}; + +use crate::{ + handlers::{ErrorResponse, ErrorResponseStatus, RespExt}, + router::{AppState, UserAuth}, +}; +use atuin_server_database::Database; + +#[derive(Deserialize)] +pub struct DeleteParams {} + +#[instrument(skip_all, fields(user.id = user.id))] +pub async fn delete<DB: Database>( + _params: Query<DeleteParams>, + UserAuth(user): UserAuth, + state: State<AppState<DB>>, +) -> Result<(), ErrorResponseStatus<'static>> { + let State(AppState { + database, + settings: _, + }) = state; + + if let Err(e) = database.delete_store(&user).await { + counter!("atuin_store_delete_failed", 1); + error!("failed to delete store {e:?}"); + + return Err(ErrorResponse::reply("failed to delete store") + .with_status(StatusCode::INTERNAL_SERVER_ERROR)); + } + + counter!("atuin_store_deleted", 1); + + Ok(()) +} diff --git a/atuin-server/src/router.rs b/atuin-server/src/router.rs index 74df229a..52fc1484 100644 --- a/atuin-server/src/router.rs +++ b/atuin-server/src/router.rs @@ -127,7 +127,8 @@ pub fn router<DB: Database>(database: DB, settings: Settings<DB::Settings>) -> R .route("/record/next", get(handlers::record::next)) .route("/api/v0/record", post(handlers::v0::record::post)) .route("/api/v0/record", get(handlers::v0::record::index)) - .route("/api/v0/record/next", get(handlers::v0::record::next)); + .route("/api/v0/record/next", get(handlers::v0::record::next)) + .route("/api/v0/store", delete(handlers::v0::store::delete)); let path = settings.path.as_str(); if path.is_empty() { |
