From c603e3bebbc6f6c0ce42fda2649a8258b4630a6b Mon Sep 17 00:00:00 2001
From: Gregory Mallios <gregmallios@gmail.com>
Date: Mon, 29 Apr 2024 00:44:54 +0300
Subject: [PATCH] fix: add missing state fields

---
 soundcore-lib/src/api/state/state.rs        | 25 +++++++++++++++++----
 soundcore-lib/src/packets/response/state.rs | 13 +++++++++++
 2 files changed, 34 insertions(+), 4 deletions(-)

diff --git a/soundcore-lib/src/api/state/state.rs b/soundcore-lib/src/api/state/state.rs
index bd86897..37d5f27 100644
--- a/soundcore-lib/src/api/state/state.rs
+++ b/soundcore-lib/src/api/state/state.rs
@@ -8,6 +8,7 @@ use crate::{
         SideTone, SoundMode, TwsStatus, WearDetection,
     },
 };
+use crate::models::{AmbientSoundNotice, AutoPowerOff, BassUp, DeviceColor, HearingProtect, InEarBeep, LDAC, PowerOnBatteryNotice, PromptLanguage, SupportTwoCnn, ThreeDimensionalEffect, TouchTone};
 
 /// This is a generalized version of the state for all devices
 #[derive(Debug, Serialize, Deserialize, Eq, PartialEq, Clone, Hash, Default)]
@@ -20,12 +21,28 @@ pub struct SoundcoreDeviceState {
     pub eq_configuration: EQConfiguration,
     pub serial: Option<SerialNumber>,
     pub fw: Option<FirmwareVer>,
-    pub host_device: Option<u8>,
-    pub tws_status: Option<TwsStatus>,
     pub button_model: Option<ButtonModel>,
+    pub host_device: Option<u8>,
     pub side_tone: Option<SideTone>,
+    pub age_range: Option<AgeRange>,
+    /// HearID
     pub hearid_eq_preset: Option<u16>,
-    pub wear_detection: Option<WearDetection>,
     pub hear_id: Option<HearID>,
-    pub age_range: Option<AgeRange>,
+    pub hear_id_has_data: Option<bool>,
+    // Toggles
+    pub touch_tone: Option<TouchTone>,
+    pub tws_status: Option<TwsStatus>,
+    pub wear_detection: Option<WearDetection>,
+    pub bass_up: Option<BassUp>,
+    pub auto_power_off: Option<AutoPowerOff>,
+    pub support_two_cnn: Option<SupportTwoCnn>,
+    pub in_ear_beep: Option<InEarBeep>,
+    pub ambient_sound_notice: Option<AmbientSoundNotice>,
+    pub power_on_battery_notice: Option<PowerOnBatteryNotice>,
+    // Other
+    pub three_dimensional_effect: Option<ThreeDimensionalEffect>,
+    pub device_color: Option<DeviceColor>,
+    pub ldac: Option<LDAC>,
+    pub prompt_language: Option<PromptLanguage>,
+    pub hearing_protect: Option<HearingProtect>,
 }
diff --git a/soundcore-lib/src/packets/response/state.rs b/soundcore-lib/src/packets/response/state.rs
index 779bd46..cd0a8ac 100644
--- a/soundcore-lib/src/packets/response/state.rs
+++ b/soundcore-lib/src/packets/response/state.rs
@@ -114,6 +114,19 @@ impl From<DeviceStateResponse> for SoundcoreDeviceState {
             wear_detection: value.wear_detection,
             hear_id: value.hear_id,
             age_range: value.age_range,
+            touch_tone: value.touch_tone,
+            bass_up: value.bass_up,
+            auto_power_off: value.auto_power_off,
+            support_two_cnn: value.support_two_cnn,
+            in_ear_beep: value.in_ear_beep,
+            ambient_sound_notice: value.ambient_sound_notice,
+            power_on_battery_notice: value.power_on_battery_notice,
+            three_dimensional_effect: value.three_dimensional_effect,
+            device_color: value.device_color,
+            ldac: value.ldac,
+            prompt_language: value.prompt_language,
+            hearing_protect: value.hearing_protect,
+            hear_id_has_data: value.hear_id_has_data,
         }
     }
 }