first

src/main.rs

@@ -0,0 +1,81 @@
+mod api;
+mod data_mgt;
+mod logs;
+use actix_files::NamedFile;
+
+use actix_web::{
+    App, HttpRequest, HttpResponse, HttpServer, get, route,
+    web::{self},
+};
+use serde_json::Value;
+use std::path::PathBuf;
+
+pub static BIND_ADDR: &str = "0.0.0.0";
+pub static BIND_PORT: u16 = 80;
+pub static STATIC_PAGES: &[&str] = &["index.html", "style.css", "view.html", "logo.png"];
+pub static HTML_DIR: &str = "html/";
+pub static LOG_DIR: &str = "logs/";
+pub static DATA_STORAGE: &str = "storage/";
+
+use crate::{
+    api::{api_get_asset, api_upload},
+    logs::log_to_file,
+};
+
+#[get("/")]
+async fn index(reg: HttpRequest) -> actix_web::Result<NamedFile> {
+    let now = std::time::Instant::now();
+    let path: PathBuf = PathBuf::from(HTML_DIR.to_string() + "index.html");
+    log_to_file(&reg, now);
+    Ok(NamedFile::open(path)?)
+}
+
+#[get("/bhs/{id}")]
+async fn view_asset(req: HttpRequest) -> actix_web::Result<NamedFile> {
+    let now = std::time::Instant::now();
+    let path: PathBuf = PathBuf::from(HTML_DIR.to_string() + "view.html");
+    log_to_file(&req, now);
+    Ok(NamedFile::open(path)?)
+}
+
+#[route("/{tail:.*}", method = "GET", method = "POST")]
+async fn catch_all(req: HttpRequest, _payload: Option<web::Json<Value>>) -> actix_web::Result<HttpResponse> {
+    let now = std::time::Instant::now();
+
+    let response = match req.uri().path() {
+        path if STATIC_PAGES.contains(&&path[1..]) => {
+            let file_path = HTML_DIR.to_string() + path;
+            Ok(NamedFile::open(file_path)?.into_response(&req))
+        }
+        _ => Ok(HttpResponse::NotFound().body("Not Found")),
+    };
+
+    log_to_file(&req, now);
+    response
+}
+
+#[actix_web::main]
+async fn main() -> std::io::Result<()> {
+    println!("Starting server at http://{}:{}/", BIND_ADDR, BIND_PORT);
+    tokio::spawn(async {
+        let mut interval = tokio::time::interval(tokio::time::Duration::from_secs(60));
+        loop {
+            interval.tick().await;
+            if let Err(e) = data_mgt::clear_assets().await {
+                eprintln!("Error clearing assets: {}", e);
+            }
+        }
+    });
+    HttpServer::new(|| {
+        App::new()
+            .app_data(web::JsonConfig::default().limit(1024 * 1024 * 3))
+            .service(index)
+            .service(view_asset)
+            .service(api_get_asset)
+            .service(api_upload)
+            .service(catch_all)
+    })
+    .bind((BIND_ADDR, BIND_PORT))?
+    .run()
+    .await
+}