apidocs.rs

use utoipa::{
    openapi::{
        security::{ApiKey, ApiKeyValue, HttpAuthScheme, HttpBuilder, SecurityScheme},
        ServerBuilder,
    },
    OpenApi,
};

use crate::{
    auth::routes::AuthApiDocs, bot::routes::BotsApiDoc, channels::routes::ChannelsApiDocs,
    friends::management::FriendsManagementApiDoc, friends::messaging::FriendsMessagingApiDoc,
    guilds::routes::GuildsApiDocs, invites::routes::InvitesApiDoc, media::routes::MediaApiDocs,
    messaging::routes::MessagingApiDocs, realtime::pubsub::PubSubApiDoc,
    voice::routes::VoiceApiDoc,
};

#[derive(OpenApi)]
#[openapi(
    modifiers(&TokenSecurityAddon)
)]
pub struct ApiDocs;

pub fn setup_oapi() -> utoipa::openapi::OpenApi {
    let mut oapi = ApiDocs::openapi();
    oapi.servers = Some(vec![
        ServerBuilder::new().url("/").build(),
        ServerBuilder::new().url("/api").build(),
    ]);

    oapi.merge(AuthApiDocs::openapi());
    oapi.merge(ChannelsApiDocs::openapi());
    oapi.merge(GuildsApiDocs::openapi());
    oapi.merge(MediaApiDocs::openapi());
    oapi.merge(MessagingApiDocs::openapi());
    oapi.merge(VoiceApiDoc::openapi());
    oapi.merge(PubSubApiDoc::openapi());
    oapi.merge(BotsApiDoc::openapi());
    oapi.merge(FriendsManagementApiDoc::openapi());
    oapi.merge(FriendsMessagingApiDoc::openapi());
    oapi.merge(InvitesApiDoc::openapi());
    oapi
}

struct TokenSecurityAddon;

impl utoipa::Modify for TokenSecurityAddon {
    fn modify(&self, openapi: &mut utoipa::openapi::OpenApi) {
        openapi.components = Some(
            utoipa::openapi::ComponentsBuilder::new()
                .security_scheme(
                    "token",
                    SecurityScheme::Http(
                        HttpBuilder::new()
                            .scheme(HttpAuthScheme::Bearer)
                            .bearer_format("AccessToken")
                            .build(),
                    ),
                )
                .build(),
        )
    }
}

struct VoiceSecurityAddon;

impl utoipa::Modify for VoiceSecurityAddon {
    fn modify(&self, openapi: &mut utoipa::openapi::OpenApi) {
        openapi.components = Some(
            utoipa::openapi::ComponentsBuilder::new()
                .security_scheme(
                    "voice",
                    SecurityScheme::ApiKey(ApiKey::Header(ApiKeyValue::new("RTC-Token"))),
                )
                .build(),
        )
    }
}