From 85620b73a74780b510f2fbbca624baa9726303e3 Mon Sep 17 00:00:00 2001
From: asur4s <todaymofish@gmail.com>
Date: Wed, 21 Dec 2022 00:03:15 -0800
Subject: [PATCH] opt: get supported keyboard modes

---
 libs/hbb_common/src/lib.rs  |  1 +
 src/common.rs               |  4 ++--
 src/ui_session_interface.rs | 14 +++++++++++++-
 3 files changed, 16 insertions(+), 3 deletions(-)

diff --git a/libs/hbb_common/src/lib.rs b/libs/hbb_common/src/lib.rs
index ae564685f..4245b1d73 100644
--- a/libs/hbb_common/src/lib.rs
+++ b/libs/hbb_common/src/lib.rs
@@ -40,6 +40,7 @@ pub use tokio_socks::TargetAddr;
 pub mod password_security;
 pub use chrono;
 pub use directories_next;
+pub mod keyboard;
 
 #[cfg(feature = "quic")]
 pub type Stream = quic::Connection;
diff --git a/src/common.rs b/src/common.rs
index 0f3794261..02b4a0c10 100644
--- a/src/common.rs
+++ b/src/common.rs
@@ -693,8 +693,8 @@ pub fn is_keyboard_mode_supported(keyboard_mode: &KeyboardMode, version_number:
     match keyboard_mode {
         KeyboardMode::Legacy => true,
         KeyboardMode::Map => version_number >= hbb_common::get_version_number("1.2.0"),
-        KeyboardMode::Translate => true,
-        KeyboardMode::Auto => true,
+        KeyboardMode::Translate => false,
+        KeyboardMode::Auto => false,
     }
 }
 
diff --git a/src/ui_session_interface.rs b/src/ui_session_interface.rs
index 0cf2f2e2d..e7ac620ee 100644
--- a/src/ui_session_interface.rs
+++ b/src/ui_session_interface.rs
@@ -4,7 +4,7 @@ use crate::client::{
     load_config, send_mouse, start_video_audio_threads, FileManager, Key, LoginConfigHandler,
     QualityStatus, KEY_MAP,
 };
-use crate::common::GrabState;
+use crate::common::{is_keyboard_mode_supported, GrabState};
 use crate::keyboard;
 use crate::{client::Data, client::Interface};
 use async_trait::async_trait;
@@ -48,6 +48,10 @@ impl<T: InvokeUiSession> Session<T> {
         self.lc.read().unwrap().custom_image_quality.clone()
     }
 
+    pub fn get_peer_version(&self) -> i64 {
+        self.lc.read().unwrap().version.clone()
+    }
+
     pub fn get_keyboard_mode(&self) -> String {
         self.lc.read().unwrap().keyboard_mode.clone()
     }
@@ -198,6 +202,14 @@ impl<T: InvokeUiSession> Session<T> {
         crate::platform::is_xfce()
     }
 
+    pub fn get_supported_keyboard_modes(&self) -> Vec<KeyboardMode> {
+        let version = self.get_peer_version();
+        KeyboardMode::iter()
+            .filter(|&mode| is_keyboard_mode_supported(mode, version))
+            .map(|&mode| mode)
+            .collect::<Vec<_>>()
+    }
+
     pub fn remove_port_forward(&self, port: i32) {
         let mut config = self.load_config();
         config.port_forwards = config