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 record;
|
||||||
pub mod status;
|
pub mod status;
|
||||||
pub mod user;
|
pub mod user;
|
||||||
|
pub mod v0;
|
||||||
|
|
||||||
const VERSION: &str = env!("CARGO_PKG_VERSION");
|
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 http::StatusCode;
|
||||||
use metrics::counter;
|
|
||||||
use serde::Deserialize;
|
use serde::Deserialize;
|
||||||
use tracing::{error, instrument};
|
use serde_json::json;
|
||||||
|
use tracing::instrument;
|
||||||
|
|
||||||
use super::{ErrorResponse, ErrorResponseStatus, RespExt};
|
use super::{ErrorResponse, ErrorResponseStatus, RespExt};
|
||||||
use crate::router::{AppState, UserAuth};
|
use crate::router::UserAuth;
|
||||||
use atuin_server_database::Database;
|
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))]
|
#[instrument(skip_all, fields(user.id = user.id))]
|
||||||
pub async fn post<DB: Database>(
|
pub async fn post<DB: Database>(
|
||||||
UserAuth(user): UserAuth,
|
UserAuth(user): UserAuth,
|
||||||
state: State<AppState<DB>>,
|
|
||||||
Json(records): Json<Vec<Record<EncryptedData>>>,
|
|
||||||
) -> Result<(), ErrorResponseStatus<'static>> {
|
) -> Result<(), ErrorResponseStatus<'static>> {
|
||||||
let State(AppState { database, settings }) = state;
|
// 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
|
||||||
|
|
||||||
tracing::debug!(
|
// If the client has no local records, they will see the empty index and do nothing. For the
|
||||||
count = records.len(),
|
// vast majority of users, this is the case.
|
||||||
user = user.username,
|
return Err(
|
||||||
"request to add records"
|
ErrorResponse::reply("record store deprecated; please upgrade")
|
||||||
|
.with_status(StatusCode::BAD_REQUEST),
|
||||||
);
|
);
|
||||||
|
|
||||||
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))]
|
#[instrument(skip_all, fields(user.id = user.id))]
|
||||||
pub async fn index<DB: Database>(
|
pub async fn index<DB: Database>(UserAuth(user): UserAuth) -> axum::response::Response {
|
||||||
UserAuth(user): UserAuth,
|
let ret = json!({
|
||||||
state: State<AppState<DB>>,
|
"hosts": {}
|
||||||
) -> Result<Json<RecordStatus>, ErrorResponseStatus<'static>> {
|
});
|
||||||
let State(AppState {
|
|
||||||
database,
|
|
||||||
settings: _,
|
|
||||||
}) = state;
|
|
||||||
|
|
||||||
let record_index = match database.status(&user).await {
|
ret.to_string().into_response()
|
||||||
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))]
|
#[instrument(skip_all, fields(user.id = user.id))]
|
||||||
pub async fn next<DB: Database>(
|
pub async fn next(
|
||||||
params: Query<NextParams>,
|
|
||||||
UserAuth(user): UserAuth,
|
UserAuth(user): UserAuth,
|
||||||
state: State<AppState<DB>>,
|
|
||||||
) -> Result<Json<Vec<Record<EncryptedData>>>, ErrorResponseStatus<'static>> {
|
) -> Result<Json<Vec<Record<EncryptedData>>>, ErrorResponseStatus<'static>> {
|
||||||
let State(AppState {
|
let records = Vec::new();
|
||||||
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))
|
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("/sync/status", get(handlers::status::status))
|
||||||
.route("/history", post(handlers::history::add))
|
.route("/history", post(handlers::history::add))
|
||||||
.route("/history", delete(handlers::history::delete))
|
.route("/history", delete(handlers::history::delete))
|
||||||
.route("/record", post(handlers::record::post))
|
.route("/api/v0/record", post(handlers::v0::record::post))
|
||||||
.route("/record", get(handlers::record::index))
|
.route("/api/v0/record", get(handlers::v0::record::index))
|
||||||
.route("/record/next", get(handlers::record::next))
|
.route("/api/v0/record/next", get(handlers::record::next))
|
||||||
.route("/user/:username", get(handlers::user::get))
|
.route("/user/:username", get(handlers::user::get))
|
||||||
.route("/account", delete(handlers::user::delete))
|
.route("/account", delete(handlers::user::delete))
|
||||||
.route("/register", post(handlers::user::register))
|
.route("/register", post(handlers::user::register))
|
||||||
|
Loading…
Reference in New Issue
Block a user