add webui for veloren-server with working players

Cargo.lock

@@ -7301,11 +7301,13 @@ dependencies = [
  "mimalloc",
  "num_cpus",
  "prometheus",
+ "rand 0.8.5",
  "ratatui",
  "ron",
  "serde",
  "shell-words",
  "signal-hook",
+ "specs",
  "tokio",
  "tracing",
  "veloren-common",

server-cli/Cargo.toml

@@ -47,6 +47,9 @@ tracing = { workspace = true }
 ron = { workspace = true }
 serde = { workspace = true, features = [ "rc", "derive" ]}
 ratatui = { version = "0.26.0", features = ["crossterm"] }
+rand = { workspace = true }
+# ECS
+specs = { workspace = true }
 
 #HTTP
 axum = { version = "0.6.20" }

server-cli/src/cli.rs

@@ -71,6 +71,15 @@ pub enum Message {
     },
     /// Disconnects all connected clients
     DisconnectAllClients,
+    /// returns active player names
+    ListPlayers,
+    ListLogs,
+}
+
+#[derive(Debug, Clone)]
+pub enum MessageReturn {
+    Players(Vec<String>),
+    Logs(Vec<String>),
 }
 
 #[derive(Parser)]

server-cli/src/main.rs

@@ -18,18 +18,21 @@ mod tui_runner;
 mod tuilog;
 mod web;
 use crate::{
-    cli::{Admin, ArgvApp, ArgvCommand, Message, SharedCommand, Shutdown},
+    cli::{
+        Admin, ArgvApp, ArgvCommand, BenchParams, Message, MessageReturn, SharedCommand, Shutdown,
+    },
+    settings::Settings,
     shutdown_coordinator::ShutdownCoordinator,
     tui_runner::Tui,
     tuilog::TuiLog,
 };
-use common::{clock::Clock, consts::MIN_RECOMMENDED_TOKIO_THREADS};
+use common::{clock::Clock, comp::Player, consts::MIN_RECOMMENDED_TOKIO_THREADS};
 use common_base::span;
 use core::sync::atomic::{AtomicUsize, Ordering};
 use server::{persistence::DatabaseSettings, settings::Protocol, Event, Input, Server};
 use std::{
     io,
-    sync::{atomic::AtomicBool, mpsc, Arc},
+    sync::{atomic::AtomicBool, Arc},
     time::{Duration, Instant},
 };
 use tokio::sync::Notify;
