File size: 2,131 Bytes
d90101d | 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 | use std::sync::Arc;
use forge_app::{AuthService, EnvironmentInfra, HttpInfra, User, UserUsage};
use reqwest::Url;
use reqwest::header::{AUTHORIZATION, HeaderMap, HeaderValue};
const USER_INFO_ROUTE: &str = "auth/user";
const USER_USAGE_ROUTE: &str = "auth/usage";
#[derive(Default, Clone)]
pub struct ForgeAuthService<I> {
infra: Arc<I>,
}
impl<I: HttpInfra + EnvironmentInfra<Config = forge_config::ForgeConfig>> ForgeAuthService<I> {
pub fn new(infra: Arc<I>) -> Self {
Self { infra }
}
async fn user_info(&self, api_key: &str) -> anyhow::Result<User> {
let services_url = self.infra.get_config()?.services_url;
let url = format!("{services_url}{USER_INFO_ROUTE}");
let url = Url::parse(&url)?;
let mut headers = HeaderMap::new();
headers.insert(
AUTHORIZATION,
HeaderValue::from_str(&format!("Bearer {api_key}"))?,
);
let response = self
.infra
.http_get(&url, Some(headers))
.await?
.error_for_status()?;
Ok(serde_json::from_slice(&response.bytes().await?)?)
}
async fn user_usage(&self, api_key: &str) -> anyhow::Result<UserUsage> {
let services_url = self.infra.get_config()?.services_url;
let url = Url::parse(&format!("{services_url}{USER_USAGE_ROUTE}"))?;
let mut headers = HeaderMap::new();
headers.insert(
AUTHORIZATION,
HeaderValue::from_str(&format!("Bearer {api_key}"))?,
);
let response = self
.infra
.http_get(&url, Some(headers))
.await?
.error_for_status()?;
Ok(serde_json::from_slice(&response.bytes().await?)?)
}
}
#[async_trait::async_trait]
impl<I: HttpInfra + EnvironmentInfra<Config = forge_config::ForgeConfig>> AuthService
for ForgeAuthService<I>
{
async fn user_info(&self, api_key: &str) -> anyhow::Result<User> {
self.user_info(api_key).await
}
async fn user_usage(&self, api_key: &str) -> anyhow::Result<UserUsage> {
self.user_usage(api_key).await
}
}
|