mirror of
https://github.com/atuinsh/atuin.git
synced 2024-11-25 01:34:13 +01:00
versioning
This commit is contained in:
parent
012b8723af
commit
c4787ddaec
@ -8,6 +8,7 @@ pub mod history;
|
||||
pub mod record;
|
||||
pub mod status;
|
||||
pub mod user;
|
||||
pub mod v0;
|
||||
|
||||
const VERSION: &str = env!("CARGO_PKG_VERSION");
|
||||
|
||||
|
@ -1,109 +1,47 @@
|
||||
use axum::{extract::Query, extract::State, Json};
|
||||
use axum::{extract::Query, response::IntoResponse, Json};
|
||||
use http::StatusCode;
|
||||
use metrics::counter;
|
||||
use serde::Deserialize;
|
||||
use tracing::{error, instrument};
|
||||
use serde_json::json;
|
||||
use tracing::instrument;
|
||||
|
||||
use super::{ErrorResponse, ErrorResponseStatus, RespExt};
|
||||
use crate::router::{AppState, UserAuth};
|
||||
use crate::router::UserAuth;
|
||||
use atuin_server_database::Database;
|
||||
|
||||
use atuin_common::record::{EncryptedData, HostId, Record, RecordIdx, RecordStatus};
|
||||
use atuin_common::record::{EncryptedData, HostId, Record, RecordId};
|
||||
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn post<DB: Database>(
|
||||
UserAuth(user): UserAuth,
|
||||
state: State<AppState<DB>>,
|
||||
Json(records): Json<Vec<Record<EncryptedData>>>,
|
||||
) -> Result<(), ErrorResponseStatus<'static>> {
|
||||
let State(AppState { database, settings }) = state;
|
||||
|
||||
tracing::debug!(
|
||||
count = records.len(),
|
||||
user = user.username,
|
||||
"request to add records"
|
||||
);
|
||||
|
||||
counter!("atuin_record_uploaded", records.len() as u64);
|
||||
|
||||
let too_big = records
|
||||
.iter()
|
||||
.any(|r| r.data.data.len() >= settings.max_record_size || settings.max_record_size == 0);
|
||||
|
||||
if too_big {
|
||||
counter!("atuin_record_too_large", 1);
|
||||
// anyone who has actually used the old record store (a very small number) will see this error
|
||||
// upon trying to sync.
|
||||
// 1. The status endpoint will say that the server has nothing
|
||||
// 2. The client will try to upload local records
|
||||
// 3. Sync will fail with this error
|
||||
|
||||
// If the client has no local records, they will see the empty index and do nothing. For the
|
||||
// vast majority of users, this is the case.
|
||||
return Err(
|
||||
ErrorResponse::reply("could not add records; record too large")
|
||||
ErrorResponse::reply("record store deprecated; please upgrade")
|
||||
.with_status(StatusCode::BAD_REQUEST),
|
||||
);
|
||||
}
|
||||
|
||||
if let Err(e) = database.add_records(&user, &records).await {
|
||||
error!("failed to add record: {}", e);
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn index<DB: Database>(UserAuth(user): UserAuth) -> axum::response::Response {
|
||||
let ret = json!({
|
||||
"hosts": {}
|
||||
});
|
||||
|
||||
return Err(ErrorResponse::reply("failed to add record")
|
||||
.with_status(StatusCode::INTERNAL_SERVER_ERROR));
|
||||
};
|
||||
|
||||
Ok(())
|
||||
ret.to_string().into_response()
|
||||
}
|
||||
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn index<DB: Database>(
|
||||
pub async fn next(
|
||||
UserAuth(user): UserAuth,
|
||||
state: State<AppState<DB>>,
|
||||
) -> Result<Json<RecordStatus>, ErrorResponseStatus<'static>> {
|
||||
let State(AppState {
|
||||
database,
|
||||
settings: _,
|
||||
}) = state;
|
||||
|
||||
let record_index = match database.status(&user).await {
|
||||
Ok(index) => index,
|
||||
Err(e) => {
|
||||
error!("failed to get record index: {}", e);
|
||||
|
||||
return Err(ErrorResponse::reply("failed to calculate record index")
|
||||
.with_status(StatusCode::INTERNAL_SERVER_ERROR));
|
||||
}
|
||||
};
|
||||
|
||||
Ok(Json(record_index))
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
pub struct NextParams {
|
||||
host: HostId,
|
||||
tag: String,
|
||||
start: Option<RecordIdx>,
|
||||
count: u64,
|
||||
}
|
||||
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn next<DB: Database>(
|
||||
params: Query<NextParams>,
|
||||
UserAuth(user): UserAuth,
|
||||
state: State<AppState<DB>>,
|
||||
) -> Result<Json<Vec<Record<EncryptedData>>>, ErrorResponseStatus<'static>> {
|
||||
let State(AppState {
|
||||
database,
|
||||
settings: _,
|
||||
}) = state;
|
||||
let params = params.0;
|
||||
|
||||
let records = match database
|
||||
.next_records(&user, params.host, params.tag, params.start, params.count)
|
||||
.await
|
||||
{
|
||||
Ok(records) => records,
|
||||
Err(e) => {
|
||||
error!("failed to get record index: {}", e);
|
||||
|
||||
return Err(ErrorResponse::reply("failed to calculate record index")
|
||||
.with_status(StatusCode::INTERNAL_SERVER_ERROR));
|
||||
}
|
||||
};
|
||||
let records = Vec::new();
|
||||
|
||||
Ok(Json(records))
|
||||
}
|
||||
|
1
atuin-server/src/handlers/v0/mod.rs
Normal file
1
atuin-server/src/handlers/v0/mod.rs
Normal file
@ -0,0 +1 @@
|
||||
pub(crate) mod record;
|
111
atuin-server/src/handlers/v0/record.rs
Normal file
111
atuin-server/src/handlers/v0/record.rs
Normal file
@ -0,0 +1,111 @@
|
||||
use axum::{extract::Query, extract::State, Json};
|
||||
use 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;
|
||||
|
||||
use atuin_common::record::{EncryptedData, HostId, Record, RecordIdx, RecordStatus};
|
||||
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn post<DB: Database>(
|
||||
UserAuth(user): UserAuth,
|
||||
state: State<AppState<DB>>,
|
||||
Json(records): Json<Vec<Record<EncryptedData>>>,
|
||||
) -> Result<(), ErrorResponseStatus<'static>> {
|
||||
let State(AppState { database, settings }) = state;
|
||||
|
||||
tracing::debug!(
|
||||
count = records.len(),
|
||||
user = user.username,
|
||||
"request to add records"
|
||||
);
|
||||
|
||||
counter!("atuin_record_uploaded", records.len() as u64);
|
||||
|
||||
let too_big = records
|
||||
.iter()
|
||||
.any(|r| r.data.data.len() >= settings.max_record_size || settings.max_record_size == 0);
|
||||
|
||||
if too_big {
|
||||
counter!("atuin_record_too_large", 1);
|
||||
|
||||
return Err(
|
||||
ErrorResponse::reply("could not add records; record too large")
|
||||
.with_status(StatusCode::BAD_REQUEST),
|
||||
);
|
||||
}
|
||||
|
||||
if let Err(e) = database.add_records(&user, &records).await {
|
||||
error!("failed to add record: {}", e);
|
||||
|
||||
return Err(ErrorResponse::reply("failed to add record")
|
||||
.with_status(StatusCode::INTERNAL_SERVER_ERROR));
|
||||
};
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn index<DB: Database>(
|
||||
UserAuth(user): UserAuth,
|
||||
state: State<AppState<DB>>,
|
||||
) -> Result<Json<RecordStatus>, ErrorResponseStatus<'static>> {
|
||||
let State(AppState {
|
||||
database,
|
||||
settings: _,
|
||||
}) = state;
|
||||
|
||||
let record_index = match database.status(&user).await {
|
||||
Ok(index) => index,
|
||||
Err(e) => {
|
||||
error!("failed to get record index: {}", e);
|
||||
|
||||
return Err(ErrorResponse::reply("failed to calculate record index")
|
||||
.with_status(StatusCode::INTERNAL_SERVER_ERROR));
|
||||
}
|
||||
};
|
||||
|
||||
Ok(Json(record_index))
|
||||
}
|
||||
|
||||
#[derive(Deserialize)]
|
||||
struct NextParams {
|
||||
host: HostId,
|
||||
tag: String,
|
||||
start: Option<RecordIdx>,
|
||||
count: u64,
|
||||
}
|
||||
|
||||
#[instrument(skip_all, fields(user.id = user.id))]
|
||||
pub async fn next<DB: Database>(
|
||||
params: Query<NextParams>,
|
||||
UserAuth(user): UserAuth,
|
||||
state: State<AppState<DB>>,
|
||||
) -> Result<Json<Vec<Record<EncryptedData>>>, ErrorResponseStatus<'static>> {
|
||||
let State(AppState {
|
||||
database,
|
||||
settings: _,
|
||||
}) = state;
|
||||
let params = params.0;
|
||||
|
||||
let records = match database
|
||||
.next_records(&user, params.host, params.tag, params.start, params.count)
|
||||
.await
|
||||
{
|
||||
Ok(records) => records,
|
||||
Err(e) => {
|
||||
error!("failed to get record index: {}", e);
|
||||
|
||||
return Err(ErrorResponse::reply("failed to calculate record index")
|
||||
.with_status(StatusCode::INTERNAL_SERVER_ERROR));
|
||||
}
|
||||
};
|
||||
|
||||
Ok(Json(records))
|
||||
}
|
@ -106,9 +106,9 @@ pub fn router<DB: Database>(database: DB, settings: Settings<DB::Settings>) -> R
|
||||
.route("/sync/status", get(handlers::status::status))
|
||||
.route("/history", post(handlers::history::add))
|
||||
.route("/history", delete(handlers::history::delete))
|
||||
.route("/record", post(handlers::record::post))
|
||||
.route("/record", get(handlers::record::index))
|
||||
.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::record::next))
|
||||
.route("/user/:username", get(handlers::user::get))
|
||||
.route("/account", delete(handlers::user::delete))
|
||||
.route("/register", post(handlers::user::register))
|
||||
|
Loading…
Reference in New Issue
Block a user