continuwuity/src/main/server.rs

use std::sync::Arc;

use conduwuit_core::{
	Error, Result,
	config::Config,
	info,
	log::Log,
	utils::{stream, sys},
};
use tokio::{runtime, sync::Mutex};

use crate::{
	clap::{Args, update},
	logging::TracingFlameGuard,
};

/// Server runtime state; complete
pub(crate) struct Server {
	/// Server runtime state; public portion
	pub(crate) server: Arc<conduwuit_core::Server>,

	pub(crate) services: Mutex<Option<Arc<conduwuit_service::Services>>>,

	_tracing_flame_guard: TracingFlameGuard,

	#[cfg(feature = "sentry_telemetry")]
	_sentry_guard: Option<::sentry::ClientInitGuard>,

	#[cfg(all(conduwuit_mods, feature = "conduwuit_mods"))]
	// Module instances; TODO: move to mods::loaded mgmt vector
	pub(crate) mods: tokio::sync::RwLock<Vec<conduwuit_core::mods::Module>>,
}

impl Server {
	pub(crate) fn new(
		args: &Args,
		runtime: Option<&runtime::Handle>,
	) -> Result<Arc<Self>, Error> {
		let _runtime_guard = runtime.map(runtime::Handle::enter);

		let config_paths = args.config.clone().unwrap_or_default();

		let config = Config::load(&config_paths)
			.and_then(|raw| update(raw, args))
			.and_then(|raw| Config::new(&raw))?;

		let (tracing_reload_handle, tracing_flame_guard, capture) =
			crate::logging::init(&config)?;

		config.check()?;

		#[cfg(feature = "sentry_telemetry")]
		let sentry_guard = crate::sentry::init(&config);

		#[cfg(unix)]
		sys::maximize_fd_limit()
			.expect("Unable to increase maximum soft and hard file descriptor limit");

		let (_old_width, _new_width) = stream::set_width(config.stream_width_default);
		let (_old_amp, _new_amp) = stream::set_amplification(config.stream_amplification);

		info!(
			server_name = %config.server_name,
			database_path = ?config.database_path,
			log_levels = %config.log,
			"{}",
			conduwuit_core::version(),
		);

		Ok(Arc::new(Self {
			server: Arc::new(conduwuit_core::Server::new(config, runtime.cloned(), Log {
				reload: tracing_reload_handle,
				capture,
			})),

			services: None.into(),

			_tracing_flame_guard: tracing_flame_guard,

			#[cfg(feature = "sentry_telemetry")]
			_sentry_guard: sentry_guard,

			#[cfg(all(conduwuit_mods, feature = "conduwuit_mods"))]
			mods: tokio::sync::RwLock::new(Vec::new()),
		}))
	}
}