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, option_zip, const_type_name, slice_partition_dedup)]
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, RepositionToFreeSpace},
62 state_ext::StateExt,
63 sys::sentinel::DeletedEntities,
64};
65use authc::Uuid;
66use censor::Censor;
67#[cfg(not(feature = "worldgen"))]
68use common::grid::Grid;
69#[cfg(feature = "worldgen")]
70use common::terrain::CoordinateConversions;
71#[cfg(feature = "worldgen")]
72use common::terrain::TerrainChunkSize;
73use common::{
74 assets::AssetExt,
75 calendar::Calendar,
76 character::{CharacterId, CharacterItem},
77 cmd::ServerChatCommand,
78 comp::{self, ChatType, Content},
79 event::{
80 ClientDisconnectEvent, ClientDisconnectWithoutPersistenceEvent, EventBus, ExitIngameEvent,
81 UpdateCharacterDataEvent,
82 },
83 link::Is,
84 mounting::{Volume, VolumeRider},
85 region::RegionMap,
86 resources::{BattleMode, GameMode, Time, TimeOfDay},
87 shared_server_config::ServerConstants,
88 slowjob::SlowJobPool,
89 terrain::TerrainChunk,
90 uid::Uid,
91 vol::RectRasterableVol,
92};
93use common_base::prof_span;
94use common_ecs::run_now;
95use common_net::{
96 msg::{ClientType, DisconnectReason, PlayerListUpdate, ServerGeneral, ServerInfo, ServerMsg},
97 sync::WorldSyncExt,
98};
99use common_state::{AreasContainer, BattleModeChangeArea, BlockDiff, BuildArea, State};
100use common_systems::add_local_systems;
101use metrics::{EcsSystemMetrics, GameplayMetrics, PhysicsMetrics, TickMetrics};
102use network::{ListenAddr, Network, Pid};
103use persistence::{
104 character_loader::{CharacterLoader, CharacterUpdaterMessage},
105 character_updater::CharacterUpdater,
106};
107use prometheus::Registry;
108use rustls::pki_types::{CertificateDer, PrivateKeyDer};
109use settings::banlist::NormalizedIpAddr;
110use specs::{
111 Builder, Entity as EcsEntity, Entity, Join, LendJoin, WorldExt, shred::SendDispatcher,
112};
113use std::{
114 ops::{Deref, DerefMut},
115 sync::{Arc, Mutex},
116 time::{Duration, Instant},
117};
118#[cfg(not(feature = "worldgen"))]
119use test_world::{IndexOwned, World};
120use tokio::runtime::Runtime;
121use tracing::{debug, error, info, trace, warn};
122use vek::*;
123use veloren_query_server::server::QueryServer;
124pub use world::{WorldGenerateStage, civ::WorldCivStage, sim::WorldSimStage};
125
126use crate::{
127 persistence::{DatabaseSettings, SqlLogMode},
128 sys::terrain,
129};
130use hashbrown::HashMap;
131use std::sync::RwLock;
132
133use crate::settings::Protocol;
134
135#[cfg(feature = "plugins")]
136use {
137 common::uid::IdMaps,
138 common_state::plugin::{PluginMgr, memory_manager::EcsWorld},
139};
140
141use crate::{chat::ChatCache, persistence::character_loader::CharacterScreenResponseKind};
142use common::comp::Anchor;
143#[cfg(feature = "worldgen")]
144pub use world::{
145 IndexOwned, World,
146 sim::{DEFAULT_WORLD_MAP, DEFAULT_WORLD_SEED, FileOpts, GenOpts, WorldOpts},
147};
148
149const BATTLE_MODE_COOLDOWN: f64 = 60.0 * 5.0;
154
155#[derive(Copy, Clone)]
159pub struct SpawnPoint(pub Vec3<f32>);
160
161impl Default for SpawnPoint {
162 fn default() -> Self { Self(Vec3::new(0.0, 0.0, 256.0)) }
163}
164
165pub const MIN_VD: u32 = 6;
170
171#[derive(Copy, Clone, Default)]
174pub struct Tick(u64);
175
176#[derive(Clone)]
177pub struct HwStats {
178 hardware_threads: u32,
179 rayon_threads: u32,
180}
181
182#[derive(Clone, Copy, PartialEq)]
183enum DisconnectType {
184 WithPersistence,
185 WithoutPersistence,
186}
187
188#[derive(Copy, Clone)]
190pub struct TickStart(Instant);
191
192#[derive(Clone, Default, Debug)]
194pub struct BattleModeBuffer {
195 map: HashMap<CharacterId, (BattleMode, Time)>,
196}
197
198impl BattleModeBuffer {
199 pub fn push(&mut self, char_id: CharacterId, save: (BattleMode, Time)) {
200 self.map.insert(char_id, save);
201 }
202
203 pub fn get(&self, char_id: &CharacterId) -> Option<&(BattleMode, Time)> {
204 self.map.get(char_id)
205 }
206
207 pub fn pop(&mut self, char_id: &CharacterId) -> Option<(BattleMode, Time)> {
208 self.map.remove(char_id)
209 }
210}
211
212pub struct RecentClientIPs {
215 pub last_addrs: schnellru::LruMap<Uuid, NormalizedIpAddr>,
216}
217
218impl Default for RecentClientIPs {
219 fn default() -> Self {
220 Self {
221 last_addrs: schnellru::LruMap::new(schnellru::ByLength::new(1000)),
222 }
223 }
224}
225
226pub struct ChunkRequest {
227 entity: EcsEntity,
228 key: Vec2<i32>,
229}
230
231#[derive(Debug)]
232pub enum ServerInitStage {
233 DbMigrations,
234 DbVacuum,
235 WorldGen(WorldGenerateStage),
236 StartingSystems,
237}
238
239pub struct Server {
240 state: State,
241 world: Arc<World>,
242 index: IndexOwned,
243
244 connection_handler: ConnectionHandler,
245
246 runtime: Arc<Runtime>,
247
248 metrics_registry: Arc<Registry>,
249 chat_cache: ChatCache,
250 database_settings: Arc<RwLock<DatabaseSettings>>,
251 disconnect_all_clients_requested: bool,
252
253 event_dispatcher: SendDispatcher<'static>,
254}
255
256impl Server {
257 pub fn new(
259 settings: Settings,
260 editable_settings: EditableSettings,
261 database_settings: DatabaseSettings,
262 data_dir: &std::path::Path,
263 report_stage: &(dyn Fn(ServerInitStage) + Send + Sync),
264 runtime: Arc<Runtime>,
265 ) -> Result<Self, Error> {
266 prof_span!("Server::new");
267 info!("Server data dir is: {}", data_dir.display());
268 if settings.auth_server_address.is_none() {
269 info!("Authentication is disabled");
270 }
271
272 report_stage(ServerInitStage::DbMigrations);
273 debug!("Running DB migrations...");
275 persistence::run_migrations(&database_settings);
276
277 report_stage(ServerInitStage::DbVacuum);
278 debug!("Vacuuming database...");
280 persistence::vacuum_database(&database_settings);
281
282 let database_settings = Arc::new(RwLock::new(database_settings));
283
284 let registry = Arc::new(Registry::new());
285 let chunk_gen_metrics = metrics::ChunkGenMetrics::new(®istry).unwrap();
286 let job_metrics = metrics::JobMetrics::new(®istry).unwrap();
287 let network_request_metrics = metrics::NetworkRequestMetrics::new(®istry).unwrap();
288 let player_metrics = metrics::PlayerMetrics::new(®istry).unwrap();
289 let ecs_system_metrics = EcsSystemMetrics::new(®istry).unwrap();
290 let tick_metrics = TickMetrics::new(®istry).unwrap();
291 let physics_metrics = PhysicsMetrics::new(®istry).unwrap();
292 let server_event_metrics = metrics::ServerEventMetrics::new(®istry).unwrap();
293 let gameplay_metrics = GameplayMetrics::new(®istry).unwrap();
294 let query_server_metrics = metrics::QueryServerMetrics::new(®istry).unwrap();
295
296 let battlemode_buffer = BattleModeBuffer::default();
297
298 let pools = State::pools(GameMode::Server);
299
300 #[cfg(feature = "plugins")]
302 let plugin_mgr = PluginMgr::from_asset_or_default();
303
304 debug!("Generating world, seed: {}", settings.world_seed);
305 #[cfg(feature = "worldgen")]
306 let (world, index) = World::generate(
307 settings.world_seed,
308 WorldOpts {
309 seed_elements: true,
310 world_file: if let Some(ref opts) = settings.map_file {
311 opts.clone()
312 } else {
313 FileOpts::LoadAsset(DEFAULT_WORLD_MAP.into())
315 },
316 calendar: Some(settings.calendar_mode.calendar_now()),
317 },
318 &pools,
319 &|stage| {
320 report_stage(ServerInitStage::WorldGen(stage));
321 },
322 );
323 #[cfg(not(feature = "worldgen"))]
324 let (world, index) = World::generate(settings.world_seed);
325
326 #[cfg(feature = "worldgen")]
327 let map = world.get_map_data(index.as_index_ref(), &pools);
328 #[cfg(not(feature = "worldgen"))]
329 let map = common_net::msg::WorldMapMsg {
330 dimensions_lg: Vec2::zero(),
331 max_height: 1.0,
332 rgba: Grid::new(Vec2::new(1, 1), 1),
333 horizons: [(vec![0], vec![0]), (vec![0], vec![0])],
334 alt: Grid::new(Vec2::new(1, 1), 1),
335 sites: Vec::new(),
336 possible_starting_sites: Vec::new(),
337 pois: Vec::new(),
338 default_chunk: Arc::new(world.generate_oob_chunk()),
339 };
340
341 #[cfg(feature = "worldgen")]
342 let map_size_lg = world.sim().map_size_lg();
343 #[cfg(not(feature = "worldgen"))]
344 let map_size_lg = world.map_size_lg();
345
346 let lod = lod::Lod::from_world(&world, index.as_index_ref(), &pools);
347
348 report_stage(ServerInitStage::StartingSystems);
349
350 let mut state = State::server(
351 Arc::clone(&pools),
352 map_size_lg,
353 Arc::clone(&map.default_chunk),
354 |dispatcher_builder| {
355 add_local_systems(dispatcher_builder);
356 sys::msg::add_server_systems(dispatcher_builder);
357 sys::add_server_systems(dispatcher_builder);
358 #[cfg(feature = "worldgen")]
359 {
360 rtsim::add_server_systems(dispatcher_builder);
361 weather::add_server_systems(dispatcher_builder);
362 }
363 },
364 #[cfg(feature = "plugins")]
365 plugin_mgr,
366 );
367 events::register_event_busses(state.ecs_mut());
368 state.ecs_mut().insert(battlemode_buffer);
369 state.ecs_mut().insert(RecentClientIPs::default());
370 state.ecs_mut().insert(settings.clone());
371 state.ecs_mut().insert(editable_settings);
372 state.ecs_mut().insert(DataDir {
373 path: data_dir.to_owned(),
374 });
375
376 state.ecs_mut().insert(Vec::<ChunkRequest>::new());
377 state
378 .ecs_mut()
379 .insert(EventBus::<chunk_serialize::ChunkSendEntry>::default());
380 state.ecs_mut().insert(Locations::default());
381 state.ecs_mut().insert(LoginProvider::new(
382 settings.auth_server_address.clone(),
383 Arc::clone(&runtime),
384 ));
385 state.ecs_mut().insert(HwStats {
386 hardware_threads: num_cpus::get() as u32,
387 rayon_threads: num_cpus::get() as u32,
388 });
389 state.ecs_mut().insert(ServerConstants {
390 day_cycle_coefficient: settings.day_cycle_coefficient(),
391 });
392 state.ecs_mut().insert(Tick(0));
393 state.ecs_mut().insert(TickStart(Instant::now()));
394 state.ecs_mut().insert(job_metrics);
395 state.ecs_mut().insert(network_request_metrics);
396 state.ecs_mut().insert(player_metrics);
397 state.ecs_mut().insert(ecs_system_metrics);
398 state.ecs_mut().insert(tick_metrics);
399 state.ecs_mut().insert(physics_metrics);
400 state.ecs_mut().insert(server_event_metrics);
401 state.ecs_mut().insert(gameplay_metrics);
402 state.ecs_mut().insert(query_server_metrics);
403 if settings.experimental_terrain_persistence {
404 #[cfg(feature = "persistent_world")]
405 {
406 warn!(
407 "Experimental terrain persistence support is enabled. This feature may break, \
408 be disabled, or otherwise change under your feet at *any time*. \
409 Additionally, it is expected to be replaced in the future *without* \
410 migration or warning. You have been warned."
411 );
412 state
413 .ecs_mut()
414 .insert(TerrainPersistence::new(data_dir.to_owned()));
415 }
416 #[cfg(not(feature = "persistent_world"))]
417 error!(
418 "Experimental terrain persistence support was requested, but the server was not \
419 compiled with the feature. Terrain modifications will *not* be persisted."
420 );
421 }
422 {
423 let pool = state.ecs_mut().write_resource::<SlowJobPool>();
424 pool.configure("CHUNK_DROP", |_n| 1);
425 pool.configure("CHUNK_GENERATOR", |n| n / 2 + n / 4);
426 pool.configure("CHUNK_SERIALIZER", |n| n / 2);
427 pool.configure("RTSIM_SAVE", |_| 1);
428 pool.configure("WEATHER", |_| 1);
429 }
430 state
431 .ecs_mut()
432 .insert(ChunkGenerator::new(chunk_gen_metrics));
433 {
434 let (sender, receiver) =
435 crossbeam_channel::bounded::<chunk_serialize::SerializedChunk>(10_000);
436 state.ecs_mut().insert(sender);
437 state.ecs_mut().insert(receiver);
438 }
439
440 state.ecs_mut().insert(CharacterUpdater::new(
441 Arc::<RwLock<DatabaseSettings>>::clone(&database_settings),
442 )?);
443
444 let ability_map = comp::item::tool::AbilityMap::<comp::AbilityItem>::load_expect_cloned(
445 "common.abilities.ability_set_manifest",
446 );
447 state.ecs_mut().insert(ability_map);
448
449 let msm = comp::inventory::item::MaterialStatManifest::load().cloned();
450 state.ecs_mut().insert(msm);
451
452 let rbm = common::recipe::RecipeBookManifest::load().cloned();
453 state.ecs_mut().insert(rbm);
454
455 state.ecs_mut().insert(CharacterLoader::new(
456 Arc::<RwLock<DatabaseSettings>>::clone(&database_settings),
457 )?);
458
459 state
461 .ecs_mut()
462 .insert(sys::PersistenceScheduler::every(Duration::from_secs(10)));
463
464 state.ecs_mut().insert(RegionMap::new());
466
467 state.ecs_mut().register::<RegionSubscription>();
469 state.ecs_mut().register::<Client>();
470 state.ecs_mut().register::<comp::Presence>();
471 state.ecs_mut().register::<wiring::WiringElement>();
472 state.ecs_mut().register::<wiring::Circuit>();
473 state.ecs_mut().register::<Anchor>();
474 state.ecs_mut().register::<comp::Pet>();
475 state.ecs_mut().register::<login_provider::PendingLogin>();
476 state.ecs_mut().register::<RepositionToFreeSpace>();
477 state.ecs_mut().register::<common::rtsim::ActorId>();
478
479 let banned_words = settings.moderation.load_banned_words(data_dir);
481 let censor = Arc::new(Censor::Custom(banned_words.into_iter().collect()));
482 state.ecs_mut().insert(Arc::clone(&censor));
483
484 state
486 .ecs_mut()
487 .insert(AutoMod::new(&settings.moderation, censor));
488
489 state.ecs_mut().insert(map);
490
491 #[cfg(feature = "worldgen")]
492 let spawn_point = SpawnPoint({
493 let index = index.as_index_ref();
494 let center_chunk = world.sim().map_size_lg().chunks().map(i32::from) / 2;
501 let spawn_chunk = world
502 .civs()
503 .sites()
504 .filter(|site| site.is_settlement())
505 .map(|site| site.center)
506 .min_by_key(|site_pos| site_pos.distance_squared(center_chunk))
507 .unwrap_or(center_chunk);
508
509 world.find_accessible_pos(index, TerrainChunkSize::center_wpos(spawn_chunk), false)
510 });
511 #[cfg(not(feature = "worldgen"))]
512 let spawn_point = SpawnPoint::default();
513
514 state.ecs_mut().insert(spawn_point);
516
517 {
520 #[cfg(feature = "worldgen")]
521 let size = world.sim().get_size();
522 #[cfg(not(feature = "worldgen"))]
523 let size = world.map_size_lg().chunks().map(u32::from);
524
525 let world_size = size.map(|e| e as i32) * TerrainChunk::RECT_SIZE.map(|e| e as i32);
526 let world_aabb = Aabb {
527 min: Vec3::new(0, 0, -32768),
528 max: Vec3::new(world_size.x, world_size.y, 32767),
529 }
530 .made_valid();
531
532 state
533 .ecs()
534 .write_resource::<AreasContainer<BuildArea>>()
535 .insert("world".to_string(), world_aabb)
536 .expect("The initial insert should always work.");
537 }
538
539 let world = Arc::new(world);
541 state.ecs_mut().insert(Arc::clone(&world));
542 state.ecs_mut().insert(lod);
543 state.ecs_mut().insert(index.clone());
544
545 state.ecs_mut().write_resource::<TimeOfDay>().0 = settings.world.start_time;
547
548 sys::sentinel::UpdateTrackers::register(state.ecs_mut());
550
551 state.ecs_mut().insert(DeletedEntities::default());
552
553 let network = Network::new_with_registry(Pid::new(), &runtime, ®istry, 1 << 20);
556 let (chat_cache, chat_tracker) = ChatCache::new(Duration::from_secs(60), &runtime);
557 state.ecs_mut().insert(chat_tracker);
558
559 let mut printed_quic_warning = false;
560 for protocol in &settings.gameserver_protocols {
561 match protocol {
562 Protocol::Tcp { address } => {
563 runtime.block_on(network.listen(ListenAddr::Tcp(*address)))?;
564 },
565 Protocol::Quic {
566 address,
567 cert_file_path,
568 key_file_path,
569 } => {
570 use rustls_pemfile::Item;
571 use std::fs;
572
573 match || -> Result<_, Box<dyn std::error::Error>> {
574 let key = fs::read(key_file_path)?;
575 let key = if key_file_path.extension().is_some_and(|x| x == "der") {
576 PrivateKeyDer::try_from(key).map_err(|_| "No valid pem key in file")?
577 } else {
578 debug!("convert pem key to der");
579 rustls_pemfile::read_all(&mut key.as_slice())
580 .find_map(|item| match item {
581 Ok(Item::Pkcs1Key(v)) => Some(PrivateKeyDer::Pkcs1(v)),
582 Ok(Item::Pkcs8Key(v)) => Some(PrivateKeyDer::Pkcs8(v)),
583 Ok(Item::Sec1Key(v)) => Some(PrivateKeyDer::Sec1(v)),
584 Ok(Item::Crl(_)) => None,
585 Ok(Item::Csr(_)) => None,
586 Ok(Item::X509Certificate(_)) => None,
587 Ok(_) => None,
588 Err(e) => {
589 tracing::warn!(?e, "error while reading key_file");
590 None
591 },
592 })
593 .ok_or("No valid pem key in file")?
594 };
595 let cert_chain = fs::read(cert_file_path)?;
596 let cert_chain = if cert_file_path.extension().is_some_and(|x| x == "der") {
597 vec![CertificateDer::from(cert_chain)]
598 } else {
599 debug!("convert pem cert to der");
600 rustls_pemfile::certs(&mut cert_chain.as_slice())
601 .filter_map(|item| match item {
602 Ok(cert) => Some(cert),
603 Err(e) => {
604 tracing::warn!(?e, "error while reading cert_file");
605 None
606 },
607 })
608 .collect()
609 };
610 let server_config = quinn::ServerConfig::with_single_cert(cert_chain, key)?;
611 Ok(server_config)
612 }() {
613 Ok(server_config) => {
614 runtime.block_on(
615 network.listen(ListenAddr::Quic(*address, server_config.clone())),
616 )?;
617
618 if !printed_quic_warning {
619 warn!(
620 "QUIC is enabled. This is experimental and not recommended in \
621 production"
622 );
623 printed_quic_warning = true;
624 }
625 },
626 Err(e) => {
627 error!(
628 ?e,
629 "Failed to load the TLS certificate, running without QUIC {}",
630 *address
631 );
632 },
633 }
634 },
635 }
636 }
637
638 if let Some(addr) = settings.query_address {
639 use veloren_query_server::proto::ServerInfo;
640
641 const QUERY_SERVER_RATELIMIT: u16 = 120;
642
643 let (query_server_info_tx, query_server_info_rx) =
644 tokio::sync::watch::channel(ServerInfo {
645 git_hash: *common::util::GIT_HASH,
646 git_timestamp: *common::util::GIT_TIMESTAMP,
647 players_count: 0,
648 player_cap: settings.max_players,
649 battlemode: settings.gameplay.battle_mode.into(),
650 });
651 let mut query_server =
652 QueryServer::new(addr, query_server_info_rx, QUERY_SERVER_RATELIMIT);
653 let query_server_metrics =
654 Arc::new(Mutex::new(veloren_query_server::server::Metrics::default()));
655 let query_server_metrics2 = Arc::clone(&query_server_metrics);
656 runtime.spawn(async move {
657 let err = query_server.run(query_server_metrics2).await.err();
658 error!(?err, "Query server stopped unexpectedly");
659 });
660 state.ecs_mut().insert(query_server_info_tx);
661 state.ecs_mut().insert(query_server_metrics);
662 }
663
664 runtime.block_on(network.listen(ListenAddr::Mpsc(14004)))?;
665
666 let connection_handler = ConnectionHandler::new(network, &runtime);
667
668 #[cfg(feature = "worldgen")]
670 {
671 match rtsim::RtSim::new(
672 &settings.world,
673 index.as_index_ref(),
674 &world,
675 data_dir.to_owned(),
676 ) {
677 Ok(rtsim) => {
678 state.ecs_mut().insert(rtsim.state().data().time_of_day);
679 state.ecs_mut().insert(rtsim);
680 },
681 Err(err) => {
682 error!("Failed to load rtsim: {}", err);
683 return Err(Error::RtsimError(err));
684 },
685 }
686 weather::init(&mut state);
687 }
688
689 let this = Self {
690 state,
691 world,
692 index,
693 connection_handler,
694 runtime,
695
696 metrics_registry: registry,
697 chat_cache,
698 database_settings,
699 disconnect_all_clients_requested: false,
700
701 event_dispatcher: Self::create_event_dispatcher(pools),
702 };
703
704 debug!(?settings, "created veloren server with");
705
706 info!("Server version: {}", *common::util::DISPLAY_VERSION);
707
708 Ok(this)
709 }
710
711 pub fn get_server_info(&self) -> ServerInfo {
712 let settings = self.state.ecs().fetch::<Settings>();
713
714 ServerInfo {
715 name: settings.server_name.clone(),
716 git_hash: *common::util::GIT_HASH,
717 git_timestamp: *common::util::GIT_TIMESTAMP,
718 auth_provider: settings.auth_server_address.clone(),
719 }
720 }
721
722 pub fn settings(&self) -> impl Deref<Target = Settings> + '_ {
724 self.state.ecs().fetch::<Settings>()
725 }
726
727 pub fn settings_mut(&self) -> impl DerefMut<Target = Settings> + '_ {
729 self.state.ecs().fetch_mut::<Settings>()
730 }
731
732 pub fn editable_settings_mut(&self) -> impl DerefMut<Target = EditableSettings> + '_ {
734 self.state.ecs().fetch_mut::<EditableSettings>()
735 }
736
737 pub fn editable_settings(&self) -> impl Deref<Target = EditableSettings> + '_ {
739 self.state.ecs().fetch::<EditableSettings>()
740 }
741
742 pub fn data_dir(&self) -> impl Deref<Target = DataDir> + '_ {
744 self.state.ecs().fetch::<DataDir>()
745 }
746
747 pub fn state(&self) -> &State { &self.state }
749
750 pub fn state_mut(&mut self) -> &mut State { &mut self.state }
752
753 pub fn world(&self) -> &World { &self.world }
755
756 pub fn metrics_registry(&self) -> &Arc<Registry> { &self.metrics_registry }
758
759 pub fn chat_cache(&self) -> &ChatCache { &self.chat_cache }
761
762 fn get_location_names(
764 &self,
765 character_list: Vec<CharacterItem<Vec3<f32>>>,
766 ) -> Vec<CharacterItem<Content>> {
767 character_list
768 .into_iter()
769 .map(|c| {
770 #[expect(deprecated, reason = "i18n location name")]
771 let name = c
772 .location
773 .as_ref()
774 .and_then(|wpos| {
775 self.world
776 .get_location_name(self.index.as_index_ref(), wpos.xy().as_::<i32>())
777 })
778 .map(Content::legacy);
779 CharacterItem {
780 character: c.character,
781 body: c.body,
782 hardcore: c.hardcore,
783 inventory: c.inventory,
784 location: name,
785 }
786 })
787 .collect()
788 }
789
790 pub fn tick(&mut self, _input: Input, dt: Duration) -> Result<Vec<Event>, Error> {
793 self.state.ecs().write_resource::<Tick>().0 += 1;
794 self.state.ecs().write_resource::<TickStart>().0 = Instant::now();
795
796 let new_calendar = self
800 .state
801 .ecs()
802 .read_resource::<Settings>()
803 .calendar_mode
804 .calendar_now();
805 *self.state.ecs_mut().write_resource::<Calendar>() = new_calendar;
806
807 #[cfg(feature = "hot-site")]
808 if let Ok(lib) = world::LIB.lock()
809 && let Some(lib) = &*lib
810 {
811 static LAST_COUNT: std::sync::atomic::AtomicU64 = std::sync::atomic::AtomicU64::new(0);
812 let last_count = LAST_COUNT.load(std::sync::atomic::Ordering::Relaxed);
813
814 let new_count = lib.reload_count();
815
816 if new_count > last_count {
817 LAST_COUNT.store(new_count, std::sync::atomic::Ordering::Relaxed);
818
819 let count = cmd::reload_chunks_inner(self, Vec3::zero(), None, true);
820
821 tracing::info!("Reloaded {count} chunks");
822 }
823 }
824
825 let mut frontend_events = Vec::new();
848
849 let before_new_connections = Instant::now();
852
853 self.handle_new_connections(&mut frontend_events);
855
856 let before_state_tick = Instant::now();
857
858 fn on_block_update(ecs: &specs::World, changes: Vec<BlockDiff>) {
859 if changes
861 .iter()
862 .any(|c| c.old.get_rtsim_resource() != c.new.get_rtsim_resource())
863 {
864 ecs.write_resource::<rtsim::RtSim>().hook_block_update(
865 &ecs.read_resource::<Arc<world::World>>(),
866 ecs.read_resource::<world::IndexOwned>().as_index_ref(),
867 changes,
868 );
869 }
870 }
871
872 let mut state_tick_metrics = Default::default();
876 let server_constants = (*self.state.ecs().read_resource::<ServerConstants>()).clone();
877 self.state.tick(
878 dt,
879 false,
880 Some(&mut state_tick_metrics),
881 &server_constants,
882 on_block_update,
883 );
884
885 let before_handle_events = Instant::now();
886
887 let disconnect_type = self.disconnect_all_clients_if_requested();
890
891 self.state.maintain_links();
893
894 frontend_events.append(&mut self.handle_events());
896
897 let before_update_terrain_and_regions = Instant::now();
898
899 self.update_region_map();
904 self.state.apply_terrain_changes(on_block_update);
907
908 let before_sync = Instant::now();
909
910 sys::run_sync_systems(self.state.ecs_mut());
912
913 let before_world_tick = Instant::now();
914
915 self.world.tick(dt);
917
918 let before_entity_cleanup = Instant::now();
919
920 if let Some(DisconnectType::WithoutPersistence) = disconnect_type {
925 run_now::<terrain::Sys>(self.state.ecs_mut());
926 }
927
928 #[cfg(feature = "worldgen")]
930 {
931 let mut rtsim = self.state.ecs().write_resource::<rtsim::RtSim>();
932 let world = self.state.ecs().read_resource::<Arc<World>>();
933 for chunk in &self.state.terrain_changes().removed_chunks {
934 rtsim.hook_unload_chunk(*chunk, &world);
935 }
936 }
937
938 let anchors = self.state.ecs().read_storage::<Anchor>();
949 let anchored_anchor_entities: Vec<Entity> = (
950 &self.state.ecs().entities(),
951 &self.state.ecs().read_storage::<Anchor>(),
952 )
953 .join()
954 .filter_map(|(_, anchor)| match anchor {
955 Anchor::Entity(anchor_entity) => Some(*anchor_entity),
956 _ => None,
957 })
958 .filter(|anchor_entity| match anchors.get(*anchor_entity) {
962 Some(Anchor::Entity(_)) => true,
963 Some(Anchor::Chunk(_)) | None => false
964 })
965 .collect();
966 drop(anchors);
967
968 for entity in anchored_anchor_entities {
969 if cfg!(debug_assertions) {
970 panic!("Entity anchor chain detected");
971 }
972 error!(
973 "Detected an anchor entity that itself has an anchor entity - anchor chains are \
974 not currently supported. The entity's Anchor component has been deleted"
975 );
976 self.state.delete_component::<Anchor>(entity);
977 }
978
979 let to_delete = {
982 let terrain = self.state.terrain();
983 (
984 &self.state.ecs().entities(),
985 &self.state.ecs().read_storage::<comp::Pos>(),
986 !&self.state.ecs().read_storage::<comp::Presence>(),
987 self.state.ecs().read_storage::<Anchor>().maybe(),
988 self.state.ecs().read_storage::<Is<VolumeRider>>().maybe(),
989 )
990 .join()
991 .filter(|(_, pos, _, anchor, is_volume_rider)| {
992 let pos = is_volume_rider
993 .and_then(|is_volume_rider| match is_volume_rider.pos.kind {
994 Volume::Terrain => None,
995 Volume::Entity(e) => {
996 let e = self.state.ecs().entity_from_uid(e)?;
997 let pos = self
998 .state
999 .ecs()
1000 .read_storage::<comp::Pos>()
1001 .get(e)
1002 .copied()?;
1003
1004 Some(pos.0)
1005 },
1006 })
1007 .unwrap_or(pos.0);
1008 let chunk_key = terrain.pos_key(pos.map(|e| e.floor() as i32));
1009 match anchor {
1010 Some(Anchor::Chunk(hc)) => {
1011 terrain.get_key_real(chunk_key).is_none()
1016 && terrain.get_key_real(*hc).is_none()
1017 },
1018 Some(Anchor::Entity(entity)) => !self.state.ecs().is_alive(*entity),
1019 None => terrain.get_key_real(chunk_key).is_none(),
1020 }
1021 })
1022 .map(|(entity, _, _, _, _)| entity)
1023 .collect::<Vec<_>>()
1024 };
1025
1026 #[cfg(feature = "worldgen")]
1027 {
1028 let mut rtsim = self.state.ecs().write_resource::<rtsim::RtSim>();
1029 let rtsim_actors = self.state.ecs().read_storage();
1030 for entity in &to_delete {
1031 if let Some(actor) = rtsim_actors.get(*entity) {
1032 rtsim.hook_rtsim_entity_unload(*actor);
1033 }
1034 }
1035 }
1036
1037 for entity in to_delete {
1039 if let Err(e) = self.state.delete_entity_recorded(entity) {
1040 error!(?e, "Failed to delete agent outside the terrain");
1041 }
1042 }
1043
1044 if let Some(DisconnectType::WithoutPersistence) = disconnect_type {
1045 info!(
1046 "Disconnection of all players without persistence complete, signalling to \
1047 persistence thread that character updates may continue to be processed"
1048 );
1049 self.state
1050 .ecs()
1051 .fetch_mut::<CharacterUpdater>()
1052 .disconnected_success();
1053 }
1054
1055 let before_persistence_updates = Instant::now();
1057
1058 let character_loader = self.state.ecs().read_resource::<CharacterLoader>();
1059
1060 let mut character_updater = self.state.ecs().write_resource::<CharacterUpdater>();
1061 let updater_messages: Vec<CharacterUpdaterMessage> = character_updater.messages().collect();
1062
1063 character_loader
1065 .messages()
1066 .chain(updater_messages)
1067 .for_each(|message| match message {
1068 CharacterUpdaterMessage::DatabaseBatchCompletion(batch_id) => {
1069 character_updater.process_batch_completion(batch_id);
1070 },
1071 CharacterUpdaterMessage::CharacterScreenResponse(response) => {
1072 match response.response_kind {
1073 CharacterScreenResponseKind::CharacterList(result) => match result {
1074 Ok(list) => self.notify_client(
1075 response.target_entity,
1076 ServerGeneral::CharacterListUpdate(self.get_location_names(list)),
1077 ),
1078 Err(error) => self.notify_client(
1079 response.target_entity,
1080 ServerGeneral::CharacterActionError(error.to_string()),
1081 ),
1082 },
1083 CharacterScreenResponseKind::CharacterCreation(result) => match result {
1084 Ok((character_id, list)) => {
1085 self.notify_client(
1086 response.target_entity,
1087 ServerGeneral::CharacterListUpdate(
1088 self.get_location_names(list),
1089 ),
1090 );
1091 self.notify_client(
1092 response.target_entity,
1093 ServerGeneral::CharacterCreated(character_id),
1094 );
1095 },
1096 Err(error) => self.notify_client(
1097 response.target_entity,
1098 ServerGeneral::CharacterActionError(error.to_string()),
1099 ),
1100 },
1101 CharacterScreenResponseKind::CharacterEdit(result) => match result {
1102 Ok((character_id, list)) => {
1103 self.notify_client(
1104 response.target_entity,
1105 ServerGeneral::CharacterListUpdate(
1106 self.get_location_names(list),
1107 ),
1108 );
1109 self.notify_client(
1110 response.target_entity,
1111 ServerGeneral::CharacterEdited(character_id),
1112 );
1113 },
1114 Err(error) => self.notify_client(
1115 response.target_entity,
1116 ServerGeneral::CharacterActionError(error.to_string()),
1117 ),
1118 },
1119 CharacterScreenResponseKind::CharacterData(result) => {
1120 match *result {
1121 Ok((character_data, skill_set_persistence_load_error)) => {
1122 let PersistedComponents {
1123 body,
1124 hardcore,
1125 stats,
1126 skill_set,
1127 inventory,
1128 waypoint,
1129 pets,
1130 active_abilities,
1131 map_marker,
1132 } = character_data;
1133 let character_data = (
1134 body,
1135 hardcore,
1136 stats,
1137 skill_set,
1138 inventory,
1139 waypoint,
1140 pets,
1141 active_abilities,
1142 map_marker,
1143 );
1144 self.state.emit_event_now(UpdateCharacterDataEvent {
1147 entity: response.target_entity,
1148 components: character_data,
1149 metadata: skill_set_persistence_load_error,
1150 })
1151 },
1152 Err(error) => {
1153 self.notify_client(
1157 response.target_entity,
1158 ServerGeneral::CharacterDataLoadResult(Err(
1159 error.to_string()
1160 )),
1161 );
1162
1163 self.state.emit_event_now(ExitIngameEvent {
1165 entity: response.target_entity,
1166 })
1167 },
1168 }
1169 },
1170 }
1171 },
1172 });
1173
1174 drop(character_loader);
1175 drop(character_updater);
1176
1177 {
1178 let index = &mut self.index;
1183 let world = &mut self.world;
1184 let ecs = self.state.ecs_mut();
1185 let slow_jobs = ecs.write_resource::<SlowJobPool>();
1186
1187 index.reload_if_changed(|index| {
1188 let mut chunk_generator = ecs.write_resource::<ChunkGenerator>();
1189 let client = ecs.read_storage::<Client>();
1190 let mut terrain = ecs.write_resource::<common::terrain::TerrainGrid>();
1191 #[cfg(feature = "worldgen")]
1192 let rtsim = ecs.read_resource::<rtsim::RtSim>();
1193 #[cfg(not(feature = "worldgen"))]
1194 let rtsim = ();
1195
1196 chunk_generator.cancel_all();
1198
1199 if client.is_empty() {
1200 terrain.clear();
1202 } else {
1203 terrain.iter().for_each(|(pos, _)| {
1205 chunk_generator.generate_chunk(
1206 None,
1207 pos,
1208 &slow_jobs,
1209 Arc::clone(world),
1210 &rtsim,
1211 index.clone(),
1212 (
1213 *ecs.read_resource::<TimeOfDay>(),
1214 (*ecs.read_resource::<Calendar>()).clone(),
1215 ),
1216 );
1217 });
1218 }
1219 });
1220 }
1221
1222 let end_of_server_tick = Instant::now();
1223
1224 run_now::<sys::metrics::Sys>(self.state.ecs());
1226
1227 {
1228 let tick_metrics = self.state.ecs().read_resource::<TickMetrics>();
1230
1231 let tt = &tick_metrics.tick_time;
1232 tt.with_label_values(&["new connections"])
1233 .set((before_state_tick - before_new_connections).as_nanos() as i64);
1234 tt.with_label_values(&["handle server events"])
1235 .set((before_update_terrain_and_regions - before_handle_events).as_nanos() as i64);
1236 tt.with_label_values(&["update terrain and region map"])
1237 .set((before_sync - before_update_terrain_and_regions).as_nanos() as i64);
1238 tt.with_label_values(&["state"])
1239 .set((before_handle_events - before_state_tick).as_nanos() as i64);
1240 tt.with_label_values(&["world tick"])
1241 .set((before_entity_cleanup - before_world_tick).as_nanos() as i64);
1242 tt.with_label_values(&["entity cleanup"])
1243 .set((before_persistence_updates - before_entity_cleanup).as_nanos() as i64);
1244 tt.with_label_values(&["persistence_updates"])
1245 .set((end_of_server_tick - before_persistence_updates).as_nanos() as i64);
1246 for (label, duration) in state_tick_metrics.timings {
1247 tick_metrics
1248 .state_tick_time
1249 .with_label_values(&[label])
1250 .set(duration.as_nanos() as i64);
1251 }
1252 tick_metrics.tick_time_hist.observe(
1253 end_of_server_tick
1254 .duration_since(before_state_tick)
1255 .as_secs_f64(),
1256 );
1257 }
1258
1259 Ok(frontend_events)
1262 }
1263
1264 pub fn cleanup(&mut self) {
1266 self.state.cleanup();
1268
1269 #[cfg(feature = "persistent_world")]
1271 self.state
1272 .ecs()
1273 .try_fetch_mut::<TerrainPersistence>()
1274 .map(|mut t| t.maintain());
1275 }
1276
1277 fn update_region_map(&mut self) {
1279 prof_span!("Server::update_region_map");
1280 let ecs = self.state().ecs();
1281 ecs.write_resource::<RegionMap>().tick(
1282 ecs.read_storage::<comp::Pos>(),
1283 ecs.read_storage::<comp::Vel>(),
1284 ecs.read_storage::<comp::Presence>(),
1285 ecs.entities(),
1286 );
1287 }
1288
1289 fn initialize_client(&mut self, client: connection_handler::IncomingClient) -> Entity {
1290 let entity = self
1291 .state
1292 .ecs_mut()
1293 .create_entity_synced()
1294 .with(client)
1295 .build();
1296 self.state
1297 .ecs()
1298 .read_resource::<metrics::PlayerMetrics>()
1299 .clients_connected
1300 .inc();
1301 entity
1302 }
1303
1304 fn disconnect_all_clients_if_requested(&mut self) -> Option<DisconnectType> {
1308 let mut character_updater = self.state.ecs().fetch_mut::<CharacterUpdater>();
1309
1310 let disconnect_type = self.get_disconnect_all_clients_requested(&mut character_updater);
1311 if let Some(disconnect_type) = disconnect_type {
1312 let with_persistence = disconnect_type == DisconnectType::WithPersistence;
1313 let clients = self.state.ecs().read_storage::<Client>();
1314 let entities = self.state.ecs().entities();
1315
1316 info!(
1317 "Disconnecting all clients ({} persistence) as requested",
1318 if with_persistence { "with" } else { "without" }
1319 );
1320 for (_, entity) in (&clients, &entities).join() {
1321 info!("Emitting client disconnect event for entity: {:?}", entity);
1322 if with_persistence {
1323 self.state.emit_event_now(ClientDisconnectEvent(
1324 entity,
1325 comp::DisconnectReason::Kicked,
1326 ))
1327 } else {
1328 self.state
1329 .emit_event_now(ClientDisconnectWithoutPersistenceEvent(entity))
1330 };
1331 }
1332
1333 self.disconnect_all_clients_requested = false;
1334 }
1335
1336 disconnect_type
1337 }
1338
1339 fn get_disconnect_all_clients_requested(
1340 &self,
1341 character_updater: &mut CharacterUpdater,
1342 ) -> Option<DisconnectType> {
1343 let without_persistence_requested = character_updater.disconnect_all_clients_requested();
1344 let with_persistence_requested = self.disconnect_all_clients_requested;
1345
1346 if without_persistence_requested {
1347 return Some(DisconnectType::WithoutPersistence);
1348 };
1349 if with_persistence_requested {
1350 return Some(DisconnectType::WithPersistence);
1351 };
1352 None
1353 }
1354
1355 fn handle_new_connections(&mut self, frontend_events: &mut Vec<Event>) {
1357 while let Ok(sender) = self.connection_handler.info_requester_receiver.try_recv() {
1358 trace!("sending info to connection_handler");
1360 let _ = sender.send(connection_handler::ServerInfoPacket {
1361 info: self.get_server_info(),
1362 time: self.state.get_time(),
1363 });
1364 }
1365
1366 while let Ok(incoming) = self.connection_handler.client_receiver.try_recv() {
1367 let entity = self.initialize_client(incoming);
1368 frontend_events.push(Event::ClientConnected { entity });
1369 }
1370 }
1371
1372 pub fn notify_client<S>(&self, entity: EcsEntity, msg: S)
1373 where
1374 S: Into<ServerMsg>,
1375 {
1376 if let Some(client) = self.state.ecs().read_storage::<Client>().get(entity) {
1377 client.send_fallible(msg);
1378 }
1379 }
1380
1381 pub fn notify_players(&mut self, msg: ServerGeneral) { self.state.notify_players(msg); }
1382
1383 fn process_command(&mut self, entity: EcsEntity, name: String, args: Vec<String>) {
1384 if let Ok(command) = name.parse::<ServerChatCommand>() {
1386 command.execute(self, entity, args);
1387 } else {
1388 #[cfg(feature = "plugins")]
1389 {
1390 let mut plugin_manager = self.state.ecs().write_resource::<PluginMgr>();
1391 let ecs_world = EcsWorld {
1392 entities: &self.state.ecs().entities(),
1393 health: self.state.ecs().read_component().into(),
1394 uid: self.state.ecs().read_component().into(),
1395 id_maps: &self.state.ecs().read_resource::<IdMaps>().into(),
1396 player: self.state.ecs().read_component().into(),
1397 };
1398 let uid = if let Some(uid) = ecs_world.uid.get(entity).copied() {
1399 uid
1400 } else {
1401 self.notify_client(
1402 entity,
1403 ServerGeneral::server_msg(
1404 comp::ChatType::CommandError,
1405 common::comp::Content::Plain(
1406 "Can't get player UUID (player may be disconnected?)".to_string(),
1407 ),
1408 ),
1409 );
1410 return;
1411 };
1412 match plugin_manager.command_event(&ecs_world, &name, args.as_slice(), uid) {
1413 Err(common_state::plugin::CommandResults::UnknownCommand) => self
1414 .notify_client(
1415 entity,
1416 ServerGeneral::server_msg(
1417 comp::ChatType::CommandError,
1418 common::comp::Content::Plain(format!(
1419 "Unknown command '/{name}'.\nType '/help' for available \
1420 commands",
1421 )),
1422 ),
1423 ),
1424 Ok(value) => {
1425 self.notify_client(
1426 entity,
1427 ServerGeneral::server_msg(
1428 comp::ChatType::CommandInfo,
1429 common::comp::Content::Plain(value.join("\n")),
1430 ),
1431 );
1432 },
1433 Err(common_state::plugin::CommandResults::PluginError(err)) => {
1434 self.notify_client(
1435 entity,
1436 ServerGeneral::server_msg(
1437 comp::ChatType::CommandError,
1438 common::comp::Content::Plain(format!(
1439 "Error occurred while executing command '/{name}'.\n{err}"
1440 )),
1441 ),
1442 );
1443 },
1444 Err(common_state::plugin::CommandResults::HostError(err)) => {
1445 error!(?err, ?name, ?args, "Can't execute command");
1446 self.notify_client(
1447 entity,
1448 ServerGeneral::server_msg(
1449 comp::ChatType::CommandError,
1450 common::comp::Content::Plain(format!(
1451 "Internal error {err:?} while executing '/{name}'.\nContact \
1452 the server administrator",
1453 )),
1454 ),
1455 );
1456 },
1457 }
1458 }
1459 }
1460 }
1461
1462 fn entity_admin_role(&self, entity: EcsEntity) -> Option<comp::AdminRole> {
1463 self.state
1464 .read_component_copied::<comp::Admin>(entity)
1465 .map(|admin| admin.0)
1466 }
1467
1468 pub fn number_of_players(&self) -> i64 {
1469 self.state.ecs().read_storage::<Client>().join().count() as i64
1470 }
1471
1472 pub fn add_admin(&mut self, username: &str, role: comp::AdminRole) {
1475 let mut editable_settings = self.editable_settings_mut();
1476 let login_provider = self.state.ecs().fetch::<LoginProvider>();
1477 let data_dir = self.data_dir();
1478 if let Some(entity) = add_admin(
1479 username,
1480 role,
1481 &login_provider,
1482 &mut editable_settings,
1483 &data_dir.path,
1484 )
1485 .and_then(|uuid| {
1486 let state = &self.state;
1487 (
1488 &state.ecs().entities(),
1489 &state.read_storage::<comp::Player>(),
1490 )
1491 .join()
1492 .find(|(_, player)| player.uuid() == uuid)
1493 .map(|(e, _)| e)
1494 }) {
1495 drop((data_dir, login_provider, editable_settings));
1496 self.state
1499 .write_component_ignore_entity_dead(entity, comp::Admin(role));
1500 };
1501 }
1502
1503 pub fn remove_admin(&self, username: &str) {
1506 let mut editable_settings = self.editable_settings_mut();
1507 let login_provider = self.state.ecs().fetch::<LoginProvider>();
1508 let data_dir = self.data_dir();
1509 if let Some(entity) = remove_admin(
1510 username,
1511 &login_provider,
1512 &mut editable_settings,
1513 &data_dir.path,
1514 )
1515 .and_then(|uuid| {
1516 let state = &self.state;
1517 (
1518 &state.ecs().entities(),
1519 &state.read_storage::<comp::Player>(),
1520 )
1521 .join()
1522 .find(|(_, player)| player.uuid() == uuid)
1523 .map(|(e, _)| e)
1524 }) {
1525 self.state
1527 .ecs()
1528 .write_storage::<comp::Admin>()
1529 .remove(entity);
1530 };
1531 }
1532
1533 #[cfg(feature = "worldgen")]
1538 pub fn create_centered_persister(&mut self, view_distance: u32) {
1539 let world_dims_chunks = self.world.sim().get_size();
1540 let world_dims_blocks = TerrainChunkSize::blocks(world_dims_chunks);
1541 let pos = comp::Pos(Vec3::from(world_dims_blocks.map(|e| e as f32 / 2.0)));
1549 self.state
1550 .create_persister(pos, view_distance, &self.world, &self.index)
1551 .build();
1552 }
1553
1554 pub fn chunks_pending(&mut self) -> bool {
1556 self.state_mut()
1557 .mut_resource::<ChunkGenerator>()
1558 .pending_chunks()
1559 .next()
1560 .is_some()
1561 }
1562
1563 pub fn set_sql_log_mode(&mut self, sql_log_mode: SqlLogMode) {
1565 let mut database_settings = self.database_settings.write().unwrap();
1571 database_settings.sql_log_mode = sql_log_mode;
1572 drop(database_settings);
1575 info!("SQL log mode changed to {:?}", sql_log_mode);
1576 }
1577
1578 pub fn disconnect_all_clients(&mut self) {
1579 info!("Disconnecting all clients due to local console command");
1580 self.disconnect_all_clients_requested = true;
1581 }
1582
1583 pub fn get_battle_mode_for(&mut self, client: EcsEntity) {
1589 let ecs = self.state.ecs();
1590 let time = ecs.read_resource::<Time>();
1591 let settings = ecs.read_resource::<Settings>();
1592 let players = ecs.read_storage::<comp::Player>();
1593 let get_player_result = players.get(client).ok_or_else(|| {
1594 error!("Can't get player component for client.");
1595
1596 Content::Plain("Can't get player component for client.".to_string())
1597 });
1598 let player = match get_player_result {
1599 Ok(player) => player,
1600 Err(content) => {
1601 self.notify_client(
1602 client,
1603 ServerGeneral::server_msg(ChatType::CommandError, content),
1604 );
1605 return;
1606 },
1607 };
1608
1609 let mut msg = format!("Current battle mode: {:?}.", player.battle_mode);
1610
1611 if settings.gameplay.battle_mode.allow_choosing() {
1612 msg.push_str(" Possible to change.");
1613 } else {
1614 msg.push_str(" Global.");
1615 }
1616
1617 if let Some(change) = player.last_battlemode_change {
1618 let Time(time) = *time;
1619 let Time(change) = change;
1620 let elapsed = time - change;
1621 let next = BATTLE_MODE_COOLDOWN - elapsed;
1622
1623 if next > 0.0 {
1624 let notice = format!(" Next change will be available in: {:.0} seconds", next);
1625 msg.push_str(¬ice);
1626 }
1627 }
1628
1629 self.notify_client(
1630 client,
1631 ServerGeneral::server_msg(ChatType::CommandInfo, Content::Plain(msg)),
1632 );
1633 }
1634
1635 pub fn set_battle_mode_for(&mut self, client: EcsEntity, battle_mode: BattleMode) {
1641 let ecs = self.state.ecs();
1642 let time = ecs.read_resource::<Time>();
1643 let settings = ecs.read_resource::<Settings>();
1644
1645 if !settings.gameplay.battle_mode.allow_choosing() {
1646 self.notify_client(
1647 client,
1648 ServerGeneral::server_msg(
1649 ChatType::CommandInfo,
1650 Content::localized("command-disabled-by-settings"),
1651 ),
1652 );
1653
1654 return;
1655 }
1656
1657 let pos = if let Some(pos) = self
1658 .state
1659 .ecs()
1660 .read_storage::<comp::Pos>()
1661 .get(client)
1662 .copied()
1663 {
1664 pos
1665 } else {
1666 self.notify_client(
1667 client,
1668 ServerGeneral::server_msg(
1669 ChatType::CommandInfo,
1670 Content::localized_with_args("command-position-unavailable", [(
1671 "target", "target",
1672 )]),
1673 ),
1674 );
1675
1676 return;
1677 };
1678
1679 #[cfg(feature = "worldgen")]
1680 let in_town = {
1681 let wpos = pos.0.xy().map(|x| x as i32);
1682 let chunk_pos = wpos.wpos_to_cpos();
1683 self.world.civs().sites().any(|site| {
1684 const RADIUS: f32 = 9.0;
1686 let delta = site
1687 .center
1688 .map(|x| x as f32)
1689 .distance(chunk_pos.map(|x| x as f32));
1690 delta < RADIUS
1691 })
1692 };
1693
1694 #[cfg(not(feature = "worldgen"))]
1695 let in_town = true;
1696
1697 let in_battlemode_change = {
1698 let areas = self
1699 .state
1700 .ecs()
1701 .read_resource::<AreasContainer<BattleModeChangeArea>>();
1702
1703 areas
1704 .areas()
1705 .iter()
1706 .any(|(_id, aabb)| aabb.contains_point(pos.0.as_()))
1707 };
1708
1709 if !in_town && !in_battlemode_change {
1710 self.notify_client(
1711 client,
1712 ServerGeneral::server_msg(
1713 ChatType::CommandInfo,
1714 Content::localized("command-battlemode-intown"),
1715 ),
1716 );
1717
1718 return;
1719 }
1720
1721 let mut players = ecs.write_storage::<comp::Player>();
1722 let mut player = if let Some(info) = players.get_mut(client) {
1723 info
1724 } else {
1725 error!("Failed to get info for player.");
1726
1727 return;
1728 };
1729
1730 if let Some(Time(last_change)) = player.last_battlemode_change {
1731 let Time(time) = *time;
1732 let elapsed = time - last_change;
1733 if elapsed < BATTLE_MODE_COOLDOWN {
1734 let next = BATTLE_MODE_COOLDOWN - elapsed;
1735
1736 self.notify_client(
1737 client,
1738 ServerGeneral::server_msg(
1739 ChatType::CommandInfo,
1740 Content::Plain(format!(
1741 "Next change will be available in {next:.0} seconds."
1742 )),
1743 ),
1744 );
1745
1746 return;
1747 }
1748 }
1749
1750 if player.battle_mode == battle_mode {
1751 self.notify_client(
1752 client,
1753 ServerGeneral::server_msg(
1754 ChatType::CommandInfo,
1755 Content::localized("command-battlemode-same"),
1756 ),
1757 );
1758
1759 return;
1760 }
1761
1762 player.battle_mode = battle_mode;
1763 player.last_battlemode_change = Some(*time);
1764
1765 self.notify_client(
1766 client,
1767 ServerGeneral::server_msg(
1768 ChatType::CommandInfo,
1769 Content::localized_with_args("command-battlemode-updated", [(
1770 "battlemode",
1771 format!("{battle_mode:?}"),
1772 )]),
1773 ),
1774 );
1775
1776 drop(players);
1777
1778 let uid = ecs.read_storage::<Uid>().get(client).copied().unwrap();
1779
1780 self.state().notify_players(ServerGeneral::PlayerListUpdate(
1781 PlayerListUpdate::UpdateBattleMode(uid, battle_mode),
1782 ));
1783 }
1784}
1785
1786impl Drop for Server {
1787 fn drop(&mut self) {
1788 self.state
1789 .notify_players(ServerGeneral::Disconnect(DisconnectReason::Shutdown));
1790
1791 #[cfg(feature = "persistent_world")]
1792 self.state
1793 .ecs()
1794 .try_fetch_mut::<TerrainPersistence>()
1795 .map(|mut terrain_persistence| {
1796 info!("Unloading terrain persistence...");
1797 terrain_persistence.unload_all()
1798 });
1799
1800 #[cfg(feature = "worldgen")]
1801 {
1802 debug!("Saving rtsim state...");
1803 self.state.ecs().write_resource::<rtsim::RtSim>().save(true);
1804 }
1805 }
1806}
1807
1808#[must_use]
1809pub fn handle_edit<T, S: settings::EditableSetting>(
1810 data: T,
1811 result: Option<(String, Result<(), settings::SettingError<S>>)>,
1812) -> Option<T> {
1813 use crate::settings::SettingError;
1814 let (info, result) = result?;
1815 match result {
1816 Ok(()) => {
1817 info!("{}", info);
1818 Some(data)
1819 },
1820 Err(SettingError::Io(err)) => {
1821 warn!(
1822 ?err,
1823 "Failed to write settings file to disk, but succeeded in memory (success message: \
1824 {})",
1825 info,
1826 );
1827 Some(data)
1828 },
1829 Err(SettingError::Integrity(err)) => {
1830 error!(?err, "Encountered an error while validating the request",);
1831 None
1832 },
1833 }
1834}
1835
1836#[must_use]
1841pub fn add_admin(
1842 username: &str,
1843 role: comp::AdminRole,
1844 login_provider: &LoginProvider,
1845 editable_settings: &mut EditableSettings,
1846 data_dir: &std::path::Path,
1847) -> Option<common::uuid::Uuid> {
1848 use crate::settings::EditableSetting;
1849 let role_ = role.into();
1850 match login_provider.username_to_uuid(username) {
1851 Ok(uuid) => handle_edit(
1852 uuid,
1853 editable_settings.admins.edit(data_dir, |admins| {
1854 match admins.insert(uuid, settings::AdminRecord {
1855 username_when_admined: Some(username.into()),
1856 date: chrono::Utc::now(),
1857 role: role_,
1858 }) {
1859 None => Some(format!(
1860 "Successfully added {} ({}) as {:?}!",
1861 username, uuid, role
1862 )),
1863 Some(old_admin) if old_admin.role == role_ => {
1864 info!("{} ({}) already has role: {:?}!", username, uuid, role);
1865 None
1866 },
1867 Some(old_admin) => Some(format!(
1868 "{} ({}) role changed from {:?} to {:?}!",
1869 username, uuid, old_admin.role, role
1870 )),
1871 }
1872 }),
1873 ),
1874 Err(err) => {
1875 error!(
1876 ?err,
1877 "Could not find uuid for this name; either the user does not exist or there was \
1878 an error communicating with the auth server."
1879 );
1880 None
1881 },
1882 }
1883}
1884
1885#[must_use]
1890pub fn remove_admin(
1891 username: &str,
1892 login_provider: &LoginProvider,
1893 editable_settings: &mut EditableSettings,
1894 data_dir: &std::path::Path,
1895) -> Option<common::uuid::Uuid> {
1896 use crate::settings::EditableSetting;
1897 match login_provider.username_to_uuid(username) {
1898 Ok(uuid) => handle_edit(
1899 uuid,
1900 editable_settings.admins.edit(data_dir, |admins| {
1901 if let Some(admin) = admins.remove(&uuid) {
1902 Some(format!(
1903 "Successfully removed {} ({}) with role {:?} from the admins list",
1904 username, uuid, admin.role,
1905 ))
1906 } else {
1907 info!("{} ({}) is not an admin!", username, uuid);
1908 None
1909 }
1910 }),
1911 ),
1912 Err(err) => {
1913 error!(
1914 ?err,
1915 "Could not find uuid for this name; either the user does not exist or there was \
1916 an error communicating with the auth server."
1917 );
1918 None
1919 },
1920 }
1921}