refactor: adapters configured from config crate
This commit is contained in:
14
Cargo.lock
generated
14
Cargo.lock
generated
@@ -11,6 +11,7 @@ dependencies = [
|
||||
"async-trait",
|
||||
"axum",
|
||||
"chrono",
|
||||
"config",
|
||||
"domain",
|
||||
"k-ap",
|
||||
"serde",
|
||||
@@ -663,6 +664,7 @@ dependencies = [
|
||||
"argon2",
|
||||
"async-trait",
|
||||
"chrono",
|
||||
"config",
|
||||
"domain",
|
||||
"jsonwebtoken",
|
||||
"rand_core 0.6.4",
|
||||
@@ -1128,6 +1130,7 @@ dependencies = [
|
||||
"async-trait",
|
||||
"auth",
|
||||
"chrono",
|
||||
"config",
|
||||
"domain",
|
||||
"infra-wiring",
|
||||
"jellyfin",
|
||||
@@ -1887,6 +1890,7 @@ name = "event-publisher"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"async-trait",
|
||||
"config",
|
||||
"domain",
|
||||
"futures",
|
||||
"tokio",
|
||||
@@ -2693,6 +2697,7 @@ version = "0.1.0"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"async-trait",
|
||||
"config",
|
||||
"domain",
|
||||
"image",
|
||||
"object-storage",
|
||||
@@ -2771,6 +2776,7 @@ name = "infra-wiring"
|
||||
version = "0.1.0"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"config",
|
||||
"sqlx",
|
||||
]
|
||||
|
||||
@@ -3246,6 +3252,7 @@ version = "0.1.0"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"async-trait",
|
||||
"config",
|
||||
"domain",
|
||||
"reqwest 0.13.3",
|
||||
"serde",
|
||||
@@ -3350,6 +3357,7 @@ dependencies = [
|
||||
"async-nats",
|
||||
"async-trait",
|
||||
"chrono",
|
||||
"config",
|
||||
"domain",
|
||||
"event-payload",
|
||||
"futures",
|
||||
@@ -3571,6 +3579,7 @@ dependencies = [
|
||||
"anyhow",
|
||||
"async-trait",
|
||||
"bytes",
|
||||
"config",
|
||||
"domain",
|
||||
"futures",
|
||||
"infer",
|
||||
@@ -3949,6 +3958,7 @@ version = "0.1.0"
|
||||
dependencies = [
|
||||
"anyhow",
|
||||
"async-trait",
|
||||
"config",
|
||||
"domain",
|
||||
"reqwest 0.13.3",
|
||||
]
|
||||
@@ -4946,6 +4956,7 @@ dependencies = [
|
||||
"axum",
|
||||
"bytes",
|
||||
"composition",
|
||||
"config",
|
||||
"domain",
|
||||
"dotenvy",
|
||||
"export",
|
||||
@@ -5208,6 +5219,7 @@ dependencies = [
|
||||
"async-trait",
|
||||
"bytes",
|
||||
"chrono",
|
||||
"config",
|
||||
"domain",
|
||||
"event-payload",
|
||||
"futures",
|
||||
@@ -5828,6 +5840,7 @@ dependencies = [
|
||||
"anyhow",
|
||||
"async-trait",
|
||||
"chrono",
|
||||
"config",
|
||||
"domain",
|
||||
"reqwest 0.13.3",
|
||||
"serde",
|
||||
@@ -7175,6 +7188,7 @@ dependencies = [
|
||||
"async-trait",
|
||||
"auth",
|
||||
"composition",
|
||||
"config",
|
||||
"domain",
|
||||
"dotenvy",
|
||||
"export",
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
k-ap = { version = "0.5.0", registry = "gitea" }
|
||||
domain = { workspace = true }
|
||||
axum = { workspace = true }
|
||||
|
||||
@@ -68,6 +68,7 @@ pub struct ActivityPubDeps {
|
||||
pub follow_query: std::sync::Arc<dyn domain::ports::FollowQuery>,
|
||||
pub instance: domain::value_objects::InstanceIdentity,
|
||||
pub allow_registration: bool,
|
||||
pub federation: config::FederationConfig,
|
||||
pub event_publisher: std::sync::Arc<dyn domain::ports::EventPublisher>,
|
||||
}
|
||||
|
||||
@@ -92,6 +93,7 @@ pub async fn wire(deps: ActivityPubDeps) -> anyhow::Result<ActivityPubWire> {
|
||||
follow_query: _,
|
||||
instance,
|
||||
allow_registration,
|
||||
federation,
|
||||
event_publisher,
|
||||
} = deps;
|
||||
let review_handler = std::sync::Arc::new(ReviewObjectHandler {
|
||||
@@ -118,9 +120,7 @@ pub async fn wire(deps: ActivityPubDeps) -> anyhow::Result<ActivityPubWire> {
|
||||
goal: goal_handler,
|
||||
});
|
||||
|
||||
let federation_debug = std::env::var("FEDERATION_DEBUG")
|
||||
.map(|v| v == "true" || v == "1")
|
||||
.unwrap_or(false);
|
||||
let federation_debug = federation.debug;
|
||||
|
||||
if federation_debug {
|
||||
tracing::warn!(
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
anyhow = { workspace = true }
|
||||
|
||||
@@ -22,19 +22,17 @@ pub struct AuthConfig {
|
||||
}
|
||||
|
||||
impl AuthConfig {
|
||||
pub fn from_env() -> anyhow::Result<Self> {
|
||||
let secret = std::env::var("JWT_SECRET")
|
||||
.map_err(|_| anyhow::anyhow!("JWT_SECRET env var is required"))?;
|
||||
pub fn from_config(config: &config::AuthConfig) -> anyhow::Result<Self> {
|
||||
let secret = config
|
||||
.jwt_secret
|
||||
.clone()
|
||||
.ok_or_else(|| anyhow::anyhow!("JWT_SECRET env var is required"))?;
|
||||
if secret.is_empty() {
|
||||
anyhow::bail!("JWT_SECRET must not be empty");
|
||||
}
|
||||
let ttl_seconds = std::env::var("JWT_TTL_SECONDS")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(900u64);
|
||||
Ok(Self {
|
||||
secret,
|
||||
ttl_seconds,
|
||||
ttl_seconds: config.jwt_ttl_seconds,
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -107,11 +105,13 @@ impl PasswordHasher for Argon2PasswordHasher {
|
||||
}
|
||||
}
|
||||
|
||||
pub fn create() -> anyhow::Result<(
|
||||
pub fn create(
|
||||
config: &config::AuthConfig,
|
||||
) -> anyhow::Result<(
|
||||
std::sync::Arc<dyn domain::ports::AuthService>,
|
||||
std::sync::Arc<dyn domain::ports::PasswordHasher>,
|
||||
)> {
|
||||
let config = AuthConfig::from_env()?;
|
||||
let config = AuthConfig::from_config(config)?;
|
||||
Ok((
|
||||
std::sync::Arc::new(JwtAuthService::new(config)),
|
||||
std::sync::Arc::new(Argon2PasswordHasher),
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
tokio = { workspace = true }
|
||||
|
||||
@@ -16,12 +16,10 @@ pub struct EventPublisherConfig {
|
||||
}
|
||||
|
||||
impl EventPublisherConfig {
|
||||
pub fn from_env() -> Self {
|
||||
let channel_buffer = std::env::var("EVENT_CHANNEL_BUFFER")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(128);
|
||||
Self { channel_buffer }
|
||||
pub fn from_config(config: &config::EventsConfig) -> Self {
|
||||
Self {
|
||||
channel_buffer: config.channel_buffer,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
anyhow = { workspace = true }
|
||||
|
||||
@@ -18,11 +18,8 @@ pub struct ConversionConfig {
|
||||
}
|
||||
|
||||
impl ConversionConfig {
|
||||
pub fn from_env() -> anyhow::Result<Option<Self>> {
|
||||
Self::from_vars(
|
||||
std::env::var("IMAGE_CONVERSION_ENABLED").ok().as_deref(),
|
||||
std::env::var("IMAGE_CONVERSION_FORMAT").ok().as_deref(),
|
||||
)
|
||||
pub fn from_config(config: &::config::ImageConversionConfig) -> anyhow::Result<Option<Self>> {
|
||||
Self::from_vars(config.enabled.then_some("true"), config.format.as_deref())
|
||||
}
|
||||
|
||||
fn from_vars(enabled: Option<&str>, format: Option<&str>) -> anyhow::Result<Option<Self>> {
|
||||
|
||||
@@ -14,12 +14,13 @@ use std::sync::Arc;
|
||||
type ConversionPair = (Arc<dyn EventHandler>, Arc<dyn PeriodicJob>);
|
||||
|
||||
pub fn build(
|
||||
config: &::config::ImageConversionConfig,
|
||||
object_storage: Arc<dyn ObjectStorage>,
|
||||
image_ref_command: Arc<dyn ImageRefCommand>,
|
||||
image_ref_query: Arc<dyn ImageRefQuery>,
|
||||
event_publisher: Arc<dyn EventPublisher>,
|
||||
) -> anyhow::Result<Option<ConversionPair>> {
|
||||
let config = match ConversionConfig::from_env()? {
|
||||
let config = match ConversionConfig::from_config(config)? {
|
||||
Some(c) => c,
|
||||
None => return Ok(None),
|
||||
};
|
||||
|
||||
@@ -43,3 +43,15 @@ fn avif_extension() {
|
||||
fn webp_extension() {
|
||||
assert_eq!(Format::Webp.extension(), ".webp");
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_config_follows_section() {
|
||||
let disabled = ::config::ImageConversionConfig::default();
|
||||
assert!(ConversionConfig::from_config(&disabled).unwrap().is_none());
|
||||
let enabled = ::config::ImageConversionConfig {
|
||||
enabled: true,
|
||||
format: Some("webp".into()),
|
||||
};
|
||||
let cfg = ConversionConfig::from_config(&enabled).unwrap().unwrap();
|
||||
assert_eq!(cfg.format, Format::Webp);
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
anyhow = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
reqwest = { workspace = true }
|
||||
|
||||
@@ -68,12 +68,16 @@ impl MetadataClient for MetadataClientImpl {
|
||||
}
|
||||
}
|
||||
|
||||
pub fn create() -> anyhow::Result<std::sync::Arc<dyn domain::ports::MetadataClient>> {
|
||||
pub fn create(
|
||||
config: &config::MetadataConfig,
|
||||
) -> anyhow::Result<std::sync::Arc<dyn domain::ports::MetadataClient>> {
|
||||
use anyhow::Context;
|
||||
if let Ok(key) = std::env::var("TMDB_API_KEY") {
|
||||
if let Some(key) = config.tmdb_api_key.clone() {
|
||||
Ok(std::sync::Arc::new(MetadataClientImpl::new_tmdb(key)))
|
||||
} else {
|
||||
let key = std::env::var("OMDB_API_KEY")
|
||||
let key = config
|
||||
.omdb_api_key
|
||||
.clone()
|
||||
.context("either TMDB_API_KEY or OMDB_API_KEY must be set")?;
|
||||
Ok(std::sync::Arc::new(MetadataClientImpl::new_omdb(key)))
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
async-nats = "0.48.0"
|
||||
|
||||
domain = { workspace = true }
|
||||
|
||||
@@ -14,13 +14,13 @@ pub struct NatsConfig {
|
||||
}
|
||||
|
||||
impl NatsConfig {
|
||||
pub fn from_env() -> anyhow::Result<Self> {
|
||||
pub fn from_config(config: &::config::NatsConfig) -> anyhow::Result<Self> {
|
||||
Self::from_vars(
|
||||
std::env::var("NATS_URL").ok().as_deref(),
|
||||
std::env::var("NATS_MODE").ok().as_deref(),
|
||||
std::env::var("NATS_SUBJECT_PREFIX").ok().as_deref(),
|
||||
std::env::var("NATS_STREAM_NAME").ok().as_deref(),
|
||||
std::env::var("NATS_CONSUMER_NAME").ok().as_deref(),
|
||||
config.url.as_deref(),
|
||||
Some(&config.mode),
|
||||
Some(&config.subject_prefix),
|
||||
Some(&config.stream_name),
|
||||
Some(&config.consumer_name),
|
||||
)
|
||||
}
|
||||
|
||||
|
||||
@@ -28,3 +28,20 @@ fn invalid_mode_errors() {
|
||||
NatsConfig::from_vars(Some("nats://test:4222"), Some("kafka"), None, None, None).is_err()
|
||||
);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_config_requires_url() {
|
||||
assert!(NatsConfig::from_config(&::config::NatsConfig::default()).is_err());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn from_config_uses_section_values() {
|
||||
let section = ::config::NatsConfig {
|
||||
url: Some("nats://test:4222".into()),
|
||||
mode: "core".into(),
|
||||
..Default::default()
|
||||
};
|
||||
let cfg = NatsConfig::from_config(§ion).unwrap();
|
||||
assert_eq!(cfg.url, "nats://test:4222");
|
||||
assert_eq!(cfg.mode, NatsMode::Core);
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
anyhow = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
|
||||
@@ -5,21 +5,25 @@ use std::sync::Arc;
|
||||
pub struct StorageConfig(Arc<dyn ObjectStore>);
|
||||
|
||||
impl StorageConfig {
|
||||
pub fn from_env() -> anyhow::Result<Self> {
|
||||
let backend = std::env::var("IMAGE_STORAGE_BACKEND").unwrap_or_else(|_| "local".into());
|
||||
|
||||
let store: Arc<dyn ObjectStore> = match backend.as_str() {
|
||||
pub fn from_config(config: &::config::StorageConfig) -> anyhow::Result<Self> {
|
||||
let store: Arc<dyn ObjectStore> = match config.backend.as_str() {
|
||||
"s3" => build_s3_store(
|
||||
&std::env::var("MINIO_ENDPOINT").context("MINIO_ENDPOINT required")?,
|
||||
&std::env::var("MINIO_ACCESS_KEY_ID").context("MINIO_ACCESS_KEY_ID required")?,
|
||||
&std::env::var("MINIO_SECRET_ACCESS_KEY")
|
||||
config
|
||||
.endpoint
|
||||
.as_deref()
|
||||
.context("MINIO_ENDPOINT required")?,
|
||||
config
|
||||
.access_key_id
|
||||
.as_deref()
|
||||
.context("MINIO_ACCESS_KEY_ID required")?,
|
||||
config
|
||||
.secret_access_key
|
||||
.as_deref()
|
||||
.context("MINIO_SECRET_ACCESS_KEY required")?,
|
||||
&std::env::var("MINIO_BUCKET").context("MINIO_BUCKET required")?,
|
||||
&std::env::var("MINIO_REGION").unwrap_or_else(|_| "minio".to_string()),
|
||||
)?,
|
||||
"local" => build_local_store(
|
||||
&std::env::var("IMAGE_STORAGE_PATH").unwrap_or_else(|_| "./images".into()),
|
||||
config.bucket.as_deref().context("MINIO_BUCKET required")?,
|
||||
&config.region,
|
||||
)?,
|
||||
"local" => build_local_store(&config.path)?,
|
||||
other => {
|
||||
anyhow::bail!("Unknown IMAGE_STORAGE_BACKEND: {other:?}. Valid values: s3, local")
|
||||
}
|
||||
|
||||
@@ -127,9 +127,9 @@ impl EventHandler for ImageCleanupHandler {
|
||||
}
|
||||
}
|
||||
|
||||
pub fn create() -> anyhow::Result<Arc<dyn ObjectStorage>> {
|
||||
pub fn create(config: &::config::StorageConfig) -> anyhow::Result<Arc<dyn ObjectStorage>> {
|
||||
Ok(Arc::new(ObjectStorageAdapter::from_config(
|
||||
StorageConfig::from_env()?,
|
||||
StorageConfig::from_config(config)?,
|
||||
)))
|
||||
}
|
||||
|
||||
|
||||
@@ -15,3 +15,22 @@ fn local_store_succeeds_if_dir_already_exists() {
|
||||
let result = build_local_store(dir.to_str().unwrap());
|
||||
assert!(result.is_ok());
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn s3_backend_requires_endpoint() {
|
||||
let section = ::config::StorageConfig {
|
||||
backend: "s3".into(),
|
||||
..Default::default()
|
||||
};
|
||||
let err = StorageConfig::from_config(§ion).err().unwrap();
|
||||
assert!(err.to_string().contains("MINIO_ENDPOINT required"));
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn unknown_backend_is_error() {
|
||||
let section = ::config::StorageConfig {
|
||||
backend: "ftp".into(),
|
||||
..Default::default()
|
||||
};
|
||||
assert!(StorageConfig::from_config(§ion).is_err());
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
async-trait = { workspace = true }
|
||||
reqwest = { workspace = true }
|
||||
|
||||
@@ -3,11 +3,9 @@ pub struct PosterFetcherConfig {
|
||||
}
|
||||
|
||||
impl PosterFetcherConfig {
|
||||
pub fn from_env() -> Self {
|
||||
let timeout_seconds = std::env::var("POSTER_FETCH_TIMEOUT_SECONDS")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(30);
|
||||
Self { timeout_seconds }
|
||||
pub fn from_config(config: &::config::MetadataConfig) -> Self {
|
||||
Self {
|
||||
timeout_seconds: config.poster_fetch_timeout_seconds,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -59,14 +59,18 @@ impl ImageFetcher for ReqwestPosterFetcher {
|
||||
}
|
||||
}
|
||||
|
||||
pub fn create() -> anyhow::Result<std::sync::Arc<dyn domain::ports::PosterFetcherClient>> {
|
||||
pub fn create(
|
||||
config: &::config::MetadataConfig,
|
||||
) -> anyhow::Result<std::sync::Arc<dyn domain::ports::PosterFetcherClient>> {
|
||||
Ok(std::sync::Arc::new(ReqwestPosterFetcher::new(
|
||||
PosterFetcherConfig::from_env(),
|
||||
PosterFetcherConfig::from_config(config),
|
||||
)?))
|
||||
}
|
||||
|
||||
pub fn create_image_fetcher() -> anyhow::Result<std::sync::Arc<dyn domain::ports::ImageFetcher>> {
|
||||
pub fn create_image_fetcher(
|
||||
config: &::config::MetadataConfig,
|
||||
) -> anyhow::Result<std::sync::Arc<dyn domain::ports::ImageFetcher>> {
|
||||
Ok(std::sync::Arc::new(ReqwestPosterFetcher::new(
|
||||
PosterFetcherConfig::from_env(),
|
||||
PosterFetcherConfig::from_config(config),
|
||||
)?))
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
sqlx = { workspace = true, features = ["macros", "chrono"] }
|
||||
|
||||
domain = { workspace = true }
|
||||
|
||||
@@ -22,20 +22,11 @@ pub struct DbEventQueueConfig {
|
||||
}
|
||||
|
||||
impl DbEventQueueConfig {
|
||||
pub fn from_env() -> Self {
|
||||
pub fn from_config(config: &config::EventsConfig) -> Self {
|
||||
Self {
|
||||
poll_interval_ms: std::env::var("EVENT_QUEUE_POLL_INTERVAL_MS")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(500),
|
||||
batch_size: std::env::var("EVENT_QUEUE_BATCH_SIZE")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(10),
|
||||
max_attempts: std::env::var("EVENT_QUEUE_MAX_ATTEMPTS")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(5),
|
||||
poll_interval_ms: config.queue_poll_interval_ms,
|
||||
batch_size: config.queue_batch_size,
|
||||
max_attempts: config.queue_max_attempts,
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -54,15 +45,19 @@ impl SqliteEventQueue {
|
||||
})
|
||||
}
|
||||
|
||||
pub async fn create_publisher(pool: SqlitePool) -> anyhow::Result<Arc<dyn EventPublisher>> {
|
||||
let q = Self::create(pool, DbEventQueueConfig::from_env()).await?;
|
||||
pub async fn create_publisher(
|
||||
pool: SqlitePool,
|
||||
config: &config::EventsConfig,
|
||||
) -> anyhow::Result<Arc<dyn EventPublisher>> {
|
||||
let q = Self::create(pool, DbEventQueueConfig::from_config(config)).await?;
|
||||
Ok(Arc::new(q))
|
||||
}
|
||||
|
||||
pub async fn create_channel(
|
||||
pool: SqlitePool,
|
||||
config: &config::EventsConfig,
|
||||
) -> anyhow::Result<(Arc<dyn EventPublisher>, Arc<dyn EventConsumer>)> {
|
||||
let q = Self::create(pool, DbEventQueueConfig::from_env()).await?;
|
||||
let q = Self::create(pool, DbEventQueueConfig::from_config(config)).await?;
|
||||
Ok((Arc::new(q.clone()), Arc::new(q)))
|
||||
}
|
||||
}
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition.workspace = true
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
reqwest = { workspace = true }
|
||||
serde = { workspace = true }
|
||||
|
||||
@@ -7,9 +7,11 @@ pub struct TmdbEnrichmentClient {
|
||||
}
|
||||
|
||||
impl TmdbEnrichmentClient {
|
||||
pub fn from_env() -> Result<Self, DomainError> {
|
||||
let api_key = std::env::var("TMDB_API_KEY")
|
||||
.map_err(|_| DomainError::InfrastructureError("TMDB_API_KEY is not set".into()))?;
|
||||
pub fn from_config(config: &config::MetadataConfig) -> Result<Self, DomainError> {
|
||||
let api_key = config
|
||||
.tmdb_api_key
|
||||
.clone()
|
||||
.ok_or_else(|| DomainError::InfrastructureError("TMDB_API_KEY is not set".into()))?;
|
||||
Ok(Self {
|
||||
api_key,
|
||||
http: reqwest::Client::new(),
|
||||
|
||||
@@ -129,6 +129,8 @@ async fn following_filter_separates_local_and_remote() {
|
||||
base_url: "http://localhost:3000".into(),
|
||||
rate_limit: 20,
|
||||
refresh_ttl_seconds: 2_592_000,
|
||||
secure_cookies: false,
|
||||
cors_origins: String::new(),
|
||||
wrapup: crate::config::WrapUpConfig {
|
||||
font_path: None,
|
||||
logo_path: None,
|
||||
|
||||
@@ -136,6 +136,8 @@ impl TestContextBuilder {
|
||||
base_url: "http://localhost:3000".into(),
|
||||
rate_limit: 20,
|
||||
refresh_ttl_seconds: 2_592_000,
|
||||
secure_cookies: false,
|
||||
cors_origins: String::new(),
|
||||
wrapup: crate::config::WrapUpConfig {
|
||||
font_path: None,
|
||||
logo_path: None,
|
||||
|
||||
@@ -9,6 +9,7 @@ nats = ["dep:nats", "infra-wiring/nats"]
|
||||
federation = ["application/federation", "dep:sqlite-federation", "dep:activitypub"]
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
application = { workspace = true }
|
||||
infra-wiring = { workspace = true }
|
||||
|
||||
@@ -103,20 +103,28 @@ pub async fn build_database_adapters(
|
||||
})
|
||||
}
|
||||
|
||||
pub fn build_auth_adapters() -> anyhow::Result<(Arc<dyn AuthService>, Arc<dyn PasswordHasher>)> {
|
||||
auth::create()
|
||||
pub fn build_auth_adapters(
|
||||
config: &config::AuthConfig,
|
||||
) -> anyhow::Result<(Arc<dyn AuthService>, Arc<dyn PasswordHasher>)> {
|
||||
auth::create(config)
|
||||
}
|
||||
|
||||
pub fn build_metadata_client() -> anyhow::Result<Arc<dyn MetadataClient>> {
|
||||
metadata::create()
|
||||
pub fn build_metadata_client(
|
||||
config: &config::MetadataConfig,
|
||||
) -> anyhow::Result<Arc<dyn MetadataClient>> {
|
||||
metadata::create(config)
|
||||
}
|
||||
|
||||
pub fn build_poster_fetcher() -> anyhow::Result<Arc<dyn PosterFetcherClient>> {
|
||||
poster_fetcher::create()
|
||||
pub fn build_poster_fetcher(
|
||||
config: &config::MetadataConfig,
|
||||
) -> anyhow::Result<Arc<dyn PosterFetcherClient>> {
|
||||
poster_fetcher::create(config)
|
||||
}
|
||||
|
||||
pub fn build_object_storage() -> anyhow::Result<Arc<dyn ObjectStorage>> {
|
||||
object_storage::create()
|
||||
pub fn build_object_storage(
|
||||
config: &config::StorageConfig,
|
||||
) -> anyhow::Result<Arc<dyn ObjectStorage>> {
|
||||
object_storage::create(config)
|
||||
}
|
||||
|
||||
pub fn build_profile_fields_repo(
|
||||
|
||||
@@ -346,6 +346,8 @@ fn test_config() -> AppConfig {
|
||||
base_url: "http://localhost:3000".into(),
|
||||
rate_limit: 20,
|
||||
refresh_ttl_seconds: 2_592_000,
|
||||
secure_cookies: false,
|
||||
cors_origins: String::new(),
|
||||
wrapup: infra_wiring::WrapUpConfig {
|
||||
font_path: None,
|
||||
logo_path: None,
|
||||
|
||||
@@ -4,6 +4,7 @@ version = "0.1.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
sqlx = { workspace = true }
|
||||
anyhow = { workspace = true }
|
||||
|
||||
|
||||
@@ -4,6 +4,8 @@ pub struct AppConfig {
|
||||
pub base_url: String,
|
||||
pub rate_limit: u64,
|
||||
pub refresh_ttl_seconds: u64,
|
||||
pub secure_cookies: bool,
|
||||
pub cors_origins: String,
|
||||
pub wrapup: WrapUpConfig,
|
||||
}
|
||||
|
||||
@@ -14,39 +16,26 @@ pub struct WrapUpConfig {
|
||||
pub bg_dir: Option<String>,
|
||||
}
|
||||
|
||||
impl AppConfig {
|
||||
pub fn from_env() -> Self {
|
||||
let allow_registration = std::env::var("ALLOW_REGISTRATION")
|
||||
.map(|v| v == "true" || v == "1")
|
||||
.unwrap_or(false);
|
||||
let base_url = std::env::var("BASE_URL")
|
||||
.unwrap_or_else(|_| "http://localhost:3000".to_string())
|
||||
.trim_end_matches('/')
|
||||
.to_string();
|
||||
let rate_limit = std::env::var("RATE_LIMIT")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(60);
|
||||
let refresh_ttl_seconds = std::env::var("REFRESH_TTL_SECONDS")
|
||||
.ok()
|
||||
.and_then(|v| v.parse().ok())
|
||||
.unwrap_or(2_592_000u64);
|
||||
impl From<&::config::AppConfig> for AppConfig {
|
||||
fn from(config: &::config::AppConfig) -> Self {
|
||||
Self {
|
||||
allow_registration,
|
||||
base_url,
|
||||
rate_limit,
|
||||
refresh_ttl_seconds,
|
||||
wrapup: WrapUpConfig::from_env(),
|
||||
allow_registration: config.server.allow_registration,
|
||||
base_url: config.server.base_url.clone(),
|
||||
rate_limit: config.server.rate_limit,
|
||||
refresh_ttl_seconds: config.auth.refresh_ttl_seconds,
|
||||
secure_cookies: config.server.secure_cookies,
|
||||
cors_origins: config.server.cors_origins.clone(),
|
||||
wrapup: WrapUpConfig::from(&config.wrapup),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl WrapUpConfig {
|
||||
pub fn from_env() -> Self {
|
||||
impl From<&::config::WrapUpConfig> for WrapUpConfig {
|
||||
fn from(config: &::config::WrapUpConfig) -> Self {
|
||||
Self {
|
||||
font_path: std::env::var("WRAPUP_FONT_PATH").ok(),
|
||||
logo_path: std::env::var("WRAPUP_LOGO_PATH").ok(),
|
||||
bg_dir: std::env::var("WRAPUP_BG_DIR").ok(),
|
||||
font_path: config.font_path.clone(),
|
||||
logo_path: config.logo_path.clone(),
|
||||
bg_dir: config.bg_dir.clone(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -14,11 +14,8 @@ pub enum EventBusBackend {
|
||||
}
|
||||
|
||||
impl EventBusBackend {
|
||||
pub fn from_env() -> anyhow::Result<Self> {
|
||||
match std::env::var("EVENT_BUS_BACKEND")
|
||||
.unwrap_or_else(|_| "db".to_string())
|
||||
.as_str()
|
||||
{
|
||||
pub fn from_config(config: &::config::EventsConfig) -> anyhow::Result<Self> {
|
||||
match config.bus_backend.as_str() {
|
||||
"db" => Ok(Self::Db),
|
||||
#[cfg(feature = "nats")]
|
||||
"nats" => Ok(Self::Nats),
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
use axum::{
|
||||
extract::Request,
|
||||
extract::{Request, State},
|
||||
http::{HeaderValue, header},
|
||||
middleware::Next,
|
||||
response::Response,
|
||||
@@ -19,15 +19,19 @@ pub fn extract_from_cookie(headers: &axum::http::HeaderMap) -> Option<String> {
|
||||
})
|
||||
}
|
||||
|
||||
pub(crate) fn secure_flag() -> &'static str {
|
||||
if std::env::var("SECURE_COOKIES").as_deref() == Ok("true") {
|
||||
"; Secure"
|
||||
} else {
|
||||
""
|
||||
}
|
||||
/// Router state for [`csrf_middleware`]: whether cookies carry the `Secure` attribute.
|
||||
#[derive(Clone, Copy)]
|
||||
pub struct SecureCookies(pub bool);
|
||||
|
||||
pub(crate) fn secure_flag(secure_cookies: bool) -> &'static str {
|
||||
if secure_cookies { "; Secure" } else { "" }
|
||||
}
|
||||
|
||||
pub async fn csrf_middleware(mut req: Request, next: Next) -> Response {
|
||||
pub async fn csrf_middleware(
|
||||
State(SecureCookies(secure_cookies)): State<SecureCookies>,
|
||||
mut req: Request,
|
||||
next: Next,
|
||||
) -> Response {
|
||||
let existing = extract_from_cookie(req.headers());
|
||||
let (token, needs_set) = match existing {
|
||||
Some(t) => (t, false),
|
||||
@@ -42,7 +46,7 @@ pub async fn csrf_middleware(mut req: Request, next: Next) -> Response {
|
||||
let cookie = format!(
|
||||
"csrf={}; HttpOnly; Path=/; SameSite=Strict{}",
|
||||
token,
|
||||
secure_flag()
|
||||
secure_flag(secure_cookies)
|
||||
);
|
||||
if let Ok(val) = HeaderValue::from_str(&cookie) {
|
||||
response.headers_mut().append(header::SET_COOKIE, val);
|
||||
|
||||
@@ -25,12 +25,16 @@ use template_askama::{LoginTemplate, RegisterTemplate};
|
||||
|
||||
// ── HTML helpers ─────────────────────────────────────────────────────────────
|
||||
|
||||
fn set_cookie_header(token: &str, max_age: i64) -> (axum::http::HeaderName, HeaderValue) {
|
||||
fn set_cookie_header(
|
||||
token: &str,
|
||||
max_age: i64,
|
||||
secure_cookies: bool,
|
||||
) -> (axum::http::HeaderName, HeaderValue) {
|
||||
let val = format!(
|
||||
"token={}; HttpOnly; Path=/; SameSite=Strict; Max-Age={}{}",
|
||||
token,
|
||||
max_age,
|
||||
secure_flag()
|
||||
secure_flag(secure_cookies)
|
||||
);
|
||||
(
|
||||
SET_COOKIE,
|
||||
@@ -177,17 +181,18 @@ pub async fn post_login(
|
||||
{
|
||||
Ok(result) => {
|
||||
let max_age = (result.expires_at - Utc::now()).num_seconds().max(0);
|
||||
let cookie = set_cookie_header(&result.token, max_age);
|
||||
let cookie =
|
||||
set_cookie_header(&result.token, max_age, state.app_ctx.config.secure_cookies);
|
||||
([cookie], Redirect::to("/")).into_response()
|
||||
}
|
||||
Err(_) => Redirect::to("/login?error=Invalid+credentials").into_response(),
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn get_logout() -> impl IntoResponse {
|
||||
pub async fn get_logout(State(state): State<AppState>) -> impl IntoResponse {
|
||||
let val = format!(
|
||||
"token=; HttpOnly; Path=/; SameSite=Strict; Max-Age=0{}",
|
||||
secure_flag()
|
||||
secure_flag(state.app_ctx.config.secure_cookies)
|
||||
);
|
||||
let cookie = (
|
||||
SET_COOKIE,
|
||||
@@ -246,7 +251,8 @@ pub async fn post_register(
|
||||
{
|
||||
Ok(result) => {
|
||||
let max_age = (result.expires_at - Utc::now()).num_seconds().max(0);
|
||||
let cookie = set_cookie_header(&result.token, max_age);
|
||||
let cookie =
|
||||
set_cookie_header(&result.token, max_age, state.app_ctx.config.secure_cookies);
|
||||
([cookie], Redirect::to("/")).into_response()
|
||||
}
|
||||
Err(_) => {
|
||||
|
||||
@@ -12,6 +12,8 @@ use crate::{handlers, state::AppState};
|
||||
|
||||
pub fn build_router(state: AppState, ap_router: Router) -> Router {
|
||||
let rate_limit = state.app_ctx.config.rate_limit;
|
||||
let secure_cookies = state.app_ctx.config.secure_cookies;
|
||||
let cors_origins = state.app_ctx.config.cors_origins.clone();
|
||||
|
||||
let ap_cfg = GovernorConfigBuilder::default()
|
||||
.with_extractor(PeerIp::default())
|
||||
@@ -23,8 +25,8 @@ pub fn build_router(state: AppState, ap_router: Router) -> Router {
|
||||
|
||||
Router::new()
|
||||
.route("/health", routing::get(health_handler))
|
||||
.merge(html_routes(rate_limit))
|
||||
.merge(api_routes(rate_limit))
|
||||
.merge(html_routes(rate_limit, secure_cookies))
|
||||
.merge(api_routes(rate_limit, &cors_origins))
|
||||
.nest_service("/static", ServeDir::new("static"))
|
||||
.nest_service(
|
||||
"/app",
|
||||
@@ -44,7 +46,7 @@ fn per_minute(n: u64) -> Quota {
|
||||
Quota::requests_per_minute(n)
|
||||
}
|
||||
|
||||
fn html_routes(rate_limit: u64) -> Router<AppState> {
|
||||
fn html_routes(rate_limit: u64, secure_cookies: bool) -> Router<AppState> {
|
||||
let auth = Router::new()
|
||||
.route(
|
||||
"/login",
|
||||
@@ -188,7 +190,10 @@ fn html_routes(rate_limit: u64) -> Router<AppState> {
|
||||
#[cfg(feature = "federation")]
|
||||
let base = base.merge(federation_html_routes());
|
||||
|
||||
base.layer(axum::middleware::from_fn(crate::csrf::csrf_middleware))
|
||||
base.layer(axum::middleware::from_fn_with_state(
|
||||
crate::csrf::SecureCookies(secure_cookies),
|
||||
crate::csrf::csrf_middleware,
|
||||
))
|
||||
}
|
||||
|
||||
fn social_html_routes() -> Router<AppState> {
|
||||
@@ -278,11 +283,10 @@ fn federation_html_routes() -> Router<AppState> {
|
||||
)
|
||||
}
|
||||
|
||||
fn cors_layer() -> CorsLayer {
|
||||
fn cors_layer(origins: &str) -> CorsLayer {
|
||||
use axum::http::{HeaderName, Method};
|
||||
use tower_http::cors::AllowOrigin;
|
||||
|
||||
let origins = std::env::var("CORS_ORIGINS").unwrap_or_default();
|
||||
let layer = CorsLayer::new()
|
||||
.allow_methods([
|
||||
Method::GET,
|
||||
@@ -319,7 +323,7 @@ fn cors_layer() -> CorsLayer {
|
||||
}
|
||||
}
|
||||
|
||||
fn api_routes(rate_limit: u64) -> Router<AppState> {
|
||||
fn api_routes(rate_limit: u64, cors_origins: &str) -> Router<AppState> {
|
||||
let cfg = GovernorConfigBuilder::default()
|
||||
.with_extractor(PeerIp::default())
|
||||
.expect_connect_info()
|
||||
@@ -506,7 +510,7 @@ fn api_routes(rate_limit: u64) -> Router<AppState> {
|
||||
Router::new()
|
||||
.nest("/api/v1", base.layer(GovernorLayer::new(cfg)))
|
||||
.nest("/api/v1", webhook_routes)
|
||||
.layer(cors_layer())
|
||||
.layer(cors_layer(cors_origins))
|
||||
}
|
||||
|
||||
fn social_api_routes() -> Router<AppState> {
|
||||
|
||||
@@ -843,6 +843,8 @@ pub(crate) fn build_test_state_from(
|
||||
base_url: "http://localhost:3000".to_string(),
|
||||
rate_limit: 20,
|
||||
refresh_ttl_seconds: 2_592_000,
|
||||
secure_cookies: false,
|
||||
cors_origins: String::new(),
|
||||
wrapup: application::config::WrapUpConfig {
|
||||
font_path: None,
|
||||
logo_path: None,
|
||||
|
||||
@@ -20,6 +20,7 @@ sqlite = []
|
||||
sqlite-federation = ["federation"]
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
axum = { workspace = true }
|
||||
tokio = { workspace = true }
|
||||
anyhow = { workspace = true }
|
||||
|
||||
@@ -5,7 +5,6 @@ use anyhow::Context;
|
||||
use tokio::net::TcpListener;
|
||||
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
||||
|
||||
use application::config::AppConfig;
|
||||
use composition::Repositories;
|
||||
use export::ExportAdapter;
|
||||
use importer::ImporterDocumentParser;
|
||||
@@ -21,15 +20,15 @@ async fn main() -> anyhow::Result<()> {
|
||||
dotenvy::dotenv().ok();
|
||||
init_tracing();
|
||||
|
||||
let (state, ap_router) = wire_dependencies()
|
||||
let config = config::load().context("Failed to load configuration")?;
|
||||
|
||||
let (state, ap_router) = wire_dependencies(&config)
|
||||
.await
|
||||
.context("Failed to wire dependencies")?;
|
||||
|
||||
let app = openapi::serve(routes::build_router(state, ap_router));
|
||||
|
||||
let host = std::env::var("HOST").unwrap_or_else(|_| "0.0.0.0".to_string());
|
||||
let port = std::env::var("PORT").unwrap_or_else(|_| "3000".to_string());
|
||||
let addr = format!("{}:{}", host, port);
|
||||
let addr = format!("{}:{}", config.server.host, config.server.port);
|
||||
let listener = TcpListener::bind(&addr).await?;
|
||||
tracing::info!("Listening on {}", addr);
|
||||
axum::serve(
|
||||
@@ -41,24 +40,24 @@ async fn main() -> anyhow::Result<()> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
|
||||
let app_config = AppConfig::from_env();
|
||||
async fn wire_dependencies(config: &config::AppConfig) -> anyhow::Result<(AppState, axum::Router)> {
|
||||
let app_config = application::config::AppConfig::from(config);
|
||||
let instance = domain::value_objects::InstanceIdentity::new(app_config.base_url.clone());
|
||||
let database_url = std::env::var("DATABASE_URL").context("DATABASE_URL must be set")?;
|
||||
let database_url = config.database.url.as_str();
|
||||
composition::factory::warn_if_database_backend_set();
|
||||
|
||||
let (auth_service, password_hasher) = composition::factory::build_auth_adapters()?;
|
||||
let metadata_client = composition::factory::build_metadata_client()?;
|
||||
let poster_fetcher = composition::factory::build_poster_fetcher()?;
|
||||
let object_storage = composition::factory::build_object_storage()?;
|
||||
let (auth_service, password_hasher) = composition::factory::build_auth_adapters(&config.auth)?;
|
||||
let metadata_client = composition::factory::build_metadata_client(&config.metadata)?;
|
||||
let poster_fetcher = composition::factory::build_poster_fetcher(&config.metadata)?;
|
||||
let object_storage = composition::factory::build_object_storage(&config.storage)?;
|
||||
|
||||
let db = composition::factory::build_database_adapters(&database_url, &instance).await?;
|
||||
let db = composition::factory::build_database_adapters(database_url, &instance).await?;
|
||||
#[cfg(feature = "federation")]
|
||||
let ap_content_repo = db.ap_content;
|
||||
let db_pool = db.db_pool;
|
||||
|
||||
// Wire up event channel, federation service, and ap_router
|
||||
let event_bus = EventBusBackend::from_env()?;
|
||||
let event_bus = EventBusBackend::from_config(&config.events)?;
|
||||
|
||||
#[cfg(feature = "federation")]
|
||||
let (
|
||||
@@ -78,7 +77,7 @@ async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
|
||||
}
|
||||
};
|
||||
|
||||
let ep = create_event_publisher(event_bus, &db_pool).await?;
|
||||
let ep = create_event_publisher(event_bus, &db_pool, config).await?;
|
||||
|
||||
let ap = activitypub::wire(activitypub::ActivityPubDeps {
|
||||
activity_repo: fed_repos.activity,
|
||||
@@ -100,6 +99,7 @@ async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
|
||||
follow_query: Arc::clone(&fed_repos.follow_query),
|
||||
instance: instance.clone(),
|
||||
allow_registration: app_config.allow_registration,
|
||||
federation: config.federation.clone(),
|
||||
event_publisher: Arc::clone(&ep),
|
||||
})
|
||||
.await?;
|
||||
@@ -136,7 +136,7 @@ async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
|
||||
};
|
||||
|
||||
#[cfg(not(feature = "federation"))]
|
||||
let event_publisher_arc = create_event_publisher(event_bus, &db_pool).await?;
|
||||
let event_publisher_arc = create_event_publisher(event_bus, &db_pool, config).await?;
|
||||
#[cfg(not(feature = "federation"))]
|
||||
let ap_router = axum::Router::new();
|
||||
#[cfg(not(feature = "federation"))]
|
||||
@@ -252,9 +252,7 @@ async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
|
||||
|
||||
let state = AppState {
|
||||
app_ctx,
|
||||
rss_renderer: Arc::new(RssAdapter::new(
|
||||
std::env::var("BASE_URL").unwrap_or_else(|_| "http://localhost:3000".into()),
|
||||
)),
|
||||
rss_renderer: Arc::new(RssAdapter::new(config.server.base_url.clone())),
|
||||
};
|
||||
Ok((state, ap_router))
|
||||
}
|
||||
@@ -262,19 +260,20 @@ async fn wire_dependencies() -> anyhow::Result<(AppState, axum::Router)> {
|
||||
async fn create_event_publisher(
|
||||
event_bus: EventBusBackend,
|
||||
db_pool: &composition::DbPool,
|
||||
config: &config::AppConfig,
|
||||
) -> anyhow::Result<Arc<dyn EventPublisher>> {
|
||||
match event_bus {
|
||||
EventBusBackend::Db => {
|
||||
tracing::info!("event bus: DB queue");
|
||||
Ok(match db_pool {
|
||||
composition::DbPool::Sqlite(pool) => {
|
||||
sqlite::SqliteEventQueue::create_publisher(pool.clone()).await?
|
||||
sqlite::SqliteEventQueue::create_publisher(pool.clone(), &config.events).await?
|
||||
}
|
||||
})
|
||||
}
|
||||
#[cfg(feature = "nats")]
|
||||
EventBusBackend::Nats => {
|
||||
let cfg = nats::NatsConfig::from_env()
|
||||
let cfg = nats::NatsConfig::from_config(&config.nats)
|
||||
.context("EVENT_BUS_BACKEND=nats requires NATS_URL to be set")?;
|
||||
tracing::info!("event bus: NATS ({})", cfg.url);
|
||||
Ok(nats::create_publisher(cfg).await?)
|
||||
|
||||
@@ -450,6 +450,8 @@ async fn test_app_with(auth: impl AuthService + 'static) -> (Router, SqlitePool)
|
||||
base_url: "http://localhost:3000".to_string(),
|
||||
rate_limit: 20,
|
||||
refresh_ttl_seconds: 2_592_000,
|
||||
secure_cookies: false,
|
||||
cors_origins: String::new(),
|
||||
wrapup: application::config::WrapUpConfig {
|
||||
font_path: None,
|
||||
logo_path: None,
|
||||
|
||||
@@ -17,6 +17,7 @@ sqlite = []
|
||||
sqlite-federation = ["federation"]
|
||||
|
||||
[dependencies]
|
||||
config = { workspace = true }
|
||||
domain = { workspace = true }
|
||||
application = { workspace = true }
|
||||
composition = { workspace = true }
|
||||
|
||||
@@ -8,19 +8,22 @@ use infra_wiring::{DbPool, EventBusBackend};
|
||||
|
||||
pub async fn create(
|
||||
db_pool: &DbPool,
|
||||
config: &config::AppConfig,
|
||||
) -> anyhow::Result<(Arc<dyn EventPublisher>, Arc<dyn EventConsumer>)> {
|
||||
match EventBusBackend::from_env()? {
|
||||
match EventBusBackend::from_config(&config.events)? {
|
||||
EventBusBackend::Db => {
|
||||
tracing::info!("event bus: DB queue");
|
||||
match db_pool {
|
||||
DbPool::Sqlite(pool) => {
|
||||
Ok(sqlite::SqliteEventQueue::create_channel(pool.clone()).await?)
|
||||
}
|
||||
DbPool::Sqlite(pool) => Ok(sqlite::SqliteEventQueue::create_channel(
|
||||
pool.clone(),
|
||||
&config.events,
|
||||
)
|
||||
.await?),
|
||||
}
|
||||
}
|
||||
#[cfg(feature = "nats")]
|
||||
EventBusBackend::Nats => {
|
||||
let cfg = nats::NatsConfig::from_env()
|
||||
let cfg = nats::NatsConfig::from_config(&config.nats)
|
||||
.context("EVENT_BUS_BACKEND=nats requires NATS_URL to be set")?;
|
||||
tracing::info!("event bus: NATS ({})", cfg.url);
|
||||
Ok(nats::create_channel(cfg).await?)
|
||||
|
||||
@@ -6,8 +6,7 @@ use std::sync::Arc;
|
||||
|
||||
use anyhow::Context;
|
||||
use application::{
|
||||
MovieDiscoveryIndexer, SearchCleanupHandler, SearchReindexHandler, config::AppConfig,
|
||||
worker::WorkerService,
|
||||
MovieDiscoveryIndexer, SearchCleanupHandler, SearchReindexHandler, worker::WorkerService,
|
||||
};
|
||||
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt};
|
||||
|
||||
@@ -18,18 +17,18 @@ async fn main() -> anyhow::Result<()> {
|
||||
dotenvy::dotenv().ok();
|
||||
init_tracing();
|
||||
|
||||
let database_url = std::env::var("DATABASE_URL").context("DATABASE_URL must be set")?;
|
||||
let config = config::load().context("Failed to load configuration")?;
|
||||
composition::factory::warn_if_database_backend_set();
|
||||
let app_config = AppConfig::from_env();
|
||||
let app_config = application::config::AppConfig::from(&config);
|
||||
let instance = domain::value_objects::InstanceIdentity::new(app_config.base_url.clone());
|
||||
|
||||
let metadata_client = metadata::create()?;
|
||||
let poster_fetcher = poster_fetcher::create()?;
|
||||
let object_storage = object_storage::create()?;
|
||||
let metadata_client = metadata::create(&config.metadata)?;
|
||||
let poster_fetcher = poster_fetcher::create(&config.metadata)?;
|
||||
let object_storage = object_storage::create(&config.storage)?;
|
||||
|
||||
// TMDb client detection happens before `build_worker_deps` so the optional
|
||||
// person-enrichment port can be handed to `WorkerServices` up front.
|
||||
let tmdb_client = match tmdb_enrichment::TmdbEnrichmentClient::from_env() {
|
||||
let tmdb_client = match tmdb_enrichment::TmdbEnrichmentClient::from_config(&config.metadata) {
|
||||
Ok(client) => {
|
||||
tracing::info!("TMDb enrichment enabled");
|
||||
Some(Arc::new(client))
|
||||
@@ -43,8 +42,8 @@ async fn main() -> anyhow::Result<()> {
|
||||
.clone()
|
||||
.map(|c| c as Arc<dyn PersonEnrichmentClient>);
|
||||
|
||||
let db = composition::factory::build_database_adapters(&database_url, &instance).await?;
|
||||
let (event_publisher_arc, consumer_arc) = event_bus::create(&db.db_pool).await?;
|
||||
let db = composition::factory::build_database_adapters(&config.database.url, &instance).await?;
|
||||
let (event_publisher_arc, consumer_arc) = event_bus::create(&db.db_pool, &config).await?;
|
||||
|
||||
let worker_services = application::WorkerServices {
|
||||
object_storage: Arc::clone(&object_storage),
|
||||
@@ -113,7 +112,7 @@ async fn main() -> anyhow::Result<()> {
|
||||
let (enrichment_handler, person_enrichment_handler, enrichment_job): EnrichmentParts =
|
||||
match tmdb_client {
|
||||
Some(client) => {
|
||||
let image_fetcher = poster_fetcher::create_image_fetcher()?;
|
||||
let image_fetcher = poster_fetcher::create_image_fetcher(&config.metadata)?;
|
||||
let handler = Arc::new(application::movies::MovieEnrichmentHandler::new(
|
||||
Arc::clone(&client) as Arc<dyn MovieEnrichmentClient>,
|
||||
worker_deps.enrich_movie.movie_query.clone(),
|
||||
@@ -140,6 +139,7 @@ async fn main() -> anyhow::Result<()> {
|
||||
// ── Image conversion ──────────────────────────────────────────────────────
|
||||
|
||||
let conversion = image_converter::build(
|
||||
&config.image_conversion,
|
||||
Arc::clone(&object_storage),
|
||||
image_ref_command,
|
||||
image_ref_query,
|
||||
@@ -257,6 +257,7 @@ async fn main() -> anyhow::Result<()> {
|
||||
follow_query: fed_repos.follow_query,
|
||||
instance: instance.clone(),
|
||||
allow_registration,
|
||||
federation: config.federation.clone(),
|
||||
event_publisher: Arc::clone(&event_publisher),
|
||||
federation_settings: std::sync::Arc::clone(&db.federation_settings),
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user