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