1#![deny(unsafe_code)]
2#![expect(
3 clippy::option_map_unit_fn,
4 clippy::needless_pass_by_ref_mut )]
6#![deny(clippy::clone_on_ref_ptr)]
7#![feature(box_patterns, let_chains, option_zip, const_type_name)]
8
9pub mod automod;
10mod character_creator;
11pub mod chat;
12pub mod chunk_generator;
13mod chunk_serialize;
14pub mod client;
15pub mod cmd;
16pub mod connection_handler;
17mod data_dir;
18pub mod error;
19pub mod events;
20pub mod input;
21pub mod location;
22pub mod lod;
23pub mod login_provider;
24pub mod metrics;
25pub mod persistence;
26mod pet;
27pub mod presence;
28pub mod rtsim;
29pub mod settings;
30pub mod state_ext;
31pub mod sys;
32#[cfg(feature = "persistent_world")]
33pub mod terrain_persistence;
34#[cfg(not(feature = "worldgen"))] mod test_world;
35
36#[cfg(feature = "worldgen")] mod weather;
37
38pub mod wiring;
39
40pub use crate::{
42 data_dir::DEFAULT_DATA_DIR_NAME,
43 error::Error,
44 events::Event,
45 input::Input,
46 settings::{CalendarMode, EditableSettings, Settings},
47};
48
49#[cfg(feature = "persistent_world")]
50use crate::terrain_persistence::TerrainPersistence;
51use crate::{
52 automod::AutoMod,
53 chunk_generator::ChunkGenerator,
54 client::Client,
55 cmd::ChatCommandExt,
56 connection_handler::ConnectionHandler,
57 data_dir::DataDir,
58 location::Locations,
59 login_provider::LoginProvider,
60 persistence::PersistedComponents,
61 presence::{RegionSubscription, RepositionOnChunkLoad},
62 state_ext::StateExt,
63 sys::sentinel::DeletedEntities,
64};
65use censor::Censor;
66#[cfg(not(feature = "worldgen"))]
67use common::grid::Grid;
68#[cfg(feature = "worldgen")]
69use common::terrain::TerrainChunkSize;
70use common::{
71 assets::AssetExt,
72 calendar::Calendar,
73 character::{CharacterId, CharacterItem},
74 cmd::ServerChatCommand,
75 comp,
76 event::{
77 ClientDisconnectEvent, ClientDisconnectWithoutPersistenceEvent, EventBus, ExitIngameEvent,
78 UpdateCharacterDataEvent, register_event_busses,
79 },
80 link::Is,
81 mounting::{Volume, VolumeRider},
82 region::RegionMap,
83 resources::{BattleMode, GameMode, Time, TimeOfDay},
84 rtsim::RtSimEntity,
85 shared_server_config::ServerConstants,
86 slowjob::SlowJobPool,
87 terrain::TerrainChunk,
88 util::GIT_DATE_TIMESTAMP,
89 vol::RectRasterableVol,
90};
91use common_base::prof_span;
92use common_ecs::run_now;
93use common_net::{
94 msg::{ClientType, DisconnectReason, ServerGeneral, ServerInfo, ServerMsg},
95 sync::WorldSyncExt,
96};
97use common_state::{AreasContainer, BlockDiff, BuildArea, State};
98use common_systems::add_local_systems;
99use metrics::{EcsSystemMetrics, PhysicsMetrics, TickMetrics};
100use network::{ListenAddr, Network, Pid};
101use persistence::{
102 character_loader::{CharacterLoader, CharacterUpdaterMessage},
103 character_updater::CharacterUpdater,
104};
105use prometheus::Registry;
106use rustls::pki_types::{CertificateDer, PrivateKeyDer};
107use specs::{
108 Builder, Entity as EcsEntity, Entity, Join, LendJoin, WorldExt, shred::SendDispatcher,
109};
110use std::{
111 ops::{Deref, DerefMut},
112 sync::{Arc, Mutex},
113 time::{Duration, Instant},
114};
115#[cfg(not(feature = "worldgen"))]
116use test_world::{IndexOwned, World};
117use tokio::runtime::Runtime;
118use tracing::{debug, error, info, trace, warn};
119use vek::*;
120use veloren_query_server::server::QueryServer;
121pub use world::{WorldGenerateStage, civ::WorldCivStage, sim::WorldSimStage};
122
123use crate::{
124 persistence::{DatabaseSettings, SqlLogMode},
125 sys::terrain,
126};
127use hashbrown::HashMap;
128use std::sync::RwLock;
129
130use crate::settings::Protocol;
131
132#[cfg(feature = "plugins")]
133use {
134 common::uid::IdMaps,
135 common_state::plugin::{PluginMgr, memory_manager::EcsWorld},
136};
137
138use crate::{chat::ChatCache, persistence::character_loader::CharacterScreenResponseKind};
139use common::comp::Anchor;
140#[cfg(feature = "worldgen")]
141pub use world::{
142 IndexOwned, World,
143 sim::{DEFAULT_WORLD_MAP, DEFAULT_WORLD_SEED, FileOpts, GenOpts, WorldOpts},
144};
145
146#[derive(Copy, Clone)]
150pub struct SpawnPoint(pub Vec3<f32>);
151
152impl Default for SpawnPoint {
153 fn default() -> Self { Self(Vec3::new(0.0, 0.0, 256.0)) }
154}
155
156pub const MIN_VD: u32 = 6;
161
162#[derive(Copy, Clone, Default)]
165pub struct Tick(u64);
166
167#[derive(Clone)]
168pub struct HwStats {
169 hardware_threads: u32,
170 rayon_threads: u32,
171}
172
173#[derive(Clone, Copy, PartialEq)]
174enum DisconnectType {
175 WithPersistence,
176 WithoutPersistence,
177}
178
179#[derive(Copy, Clone)]
181pub struct TickStart(Instant);
182
183#[derive(Clone, Default, Debug)]
185pub struct BattleModeBuffer {
186 map: HashMap<CharacterId, (BattleMode, Time)>,
187}
188
189impl BattleModeBuffer {
190 pub fn push(&mut self, char_id: CharacterId, save: (BattleMode, Time)) {
191 self.map.insert(char_id, save);
192 }
193
194 pub fn get(&self, char_id: &CharacterId) -> Option<&(BattleMode, Time)> {
195 self.map.get(char_id)
196 }
197
198 pub fn pop(&mut self, char_id: &CharacterId) -> Option<(BattleMode, Time)> {
199 self.map.remove(char_id)
200 }
201}
202
203pub struct ChunkRequest {
204 entity: EcsEntity,
205 key: Vec2<i32>,
206}
207
208#[derive(Debug)]
209pub enum ServerInitStage {
210 DbMigrations,
211 DbVacuum,
212 WorldGen(WorldGenerateStage),
213 StartingSystems,
214}
215
216pub struct Server {
217 state: State,
218 world: Arc<World>,
219 index: IndexOwned,
220
221 connection_handler: ConnectionHandler,
222
223 runtime: Arc<Runtime>,
224
225 metrics_registry: Arc<Registry>,
226 chat_cache: ChatCache,
227 database_settings: Arc<RwLock<DatabaseSettings>>,
228 disconnect_all_clients_requested: bool,
229
230 event_dispatcher: SendDispatcher<'static>,
231}
232
233impl Server {
234 pub fn new(
236 settings: Settings,
237 editable_settings: EditableSettings,
238 database_settings: DatabaseSettings,
239 data_dir: &std::path::Path,
240 report_stage: &(dyn Fn(ServerInitStage) + Send + Sync),
241 runtime: Arc<Runtime>,
242 ) -> Result<Self, Error> {
243 prof_span!("Server::new");
244 info!("Server data dir is: {}", data_dir.display());
245 if settings.auth_server_address.is_none() {
246 info!("Authentication is disabled");
247 }
248
249 report_stage(ServerInitStage::DbMigrations);
250 debug!("Running DB migrations...");
252 persistence::run_migrations(&database_settings);
253
254 report_stage(ServerInitStage::DbVacuum);
255 debug!("Vacuuming database...");
257 persistence::vacuum_database(&database_settings);
258
259 let database_settings = Arc::new(RwLock::new(database_settings));
260
261 let registry = Arc::new(Registry::new());
262 let chunk_gen_metrics = metrics::ChunkGenMetrics::new(®istry).unwrap();
263 let job_metrics = metrics::JobMetrics::new(®istry).unwrap();
264 let network_request_metrics = metrics::NetworkRequestMetrics::new(®istry).unwrap();
265 let player_metrics = metrics::PlayerMetrics::new(®istry).unwrap();
266 let ecs_system_metrics = EcsSystemMetrics::new(®istry).unwrap();
267 let tick_metrics = TickMetrics::new(®istry).unwrap();
268 let physics_metrics = PhysicsMetrics::new(®istry).unwrap();
269 let server_event_metrics = metrics::ServerEventMetrics::new(®istry).unwrap();
270 let query_server_metrics = metrics::QueryServerMetrics::new(®istry).unwrap();
271
272 let battlemode_buffer = BattleModeBuffer::default();
273
274 let pools = State::pools(GameMode::Server);
275
276 #[cfg(feature = "plugins")]
278 let plugin_mgr = PluginMgr::from_asset_or_default();
279
280 debug!("Generating world, seed: {}", settings.world_seed);
281 #[cfg(feature = "worldgen")]
282 let (world, index) = World::generate(
283 settings.world_seed,
284 WorldOpts {
285 seed_elements: true,
286 world_file: if let Some(ref opts) = settings.map_file {
287 opts.clone()
288 } else {
289 FileOpts::LoadAsset(DEFAULT_WORLD_MAP.into())
291 },
292 calendar: Some(settings.calendar_mode.calendar_now()),
293 },
294 &pools,
295 &|stage| {
296 report_stage(ServerInitStage::WorldGen(stage));
297 },
298 );
299 #[cfg(not(feature = "worldgen"))]
300 let (world, index) = World::generate(settings.world_seed);
301
302 #[cfg(feature = "worldgen")]
303 let map = world.get_map_data(index.as_index_ref(), &pools);
304 #[cfg(not(feature = "worldgen"))]
305 let map = common_net::msg::WorldMapMsg {
306 dimensions_lg: Vec2::zero(),
307 max_height: 1.0,
308 rgba: Grid::new(Vec2::new(1, 1), 1),
309 horizons: [(vec![0], vec![0]), (vec![0], vec![0])],
310 alt: Grid::new(Vec2::new(1, 1), 1),
311 sites: Vec::new(),
312 possible_starting_sites: Vec::new(),
313 pois: Vec::new(),
314 default_chunk: Arc::new(world.generate_oob_chunk()),
315 };
316
317 #[cfg(feature = "worldgen")]
318 let map_size_lg = world.sim().map_size_lg();
319 #[cfg(not(feature = "worldgen"))]
320 let map_size_lg = world.map_size_lg();
321
322 let lod = lod::Lod::from_world(&world, index.as_index_ref(), &pools);
323
324 report_stage(ServerInitStage::StartingSystems);
325
326 let mut state = State::server(
327 Arc::clone(&pools),
328 map_size_lg,
329 Arc::clone(&map.default_chunk),
330 |dispatcher_builder| {
331 add_local_systems(dispatcher_builder);
332 sys::msg::add_server_systems(dispatcher_builder);
333 sys::add_server_systems(dispatcher_builder);
334 #[cfg(feature = "worldgen")]
335 {
336 rtsim::add_server_systems(dispatcher_builder);
337 weather::add_server_systems(dispatcher_builder);
338 }
339 },
340 #[cfg(feature = "plugins")]
341 plugin_mgr,
342 );
343 register_event_busses(state.ecs_mut());
344 state.ecs_mut().insert(battlemode_buffer);
345 state.ecs_mut().insert(settings.clone());
346 state.ecs_mut().insert(editable_settings);
347 state.ecs_mut().insert(DataDir {
348 path: data_dir.to_owned(),
349 });
350
351 register_event_busses(state.ecs_mut());
352 state.ecs_mut().insert(Vec::<ChunkRequest>::new());
353 state
354 .ecs_mut()
355 .insert(EventBus::<chunk_serialize::ChunkSendEntry>::default());
356 state.ecs_mut().insert(Locations::default());
357 state.ecs_mut().insert(LoginProvider::new(
358 settings.auth_server_address.clone(),
359 Arc::clone(&runtime),
360 ));
361 state.ecs_mut().insert(HwStats {
362 hardware_threads: num_cpus::get() as u32,
363 rayon_threads: num_cpus::get() as u32,
364 });
365 state.ecs_mut().insert(ServerConstants {
366 day_cycle_coefficient: settings.day_cycle_coefficient(),
367 });
368 state.ecs_mut().insert(Tick(0));
369 state.ecs_mut().insert(TickStart(Instant::now()));
370 state.ecs_mut().insert(job_metrics);
371 state.ecs_mut().insert(network_request_metrics);
372 state.ecs_mut().insert(player_metrics);
373 state.ecs_mut().insert(ecs_system_metrics);
374 state.ecs_mut().insert(tick_metrics);
375 state.ecs_mut().insert(physics_metrics);
376 state.ecs_mut().insert(server_event_metrics);
377 state.ecs_mut().insert(query_server_metrics);
378 if settings.experimental_terrain_persistence {
379 #[cfg(feature = "persistent_world")]
380 {
381 warn!(
382 "Experimental terrain persistence support is enabled. This feature may break, \
383 be disabled, or otherwise change under your feet at *any time*. \
384 Additionally, it is expected to be replaced in the future *without* \
385 migration or warning. You have been warned."
386 );
387 state
388 .ecs_mut()
389 .insert(TerrainPersistence::new(data_dir.to_owned()));
390 }
391 #[cfg(not(feature = "persistent_world"))]
392 error!(
393 "Experimental terrain persistence support was requested, but the server was not \
394 compiled with the feature. Terrain modifications will *not* be persisted."
395 );
396 }
397 {
398 let pool = state.ecs_mut().write_resource::<SlowJobPool>();
399 pool.configure("CHUNK_DROP", |_n| 1);
400 pool.configure("CHUNK_GENERATOR", |n| n / 2 + n / 4);
401 pool.configure("CHUNK_SERIALIZER", |n| n / 2);
402 pool.configure("RTSIM_SAVE", |_| 1);
403 pool.configure("WEATHER", |_| 1);
404 }
405 state
406 .ecs_mut()
407 .insert(ChunkGenerator::new(chunk_gen_metrics));
408 {
409 let (sender, receiver) =
410 crossbeam_channel::bounded::<chunk_serialize::SerializedChunk>(10_000);
411 state.ecs_mut().insert(sender);
412 state.ecs_mut().insert(receiver);
413 }
414
415 state.ecs_mut().insert(CharacterUpdater::new(
416 Arc::<RwLock<DatabaseSettings>>::clone(&database_settings),
417 )?);
418
419 let ability_map = comp::item::tool::AbilityMap::<comp::AbilityItem>::load_expect_cloned(
420 "common.abilities.ability_set_manifest",
421 );
422 state.ecs_mut().insert(ability_map);
423
424 let msm = comp::inventory::item::MaterialStatManifest::load().cloned();
425 state.ecs_mut().insert(msm);
426
427 let rbm = common::recipe::RecipeBookManifest::load().cloned();
428 state.ecs_mut().insert(rbm);
429
430 state.ecs_mut().insert(CharacterLoader::new(
431 Arc::<RwLock<DatabaseSettings>>::clone(&database_settings),
432 )?);
433
434 state
436 .ecs_mut()
437 .insert(sys::PersistenceScheduler::every(Duration::from_secs(10)));
438
439 state.ecs_mut().insert(RegionMap::new());
441
442 state.ecs_mut().register::<RegionSubscription>();
444 state.ecs_mut().register::<Client>();
445 state.ecs_mut().register::<comp::Presence>();
446 state.ecs_mut().register::<wiring::WiringElement>();
447 state.ecs_mut().register::<wiring::Circuit>();
448 state.ecs_mut().register::<Anchor>();
449 state.ecs_mut().register::<comp::Pet>();
450 state.ecs_mut().register::<login_provider::PendingLogin>();
451 state.ecs_mut().register::<RepositionOnChunkLoad>();
452 state.ecs_mut().register::<RtSimEntity>();
453
454 let banned_words = settings.moderation.load_banned_words(data_dir);
456 let censor = Arc::new(Censor::Custom(banned_words.into_iter().collect()));
457 state.ecs_mut().insert(Arc::clone(&censor));
458
459 state
461 .ecs_mut()
462 .insert(AutoMod::new(&settings.moderation, censor));
463
464 state.ecs_mut().insert(map);
465
466 #[cfg(feature = "worldgen")]
467 let spawn_point = SpawnPoint({
468 use world::civ::SiteKind;
469
470 let index = index.as_index_ref();
471 let center_chunk = world.sim().map_size_lg().chunks().map(i32::from) / 2;
478 let spawn_chunk = world
479 .civs()
480 .sites()
481 .filter(|site| matches!(site.kind, SiteKind::Settlement | SiteKind::Refactor))
482 .map(|site| site.center)
483 .min_by_key(|site_pos| site_pos.distance_squared(center_chunk))
484 .unwrap_or(center_chunk);
485
486 world.find_accessible_pos(index, TerrainChunkSize::center_wpos(spawn_chunk), false)
487 });
488 #[cfg(not(feature = "worldgen"))]
489 let spawn_point = SpawnPoint::default();
490
491 state.ecs_mut().insert(spawn_point);
493
494 {
497 #[cfg(feature = "worldgen")]
498 let size = world.sim().get_size();
499 #[cfg(not(feature = "worldgen"))]
500 let size = world.map_size_lg().chunks().map(u32::from);
501
502 let world_size = size.map(|e| e as i32) * TerrainChunk::RECT_SIZE.map(|e| e as i32);
503 let world_aabb = Aabb {
504 min: Vec3::new(0, 0, -32768),
505 max: Vec3::new(world_size.x, world_size.y, 32767),
506 }
507 .made_valid();
508
509 state
510 .ecs()
511 .write_resource::<AreasContainer<BuildArea>>()
512 .insert("world".to_string(), world_aabb)
513 .expect("The initial insert should always work.");
514 }
515
516 let world = Arc::new(world);
518 state.ecs_mut().insert(Arc::clone(&world));
519 state.ecs_mut().insert(lod);
520 state.ecs_mut().insert(index.clone());
521
522 state.ecs_mut().write_resource::<TimeOfDay>().0 = settings.world.start_time;
524
525 sys::sentinel::UpdateTrackers::register(state.ecs_mut());
527
528 state.ecs_mut().insert(DeletedEntities::default());
529
530 let network = Network::new_with_registry(Pid::new(), &runtime, ®istry);
531 let (chat_cache, chat_tracker) = ChatCache::new(Duration::from_secs(60), &runtime);
532 state.ecs_mut().insert(chat_tracker);
533
534 let mut printed_quic_warning = false;
535 for protocol in &settings.gameserver_protocols {
536 match protocol {
537 Protocol::Tcp { address } => {
538 runtime.block_on(network.listen(ListenAddr::Tcp(*address)))?;
539 },
540 Protocol::Quic {
541 address,
542 cert_file_path,
543 key_file_path,
544 } => {
545 use rustls_pemfile::Item;
546 use std::fs;
547
548 match || -> Result<_, Box<dyn std::error::Error>> {
549 let key = fs::read(key_file_path)?;
550 let key = if key_file_path.extension().is_some_and(|x| x == "der") {
551 PrivateKeyDer::try_from(key).map_err(|_| "No valid pem key in file")?
552 } else {
553 debug!("convert pem key to der");
554 rustls_pemfile::read_all(&mut key.as_slice())
555 .find_map(|item| match item {
556 Ok(Item::Pkcs1Key(v)) => Some(PrivateKeyDer::Pkcs1(v)),
557 Ok(Item::Pkcs8Key(v)) => Some(PrivateKeyDer::Pkcs8(v)),
558 Ok(Item::Sec1Key(v)) => Some(PrivateKeyDer::Sec1(v)),
559 Ok(Item::Crl(_)) => None,
560 Ok(Item::Csr(_)) => None,
561 Ok(Item::X509Certificate(_)) => None,
562 Ok(_) => None,
563 Err(e) => {
564 tracing::warn!(?e, "error while reading key_file");
565 None
566 },
567 })
568 .ok_or("No valid pem key in file")?
569 };
570 let cert_chain = fs::read(cert_file_path)?;
571 let cert_chain = if cert_file_path.extension().is_some_and(|x| x == "der") {
572 vec![CertificateDer::from(cert_chain)]
573 } else {
574 debug!("convert pem cert to der");
575 rustls_pemfile::certs(&mut cert_chain.as_slice())
576 .filter_map(|item| match item {
577 Ok(cert) => Some(cert),
578 Err(e) => {
579 tracing::warn!(?e, "error while reading cert_file");
580 None
581 },
582 })
583 .collect()
584 };
585 let server_config = quinn::ServerConfig::with_single_cert(cert_chain, key)?;
586 Ok(server_config)
587 }() {
588 Ok(server_config) => {
589 runtime.block_on(
590 network.listen(ListenAddr::Quic(*address, server_config.clone())),
591 )?;
592
593 if !printed_quic_warning {
594 warn!(
595 "QUIC is enabled. This is experimental and not recommended in \
596 production"
597 );
598 printed_quic_warning = true;
599 }
600 },
601 Err(e) => {
602 error!(
603 ?e,
604 "Failed to load the TLS certificate, running without QUIC {}",
605 *address
606 );
607 },
608 }
609 },
610 }
611 }
612
613 if let Some(addr) = settings.query_address {
614 use veloren_query_server::proto::ServerInfo;
615
616 const QUERY_SERVER_RATELIMIT: u16 = 120;
617
618 let (query_server_info_tx, query_server_info_rx) =
619 tokio::sync::watch::channel(ServerInfo {
620 git_hash: *sys::server_info::GIT_HASH,
621 git_timestamp: *GIT_DATE_TIMESTAMP,
622 players_count: 0,
623 player_cap: settings.max_players,
624 battlemode: settings.gameplay.battle_mode.into(),
625 });
626 let mut query_server =
627 QueryServer::new(addr, query_server_info_rx, QUERY_SERVER_RATELIMIT);
628 let query_server_metrics =
629 Arc::new(Mutex::new(veloren_query_server::server::Metrics::default()));
630 let query_server_metrics2 = Arc::clone(&query_server_metrics);
631 runtime.spawn(async move {
632 let err = query_server.run(query_server_metrics2).await.err();
633 error!(?err, "Query server stopped unexpectedly");
634 });
635 state.ecs_mut().insert(query_server_info_tx);
636 state.ecs_mut().insert(query_server_metrics);
637 }
638
639 runtime.block_on(network.listen(ListenAddr::Mpsc(14004)))?;
640
641 let connection_handler = ConnectionHandler::new(network, &runtime);
642
643 #[cfg(feature = "worldgen")]
645 {
646 match rtsim::RtSim::new(
647 &settings.world,
648 index.as_index_ref(),
649 &world,
650 data_dir.to_owned(),
651 ) {
652 Ok(rtsim) => {
653 state.ecs_mut().insert(rtsim.state().data().time_of_day);
654 state.ecs_mut().insert(rtsim);
655 },
656 Err(err) => {
657 error!("Failed to load rtsim: {}", err);
658 return Err(Error::RtsimError(err));
659 },
660 }
661 weather::init(&mut state);
662 }
663
664 let this = Self {
665 state,
666 world,
667 index,
668 connection_handler,
669 runtime,
670
671 metrics_registry: registry,
672 chat_cache,
673 database_settings,
674 disconnect_all_clients_requested: false,
675
676 event_dispatcher: Self::create_event_dispatcher(pools),
677 };
678
679 debug!(?settings, "created veloren server with");
680
681 let git_hash = *common::util::GIT_HASH;
682 let git_date = common::util::GIT_DATE.clone();
683 let git_time = *common::util::GIT_TIME;
684 let version = common::util::DISPLAY_VERSION_LONG.clone();
685 info!(?version, "Server version");
686 debug!(?git_hash, ?git_date, ?git_time, "detailed Server version");
687
688 Ok(this)
689 }
690
691 pub fn get_server_info(&self) -> ServerInfo {
692 let settings = self.state.ecs().fetch::<Settings>();
693
694 ServerInfo {
695 name: settings.server_name.clone(),
696 git_hash: common::util::GIT_HASH.to_string(),
697 git_date: common::util::GIT_DATE.to_string(),
698 auth_provider: settings.auth_server_address.clone(),
699 }
700 }
701
702 pub fn settings(&self) -> impl Deref<Target = Settings> + '_ {
704 self.state.ecs().fetch::<Settings>()
705 }
706
707 pub fn settings_mut(&self) -> impl DerefMut<Target = Settings> + '_ {
709 self.state.ecs().fetch_mut::<Settings>()
710 }
711
712 pub fn editable_settings_mut(&self) -> impl DerefMut<Target = EditableSettings> + '_ {
714 self.state.ecs().fetch_mut::<EditableSettings>()
715 }
716
717 pub fn editable_settings(&self) -> impl Deref<Target = EditableSettings> + '_ {
719 self.state.ecs().fetch::<EditableSettings>()
720 }
721
722 pub fn data_dir(&self) -> impl Deref<Target = DataDir> + '_ {
724 self.state.ecs().fetch::<DataDir>()
725 }
726
727 pub fn state(&self) -> &State { &self.state }
729
730 pub fn state_mut(&mut self) -> &mut State { &mut self.state }
732
733 pub fn world(&self) -> &World { &self.world }
735
736 pub fn metrics_registry(&self) -> &Arc<Registry> { &self.metrics_registry }
738
739 pub fn chat_cache(&self) -> &ChatCache { &self.chat_cache }
741
742 fn parse_locations(&self, character_list_data: &mut [CharacterItem]) {
743 character_list_data.iter_mut().for_each(|c| {
744 let name = c
745 .location
746 .as_ref()
747 .and_then(|s| {
748 persistence::parse_waypoint(s)
749 .ok()
750 .and_then(|(waypoint, _)| waypoint.map(|w| w.get_pos()))
751 })
752 .and_then(|wpos| {
753 self.world
754 .get_location_name(self.index.as_index_ref(), wpos.xy().as_::<i32>())
755 });
756 c.location = name;
757 });
758 }
759
760 pub fn tick(&mut self, _input: Input, dt: Duration) -> Result<Vec<Event>, Error> {
763 self.state.ecs().write_resource::<Tick>().0 += 1;
764 self.state.ecs().write_resource::<TickStart>().0 = Instant::now();
765
766 let new_calendar = self
770 .state
771 .ecs()
772 .read_resource::<Settings>()
773 .calendar_mode
774 .calendar_now();
775 *self.state.ecs_mut().write_resource::<Calendar>() = new_calendar;
776
777 let mut frontend_events = Vec::new();
800
801 let before_new_connections = Instant::now();
804
805 self.handle_new_connections(&mut frontend_events);
807
808 let before_state_tick = Instant::now();
809
810 fn on_block_update(ecs: &specs::World, changes: Vec<BlockDiff>) {
811 if changes
813 .iter()
814 .any(|c| c.old.get_rtsim_resource() != c.new.get_rtsim_resource())
815 {
816 ecs.write_resource::<rtsim::RtSim>().hook_block_update(
817 &ecs.read_resource::<Arc<world::World>>(),
818 ecs.read_resource::<world::IndexOwned>().as_index_ref(),
819 changes,
820 );
821 }
822 }
823
824 let mut state_tick_metrics = Default::default();
828 let server_constants = (*self.state.ecs().read_resource::<ServerConstants>()).clone();
829 self.state.tick(
830 dt,
831 false,
832 Some(&mut state_tick_metrics),
833 &server_constants,
834 on_block_update,
835 );
836
837 let before_handle_events = Instant::now();
838
839 let disconnect_type = self.disconnect_all_clients_if_requested();
842
843 self.state.maintain_links();
845
846 frontend_events.append(&mut self.handle_events());
848
849 let before_update_terrain_and_regions = Instant::now();
850
851 self.update_region_map();
856 self.state.apply_terrain_changes(on_block_update);
859
860 let before_sync = Instant::now();
861
862 sys::run_sync_systems(self.state.ecs_mut());
864
865 let before_world_tick = Instant::now();
866
867 self.world.tick(dt);
869
870 let before_entity_cleanup = Instant::now();
871
872 if let Some(DisconnectType::WithoutPersistence) = disconnect_type {
877 run_now::<terrain::Sys>(self.state.ecs_mut());
878 }
879
880 #[cfg(feature = "worldgen")]
882 {
883 let mut rtsim = self.state.ecs().write_resource::<rtsim::RtSim>();
884 for chunk in &self.state.terrain_changes().removed_chunks {
885 rtsim.hook_unload_chunk(*chunk);
886 }
887 }
888
889 let anchors = self.state.ecs().read_storage::<Anchor>();
900 let anchored_anchor_entities: Vec<Entity> = (
901 &self.state.ecs().entities(),
902 &self.state.ecs().read_storage::<Anchor>(),
903 )
904 .join()
905 .filter_map(|(_, anchor)| match anchor {
906 Anchor::Entity(anchor_entity) => Some(*anchor_entity),
907 _ => None,
908 })
909 .filter(|anchor_entity| match anchors.get(*anchor_entity) {
913 Some(Anchor::Entity(_)) => true,
914 Some(Anchor::Chunk(_)) | None => false
915 })
916 .collect();
917 drop(anchors);
918
919 for entity in anchored_anchor_entities {
920 if cfg!(debug_assertions) {
921 panic!("Entity anchor chain detected");
922 }
923 error!(
924 "Detected an anchor entity that itself has an anchor entity - anchor chains are \
925 not currently supported. The entity's Anchor component has been deleted"
926 );
927 self.state.delete_component::<Anchor>(entity);
928 }
929
930 let to_delete = {
933 let terrain = self.state.terrain();
934 (
935 &self.state.ecs().entities(),
936 &self.state.ecs().read_storage::<comp::Pos>(),
937 !&self.state.ecs().read_storage::<comp::Presence>(),
938 self.state.ecs().read_storage::<Anchor>().maybe(),
939 self.state.ecs().read_storage::<Is<VolumeRider>>().maybe(),
940 )
941 .join()
942 .filter(|(_, pos, _, anchor, is_volume_rider)| {
943 let pos = is_volume_rider
944 .and_then(|is_volume_rider| match is_volume_rider.pos.kind {
945 Volume::Terrain => None,
946 Volume::Entity(e) => {
947 let e = self.state.ecs().entity_from_uid(e)?;
948 let pos = self
949 .state
950 .ecs()
951 .read_storage::<comp::Pos>()
952 .get(e)
953 .copied()?;
954
955 Some(pos.0)
956 },
957 })
958 .unwrap_or(pos.0);
959 let chunk_key = terrain.pos_key(pos.map(|e| e.floor() as i32));
960 match anchor {
961 Some(Anchor::Chunk(hc)) => {
962 terrain.get_key_real(chunk_key).is_none()
967 && terrain.get_key_real(*hc).is_none()
968 },
969 Some(Anchor::Entity(entity)) => !self.state.ecs().is_alive(*entity),
970 None => terrain.get_key_real(chunk_key).is_none(),
971 }
972 })
973 .map(|(entity, _, _, _, _)| entity)
974 .collect::<Vec<_>>()
975 };
976
977 #[cfg(feature = "worldgen")]
978 {
979 let mut rtsim = self.state.ecs().write_resource::<rtsim::RtSim>();
980 let rtsim_entities = self.state.ecs().read_storage();
981 for entity in &to_delete {
982 if let Some(rtsim_entity) = rtsim_entities.get(*entity) {
983 rtsim.hook_rtsim_entity_unload(*rtsim_entity);
984 }
985 }
986 }
987
988 for entity in to_delete {
990 if let Err(e) = self.state.delete_entity_recorded(entity) {
991 error!(?e, "Failed to delete agent outside the terrain");
992 }
993 }
994
995 if let Some(DisconnectType::WithoutPersistence) = disconnect_type {
996 info!(
997 "Disconnection of all players without persistence complete, signalling to \
998 persistence thread that character updates may continue to be processed"
999 );
1000 self.state
1001 .ecs()
1002 .fetch_mut::<CharacterUpdater>()
1003 .disconnected_success();
1004 }
1005
1006 let before_persistence_updates = Instant::now();
1008
1009 let character_loader = self.state.ecs().read_resource::<CharacterLoader>();
1010
1011 let mut character_updater = self.state.ecs().write_resource::<CharacterUpdater>();
1012 let updater_messages: Vec<CharacterUpdaterMessage> = character_updater.messages().collect();
1013
1014 character_loader
1016 .messages()
1017 .chain(updater_messages)
1018 .for_each(|message| match message {
1019 CharacterUpdaterMessage::DatabaseBatchCompletion(batch_id) => {
1020 character_updater.process_batch_completion(batch_id);
1021 },
1022 CharacterUpdaterMessage::CharacterScreenResponse(response) => {
1023 match response.response_kind {
1024 CharacterScreenResponseKind::CharacterList(result) => match result {
1025 Ok(mut character_list_data) => {
1026 self.parse_locations(&mut character_list_data);
1027 self.notify_client(
1028 response.target_entity,
1029 ServerGeneral::CharacterListUpdate(character_list_data),
1030 )
1031 },
1032 Err(error) => self.notify_client(
1033 response.target_entity,
1034 ServerGeneral::CharacterActionError(error.to_string()),
1035 ),
1036 },
1037 CharacterScreenResponseKind::CharacterCreation(result) => match result {
1038 Ok((character_id, mut list)) => {
1039 self.parse_locations(&mut list);
1040 self.notify_client(
1041 response.target_entity,
1042 ServerGeneral::CharacterListUpdate(list),
1043 );
1044 self.notify_client(
1045 response.target_entity,
1046 ServerGeneral::CharacterCreated(character_id),
1047 );
1048 },
1049 Err(error) => self.notify_client(
1050 response.target_entity,
1051 ServerGeneral::CharacterActionError(error.to_string()),
1052 ),
1053 },
1054 CharacterScreenResponseKind::CharacterEdit(result) => match result {
1055 Ok((character_id, mut list)) => {
1056 self.parse_locations(&mut list);
1057 self.notify_client(
1058 response.target_entity,
1059 ServerGeneral::CharacterListUpdate(list),
1060 );
1061 self.notify_client(
1062 response.target_entity,
1063 ServerGeneral::CharacterEdited(character_id),
1064 );
1065 },
1066 Err(error) => self.notify_client(
1067 response.target_entity,
1068 ServerGeneral::CharacterActionError(error.to_string()),
1069 ),
1070 },
1071 CharacterScreenResponseKind::CharacterData(result) => {
1072 match *result {
1073 Ok((character_data, skill_set_persistence_load_error)) => {
1074 let PersistedComponents {
1075 body,
1076 hardcore,
1077 stats,
1078 skill_set,
1079 inventory,
1080 waypoint,
1081 pets,
1082 active_abilities,
1083 map_marker,
1084 } = character_data;
1085 let character_data = (
1086 body,
1087 hardcore,
1088 stats,
1089 skill_set,
1090 inventory,
1091 waypoint,
1092 pets,
1093 active_abilities,
1094 map_marker,
1095 );
1096 self.state.emit_event_now(UpdateCharacterDataEvent {
1099 entity: response.target_entity,
1100 components: character_data,
1101 metadata: skill_set_persistence_load_error,
1102 })
1103 },
1104 Err(error) => {
1105 self.notify_client(
1109 response.target_entity,
1110 ServerGeneral::CharacterDataLoadResult(Err(
1111 error.to_string()
1112 )),
1113 );
1114
1115 self.state.emit_event_now(ExitIngameEvent {
1117 entity: response.target_entity,
1118 })
1119 },
1120 }
1121 },
1122 }
1123 },
1124 });
1125
1126 drop(character_loader);
1127 drop(character_updater);
1128
1129 {
1130 let index = &mut self.index;
1135 let world = &mut self.world;
1136 let ecs = self.state.ecs_mut();
1137 let slow_jobs = ecs.write_resource::<SlowJobPool>();
1138
1139 index.reload_if_changed(|index| {
1140 let mut chunk_generator = ecs.write_resource::<ChunkGenerator>();
1141 let client = ecs.read_storage::<Client>();
1142 let mut terrain = ecs.write_resource::<common::terrain::TerrainGrid>();
1143 #[cfg(feature = "worldgen")]
1144 let rtsim = ecs.read_resource::<rtsim::RtSim>();
1145 #[cfg(not(feature = "worldgen"))]
1146 let rtsim = ();
1147
1148 chunk_generator.cancel_all();
1150
1151 if client.is_empty() {
1152 terrain.clear();
1154 } else {
1155 terrain.iter().for_each(|(pos, _)| {
1157 chunk_generator.generate_chunk(
1158 None,
1159 pos,
1160 &slow_jobs,
1161 Arc::clone(world),
1162 &rtsim,
1163 index.clone(),
1164 (
1165 *ecs.read_resource::<TimeOfDay>(),
1166 (*ecs.read_resource::<Calendar>()).clone(),
1167 ),
1168 );
1169 });
1170 }
1171 });
1172 }
1173
1174 let end_of_server_tick = Instant::now();
1175
1176 run_now::<sys::metrics::Sys>(self.state.ecs());
1178
1179 {
1180 let tick_metrics = self.state.ecs().read_resource::<TickMetrics>();
1182
1183 let tt = &tick_metrics.tick_time;
1184 tt.with_label_values(&["new connections"])
1185 .set((before_state_tick - before_new_connections).as_nanos() as i64);
1186 tt.with_label_values(&["handle server events"])
1187 .set((before_update_terrain_and_regions - before_handle_events).as_nanos() as i64);
1188 tt.with_label_values(&["update terrain and region map"])
1189 .set((before_sync - before_update_terrain_and_regions).as_nanos() as i64);
1190 tt.with_label_values(&["state"])
1191 .set((before_handle_events - before_state_tick).as_nanos() as i64);
1192 tt.with_label_values(&["world tick"])
1193 .set((before_entity_cleanup - before_world_tick).as_nanos() as i64);
1194 tt.with_label_values(&["entity cleanup"])
1195 .set((before_persistence_updates - before_entity_cleanup).as_nanos() as i64);
1196 tt.with_label_values(&["persistence_updates"])
1197 .set((end_of_server_tick - before_persistence_updates).as_nanos() as i64);
1198 for (label, duration) in state_tick_metrics.timings {
1199 tick_metrics
1200 .state_tick_time
1201 .with_label_values(&[label])
1202 .set(duration.as_nanos() as i64);
1203 }
1204 tick_metrics.tick_time_hist.observe(
1205 end_of_server_tick
1206 .duration_since(before_state_tick)
1207 .as_secs_f64(),
1208 );
1209 }
1210
1211 Ok(frontend_events)
1214 }
1215
1216 pub fn cleanup(&mut self) {
1218 self.state.cleanup();
1220
1221 #[cfg(feature = "persistent_world")]
1223 self.state
1224 .ecs()
1225 .try_fetch_mut::<TerrainPersistence>()
1226 .map(|mut t| t.maintain());
1227 }
1228
1229 fn update_region_map(&mut self) {
1231 prof_span!("Server::update_region_map");
1232 let ecs = self.state().ecs();
1233 ecs.write_resource::<RegionMap>().tick(
1234 ecs.read_storage::<comp::Pos>(),
1235 ecs.read_storage::<comp::Vel>(),
1236 ecs.read_storage::<comp::Presence>(),
1237 ecs.entities(),
1238 );
1239 }
1240
1241 fn initialize_client(&mut self, client: connection_handler::IncomingClient) -> Entity {
1242 let entity = self
1243 .state
1244 .ecs_mut()
1245 .create_entity_synced()
1246 .with(client)
1247 .build();
1248 self.state
1249 .ecs()
1250 .read_resource::<metrics::PlayerMetrics>()
1251 .clients_connected
1252 .inc();
1253 entity
1254 }
1255
1256 fn disconnect_all_clients_if_requested(&mut self) -> Option<DisconnectType> {
1260 let mut character_updater = self.state.ecs().fetch_mut::<CharacterUpdater>();
1261
1262 let disconnect_type = self.get_disconnect_all_clients_requested(&mut character_updater);
1263 if let Some(disconnect_type) = disconnect_type {
1264 let with_persistence = disconnect_type == DisconnectType::WithPersistence;
1265 let clients = self.state.ecs().read_storage::<Client>();
1266 let entities = self.state.ecs().entities();
1267
1268 info!(
1269 "Disconnecting all clients ({} persistence) as requested",
1270 if with_persistence { "with" } else { "without" }
1271 );
1272 for (_, entity) in (&clients, &entities).join() {
1273 info!("Emitting client disconnect event for entity: {:?}", entity);
1274 if with_persistence {
1275 self.state.emit_event_now(ClientDisconnectEvent(
1276 entity,
1277 comp::DisconnectReason::Kicked,
1278 ))
1279 } else {
1280 self.state
1281 .emit_event_now(ClientDisconnectWithoutPersistenceEvent(entity))
1282 };
1283 }
1284
1285 self.disconnect_all_clients_requested = false;
1286 }
1287
1288 disconnect_type
1289 }
1290
1291 fn get_disconnect_all_clients_requested(
1292 &self,
1293 character_updater: &mut CharacterUpdater,
1294 ) -> Option<DisconnectType> {
1295 let without_persistence_requested = character_updater.disconnect_all_clients_requested();
1296 let with_persistence_requested = self.disconnect_all_clients_requested;
1297
1298 if without_persistence_requested {
1299 return Some(DisconnectType::WithoutPersistence);
1300 };
1301 if with_persistence_requested {
1302 return Some(DisconnectType::WithPersistence);
1303 };
1304 None
1305 }
1306
1307 fn handle_new_connections(&mut self, frontend_events: &mut Vec<Event>) {
1309 while let Ok(sender) = self.connection_handler.info_requester_receiver.try_recv() {
1310 trace!("sending info to connection_handler");
1312 let _ = sender.send(connection_handler::ServerInfoPacket {
1313 info: self.get_server_info(),
1314 time: self.state.get_time(),
1315 });
1316 }
1317
1318 while let Ok(incoming) = self.connection_handler.client_receiver.try_recv() {
1319 let entity = self.initialize_client(incoming);
1320 frontend_events.push(Event::ClientConnected { entity });
1321 }
1322 }
1323
1324 pub fn notify_client<S>(&self, entity: EcsEntity, msg: S)
1325 where
1326 S: Into<ServerMsg>,
1327 {
1328 if let Some(client) = self.state.ecs().read_storage::<Client>().get(entity) {
1329 client.send_fallible(msg);
1330 }
1331 }
1332
1333 pub fn notify_players(&mut self, msg: ServerGeneral) { self.state.notify_players(msg); }
1334
1335 pub fn generate_chunk(&mut self, entity: EcsEntity, key: Vec2<i32>) {
1336 let ecs = self.state.ecs();
1337 let slow_jobs = ecs.read_resource::<SlowJobPool>();
1338 #[cfg(feature = "worldgen")]
1339 let rtsim = ecs.read_resource::<rtsim::RtSim>();
1340 #[cfg(not(feature = "worldgen"))]
1341 let rtsim = ();
1342 ecs.write_resource::<ChunkGenerator>().generate_chunk(
1343 Some(entity),
1344 key,
1345 &slow_jobs,
1346 Arc::clone(&self.world),
1347 &rtsim,
1348 self.index.clone(),
1349 (
1350 *ecs.read_resource::<TimeOfDay>(),
1351 (*ecs.read_resource::<Calendar>()).clone(),
1352 ),
1353 );
1354 }
1355
1356 fn process_command(&mut self, entity: EcsEntity, name: String, args: Vec<String>) {
1357 if let Ok(command) = name.parse::<ServerChatCommand>() {
1359 command.execute(self, entity, args);
1360 } else {
1361 #[cfg(feature = "plugins")]
1362 {
1363 let mut plugin_manager = self.state.ecs().write_resource::<PluginMgr>();
1364 let ecs_world = EcsWorld {
1365 entities: &self.state.ecs().entities(),
1366 health: self.state.ecs().read_component().into(),
1367 uid: self.state.ecs().read_component().into(),
1368 id_maps: &self.state.ecs().read_resource::<IdMaps>().into(),
1369 player: self.state.ecs().read_component().into(),
1370 };
1371 let uid = if let Some(uid) = ecs_world.uid.get(entity).copied() {
1372 uid
1373 } else {
1374 self.notify_client(
1375 entity,
1376 ServerGeneral::server_msg(
1377 comp::ChatType::CommandError,
1378 common::comp::Content::Plain(
1379 "Can't get player UUID (player may be disconnected?)".to_string(),
1380 ),
1381 ),
1382 );
1383 return;
1384 };
1385 match plugin_manager.command_event(&ecs_world, &name, args.as_slice(), uid) {
1386 Err(common_state::plugin::CommandResults::UnknownCommand) => self
1387 .notify_client(
1388 entity,
1389 ServerGeneral::server_msg(
1390 comp::ChatType::CommandError,
1391 common::comp::Content::Plain(format!(
1392 "Unknown command '/{name}'.\nType '/help' for available \
1393 commands",
1394 )),
1395 ),
1396 ),
1397 Ok(value) => {
1398 self.notify_client(
1399 entity,
1400 ServerGeneral::server_msg(
1401 comp::ChatType::CommandInfo,
1402 common::comp::Content::Plain(value.join("\n")),
1403 ),
1404 );
1405 },
1406 Err(common_state::plugin::CommandResults::PluginError(err)) => {
1407 self.notify_client(
1408 entity,
1409 ServerGeneral::server_msg(
1410 comp::ChatType::CommandError,
1411 common::comp::Content::Plain(format!(
1412 "Error occurred while executing command '/{name}'.\n{err}"
1413 )),
1414 ),
1415 );
1416 },
1417 Err(common_state::plugin::CommandResults::HostError(err)) => {
1418 error!(?err, ?name, ?args, "Can't execute command");
1419 self.notify_client(
1420 entity,
1421 ServerGeneral::server_msg(
1422 comp::ChatType::CommandError,
1423 common::comp::Content::Plain(format!(
1424 "Internal error {err:?} while executing '/{name}'.\nContact \
1425 the server administrator",
1426 )),
1427 ),
1428 );
1429 },
1430 }
1431 }
1432 }
1433 }
1434
1435 fn entity_admin_role(&self, entity: EcsEntity) -> Option<comp::AdminRole> {
1436 self.state
1437 .read_component_copied::<comp::Admin>(entity)
1438 .map(|admin| admin.0)
1439 }
1440
1441 pub fn number_of_players(&self) -> i64 {
1442 self.state.ecs().read_storage::<Client>().join().count() as i64
1443 }
1444
1445 pub fn add_admin(&mut self, username: &str, role: comp::AdminRole) {
1448 let mut editable_settings = self.editable_settings_mut();
1449 let login_provider = self.state.ecs().fetch::<LoginProvider>();
1450 let data_dir = self.data_dir();
1451 if let Some(entity) = add_admin(
1452 username,
1453 role,
1454 &login_provider,
1455 &mut editable_settings,
1456 &data_dir.path,
1457 )
1458 .and_then(|uuid| {
1459 let state = &self.state;
1460 (
1461 &state.ecs().entities(),
1462 &state.read_storage::<comp::Player>(),
1463 )
1464 .join()
1465 .find(|(_, player)| player.uuid() == uuid)
1466 .map(|(e, _)| e)
1467 }) {
1468 drop((data_dir, login_provider, editable_settings));
1469 self.state
1472 .write_component_ignore_entity_dead(entity, comp::Admin(role));
1473 };
1474 }
1475
1476 pub fn remove_admin(&self, username: &str) {
1479 let mut editable_settings = self.editable_settings_mut();
1480 let login_provider = self.state.ecs().fetch::<LoginProvider>();
1481 let data_dir = self.data_dir();
1482 if let Some(entity) = remove_admin(
1483 username,
1484 &login_provider,
1485 &mut editable_settings,
1486 &data_dir.path,
1487 )
1488 .and_then(|uuid| {
1489 let state = &self.state;
1490 (
1491 &state.ecs().entities(),
1492 &state.read_storage::<comp::Player>(),
1493 )
1494 .join()
1495 .find(|(_, player)| player.uuid() == uuid)
1496 .map(|(e, _)| e)
1497 }) {
1498 self.state
1500 .ecs()
1501 .write_storage::<comp::Admin>()
1502 .remove(entity);
1503 };
1504 }
1505
1506 #[cfg(feature = "worldgen")]
1511 pub fn create_centered_persister(&mut self, view_distance: u32) {
1512 let world_dims_chunks = self.world.sim().get_size();
1513 let world_dims_blocks = TerrainChunkSize::blocks(world_dims_chunks);
1514 let pos = comp::Pos(Vec3::from(world_dims_blocks.map(|e| e as f32 / 2.0)));
1522 self.state
1523 .create_persister(pos, view_distance, &self.world, &self.index)
1524 .build();
1525 }
1526
1527 pub fn chunks_pending(&mut self) -> bool {
1529 self.state_mut()
1530 .mut_resource::<ChunkGenerator>()
1531 .pending_chunks()
1532 .next()
1533 .is_some()
1534 }
1535
1536 pub fn set_sql_log_mode(&mut self, sql_log_mode: SqlLogMode) {
1538 let mut database_settings = self.database_settings.write().unwrap();
1544 database_settings.sql_log_mode = sql_log_mode;
1545 drop(database_settings);
1548 info!("SQL log mode changed to {:?}", sql_log_mode);
1549 }
1550
1551 pub fn disconnect_all_clients(&mut self) {
1552 info!("Disconnecting all clients due to local console command");
1553 self.disconnect_all_clients_requested = true;
1554 }
1555}
1556
1557impl Drop for Server {
1558 fn drop(&mut self) {
1559 self.state
1560 .notify_players(ServerGeneral::Disconnect(DisconnectReason::Shutdown));
1561
1562 #[cfg(feature = "persistent_world")]
1563 self.state
1564 .ecs()
1565 .try_fetch_mut::<TerrainPersistence>()
1566 .map(|mut terrain_persistence| {
1567 info!("Unloading terrain persistence...");
1568 terrain_persistence.unload_all()
1569 });
1570
1571 #[cfg(feature = "worldgen")]
1572 {
1573 debug!("Saving rtsim state...");
1574 self.state.ecs().write_resource::<rtsim::RtSim>().save(true);
1575 }
1576 }
1577}
1578
1579#[must_use]
1580pub fn handle_edit<T, S: settings::EditableSetting>(
1581 data: T,
1582 result: Option<(String, Result<(), settings::SettingError<S>>)>,
1583) -> Option<T> {
1584 use crate::settings::SettingError;
1585 let (info, result) = result?;
1586 match result {
1587 Ok(()) => {
1588 info!("{}", info);
1589 Some(data)
1590 },
1591 Err(SettingError::Io(err)) => {
1592 warn!(
1593 ?err,
1594 "Failed to write settings file to disk, but succeeded in memory (success message: \
1595 {})",
1596 info,
1597 );
1598 Some(data)
1599 },
1600 Err(SettingError::Integrity(err)) => {
1601 error!(?err, "Encountered an error while validating the request",);
1602 None
1603 },
1604 }
1605}
1606
1607#[must_use]
1612pub fn add_admin(
1613 username: &str,
1614 role: comp::AdminRole,
1615 login_provider: &LoginProvider,
1616 editable_settings: &mut EditableSettings,
1617 data_dir: &std::path::Path,
1618) -> Option<common::uuid::Uuid> {
1619 use crate::settings::EditableSetting;
1620 let role_ = role.into();
1621 match login_provider.username_to_uuid(username) {
1622 Ok(uuid) => handle_edit(
1623 uuid,
1624 editable_settings.admins.edit(data_dir, |admins| {
1625 match admins.insert(uuid, settings::AdminRecord {
1626 username_when_admined: Some(username.into()),
1627 date: chrono::Utc::now(),
1628 role: role_,
1629 }) {
1630 None => Some(format!(
1631 "Successfully added {} ({}) as {:?}!",
1632 username, uuid, role
1633 )),
1634 Some(old_admin) if old_admin.role == role_ => {
1635 info!("{} ({}) already has role: {:?}!", username, uuid, role);
1636 None
1637 },
1638 Some(old_admin) => Some(format!(
1639 "{} ({}) role changed from {:?} to {:?}!",
1640 username, uuid, old_admin.role, role
1641 )),
1642 }
1643 }),
1644 ),
1645 Err(err) => {
1646 error!(
1647 ?err,
1648 "Could not find uuid for this name; either the user does not exist or there was \
1649 an error communicating with the auth server."
1650 );
1651 None
1652 },
1653 }
1654}
1655
1656#[must_use]
1661pub fn remove_admin(
1662 username: &str,
1663 login_provider: &LoginProvider,
1664 editable_settings: &mut EditableSettings,
1665 data_dir: &std::path::Path,
1666) -> Option<common::uuid::Uuid> {
1667 use crate::settings::EditableSetting;
1668 match login_provider.username_to_uuid(username) {
1669 Ok(uuid) => handle_edit(
1670 uuid,
1671 editable_settings.admins.edit(data_dir, |admins| {
1672 if let Some(admin) = admins.remove(&uuid) {
1673 Some(format!(
1674 "Successfully removed {} ({}) with role {:?} from the admins list",
1675 username, uuid, admin.role,
1676 ))
1677 } else {
1678 info!("{} ({}) is not an admin!", username, uuid);
1679 None
1680 }
1681 }),
1682 ),
1683 Err(err) => {
1684 error!(
1685 ?err,
1686 "Could not find uuid for this name; either the user does not exist or there was \
1687 an error communicating with the auth server."
1688 );
1689 None
1690 },
1691 }
1692}