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