Changes

src/firmware_api.rs

@@ -0,0 +1,86 @@
+use std::{fs, path::PathBuf};
+
+use actix_web::{get, put, web, HttpResponse, Responder};
+use log::{debug, info, warn};
+
+use crate::{schemas::{AppState, OTAConfigurationList}, util::get_files};
+
+// Upload Firmware file
+#[put("/firmware/{device}/{config}/{version}")]
+async fn upload_firmware(
+    data: web::Data<AppState>,
+    path: web::Path<(String, String, String)>,
+    body: web::Bytes,
+) -> impl Responder {
+    let (device, config, version) = path.into_inner();
+    let version = version.replace(".", "-");
+
+    let firmware_root_path = &data.firmwares_path;
+
+    let firmware_folder = firmware_root_path.join(&device);
+    let firmware_path = firmware_folder
+        .join(format!("firmware_{config}_{version}"))
+        .with_extension("bin");
+
+    if firmware_path.is_file() {
+        warn!("Firmware with product: {device}, config: {config} and version: {version} at path {firmware_path:?} already exists, cant upload");
+        return HttpResponse::Conflict().body(format!("{firmware_path:?}"));
+    }
+
+    info!("Uploading firmware with product: {device}, config: {config} and version: {version} to {firmware_path:?}");
+
+    fs::create_dir_all(&firmware_folder).unwrap();
+    let x = tokio::fs::write(&firmware_path, &body).await;
+    debug!("{x:?}");
+
+    HttpResponse::Ok().body(format!(
+        "Firmware version {} uploaded successfully",
+        version
+    ))
+}
+
+#[get("/firmware/{device}")]
+async fn get_firmware_json(data: web::Data<AppState>, path: web::Path<String>) -> impl Responder {
+    let device = path.into_inner();
+    let fw_path = &data.firmwares_path.join(device);
+    if fw_path.is_dir() {
+        match get_files(fw_path, &data.hostname) {
+            Ok(cfg) => HttpResponse::Ok().json(OTAConfigurationList {
+                configurations: cfg,
+            }),
+            Err(e) => HttpResponse::InternalServerError().body(e.to_string()),
+        }
+    } else {
+        HttpResponse::Ok().json(OTAConfigurationList {
+            configurations: vec![],
+        })
+    }
+}
+
+#[get("/firmware/{product}/{config}/{version}.bin")]
+async fn serve_firmware(
+    path: web::Path<(String, String, String)>,
+    data: web::Data<AppState>,
+) -> impl Responder {
+    let (product, config, version) = path.into_inner();
+    let version = version.replace(".", "-").replace("_", "-");
+    let fw_root_path = &data.firmwares_path;
+    let file_path = PathBuf::from(format!("{product}/firmware_{config}_{version}.bin"));
+    let file_path = fw_root_path.join(&file_path);
+
+    info!("Requested firmware for product: {product}, config: {config} and version: {version}, expected to be stored at {file_path:?}");
+
+    if file_path.exists() {
+        info!("File exists, serving download now");
+        HttpResponse::Ok()
+            .content_type("application/octet-stream")
+            .insert_header((
+                "Content-Disposition",
+                format!("attachment; filename=\"{product}_{config}_{version}.bin\""),
+            ))
+            .body(std::fs::read(file_path).unwrap_or_else(|_| Vec::new()))
+    } else {
+        warn!("File does not exist");
+        HttpResponse::NotFound().body("Firmware version not found")
+    }
+}