Skip to main content

veloren_server/
lib.rs

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