Add endpoint handlers

This commit is contained in:
Andras Schmelczer 2024-12-08 14:23:07 +00:00
parent f44d4ab407
commit d4b66508ee
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
5 changed files with 297 additions and 0 deletions

View file

@ -0,0 +1,67 @@
use crate::app_state::AppState;
use anyhow::Context;
use anyhow::Result;
use axum::extract::DefaultBodyLimit;
use axum::extract::WebSocketUpgrade;
use axum::response::Response;
use axum::routing::delete;
use axum::{routing::get, routing::put, Router};
use log::info;
mod delete_document;
mod fetch_latest_document_version;
mod fetch_latest_documents;
mod requests;
mod update_document;
pub async fn create_server(app_state: AppState) -> Result<()> {
let address = format!(
"{}:{}",
&app_state.config.server.host, &app_state.config.server.port
);
let app = Router::new()
.route(
"/vaults/:vault_id/documents/latest",
get(fetch_latest_documents::fetch_latest_documents),
)
.route(
"/vaults/:vault_id/documents/:document_id/versions/:parent_version_id",
put(update_document::update_document),
)
.route(
"/vaults/:vault_id/documents/:document_id",
delete(delete_document::delete_document),
)
.route(
"/vaults/:vault_id/documents/:document_id/versions/latest",
get(fetch_latest_document_version::fetch_latest_document_version),
)
.route("/ws", get(handler))
.layer(DefaultBodyLimit::max(
app_state.config.server.max_body_size_mb * 1024 * 1024,
))
.with_state(app_state);
let listener = tokio::net::TcpListener::bind(address.clone())
.await
.with_context(|| format!("Failed to bind to address: {}", address))?;
info!(
"Listening on {}",
listener
.local_addr()
.context("Failed to get local address")?
);
axum::serve(listener, app)
.await
.context("Failed to start server")
}
async fn handler(ws: WebSocketUpgrade) -> Response {
ws.protocols(["graphql-ws", "graphql-transport-ws"])
.on_upgrade(|socket| async {
// ...
})
}

View file

@ -0,0 +1,66 @@
use crate::app_state::AppState;
use crate::database::models::DocumentId;
use crate::database::models::StoredDocumentVersion;
use crate::database::models::VaultId;
use crate::errors::not_found_error;
use crate::errors::server_error;
use crate::errors::SyncServerError;
use anyhow::anyhow;
use anyhow::Context;
use axum::extract::Path;
use axum::extract::State;
use axum::Json;
use super::requests::DeleteDocumentVersion;
#[axum::debug_handler]
pub async fn delete_document(
Path((vault_id, document_id)): Path<(VaultId, DocumentId)>,
State(state): State<AppState>,
Json(request): Json<DeleteDocumentVersion>,
) -> Result<(), SyncServerError> {
let mut transaction = state
.database
.create_transaction()
.await
.map_err(server_error)?;
let latest_version = state
.database
.get_latest_document_version(&vault_id, &document_id, Some(&mut transaction))
.await
.map_err(server_error)?
.map(Ok)
.unwrap_or_else(|| {
Err(not_found_error(anyhow!(
"Latest document version of document `{}` not found",
document_id
)))
})?;
let new_version = StoredDocumentVersion {
vault_id,
document_id,
version_id: latest_version.version_id + 1,
content: vec![],
created_date: request.created_date,
updated_date: chrono::Utc::now(),
relative_path: latest_version.relative_path,
is_binary: latest_version.is_binary,
is_deleted: true,
};
state
.database
.insert_document_version(&new_version, Some(&mut transaction))
.await
.map_err(server_error)?;
transaction
.commit()
.await
.context("Failed to commit successful transaction")
.map_err(server_error)?;
Ok(())
}

View file

