Adding an online user count to main page.

- Fixes #274

2 files changed, 6 insertions, 1 deletions

diff --git a/server/src/api/site.rs b/server/src/api/site.rs
index b517568e..912378ab 100644
--- a/server/src/api/site.rs
+++ b/server/src/api/site.rs
@@ -81,6 +81,7 @@ pub struct GetSiteResponse {
   site: Option<SiteView>,
   admins: Vec<UserView>,
   banned: Vec<UserView>,
+  pub online: usize,
 }
 
 #[derive(Serialize, Deserialize)]
@@ -296,6 +297,7 @@ impl Perform<GetSiteResponse> for Oper<GetSite> {
       site: site_view,
       admins: admins,
       banned: banned,
+      online: 0
     })
   }
 }
@@ -486,6 +488,7 @@ impl Perform<GetSiteResponse> for Oper<TransferSite> {
       site: Some(site_view),
       admins: admins,
       banned: banned,
+      online: 0
     })
   }
 }
diff --git a/server/src/websocket/server.rs b/server/src/websocket/server.rs
index 08714c30..41bb1058 100644
--- a/server/src/websocket/server.rs
+++ b/server/src/websocket/server.rs
@@ -492,8 +492,10 @@ fn parse_json_message(chat: &mut ChatServer, msg: StandardMessage) -> Result<Str
       Ok(serde_json::to_string(&res)?)
     }
     UserOperation::GetSite => {
+      let online: usize = chat.sessions.len();
       let get_site: GetSite = serde_json::from_str(data)?;
-      let res = Oper::new(user_operation, get_site).perform()?;
+      let mut res = Oper::new(user_operation, get_site).perform()?;
+      res.online = online;
       Ok(serde_json::to_string(&res)?)
     }
     UserOperation::Search => {