start-os/core/src/middleware/db.rs

use axum::response::Response;
use http::HeaderValue;
use http::header::InvalidHeaderValue;
use rpc_toolkit::{Middleware, RpcRequest, RpcResponse};
use rust_i18n::t;
use serde::Deserialize;

use crate::context::RpcContext;

#[derive(Deserialize)]
pub struct Metadata {
    #[serde(default)]
    sync_db: bool,
}

#[derive(Clone)]
pub struct SyncDb {
    sync_db: bool,
}
impl SyncDb {
    pub fn new() -> Self {
        SyncDb { sync_db: false }
    }
}

impl Middleware<RpcContext> for SyncDb {
    type Metadata = Metadata;
    async fn process_rpc_request(
        &mut self,
        _: &RpcContext,
        metadata: Self::Metadata,
        _: &mut RpcRequest,
    ) -> Result<(), RpcResponse> {
        self.sync_db = metadata.sync_db;
        Ok(())
    }
    async fn process_http_response(&mut self, context: &RpcContext, response: &mut Response) {
        if let Err(e) = async {
            if self.sync_db {
                let id = context.db.sequence().await;
                response
                    .headers_mut()
                    .append("X-Patch-Sequence", HeaderValue::from_str(&id.to_string())?);
                context.sync_db.send_replace(id);
            }
            Ok::<_, InvalidHeaderValue>(())
        }
        .await
        {
            tracing::error!(
                "{}",
                t!(
                    "middleware.db.error-writing-patch-sequence-header",
                    error = e
                )
            );
            tracing::debug!("{e:?}");
        }
    }
}