6db412f
use crate::errors::PullError;
use api::users::UserDto;
use auth::Accounts;
use cache::AppRedis;
use config::quota::QuotaConfig;
use std::collections::HashMap;
#[derive(Clone)]
pub struct Actors {
accounts: Accounts,
quotas: AppRedis,
limits: QuotaConfig,
}
impl Actors {
pub fn new(accounts: Accounts, quotas: AppRedis, limits: QuotaConfig) -> Self {
Self {
accounts,
quotas,
limits,
}
}
pub async fn briefs(&self, ids: &[i64]) -> Result<HashMap<i64, UserDto>, PullError> {
Ok(auth::users::load_user_briefs(&self.accounts, ids).await?)
}
pub async fn resolve(&self, reference: &str) -> Result<i64, PullError> {
let reference = reference.trim();
if let Ok(uid) = reference.parse::<i64>() {
return Ok(uid);
}
let user = auth::users::find_by_username(&self.accounts, reference)
.await?
.ok_or_else(|| PullError::BadRequest(format!("user `{reference}` not found")))?;
Ok(user.user_id)
}
pub async fn allow_pull(&self, user: i64) -> Result<(), PullError> {
if self
.quotas
.quota_allow(
&format!("quota:pull_create:{user}"),
self.limits.pull_create_per_day,
86_400,
)
.await
{
Ok(())
} else {
Err(PullError::RateLimited(
"pull request creation quota exceeded, retry tomorrow".to_owned(),
))
}
}
pub async fn allow_comment(&self, user: i64) -> Result<(), PullError> {
if self
.quotas
.quota_allow(
&format!("quota:comment_create:{user}"),
self.limits.comment_create_per_day,
86_400,
)
.await
{
Ok(())
} else {
Err(PullError::RateLimited(
"comment creation quota exceeded, retry tomorrow".to_owned(),
))
}
}
}