@@ -216,12 +219,22 @@ fn main() -> io::Result<()> {
     let metrics_shutdown = Arc::new(Notify::new());
     let metrics_shutdown_clone = Arc::clone(&metrics_shutdown);
     let web_chat_secret = settings.web_chat_secret.clone();
+    let ui_api_secret = settings.ui_api_secret.clone().unwrap_or_else(|| {
+        // when no secret is provided we generate one that we distribute via the /ui
+        // endpoint
+        use rand::distributions::{Alphanumeric, DistString};
+        Alphanumeric.sample_string(&mut rand::thread_rng(), 32)
+    });
+
+    let (web_ui_request_s, web_ui_request_r) = tokio::sync::mpsc::channel(1000);
 
     runtime.spawn(async move {
         web::run(
             registry,
             chat,
             web_chat_secret,
+            ui_api_secret,
+            web_ui_request_s,
             settings.web_address,
             metrics_shutdown_clone.notified(),
         )
@@ -246,15 +259,39 @@ fn main() -> io::Result<()> {
         "Server is ready to accept connections."
     );
 
-    let mut shutdown_coordinator = ShutdownCoordinator::new(Arc::clone(&shutdown_signal));
-
-    // Set up an fps clock
-    let mut clock = Clock::new(Duration::from_secs_f64(1.0 / TPS as f64));
-
     if let Some(bench) = bench {
         #[cfg(feature = "worldgen")]
         server.create_centered_persister(bench.view_distance);
     }
+
+    server_loop(
+        server,
+        bench,
+        settings,
+        tui,
+        web_ui_request_r,
+        shutdown_signal,
+    )?;
+
+    metrics_shutdown.notify_one();
+
+    Ok(())
+}
+
+fn server_loop(
+    mut server: Server,
+    bench: Option<BenchParams>,
+    settings: Settings,
+    tui: Option<Tui>,
+    mut web_ui_request_r: tokio::sync::mpsc::Receiver<(
+        Message,
+        tokio::sync::oneshot::Sender<MessageReturn>,
+    )>,
+    shutdown_signal: Arc<AtomicBool>,
+) -> io::Result<()> {
+    // Set up an fps clock
+    let mut clock = Clock::new(Duration::from_secs_f64(1.0 / TPS as f64));
+    let mut shutdown_coordinator = ShutdownCoordinator::new(Arc::clone(&shutdown_signal));
     let mut bench_exit_time = None;
 
     let mut tick_no = 0u64;
@@ -296,49 +333,91 @@ fn main() -> io::Result<()> {
             trace!(?tick_no, "keepalive")
         }
 
-        if let Some(tui) = tui.as_ref() {
-            match tui.msg_r.try_recv() {
-                Ok(msg) => match msg {
-                    Message::Shutdown {
-                        command: Shutdown::Cancel,
-                    } => shutdown_coordinator.abort_shutdown(&mut server),
-                    Message::Shutdown {
-                        command: Shutdown::Graceful { seconds, reason },
-                    } => {
-                        shutdown_coordinator.initiate_shutdown(
-                            &mut server,
-                            Duration::from_secs(seconds),
-                            reason,
-                        );
-                    },
-                    Message::Shutdown {
-                        command: Shutdown::Immediate,
-                    } => {
-                        info!("Closing the server");
-                        break;
-                    },
-                    Message::Shared(SharedCommand::Admin {
-                        command: Admin::Add { username, role },
-                    }) => {
-                        server.add_admin(&username, role);
-                    },
-                    Message::Shared(SharedCommand::Admin {
-                        command: Admin::Remove { username },
-                    }) => {
-                        server.remove_admin(&username);
-                    },
-                    Message::LoadArea { view_distance } => {
-                        #[cfg(feature = "worldgen")]
-                        server.create_centered_persister(view_distance);
-                    },
-                    Message::SqlLogMode { mode } => {
-                        server.set_sql_log_mode(mode);
-                    },
-                    Message::DisconnectAllClients => {
-                        server.disconnect_all_clients();
-                    },
+        let mut handle_msg = |msg, response: tokio::sync::oneshot::Sender<MessageReturn>| {
+            use specs::{Join, WorldExt};
+            match msg {
+                Message::Shutdown {
+                    command: Shutdown::Cancel,
+                } => shutdown_coordinator.abort_shutdown(&mut server),
+                Message::Shutdown {
+                    command: Shutdown::Graceful { seconds, reason },
+                } => {
+                    shutdown_coordinator.initiate_shutdown(
+                        &mut server,
+                        Duration::from_secs(seconds),
+                        reason,
+                    );
                 },
-                Err(mpsc::TryRecvError::Empty) | Err(mpsc::TryRecvError::Disconnected) => {},
+                Message::Shutdown {
+                    command: Shutdown::Immediate,
+                } => {
+                    return true;
+                },
+                Message::Shared(SharedCommand::Admin {
+                    command: Admin::Add { username, role },
+                }) => {
+                    server.add_admin(&username, role);
+                },
+                Message::Shared(SharedCommand::Admin {
+                    command: Admin::Remove { username },
+                }) => {
+                    server.remove_admin(&username);
+                },
+                Message::LoadArea { view_distance } => {
+                    #[cfg(feature = "worldgen")]
+                    server.create_centered_persister(view_distance);
+                },
+                Message::SqlLogMode { mode } => {
+                    server.set_sql_log_mode(mode);
+                },
+                Message::DisconnectAllClients => {
+                    server.disconnect_all_clients();
+                },
+                Message::ListPlayers => {
+                    let players: Vec<String> = server
+                        .state()
+                        .ecs()
+                        .read_storage::<Player>()
+                        .join()
+                        .map(|p| p.alias.clone())
+                        .collect();
+                    let _ = response.send(MessageReturn::Players(players));
+                },
+                Message::ListLogs => {
+                    let log = LOG.inner.lock().unwrap();
+                    let lines: Vec<_> = log
+                        .lines
+                        .iter()
+                        .rev()
+                        .take(30)
+                        .map(|l| l.to_string())
+                        .collect();
+                    let _ = response.send(MessageReturn::Logs(lines));
+                },
+            }
+            false
+        };
+
+        if let Some(tui) = tui.as_ref() {
+            if let Ok(msg) = tui.msg_r.try_recv() {
+                let (sender, mut recv) = tokio::sync::oneshot::channel();
+                if handle_msg(msg, sender) {
+                    info!("Closing the server");
+                    break;
+                }
+                if let Ok(msg_answ) = recv.try_recv() {
+                    match msg_answ {
+                        MessageReturn::Players(players) => info!("Players: {:?}", players),
+                        MessageReturn::Logs(_) => info!("skipp sending logs to tui"),
+                    };
+                }
+            }
+        }
+
+        if let Ok((msg, sender)) = web_ui_request_r.try_recv() {
+            if handle_msg(msg, sender) {
+                info!("Closing the server");
+                break;
             }
         }
 
@@ -348,7 +427,5 @@ fn main() -> io::Result<()> {
         #[cfg(feature = "tracy")]
         common_base::tracy_client::frame_mark();
     }
-    metrics_shutdown.notify_one();
-
     Ok(())
 }

server-cli/src/settings.rs

@@ -34,6 +34,9 @@ pub struct Settings {
     /// SECRET API HEADER used to access the chat api, if disabled the API is
     /// unreachable
     pub web_chat_secret: Option<String>,
+    /// public SECRET API HEADER used to access the ui api, if disabled the API
+    /// is reachable localhost only
+    pub ui_api_secret: Option<String>,
     pub shutdown_signals: Vec<ShutdownSignal>,
 }
 
@@ -44,6 +47,7 @@ impl Default for Settings {
             update_shutdown_message: "The server is restarting for an update".to_owned(),
             web_address: SocketAddr::from((Ipv4Addr::LOCALHOST, 14005)),
             web_chat_secret: None,
+            ui_api_secret: None,
             shutdown_signals: if cfg!(any(target_os = "linux", target_os = "macos")) {
                 vec![ShutdownSignal::SIGUSR1]
             } else {

server-cli/src/web/mod.rs

@@ -1,3 +1,4 @@
+use crate::web::ui_api::UiRequestSender;
 use axum::{extract::State, response::IntoResponse, routing::get, Router};
 use core::{future::Future, ops::Deref};
 use hyper::{body::Body, header, http, StatusCode};
@@ -6,11 +7,15 @@ use server::chat::ChatCache;
 use std::net::SocketAddr;
 
 mod chat;
+mod ui;
+mod ui_api;
 
 pub async fn run<S, F, R>(
     registry: R,
     cache: ChatCache,
     chat_secret: Option<String>,
+    ui_secret: String,
+    web_ui_request_s: UiRequestSender,
     addr: S,
     shutdown: F,
 ) -> Result<(), hyper::Error>
@@ -25,6 +30,11 @@ where
 
     let app = Router::new()
         .nest("/chat/v1", chat::router(cache, chat_secret))
+        .nest(
+            "/ui_api/v1",
+            ui_api::router(web_ui_request_s, ui_secret.clone()),
+        )
+        .nest("/ui", ui::router(ui_secret))
         .nest("/metrics", metrics)
         .route("/health", get(|| async {}));
 

server-cli/src/web/ui.rs

@@ -0,0 +1,303 @@
+use axum::{
+    extract::{ConnectInfo, State},
+    http::{header::SET_COOKIE, HeaderValue},
+    response::{Html, IntoResponse},
+    routing::get,
+    Router,
+};
+use std::net::SocketAddr;
+
+/// Keep Size small, so we dont have to Clone much for each request.
+#[derive(Clone)]
+struct UiApiToken {
+    secret_token: String,
+}
+
+pub fn router(secret_token: String) -> Router {
+    let token = UiApiToken { secret_token };
+    Router::new().route("/", get(ui)).with_state(token)
+}
+
+async fn ui(
+    ConnectInfo(addr): ConnectInfo<SocketAddr>,
+    State(token): State<UiApiToken>,
+) -> impl IntoResponse {
+    if !addr.ip().is_loopback() {
+        return Html(
+            r#"<!DOCTYPE html>
+<html>
+<body>
+Ui is only accissable from 127.0.0.1
+</body>
+</html>
+        "#
+            .to_string(),
+        )
+        .into_response();
+    }
+
+    let mut response = Html(format!(
+        r#"<!DOCTYPE html>
+<html>
+<head>
+<script type="text/javascript">
+{}
+</script>
+<style>
+{}
+</style>
+</head>
+<body>
+
+{}
+
+</body>
+</html>"#,
+        javascript(),
+        css(),
+        inner()
+    ))
+    .into_response();
+
+    let cookie = format!("X-Secret-Token={}; SameSite=Strict", token.secret_token);
+
+    response.headers_mut().insert(
+        SET_COOKIE,
+        HeaderValue::from_str(&cookie).expect("An invalid secret-token for ui was provided"),
+    );
+    response
+}
+
+fn inner() -> &'static str {
+    r##"
+<div class="tab">
+  <button class="tablinks active" onclick="openTab(event, 'settings')">Settings</button>
+  <button class="tablinks" onclick="openTab(event, 'logs')">Logs</button>
+  <button class="tablinks" onclick="openTab(event, 'players')">Players</button>
+  <button class="tablinks" onclick="openTab(event, 'access')">Access</button>
+</div>
+
+<div id="settings" class="tabcontent">
+  <div class="flex-container">
+    <div class="first">
+      <p>Server Name:</p>
+    </div>
+    <div>
+      <p><input type="text" id="server_name" value="Veloren Alpha"></input></p>
+    </div>
+  </div>
+
+  <div class="flex-container">
+    <div class="first">
+      <p>Server Ip/Port:</p>
+    </div>
+    <div>
+      <p><input type="text" id="server_ip_port" value="0.0.0.0:14004"></input></p>
+    </div>
+  </div>
+
+  <div class="flex-container">
+    <div class="first">
+      <p>Require Auth:</p>
+    </div>
+    <div>
+      <p><input type="checkbox" checked="checked"></input> Enabled </p>
+    </div>
+  </div>
+
+  <div class="flex-container">
+    <div class="first">
+      <p>Player Limit:</p>
+    </div>
+    <div>
+      <p id="player_limit_no">20</p>
+      <p><input type="range" min="1" max="100" value="20" class="slider" id="player_limit" oninput="changeSlider(event, 'player_limit', 'player_limit_no')"></p>
+    </div>
+  </div>
+
+  <div class="flex-container">
+    <div class="first">
+      <p>Max View Distance:</p>
+    </div>
+    <div>
+      <p id="view_distance_no">30</p>
+      <p><input type="range" min="1" max="100" value="30" class="slider" id="view_distance" oninput="changeSlider(event, 'view_distance', 'view_distance_no')"></p>
+    </div>
+  </div>
+
+  <div class="flex-container">
+    <div class="first">
+      <p>Global PvP:</p>
+    </div>
+    <div>
+      <p><input type="checkbox"></input> Enable </p>
+    </div>
+  </div>
+
+  <div class="flex-container">
+    <div class="first">
+      <p>Experimental Terrain Persistence:</p>
+    </div>
+    <div>
+      <p><input type="checkbox"></input> Enable </p>
+    </div>
+  </div>
+</div>
+
+<div id="logs" class="tabcontent">
+  <h3>Server Logs</h3>
+  <div id="logs_list"></div>
+</div>
+
+<div id="players" class="tabcontent">
+  <h3>Players</h3>
+  <ul id="players_list">
+  </ul>
+</div>
+
+<div id="access" class="tabcontent">
+  <h3>Whitelist</h3>
+  <h3>Banlist</h3>
+  <h3>Admin</h3>
+</div>
+    "##
+}
+
+fn javascript() -> &'static str {
+    r#"
+function openTab(evt, cityName) {
+  // Declare all variables
+  var i, tabcontent, tablinks;
+
+  // Get all elements with class="tabcontent" and hide them
+  tabcontent = document.getElementsByClassName("tabcontent");
+  for (i = 0; i < tabcontent.length; i++) {
+    tabcontent[i].style.display = "none";
+  }
+
+  // Get all elements with class="tablinks" and remove the class "active"
+  tablinks = document.getElementsByClassName("tablinks");
+  for (i = 0; i < tablinks.length; i++) {
+    tablinks[i].className = tablinks[i].className.replace(" active", "");
+  }
+
+  // Show the current tab, and add an "active" class to the button that opened the tab
+  document.getElementById(cityName).style.display = "block";
+  evt.currentTarget.className += " active";
+}
+
+function changeSlider(evt, sliderId, showId) {
+    var slider = document.getElementById(sliderId);
+    var sliderNo = document.getElementById(showId);
+    sliderNo.innerHTML = slider.value;
+}
+
+async function update_players() {
+    const players_response = await fetch("/ui_api/v1/players");
+    const players = await players_response.json();
+
+    // remove no longer existing childs
+    var players_list = document.getElementById("players_list");
+    for (var i = players_list.children.length-1; i >= 0; i--) {
+      if (!players.includes(players_list.children[i].innerText)) {
+        console.log("remove player: " + players_list.children[i].innerText);
+        players_list.removeChild(players_list.children[i]);
+        i--;
+      }
+    }
+
+    // add non-existing elements
+    addloop: for (const player of players) {
+      for (var i = 0; i < players_list.children.length; i++) {
+        if (players_list.children[i].innerText == player) {
+          continue addloop;
+        }
+      }
+
+      var li = document.createElement("li");
+      li.appendChild(document.createTextNode(player));
+      players_list.appendChild(li);
+
+      console.log("added player: " + player);
+    }
+}
+
+async function update_logs() {
+    const logs_response = await fetch("/ui_api/v1/logs");
+    const logs = await logs_response.json();
+
+    // remove no longer existing childs
+    var logs_list = document.getElementById("logs_list");
+    while (logs_list.lastElementChild) {
+      logs_list.removeChild(logs_list.lastElementChild);
+    }
+
+    for (const log of logs) {
+      var p = document.createElement("p");
+      p.appendChild(document.createTextNode(log));
+      logs_list.appendChild(p);
+    }
+}
+
+async function loop() {
+    await update_players();
+    await update_logs();
+}
+
+var loopId = window.setInterval(loop, 1000);
+    "#
+}
+
+fn css() -> &'static str {
+    r#"
+ /* Style the tab */
+.tab {
+  overflow: hidden;
+  border: 1px solid #ccc;
+  background-color: #f1f1f1;
+}
+
+/* Style the buttons that are used to open the tab content */
+.tab button {
+  background-color: inherit;
+  float: left;
+  border: none;
+  outline: none;
+  cursor: pointer;
+  padding: 14px 16px;
+  transition: 0.3s;
+}
+
+/* Change background color of buttons on hover */
+.tab button:hover {
+  background-color: #ddd;
+}
+
+/* Create an active/current tablink class */
+.tab button.active {
+  background-color: #ccc;
+}
+
+/* Style the tab content */
+.tabcontent {
+  display: none;
+  padding: 6px 12px;
+  border: 1px solid #ccc;
+  border-top: none;
+}
+div#settings.tabcontent  {
+  display: block;
+}
+
+.flex-container {
+  display: flex;
+  margin: 4px;
+  padding: 4px;
+  background-color: #f1f1f1;
+}
+
+.flex-container .first {
+    width: 300px
+}
+    "#
+}

server-cli/src/web/ui_api.rs

@@ -0,0 +1,103 @@
+use crate::cli::{Message, MessageReturn};
+use axum::{
+    extract::{ConnectInfo, State},
+    http::header::COOKIE,
+    middleware::Next,
+    response::{IntoResponse, Response},
+    routing::get,
+    Json, Router,
+};
+use hyper::{Request, StatusCode};
+use std::{
+    collections::HashSet,
+    net::{IpAddr, SocketAddr},
+    sync::Arc,
+};
+use tokio::sync::Mutex;
+
+/// Keep Size small, so we dont have to Clone much for each request.
+#[derive(Clone)]
+struct UiApiToken {
+    secret_token: String,
+}
+
+pub(crate) type UiRequestSender =
+    tokio::sync::mpsc::Sender<(Message, tokio::sync::oneshot::Sender<MessageReturn>)>;
+
+#[derive(Clone, Default)]
+struct IpAddresses {
+    users: Arc<Mutex<HashSet<IpAddr>>>,
+}
+
+async fn validate_secret<B>(
+    State(token): State<UiApiToken>,
+    req: Request<B>,
+    next: Next<B>,
+) -> Result<Response, StatusCode> {
+    let session_cookie = req.headers().get(COOKIE).ok_or(StatusCode::UNAUTHORIZED)?;
+
+    pub const X_SECRET_TOKEN: &str = "X-Secret-Token";
+    let expected = format!("{X_SECRET_TOKEN}={}", token.secret_token);
+
+    if session_cookie.as_bytes() != expected.as_bytes() {
+        return Err(StatusCode::UNAUTHORIZED);
+    }
+
+    Ok(next.run(req).await)
+}
+
+/// Logs each new IP address that accesses this API authenticated
+async fn log_users<B>(
+    State(ip_addresses): State<IpAddresses>,
+    ConnectInfo(addr): ConnectInfo<SocketAddr>,
+    req: Request<B>,
+    next: Next<B>,
+) -> Result<Response, StatusCode> {
+    let mut ip_addresses = ip_addresses.users.lock().await;
+    let ip_addr = addr.ip();
+    if !ip_addresses.contains(&ip_addr) {
+        ip_addresses.insert(ip_addr);
+        let users_so_far = ip_addresses.len();
+        tracing::info!(?ip_addr, ?users_so_far, "Is accessing the /ui_api endpoint");
+    }
+    Ok(next.run(req).await)
+}
+
+pub fn router(web_ui_request_s: UiRequestSender, secret_token: String) -> Router {
+    let token = UiApiToken { secret_token };
+    let ip_addrs = IpAddresses::default();
+    Router::new()
+        .route("/players", get(players))
+        .route("/logs", get(logs))
+        .layer(axum::middleware::from_fn_with_state(ip_addrs, log_users))
+        .layer(axum::middleware::from_fn_with_state(token, validate_secret))
+        .with_state(web_ui_request_s)
+}
+
+async fn players(
+    State(web_ui_request_s): State<UiRequestSender>,
+) -> Result<impl IntoResponse, StatusCode> {
+    let (sender, receiver) = tokio::sync::oneshot::channel();
+    let _ = web_ui_request_s.send((Message::ListPlayers, sender)).await;
+    match receiver
+        .await
+        .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?
+    {
+        MessageReturn::Players(players) => Ok(Json(players)),
+        _ => Err(StatusCode::INTERNAL_SERVER_ERROR),
+    }
+}
+
+async fn logs(
+    State(web_ui_request_s): State<UiRequestSender>,
+) -> Result<impl IntoResponse, StatusCode> {
+    let (sender, receiver) = tokio::sync::oneshot::channel();
+    let _ = web_ui_request_s.send((Message::ListLogs, sender)).await;
+    match receiver
+        .await
+        .map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?
+    {
+        MessageReturn::Logs(logs) => Ok(Json(logs)),
+        _ => Err(StatusCode::INTERNAL_SERVER_ERROR),
+    }
+}