chore: Add retries for Keon/Handy

buttplugio · Oct 16, 2023 · 1457ad3 · 1457ad3
1 parent 8145fe3
commit 1457ad3
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 0 deletions.
diff --git a/buttplug/src/server/device/protocol/kiiroo_v2.rs b/buttplug/src/server/device/protocol/kiiroo_v2.rs
@@ -54,6 +54,10 @@ pub struct KiirooV2 {
 }
 
 impl ProtocolHandler for KiirooV2 {
+  fn keepalive_strategy(&self) -> super::ProtocolKeepaliveStrategy {
+    super::ProtocolKeepaliveStrategy::RepeatLastPacketStrategy
+  }
+
   fn handle_linear_cmd(
     &self,
     message: message::LinearCmd,

diff --git a/buttplug/src/server/device/protocol/kiiroo_v21_initialized.rs b/buttplug/src/server/device/protocol/kiiroo_v21_initialized.rs
@@ -67,6 +67,10 @@ pub struct KiirooV21Initialized {
 }
 
 impl ProtocolHandler for KiirooV21Initialized {
+  fn keepalive_strategy(&self) -> super::ProtocolKeepaliveStrategy {
+    super::ProtocolKeepaliveStrategy::RepeatLastPacketStrategy
+  }
+
   fn handle_scalar_vibrate_cmd(
     &self,
     _index: u32,

diff --git a/buttplug/src/server/device/protocol/thehandy/mod.rs b/buttplug/src/server/device/protocol/thehandy/mod.rs
@@ -5,6 +5,8 @@
 // Licensed under the BSD 3-Clause license. See LICENSE file in the project root
 // for full license information.
 
+use self::handyplug::Ping;
+
 use super::fleshlight_launch_helper;
 use crate::server::device::configuration::ProtocolDeviceAttributes;
 use crate::{
@@ -114,6 +116,21 @@ pub struct TheHandy {
 }
 
 impl ProtocolHandler for TheHandy {
+
+  fn keepalive_strategy(&self) -> super::ProtocolKeepaliveStrategy {
+    let ping_payload = handyplug::Payload {
+      messages: vec![handyplug::Message {
+        message: Some(handyplug::message::Message::Ping(Ping { id: 999 })),
+      }]
+    };
+    let mut ping_buf = vec![];
+    ping_payload
+      .encode(&mut ping_buf)
+      .expect("Infallible encode.");
+
+    super::ProtocolKeepaliveStrategy::RepeatPacketStrategy(HardwareWriteCmd::new(Endpoint::Tx, ping_buf, true))
+  }
+
   fn handle_fleshlight_launch_fw12_cmd(
     &self,
     message: message::FleshlightLaunchFW12Cmd,