First successful VNC connection

virtweb_backend/src/controllers/vm_controller.rs

@@ -1,8 +1,10 @@
+use crate::actors::vnc_actor::VNCActor;
 use crate::actors::vnc_tokens_actor::VNCTokensManager;
 use crate::controllers::{HttpResult, LibVirtReq};
 use crate::libvirt_lib_structures::{DomainState, DomainXMLUuid};
 use crate::libvirt_rest_structures::VMInfo;
-use actix_web::{web, HttpResponse};
+use actix_web::{web, HttpRequest, HttpResponse};
+use actix_web_actors::ws;
 
 #[derive(serde::Serialize)]
 struct VMInfoAndState {
@@ -231,3 +233,31 @@ pub async fn vnc_token(
         }
     })
 }
+
+#[derive(serde::Deserialize)]
+pub struct VNCTokenQuery {
+    token: String,
+}
+
+/// Start a VNC connection
+pub async fn vnc(
+    client: LibVirtReq,
+    manager: web::Data<VNCTokensManager>,
+    token: web::Query<VNCTokenQuery>,
+    req: HttpRequest,
+    stream: web::Payload,
+) -> HttpResult {
+    let domain_id = manager.consume_token(token.0.token).await?;
+    let domain = client.get_single_domain(domain_id).await?;
+
+    let socket_path = match domain.devices.graphics {
+        None => {
+            log::error!("Attempted to open VNC for a domain where VNC is disabled!");
+            return Ok(HttpResponse::ServiceUnavailable().json("VNC is not enabled!"));
+        }
+        Some(g) => g.socket,
+    };
+
+    log::info!("Start VNC connection on socket {socket_path}");
+    Ok(ws::start(VNCActor::new(&socket_path).await?, &req, stream)?)
+}