2024-01-25 14:22:11 +00:00
|
|
|
use crate::request::client_builder;
|
|
|
|
use activitypub_federation::config::{Data, FederationConfig};
|
2023-07-11 13:09:59 +00:00
|
|
|
use lemmy_db_schema::{
|
|
|
|
source::secret::Secret,
|
2024-01-25 14:22:11 +00:00
|
|
|
utils::{build_db_pool_for_tests, ActualDbPool, DbPool},
|
2023-07-11 13:09:59 +00:00
|
|
|
};
|
2022-11-28 14:29:33 +00:00
|
|
|
use lemmy_utils::{
|
|
|
|
rate_limit::RateLimitCell,
|
|
|
|
settings::{structs::Settings, SETTINGS},
|
|
|
|
};
|
2024-03-06 16:21:46 +00:00
|
|
|
use reqwest_middleware::{ClientBuilder, ClientWithMiddleware};
|
2022-12-09 15:31:47 +00:00
|
|
|
use std::sync::Arc;
|
2022-11-28 14:29:33 +00:00
|
|
|
|
2023-03-21 15:03:05 +00:00
|
|
|
#[derive(Clone)]
|
2022-11-28 14:29:33 +00:00
|
|
|
pub struct LemmyContext {
|
2023-07-11 13:09:59 +00:00
|
|
|
pool: ActualDbPool,
|
2023-03-21 15:03:05 +00:00
|
|
|
client: Arc<ClientWithMiddleware>,
|
|
|
|
secret: Arc<Secret>,
|
2022-11-28 14:29:33 +00:00
|
|
|
rate_limit_cell: RateLimitCell,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl LemmyContext {
|
|
|
|
pub fn create(
|
2023-07-11 13:09:59 +00:00
|
|
|
pool: ActualDbPool,
|
2022-11-28 14:29:33 +00:00
|
|
|
client: ClientWithMiddleware,
|
|
|
|
secret: Secret,
|
|
|
|
rate_limit_cell: RateLimitCell,
|
|
|
|
) -> LemmyContext {
|
|
|
|
LemmyContext {
|
|
|
|
pool,
|
2023-03-21 15:03:05 +00:00
|
|
|
client: Arc::new(client),
|
|
|
|
secret: Arc::new(secret),
|
2022-11-28 14:29:33 +00:00
|
|
|
rate_limit_cell,
|
|
|
|
}
|
|
|
|
}
|
2023-07-11 13:09:59 +00:00
|
|
|
pub fn pool(&self) -> DbPool<'_> {
|
|
|
|
DbPool::Pool(&self.pool)
|
|
|
|
}
|
|
|
|
pub fn inner_pool(&self) -> &ActualDbPool {
|
2022-11-28 14:29:33 +00:00
|
|
|
&self.pool
|
|
|
|
}
|
|
|
|
pub fn client(&self) -> &ClientWithMiddleware {
|
|
|
|
&self.client
|
|
|
|
}
|
|
|
|
pub fn settings(&self) -> &'static Settings {
|
|
|
|
&SETTINGS
|
|
|
|
}
|
|
|
|
pub fn secret(&self) -> &Secret {
|
|
|
|
&self.secret
|
|
|
|
}
|
2023-10-19 13:31:51 +00:00
|
|
|
pub fn rate_limit_cell(&self) -> &RateLimitCell {
|
2022-11-28 14:29:33 +00:00
|
|
|
&self.rate_limit_cell
|
|
|
|
}
|
2024-01-25 14:22:11 +00:00
|
|
|
|
2024-03-06 16:21:46 +00:00
|
|
|
/// Initialize a context for use in tests which blocks federation network calls.
|
2024-01-25 14:22:11 +00:00
|
|
|
///
|
|
|
|
/// Do not use this in production code.
|
|
|
|
pub async fn init_test_context() -> Data<LemmyContext> {
|
|
|
|
// call this to run migrations
|
|
|
|
let pool = build_db_pool_for_tests().await;
|
|
|
|
|
|
|
|
let client = client_builder(&SETTINGS).build().expect("build client");
|
|
|
|
|
2024-03-06 16:21:46 +00:00
|
|
|
let client = ClientBuilder::new(client).build();
|
2024-01-25 14:22:11 +00:00
|
|
|
let secret = Secret {
|
|
|
|
id: 0,
|
2024-05-17 00:41:57 +00:00
|
|
|
jwt_secret: String::new().into(),
|
2024-01-25 14:22:11 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
let rate_limit_cell = RateLimitCell::with_test_config();
|
|
|
|
|
|
|
|
let context = LemmyContext::create(pool, client, secret, rate_limit_cell.clone());
|
|
|
|
let config = FederationConfig::builder()
|
|
|
|
.domain(context.settings().hostname.clone())
|
|
|
|
.app_data(context)
|
2024-03-06 16:21:46 +00:00
|
|
|
// Dont allow any network fetches
|
|
|
|
.http_fetch_limit(0)
|
2024-01-25 14:22:11 +00:00
|
|
|
.build()
|
|
|
|
.await
|
|
|
|
.expect("build federation config");
|
|
|
|
config.to_request_data()
|
|
|
|
}
|
|
|
|
}
|