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