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