feat: generate readable initial user name

packages/server/Cargo.toml

@@ -26,3 +26,4 @@ include_dir = "0.7"
 mime = "0.3"
 mime_guess = "2"
 sled = "0.34"
+woothee = "0.13"

packages/server/src/controllers.rs

@@ -22,12 +22,13 @@ pub struct UploadQuery {
 
 #[post("/file/upload")]
 pub async fn file_upload(
+    req: HttpRequest,
     payload: web::Payload,
     query: web::Query<UploadQuery>,
     file_manager: web::Data<FileManager>,
     session: Session,
 ) -> ResponseResult<UserFile> {
-    let user = User::get_from_session(&session).map_err(anyhow::Error::from)?;
+    let user = User::get_from_session(&session, &req).map_err(anyhow::Error::from)?;
     let file = file_manager
         .add_from_stream(payload, query.filename.to_string(), user.id)
         .await?;
@@ -47,18 +48,19 @@ pub async fn file_download(
 }
 
 #[get("/user-info")]
-pub async fn user_info(session: Session) -> ResponseResult<User> {
-    let user = User::get_from_session(&session).map_err(anyhow::Error::from)?;
+pub async fn user_info(req: HttpRequest, session: Session) -> ResponseResult<User> {
+    let user = User::get_from_session(&session, &req).map_err(anyhow::Error::from)?;
     MyResponse::ok(user)
 }
 
 #[post("/user-info")]
 pub async fn update_user_info(
+    req: HttpRequest,
     payload: web::Json<User>,
     session: Session,
     office: web::Data<Addr<PostOffice>>,
 ) -> ResponseResult<User> {
-    let mut user = User::get_from_session(&session).map_err(anyhow::Error::from)?;
+    let mut user = User::get_from_session(&session, &req).map_err(anyhow::Error::from)?;
     // Update current user info, payload.id will be excluded
     user.update(payload.into_inner());
     user.insert_to_session(&session)
@@ -88,7 +90,7 @@ pub async fn websocket(
     stream: web::Payload,
     office: web::Data<Addr<PostOffice>>,
 ) -> Result<HttpResponse, actix_web::Error> {
-    let user = User::get_from_session(&session)?;
+    let user = User::get_from_session(&session, &req)?;
 
     log::info!(
         "CONNECT /ws: websocket connected from user: {} {}",

packages/server/src/user.rs

@@ -1,4 +1,5 @@
 use actix_session::{Session, SessionInsertError};
+use actix_web::HttpRequest;
 use serde::{Deserialize, Serialize};
 
 const USER_INFO_SESSION_KEY: &str = "user_info";
@@ -9,6 +10,15 @@ pub struct User {
     pub user_name: String,
 }
 
+fn get_default_user_name_from_ua(ua: &str) -> Option<String> {
+    let result = woothee::parser::Parser::new().parse(ua)?;
+    Some(format!(
+        "{} {}",
+        result.os.replace("Windows", "Win"),
+        result.name
+    ))
+}
+
 impl User {
     pub fn new() -> Self {
         let id = nanoid::nanoid!(16);
@@ -26,13 +36,25 @@ impl User {
         Ok(())
     }
 
-    pub fn get_from_session(session: &Session) -> Result<User, SessionInsertError> {
+    pub fn get_from_session(
+        session: &Session,
+        req: &HttpRequest,
+    ) -> Result<User, SessionInsertError> {
         let user_option = session.get::<User>(USER_INFO_SESSION_KEY).unwrap_or(None);
 
         let user = match user_option {
             Some(user) => user,
             None => {
-                let new_user = User::new();
+                let user_agent = req
+                    .headers()
+                    .get("User-Agent")
+                    .and_then(|ua| ua.to_str().ok())
+                    .unwrap_or_default();
+
+                let mut new_user = User::new();
+                if let Some(user_name) = get_default_user_name_from_ua(user_agent) {
+                    new_user.user_name = user_name;
+                }
                 new_user.insert_to_session(session)?;
                 new_user
             }