use std::collections::HashSet;
use axum::{response::IntoResponse, routing::get, Router};
use composable_tower_http::{
authorize::{
api_key::{ApiKey, DefaultApiKeyAuthorizer},
basic_auth::{BasicAuthUser, DefaultBasicAuthAuthorizer},
header::{basic_auth::DefaultBasicAuthExtractor, DefaultHeaderExtractor},
},
extension::ExtensionLayerExt,
extract::{And, Extracted, ExtractorExt},
};
#[path = "../util/util.rs"]
mod util;
async fn api_key_and_basic_auth(
Extracted(And { left, right }): Extracted<And<ApiKey, BasicAuthUser>>,
) -> impl IntoResponse {
format!("You used the api key: {:?}, and you are: {:?}", left, right)
}
#[tokio::main]
async fn main() -> anyhow::Result<()> {
util::init("and")?;
let valid_api_keys: HashSet<ApiKey> = ["api-key-1", "api-key-2"]
.into_iter()
.map(ApiKey::new)
.collect();
let api_key_authorizer =
DefaultApiKeyAuthorizer::new(DefaultHeaderExtractor::new("x-api-key"), valid_api_keys);
let basic_auth_users: HashSet<BasicAuthUser> = [("user-1", "password-1"), ("user-2", "")]
.into_iter()
.map(Into::into)
.collect();
let basic_auth_authorizer =
DefaultBasicAuthAuthorizer::new(DefaultBasicAuthExtractor::new(), basic_auth_users);
let layer = api_key_authorizer
.and(basic_auth_authorizer)
.extension_layer();
let app = Router::new()
.route("/", get(api_key_and_basic_auth))
.layer(layer)
.layer(util::trace_layer());
util::serve(app).await
}