k2db-api-server 0.1.1

Single-binary Rust server for the k2db API
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104

// SPDX-FileCopyrightText: 2026 Alexander R. Croft
// SPDX-License-Identifier: MIT

use axum::Json;
use axum::http::{HeaderValue, StatusCode, header};
use axum::response::{IntoResponse, Response};
use k2db::{K2DbError, ServiceError as K2ServiceError};
use k2db_api_contract::ProblemDetailsPayload;

#[derive(Debug, Clone)]
pub struct ApiError {
    status: StatusCode,
    payload: ProblemDetailsPayload,
}

impl ApiError {
    pub fn new(
        status: StatusCode,
        title: impl Into<String>,
        detail: impl Into<String>,
        trace: impl Into<Option<String>>,
    ) -> Self {
        let title = title.into();
        Self {
            status,
            payload: ProblemDetailsPayload {
                type_uri: Some(format!("urn:service-error:{title}")),
                title: Some(title),
                status: Some(status.as_u16()),
                detail: Some(detail.into()),
                trace: trace.into(),
                chain: None,
                extra: serde_json::Map::new(),
            },
        }
    }

    pub fn bad_request(detail: impl Into<String>, trace: &'static str) -> Self {
        Self::new(StatusCode::BAD_REQUEST, "bad_request", detail, Some(trace.to_owned()))
    }

    pub fn unauthorized(detail: impl Into<String>, trace: &'static str) -> Self {
        Self::new(StatusCode::UNAUTHORIZED, "unauthorized", detail, Some(trace.to_owned()))
    }

    pub fn forbidden(detail: impl Into<String>, trace: &'static str) -> Self {
        Self::new(StatusCode::FORBIDDEN, "forbidden", detail, Some(trace.to_owned()))
    }

    pub fn internal(detail: impl Into<String>, trace: &'static str) -> Self {
        Self::new(
            StatusCode::INTERNAL_SERVER_ERROR,
            "service_error",
            detail,
            Some(trace.to_owned()),
        )
    }

    pub fn service_unavailable(detail: impl Into<String>, trace: &'static str) -> Self {
        Self::new(
            StatusCode::SERVICE_UNAVAILABLE,
            "service_unavailable",
            detail,
            Some(trace.to_owned()),
        )
    }

    pub fn from_k2db(error: K2DbError) -> Self {
        let status = match error.service_error {
            K2ServiceError::BadRequest => StatusCode::BAD_REQUEST,
            K2ServiceError::NotFound => StatusCode::NOT_FOUND,
            K2ServiceError::ConfigurationError => StatusCode::INTERNAL_SERVER_ERROR,
            K2ServiceError::AlreadyExists => StatusCode::CONFLICT,
            K2ServiceError::ValidationError => StatusCode::BAD_REQUEST,
            K2ServiceError::BadGateway => StatusCode::BAD_GATEWAY,
            K2ServiceError::ServiceUnavailable => StatusCode::SERVICE_UNAVAILABLE,
            K2ServiceError::SystemError => StatusCode::INTERNAL_SERVER_ERROR,
        };

        let title = match error.service_error {
            K2ServiceError::BadRequest => "bad_request",
            K2ServiceError::NotFound => "not_found",
            K2ServiceError::ConfigurationError => "configuration_error",
            K2ServiceError::AlreadyExists => "already_exists",
            K2ServiceError::ValidationError => "validation_error",
            K2ServiceError::BadGateway => "bad_gateway",
            K2ServiceError::ServiceUnavailable => "service_unavailable",
            K2ServiceError::SystemError => "service_error",
        };

        Self::new(status, title, error.message, error.key)
    }
}

impl IntoResponse for ApiError {
    fn into_response(self) -> Response {
        let mut response = (self.status, Json(self.payload)).into_response();
        response.headers_mut().insert(
            header::CONTENT_TYPE,
            HeaderValue::from_static("application/problem+json"),
        );
        response
    }
}