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