@ -0,0 +1,32 @@
use crate::app_state::AppState;
use crate::database::models::DocumentId;
use crate::database::models::DocumentVersion;
use crate::database::models::VaultId;
use crate::errors::not_found_error;
use crate::errors::server_error;
use crate::errors::SyncServerError;
use anyhow::anyhow;
use axum::extract::Path;
use axum::extract::State;
use axum::Json;
#[axum::debug_handler]
pub async fn fetch_latest_document_version(
Path((vault_id, document_id)): Path<(VaultId, DocumentId)>,
State(state): State<AppState>,
) -> Result<Json<DocumentVersion>, SyncServerError> {
let latest_version = state
.database
.get_latest_document_version(&vault_id, &document_id, None)
.await
.map_err(server_error)?
.map(Ok)
.unwrap_or_else(|| {
Err(not_found_error(anyhow!(
"Latest document version of document `{}` not found",
document_id
)))
})?;
Ok(Json(latest_version.into()))
}

View file

@ -0,0 +1,22 @@
use crate::app_state::AppState;
use crate::database::models::DocumentVersionWithoutContent;
use crate::database::models::VaultId;
use crate::errors::server_error;
use crate::errors::SyncServerError;
use axum::extract::Path;
use axum::extract::State;
use axum::Json;
#[axum::debug_handler]
pub async fn fetch_latest_documents(
Path(vault_id): Path<VaultId>,
State(state): State<AppState>,
) -> Result<Json<Vec<DocumentVersionWithoutContent>>, SyncServerError> {
let latest_version = state
.database
.get_latest_documents(&vault_id, None)
.await
.map_err(server_error)?;
Ok(Json(latest_version))
}

View file

@ -0,0 +1,110 @@
use crate::app_state::AppState;
use crate::database::models::DocumentId;
use crate::database::models::DocumentVersionId;
use crate::database::models::DocumentVersionWithoutContent;
use crate::database::models::StoredDocumentVersion;
use crate::database::models::VaultId;
use crate::errors::client_error;
use crate::errors::not_found_error;
use crate::errors::server_error;
use crate::errors::SyncServerError;
use anyhow::anyhow;
use anyhow::Context;
use axum::extract::Path;
use axum::extract::State;
use axum::Json;
use sync_lib::base64_to_bytes;
use sync_lib::base64_to_string;
use super::requests::CreateDocumentVersion;
#[axum::debug_handler]
pub async fn update_document(
Path((vault_id, document_id, parent_version_id)): Path<(
VaultId,
DocumentId,
DocumentVersionId,
)>,
State(state): State<AppState>,
Json(request): Json<CreateDocumentVersion>,
) -> Result<Json<DocumentVersionWithoutContent>, SyncServerError> {
let parent = state
.database
.get_document_version(&vault_id, &document_id, &parent_version_id, None)
.await
.map_err(server_error)?
.map(Ok)
.unwrap_or_else(|| {
Err(not_found_error(anyhow!(
"Parent version with id `{}` not found",
parent_version_id
)))
})?;
let mut transaction = state
.database
.create_transaction()
.await
.map_err(server_error)?;
let latest_version = state
.database
.get_latest_document_version(&vault_id, &document_id, Some(&mut transaction))
.await
.map_err(server_error)?
.map(Ok)
.unwrap_or_else(|| {
Err(not_found_error(anyhow!(
"Latest document version of document `{}` not found",
document_id
)))
})?;
if latest_version.is_deleted {
return Err(client_error(anyhow!(
"Document `{}` is deleted",
document_id
)));
}
let merged_content = if request.is_binary {
base64_to_bytes(&request.content_base64)
.context("Failed to decode base64 content in request")
.map_err(client_error)?
} else {
reconcile::reconcile(
&parent.content_as_string(),
&latest_version.content_as_string(),
&base64_to_string(&request.content_base64)
.context("Failed to decode base64 content in request")
.map_err(client_error)?,
)
.into_bytes()
};
let new_version = StoredDocumentVersion {
vault_id,
document_id,
version_id: latest_version.version_id + 1,
content: merged_content,
created_date: request.created_date,
relative_path: request.relative_path,
updated_date: chrono::Utc::now(),
is_binary: request.is_binary,
is_deleted: false,
};
state
.database
.insert_document_version(&new_version, Some(&mut transaction))
.await
.map_err(server_error)?;
transaction
.commit()
.await
.context("Failed to commit successful transaction")
.map_err(server_error)?;
Ok(Json(new_version.into()))
}