add security.txt

src/config.rs

@@ -1,9 +1,11 @@
 use serde::Deserialize;
 
-#[derive(Deserialize)]
+#[derive(Deserialize, Clone)]
 pub struct Config {
     pub scheme: String,
     pub port: u16,
+    pub mail: Option<String>,
+    pub lang: Option<String>,
 }
 
 pub fn get_config(file_path: &str) -> Config {
@@ -20,6 +22,8 @@ pub fn get_config(file_path: &str) -> Config {
             Config {
                 scheme: "http".to_string(),
                 port: 8080,
+                mail: None,
+                lang: None,
             }
         }
     }

src/main.rs

@@ -1,10 +1,12 @@
-use actix_web::{App, HttpServer};
+use actix_web::{web, App, HttpServer};
 mod config;
-mod template;
 
 #[path = "routes/index.rs"]
 mod index;
 
+#[path = "routes/agreements.rs"]
+mod agreements;
+
 #[actix_web::main]
 async fn main() -> std::io::Result<()> {
     let config = config::get_config("config/config.toml");
@@ -12,8 +14,13 @@ async fn main() -> std::io::Result<()> {
     let addr = ("127.0.0.1", config.port);
     println!("Listening to {}://{}:{}", config.scheme, addr.0, addr.1);
 
-    HttpServer::new(|| App::new().service(index::page))
-        .bind(addr)?
-        .run()
-        .await
+    HttpServer::new(move || {
+        App::new()
+            .app_data(web::Data::new(config.clone()))
+            .service(index::page)
+            .service(agreements::security)
+    })
+    .bind(addr)?
+    .run()
+    .await
 }

src/routes/agreements.rs

@@ -0,0 +1,31 @@
+use std::net::SocketAddr;
+
+use actix_web::{routes, web, HttpRequest, HttpResponse, Responder};
+use askama::Template;
+
+use crate::config::Config;
+
+#[routes]
+#[get("/.well-known/security.txt")]
+#[get("/security.txt")]
+pub async fn security(req: HttpRequest, config: web::Data<Config>) -> impl Responder {
+    HttpResponse::Ok().body(get_security(config.get_ref().clone(), req.peer_addr()))
+}
+
+#[derive(Template)]
+#[template(path = "../templates/security.txt")]
+struct SecurityTemplate {
+    contact: String,
+    pref_lang: String,
+    url: String,
+}
+
+pub fn get_security(config: Config, addr: Option<SocketAddr>) -> std::string::String {
+    let data = SecurityTemplate {
+        contact: config.mail.unwrap_or_default(),
+        pref_lang: config.lang.unwrap_or_default(),
+        url: format!("{}/.well-known/security.txt", addr.unwrap().ip()),
+    };
+
+    data.render().unwrap()
+}

src/routes/index.rs

@@ -1,8 +1,19 @@
-use crate::template::get_index;
-
 use actix_web::{get, HttpResponse, Responder};
+use askama::Template;
 
 #[get("/")]
 pub async fn page() -> impl Responder {
     HttpResponse::Ok().body(get_index())
 }
+
+#[derive(Template)]
+#[template(path = "../templates/index.html")]
+struct IndexTemplate<'a> {
+    name: &'a str,
+}
+
+pub fn get_index() -> std::string::String {
+    let index = IndexTemplate { name: "world" }; // instantiate your struct
+
+    index.render().unwrap()
+}

src/template.rs

@@ -1,13 +0,0 @@
-use askama::Template;
-
-#[derive(Template)]
-#[template(path = "../templates/index.html")]
-struct HelloTemplate<'a> {
-    name: &'a str,
-}
-
-pub fn get_index() -> std::string::String {
-    let hello = HelloTemplate { name: "world" }; // instantiate your struct
-
-    hello.render().unwrap()
-}

templates/security.txt

@@ -0,0 +1,3 @@
+Contact: {{ contact }}
+Preferred-Languages: {{ pref_lang }}
+Canonical: {{ url }}