obscura-server 0.1.0

A server for relaying secure messages using the Signal Protocol
Documentation 
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

use crate::config::Config;
use crate::core::notification::Notifier;
use crate::storage::DbPool;
use axum::{
    Router,
    routing::{get, post},
};
use std::sync::Arc;
use tower_governor::{GovernorLayer, governor::GovernorConfigBuilder};

pub mod auth;
pub mod gateway;
pub mod keys;
pub mod messages;
pub mod middleware;

#[derive(Clone)]
pub struct AppState {
    pub pool: DbPool,
    pub config: Config,
    pub notifier: Arc<dyn Notifier>,
}

pub fn app_router(pool: DbPool, config: Config, notifier: Arc<dyn Notifier>) -> Router {
    let governor_conf = Arc::new(
        GovernorConfigBuilder::default()
            .per_second(config.rate_limit_per_second as u64)
            .burst_size(config.rate_limit_burst)
            .finish()
            .unwrap(),
    );

    let state = AppState { pool, config, notifier };

    Router::new()
        .route("/v1/accounts", post(auth::register))
        .route("/v1/sessions", post(auth::login))
        .route("/v1/keys", post(keys::upload_keys))
        .route("/v1/keys/{userId}", get(keys::get_pre_key_bundle))
        .route("/v1/messages/{recipientId}", post(messages::send_message))
        .route("/v1/gateway", get(gateway::websocket_handler))
        .layer(GovernorLayer::new(governor_conf))
        .with_state(state)
}