backend: add device type upon management connection.

fix #219

Author: ffreddow

backend/src/routes/management.rs

@@ -383,13 +383,26 @@ pub async fn management(
         ManagementDataVersion::V1(v) => (v.firmware_version.clone(), v.port),
         ManagementDataVersion::V2(v) => (v.firmware_version.clone(), v.port),
     };
+
+    let user_agent = req.headers().get("User-Agent");
+    let device_type = user_agent
+        .and_then(|h| h.to_str().ok())
+        .and_then(|ua| {
+            if ua == "ESP32 HTTP Client/1.0" {
+                None
+            } else {
+                Some(ua.to_string())
+            }
+        });
+
     let mut conn = get_connection(&state)?;
     web_block_unpacked(move || {
         match diesel::update(chargers::chargers)
             .filter(chargers::id.eq(charger_id))
             .set((
                 chargers::firmware_version.eq(fw_version),
                 chargers::webinterface_port.eq(port as i32),
+                chargers::device_type.eq(device_type),
             ))
             .execute(&mut conn)
         {
@@ -455,6 +468,7 @@ mod tests {
         let app = test::init_service(app).await;
 
         let user_id = get_test_uuid(&mail).unwrap();
+        let charger_uuid_clone = charger.uuid.clone();
         let data = ManagementDataVersion::V2(ManagementDataVersion2 {
             id: charger.uuid,
             password: charger.password,
@@ -475,13 +489,26 @@ mod tests {
         let req = test::TestRequest::put()
             .uri("/management")
             .append_header(("X-Forwarded-For", "123.123.123.3"))
+            .append_header(("User-Agent", "Tinkerforge-WARP2_Charger/2.8.0+6811d0b1"))
             .set_json(body)
             .to_request();
         let resp: ManagementResponseSchema = test::call_and_read_body_json(&app, req).await;
 
         println!("{resp:?}");
         assert_eq!([1], *resp.configured_users);
         assert_eq!(vec![user_id.to_string()], resp.configured_users_uuids);
+
+        // Verify device_type stored correctly
+        use db_connector::models::chargers::Charger as DbCharger;
+        use db_connector::schema::chargers::dsl::*;
+        let pool = db_connector::test_connection_pool();
+        let mut conn = pool.get().unwrap();
+        let db_charger: DbCharger = chargers
+            .filter(id.eq(uuid::Uuid::from_str(&charger_uuid_clone).unwrap()))
+            .select(DbCharger::as_select())
+            .get_result(&mut conn)
+            .unwrap();
+        assert_eq!(db_charger.device_type.as_deref(), Some("Tinkerforge-WARP2_Charger/2.8.0+6811d0b1"));
     }
 
     #[actix_web::test]