feat: 新增 MJPEG/H.264 VNC 初步支持

src/main.rs

@@ -31,10 +31,12 @@ use one_kvm::state::{AppState, ShutdownAction};
 use one_kvm::update::UpdateService;
 use one_kvm::utils::bind_tcp_listener;
 use one_kvm::video::codec_constraints::{
-    enforce_constraints_with_stream_manager, StreamCodecConstraints,
+    enforce_constraints_with_stream_manager, validate_third_party_codec_compatibility,
+    StreamCodecConstraints,
 };
 use one_kvm::video::format::{PixelFormat, Resolution};
 use one_kvm::video::{Streamer, VideoStreamManager};
+use one_kvm::vnc::VncService;
 use one_kvm::web;
 use one_kvm::webrtc::{WebRtcStreamer, WebRtcStreamerConfig};
 
@@ -486,7 +488,18 @@ async fn main() -> anyhow::Result<()> {
         );
     }
 
-    let rustdesk = if config.rustdesk.is_valid() {
+    let third_party_codec_config_valid = match validate_third_party_codec_compatibility(&config) {
+        Ok(()) => true,
+        Err(e) => {
+            tracing::warn!(
+                    "Third-party access codec configuration is invalid; RustDesk/VNC/RTSP will not start: {}",
+                    e
+                );
+            false
+        }
+    };
+
+    let rustdesk = if third_party_codec_config_valid && config.rustdesk.is_valid() {
         tracing::info!(
             "Initializing RustDesk service: ID={} -> {}",
             config.rustdesk.device_id,
@@ -510,7 +523,7 @@ async fn main() -> anyhow::Result<()> {
         None
     };
 
-    let rtsp = if config.rtsp.enabled {
+    let rtsp = if third_party_codec_config_valid && config.rtsp.enabled {
         tracing::info!(
             "Initializing RTSP service: rtsp://{}:{}/{}",
             config.rtsp.bind,
@@ -524,6 +537,23 @@ async fn main() -> anyhow::Result<()> {
         None
     };
 
+    let vnc = if third_party_codec_config_valid && config.vnc.enabled {
+        tracing::info!(
+            "Initializing VNC service: {}:{} ({:?})",
+            config.vnc.bind,
+            config.vnc.port,
+            config.vnc.encoding
+        );
+        Some(Arc::new(VncService::new(
+            config.vnc.clone(),
+            stream_manager.clone(),
+            hid.clone(),
+        )))
+    } else {
+        tracing::info!("VNC disabled in configuration");
+        None
+    };
+
     let update_service = Arc::new(UpdateService::new(data_dir.join("updates")));
 
     let state = AppState::new(
@@ -541,6 +571,7 @@ async fn main() -> anyhow::Result<()> {
         atx,
         audio,
         rustdesk.clone(),
+        vnc.clone(),
         rtsp.clone(),
         extensions.clone(),
         events.clone(),
@@ -573,6 +604,13 @@ async fn main() -> anyhow::Result<()> {
             tracing::info!("RustDesk service started");
         }
     }
+    if let Some(ref service) = vnc {
+        if let Err(e) = service.start().await {
+            tracing::error!("Failed to start VNC service: {}", e);
+        } else {
+            tracing::info!("VNC service started");
+        }
+    }
 
     if let Some(ref service) = rtsp {
         if let Err(e) = service.start().await {
@@ -1135,6 +1173,14 @@ async fn cleanup(state: &Arc<AppState>) {
         }
     }
 
+    if let Some(ref service) = *state.vnc.read().await {
+        if let Err(e) = service.stop().await {
+            tracing::warn!("Failed to stop VNC service: {}", e);
+        } else {
+            tracing::info!("VNC service stopped");
+        }
+    }
+
     if let Some(ref service) = *state.rtsp.read().await {
         if let Err(e) = service.stop().await {
             tracing::warn!("Failed to stop RTSP service: {}", e);