Can register new clients

src/server/web_ui.rs

@@ -1,12 +1,14 @@
 use crate::app_config::AppConfig;
 use crate::constants::{STATE_KEY, USER_SESSION_KEY};
 use crate::server::{HttpFailure, HttpResult};
-use crate::user::{User, UserConfig, UserID};
+use crate::user::{APIClient, User, UserConfig, UserID};
 use crate::utils;
 use actix_session::Session;
 use actix_web::{web, HttpResponse};
 use askama::Template;
+use ipnet::IpNet;
 use light_openid::primitives::OpenIDConfig;
+use std::str::FromStr;
 
 /// Static assets
 #[derive(rust_embed::Embed)]
@@ -36,17 +38,21 @@ struct HomeTemplate {
     error_message: Option<String>,
 }
 
-/// Update matrix token request
+/// HTTP form request
 #[derive(serde::Deserialize)]
-pub struct UpdateMatrixToken {
+pub struct FormRequest {
+    /// Update matrix token
     new_matrix_token: Option<String>,
+
+    /// Create a new client
+    new_client_desc: Option<String>,
+
+    /// Restrict new client to a given network
+    ip_network: Option<String>,
 }
 
 /// Main route
-pub async fn home(
-    session: Session,
-    update_matrix_token: Option<web::Form<UpdateMatrixToken>>,
-) -> HttpResult {
+pub async fn home(session: Session, form_req: Option<web::Form<FormRequest>>) -> HttpResult {
     // Get user information, requesting authentication if information is missing
     let Some(user): Option<User> = session.get(USER_SESSION_KEY)? else {
         // Generate auth state
@@ -73,9 +79,9 @@ pub async fn home(
         .await
         .map_err(HttpFailure::FetchUserConfig)?;
 
-    // Update matrix token, if requested
-    if let Some(update_matrix_token) = update_matrix_token {
-        if let Some(t) = update_matrix_token.0.new_matrix_token {
+    if let Some(form_req) = form_req {
+        // Update matrix token, if requested
+        if let Some(t) = form_req.0.new_matrix_token {
             if t.len() < 3 {
                 error_message = Some("Specified Matrix token is too short!".to_string());
             } else {
@@ -85,6 +91,28 @@ pub async fn home(
                 success_message = Some("Matrix token was successfully updated!".to_string());
             }
         }
+
+        // Create a new client, if requested
+        if let Some(new_token_desc) = form_req.0.new_client_desc {
+            let ip_net = match form_req.0.ip_network.as_deref() {
+                None | Some("") => None,
+                Some(e) => match IpNet::from_str(e) {
+                    Ok(n) => Some(n),
+                    Err(e) => {
+                        log::error!("Failed to parse IP network provided by user: {e}");
+                        error_message = Some(format!("Failed to parse restricted IP network: {e}"));
+                        None
+                    }
+                },
+            };
+
+            if error_message.is_none() {
+                let token = APIClient::generate(new_token_desc, ip_net);
+                success_message = Some(format!("The secret of your new token is '{}'. Be sure to write it somewhere as you will not be able to recover it later!", token.secret));
+                config.clients.push(token);
+                config.save().await?;
+            }
+        }
     }
 
     // Render page