reai-board-sdk 0.2.2

ReAI Vibe Board keyboard SDK — USB/BLE connectivity, auto-reconnect, HID protocol, mSBC decoding, USB Audio capture
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
//! Vendor GATT 客户端(V2 tokio 原生 async)。
//!
//! 通过 btleplug 连接固件的 Vendor GATT Service (0xFE60),
//! 订阅 Event (FE62) 和 Audio (FE63) 通知,通过 Command (FE61) 写入命令。
//!
//! 事件通过 `broadcast::Sender<BoardEvent>` 上报,mSBC 音频帧转给 [`AudioFrameSink`]。
//!
//! **V2 变化**:删除 V1 的 `ble_run` 同步包装(自建 current_thread runtime + std mpsc)。
//! 所有方法改为 async,直接 `.await` btleplug 原生 API,跑在调用方的 tokio runtime 上。
//! pending map 从 `std::sync::mpsc` 改为 `tokio::sync::oneshot`。

use std::collections::HashMap;
use std::sync::atomic::{AtomicBool, AtomicU64, Ordering};
use std::sync::{Arc, Mutex};
use std::time::Instant;

use anyhow::Result;
use btleplug::api::{Central as _, Peripheral as _, ScanFilter, WriteType};
use btleplug::platform::{Adapter, Peripheral};
use futures_util::StreamExt;
use tokio::sync::broadcast;

use crate::kernel::consumer_hold::ConsumerHeldTracker;
use crate::kernel::event::{
    AiVoiceKeyEvent, BoardEvent, ErrorEvent, KeySource, ModeChangeEvent, ModeSource,
};
use crate::kernel::key_aggregator::KeyStateAggregator;
use crate::kernel::protocol_gatt as protocol;
use crate::kernel::protocol_hid::{
    find_key_index_by_value, is_ai_voice_consumer_code, key_index_to_mode, WorkMode,
    CMD_AUDIO_DATA, CMD_DEVICE_DISCONNECT, CMD_STATUS, CMD_WORK_MODE_DATA,
};
#[cfg(feature = "test-mode")]
use crate::kernel::protocol_hid::{parse_factory_key_event_unscoped, CMD_AI_FACTORY_KEY_EVENT};
use crate::kernel::sink::AudioFrameSink;

/// Bluetooth CCC(Client Characteristic Configuration)描述符的标准 UUID。
///
/// 即 `00002902-0000-0000-1000-00805f9b34fb`。btleplug 在多数平台会自动订阅,
/// 但失败时需手动找到这个描述符写入 `0x01 0x00` 开启通知。
///
/// 注意:`Uuid::from_u128(0x2902)` **不**等于这个标准 UUID——前者高位全零,
/// 与 Bluetooth Base UUID 完全不同。旧代码里 `|| from_u128(0x2902)` 是永远不匹配的死分支。
const CCC_DESCRIPTOR_UUID: uuid::Uuid = uuid::Uuid::from_bytes([
    0x00, 0x00, 0x29, 0x02, 0x00, 0x00, 0x10, 0x00, 0x80, 0x00, 0x00, 0x80, 0x5f, 0x9b, 0x34, 0xfb,
]);

/// BLE 扫描发现的设备信息(手动扫描/选择连接用)
#[derive(Debug, Clone, serde::Serialize)]
pub struct BleDeviceInfo {
    /// 设备名(REAI_VB_XXXX)
    pub name: String,
    /// btleplug PeripheralId 的字符串形式(调试/去重用)
    pub id: String,
    /// 信号强度(dBm),可能为 None
    pub rssi: Option<i16>,
    /// MAC 地址(冒号分隔大写)
    pub address: String,
}

/// 发现的 GATT 特征值句柄
#[derive(Clone)]
struct GattChars {
    cmd: btleplug::api::Characteristic,
    event: btleplug::api::Characteristic,
    audio: btleplug::api::Characteristic,
}

/// 等待响应的 oneshot 注册项
struct PendingResponse {
    #[allow(dead_code)]
    cmd: u8,
    tx: tokio::sync::oneshot::Sender<Vec<u8>>,
}

/// handle_event 的有状态上下文(替代全局 static,支持单设备实例化)
#[derive(Default)]
struct GattEventState {
    /// 当前按着哪些键。
    ///
    /// 原来这里是「最近一次按下的键索引」,一次只记得住一个。BLE 与 USB 走的是同一条
    /// 单值 Consumer 流,于是按住 Tab 再转旋钮时,这个记录被旋钮顶掉,随后的 0x0000
    /// 把释放算在了旋钮头上——**按住的那个键再也等不到松手事件**,Command 一直按着,
    /// 只能靠上层 30 秒硬超时兜底。改用与 USB 共享的账本([`ConsumerHeldTracker`])。
    consumer: ConsumerHeldTracker,
    /// 按来源合并并 diff 出按下/松开事件,与 USB 用的是同一套。
    ///
    /// 延迟创建:它需要事件发送端,而发送端要等 `handle_event` 传进来。
    aggregator: Option<KeyStateAggregator>,
    ai_voice_pressed: bool,
    mode_key_pressed: usize,
}

impl GattEventState {
    /// 断开时把还按着的键全部交代成松开。
    ///
    /// 这个状态本身随通知任务一起销毁,所以「不清也不会串到下一次连接」——
    /// 但**消费方那边不会自动忘记**:只收到过按下、没收到松开的键,会一直被当成按着。
    /// 语音键卡住尤其难受(录音停不下来),所以断开时必须补齐。
    fn release_all(&mut self, event_tx: &broadcast::Sender<BoardEvent>) {
        if let Some(aggregator) = &self.aggregator {
            aggregator.report_change(KeySource::Gatt, Vec::new(), None);
        }
        if self.ai_voice_pressed {
            self.ai_voice_pressed = false;
            log::info!(target: "gatt", "通知监听退出,补发 AI 语音键释放");
            let _ = event_tx.send(BoardEvent::AiVoiceKey(AiVoiceKeyEvent { pressed: false }));
        }
    }
}

pub struct VendorGattClient {
    adapter: Adapter,
    peripheral: Mutex<Option<Peripheral>>,
    chars: Mutex<Option<GattChars>>,
    running: Arc<AtomicBool>,
    audio_sink: Arc<dyn AudioFrameSink>,
    event_tx: broadcast::Sender<BoardEvent>,
    /// 等待命令响应的 pending map(cmd → oneshot sender)
    pending_responses: Arc<Mutex<HashMap<u8, PendingResponse>>>,
    /// GATT 写入序列化锁:防止并发写入导致 pending 响应错乱。
    /// 用 tokio::sync::Mutex —— 持锁期间会 `.await send_command()`(std Mutex 跨 await 是 UB)。
    write_lock: tokio::sync::Mutex<()>,
    seq: AtomicU64,
    /// 通知监听 task 的 JoinHandle。`disconnect` 时 await,避免重连场景下
    /// 旧 task 末尾的 `pending.clear()` 清掉新 task 刚注册的 pending response。
    notify_handle: Mutex<Option<tokio::task::JoinHandle<()>>>,
    /// 音频解码线程的 JoinHandle。`disconnect` 时 join(超时保护),避免线程悬挂。
    audio_thread_handle: Mutex<Option<std::thread::JoinHandle<()>>>,
}

impl VendorGattClient {
    pub fn new(
        adapter: Adapter,
        audio_sink: Arc<dyn AudioFrameSink>,
        event_tx: broadcast::Sender<BoardEvent>,
    ) -> Self {
        Self {
            adapter,
            peripheral: Mutex::new(None),
            chars: Mutex::new(None),
            running: Arc::new(AtomicBool::new(false)),
            audio_sink,
            event_tx,
            pending_responses: Arc::new(Mutex::new(HashMap::new())),
            write_lock: tokio::sync::Mutex::new(()),
            seq: AtomicU64::new(0),
            notify_handle: Mutex::new(None),
            audio_thread_handle: Mutex::new(None),
        }
    }

    /// 扫描发现 Vendor GATT 设备,返回 (Peripheral, 设备名)。
    ///
    /// 传 target_name 精确匹配;否则匹配 REAI_VB_ 前缀。5 轮 × 2s。
    pub async fn scan_for_device(&self, target_name: Option<&str>) -> Result<(Peripheral, String)> {
        let adapter = self.adapter.clone();
        let target = target_name.map(|s| s.to_string());

        // 不用 Service UUID 过滤:固件广播包可能不含 0xFE60,改为名字前缀匹配
        adapter
            .start_scan(ScanFilter::default())
            .await
            .map_err(|e| anyhow::anyhow!("BLE 扫描启动失败: {}", e))?;

        for round in 0..5 {
            tokio::time::sleep(std::time::Duration::from_secs(2)).await;

            let peripherals = adapter
                .peripherals()
                .await
                .map_err(|e| anyhow::anyhow!("获取设备列表失败: {}", e))?;

            for p in &peripherals {
                if let Ok(Some(props)) = p.properties().await {
                    let name = props.local_name.unwrap_or_default();
                    let matched = match &target {
                        Some(t) => name == *t,
                        None => name.starts_with(protocol::VENDOR_DEVICE_PREFIX),
                    };
                    if matched {
                        log::info!(target: "gatt", "发现 Vendor GATT 设备: {} ({})", name, p.id());
                        if let Err(e) = adapter.stop_scan().await {
                            log::warn!(target: "gatt", "stop_scan 失败(可能仍占用适配器): {}", e);
                        }
                        return Ok((p.clone(), name));
                    }
                }
            }

            log::debug!(
                target: "gatt",
                "扫描轮次 {}/5 未找到{}",
                round + 1,
                target.as_ref().map(|t| format!(" {}", t)).unwrap_or_default()
            );
        }

        if let Err(e) = adapter.stop_scan().await {
            log::warn!(target: "gatt", "stop_scan 失败(可能仍占用适配器): {}", e);
        }
        Err(anyhow::anyhow!(
            "未发现 Vendor GATT 设备{}",
            target
                .as_ref()
                .map(|t| format!(" {}", t))
                .unwrap_or_default()
        ))
    }

    /// 扫描列出周围所有 Vendor GATT 设备(REAI_VB_ 前缀),返回去重后的列表。
    ///
    /// 与 [`scan_for_device`](Self::scan_for_device) 不同:后者找到第一个就返回,
    /// 本方法扫满 `timeout` 累积所有匹配设备(供手动选择)。
    /// 不连接、不订阅,纯扫描。扫完自动 stop_scan。
    pub async fn scan_all_vendor_devices(
        &self,
        timeout: std::time::Duration,
    ) -> Result<Vec<BleDeviceInfo>> {
        let adapter = self.adapter.clone();
        adapter
            .start_scan(ScanFilter::default())
            .await
            .map_err(|e| anyhow::anyhow!("BLE 扫描启动失败: {}", e))?;

        // 用 id 去重(同一设备多轮广播)
        let mut seen: std::collections::HashMap<String, BleDeviceInfo> =
            std::collections::HashMap::new();
        let deadline = tokio::time::Instant::now() + timeout;

        while tokio::time::Instant::now() < deadline {
            let peripherals = adapter
                .peripherals()
                .await
                .map_err(|e| anyhow::anyhow!("获取设备列表失败: {}", e))?;

            for p in &peripherals {
                if let Ok(Some(props)) = p.properties().await {
                    let name = props.local_name.unwrap_or_default();
                    if name.starts_with(protocol::VENDOR_DEVICE_PREFIX) {
                        let id_str = format!("{:?}", p.id());
                        seen.entry(id_str.clone()).or_insert(BleDeviceInfo {
                            name: name.clone(),
                            id: id_str,
                            rssi: props.rssi,
                            address: format!("{}", props.address),
                        });
                    }
                }
            }

            tokio::time::sleep(std::time::Duration::from_millis(500)).await;
        }

        if let Err(e) = adapter.stop_scan().await {
            log::warn!(target: "gatt", "stop_scan 失败(可能仍占用适配器): {}", e);
        }
        Ok(seen.into_values().collect())
    }

    /// 连接设备、发现服务、订阅 Event/Audio 通知。
    ///
    /// 整个「连接+服务发现+订阅」包 10s 超时:btleplug 在 macOS 遇到
    /// "Event receiver died" 时 discover_services().await 永不完成,
    /// 超时则返回失败 → hotplug 重新 scan+connect,给设备 BLE 就绪时间。
    pub async fn connect(&self, peripheral: &Peripheral) -> Result<()> {
        let p = peripheral.clone();
        let inner = async {
            p.connect()
                .await
                .map_err(|e| anyhow::anyhow!("Vendor GATT 连接失败: {}", e))?;

            log::info!(target: "gatt", "BLE 已连接,开始服务发现");

            p.discover_services()
                .await
                .map_err(|e| anyhow::anyhow!("GATT 服务发现失败: {}", e))?;

            let chars = p.characteristics();
            log::info!(target: "gatt", "GATT 服务发现完成,共 {} 个特征值", chars.len());

            let cmd = chars
                .iter()
                .find(|c| c.uuid == protocol::CMD_CHAR_UUID)
                .ok_or_else(|| anyhow::anyhow!("未找到 Command 特征值 (FE61)"))?;
            let event = chars
                .iter()
                .find(|c| c.uuid == protocol::EVENT_CHAR_UUID)
                .ok_or_else(|| anyhow::anyhow!("未找到 Event 特征值 (FE62)"))?;
            let audio = chars
                .iter()
                .find(|c| c.uuid == protocol::AUDIO_CHAR_UUID)
                .ok_or_else(|| anyhow::anyhow!("未找到 Audio 特征值 (FE63)"))?;

            subscribe_with_ccc_fallback(&p, event, "Event").await?;
            subscribe_with_ccc_fallback(&p, audio, "Audio").await?;

            Ok(GattChars {
                cmd: cmd.clone(),
                event: event.clone(),
                audio: audio.clone(),
            })
        };

        let found_chars =
            match tokio::time::timeout(std::time::Duration::from_secs(10), inner).await {
                Ok(r) => r,
                Err(_) => Err(anyhow::anyhow!(
                    "BLE connect 超时(10s)— 可能 CoreBluetooth Event receiver died,将重试"
                )),
            }?;

        *self.peripheral.lock().unwrap() = Some(peripheral.clone());
        *self.chars.lock().unwrap() = Some(found_chars);
        self.running.store(true, Ordering::SeqCst);

        Ok(())
    }

    /// 写入命令到 Command 特征值(5s 超时)
    pub async fn send_command(&self, data: &[u8]) -> Result<()> {
        let p = self
            .peripheral
            .lock()
            .unwrap()
            .clone()
            .ok_or_else(|| anyhow::anyhow!("Vendor GATT 未连接"))?;
        let chars = self
            .chars
            .lock()
            .unwrap()
            .clone()
            .ok_or_else(|| anyhow::anyhow!("GATT 特征值未发现"))?;
        let cmd_data = data.to_vec();
        let seq = self.seq.fetch_add(1, Ordering::SeqCst);

        log::debug!(target: "gatt", "[GATT:{}] TX CMD=0x{:02X} len={}", seq, cmd_data[0], cmd_data.len());
        // FE61 只有 WRITE 属性,必须用 WithResponse(Windows 严格按属性匹配)
        tokio::time::timeout(
            std::time::Duration::from_secs(5),
            p.write(&chars.cmd, &cmd_data, WriteType::WithResponse),
        )
        .await
        .map_err(|_| anyhow::anyhow!("GATT 写入超时 5s"))?
        .map_err(|e| anyhow::anyhow!("GATT 写入失败: {}", e))
    }

    /// 发送命令并等待 Event 通道的匹配响应(持 write_lock 序列化)
    pub async fn send_command_and_read_response(
        &self,
        expected_cmd: u8,
        data: &[u8],
        timeout_ms: u64,
    ) -> Result<Vec<u8>> {
        let _write_guard = self.write_lock.lock().await;
        log::info!(
            target: "gatt",
            "[GATT] cmd_and_read: CMD=0x{:02X} len={} timeout={}ms",
            expected_cmd, data.len(), timeout_ms
        );

        let (tx, rx) = tokio::sync::oneshot::channel::<Vec<u8>>();

        // 注册 pending response
        {
            let mut pending = self.pending_responses.lock().unwrap();
            if let Some(old) = pending.remove(&expected_cmd) {
                log::warn!(target: "gatt", "[GATT] 清理残留 pending CMD=0x{:02X}", expected_cmd);
                let _ = old.tx.send(vec![]); // oneshot:发送空响应解除旧等待
            }
            pending.insert(
                expected_cmd,
                PendingResponse {
                    cmd: expected_cmd,
                    tx,
                },
            );
        }

        if let Err(e) = self.send_command(data).await {
            self.pending_responses.lock().unwrap().remove(&expected_cmd);
            return Err(e);
        }

        match tokio::time::timeout(std::time::Duration::from_millis(timeout_ms), rx).await {
            Ok(Ok(response)) => Ok(response),
            Ok(Err(_)) => {
                // oneshot 关闭(notification loop 退出时 clear 了 pending)
                self.pending_responses.lock().unwrap().remove(&expected_cmd);
                Err(anyhow::anyhow!(
                    "[GATT] CMD=0x{:02X} 响应通道已关闭(notification loop 已退出)",
                    expected_cmd
                ))
            }
            Err(_) => {
                self.pending_responses.lock().unwrap().remove(&expected_cmd);
                Err(anyhow::anyhow!(
                    "[GATT] CMD=0x{:02X} 响应超时 ({}ms)",
                    expected_cmd,
                    timeout_ms
                ))
            }
        }
    }

    /// 启动通知监听循环(在调用方 tokio runtime 上 spawn)
    pub fn start_notification_loop(&self) {
        let p = match self.peripheral.lock().unwrap().clone() {
            Some(p) => p,
            None => return,
        };
        let running = self.running.clone();
        let audio_sink = self.audio_sink.clone();
        let event_tx = self.event_tx.clone();
        let pending = self.pending_responses.clone();
        let event_uuid = protocol::EVENT_CHAR_UUID;
        let audio_uuid = protocol::AUDIO_CHAR_UUID;

        // 音频处理 channel:通知循环投递帧,独立线程解码(on_msbc_frame)
        let (audio_tx, audio_rx) = std::sync::mpsc::sync_channel::<Vec<u8>>(64);
        let audio_running = running.clone();
        let audio_handler = audio_sink.clone();
        let audio_thread = std::thread::spawn(move || {
            const MSBC_FRAME_SIZE: usize = 57;
            let mut processed: u64 = 0;
            while audio_running.load(Ordering::SeqCst) {
                match audio_rx.recv_timeout(std::time::Duration::from_secs(1)) {
                    Ok(data) => {
                        if let Some((_flag, frames_data)) = protocol::parse_audio_packet(&data) {
                            for frame in frames_data.chunks(MSBC_FRAME_SIZE) {
                                if frame.len() == MSBC_FRAME_SIZE {
                                    audio_handler.on_msbc_frame(frame);
                                    processed += 1;
                                }
                            }
                        } else {
                            log::warn!(target: "gatt", "[GATT-audio] 解析失败: data_len={}", data.len());
                        }
                    }
                    Err(std::sync::mpsc::RecvTimeoutError::Timeout) => continue,
                    Err(std::sync::mpsc::RecvTimeoutError::Disconnected) => break,
                }
            }
            log::debug!(target: "gatt", "音频处理线程退出,共处理 {} 帧", processed);
        });
        if let Ok(mut slot) = self.audio_thread_handle.lock() {
            *slot = Some(audio_thread);
        }

        let notify_handle = tokio::spawn(async move {
            let mut stream = match p.notifications().await {
                Ok(s) => s,
                Err(e) => {
                    log::warn!(target: "gatt", "通知流创建失败: {}", e);
                    return;
                }
            };

            log::info!(target: "gatt", "通知监听已启动");

            let mut state = GattEventState::default();
            let mut notif_count: u64 = 0;
            let mut audio_count: u64 = 0;
            // 心跳:CoreBluetooth 断连后 notifications().next() 不立即返回 None,
            // 靠 is_connected() 主动探测。10s 无通知即检查。
            let heartbeat = std::time::Duration::from_secs(10);
            let mut last_notif = tokio::time::Instant::now();

            while running.load(Ordering::SeqCst) {
                tokio::select! {
                    notification = stream.next() => {
                        last_notif = tokio::time::Instant::now();
                        match notification {
                            Some(notif) => {
                                notif_count += 1;
                                if notif.uuid == audio_uuid {
                                    audio_count += 1;
                                    // try_send:通道满丢帧保连接,不阻塞通知循环
                                    match audio_tx.try_send(notif.value.to_vec()) {
                                        Ok(_) => {}
                                        Err(std::sync::mpsc::TrySendError::Full(_)) => {
                                            if audio_count.is_multiple_of(100) {
                                                log::warn!(target: "gatt", "[GATT] 音频帧丢弃(通道满): total={}", audio_count);
                                            }
                                        }
                                        Err(std::sync::mpsc::TrySendError::Disconnected(_)) => {
                                            log::warn!(target: "gatt", "[GATT] 音频解码通道关闭,结束 BLE 会话触发重连");
                                            break;
                                        }
                                    }
                                } else if notif.uuid == event_uuid {
                                    handle_event(&notif.value, &mut state, &event_tx, &pending);
                                } else {
                                    log::warn!(target: "gatt", "收到未知 UUID 通知: {:?}", notif.uuid);
                                }
                            }
                            None => {
                                log::info!(target: "gatt", "通知流结束(设备断开),共 {} 通知", notif_count);
                                break;
                            }
                        }
                    }
                    _ = tokio::time::sleep_until(last_notif + heartbeat) => {
                        match p.is_connected().await {
                            Ok(false) | Err(_) => {
                                log::info!(target: "gatt", "心跳:10s 无通知且设备已断开");
                                break;
                            }
                            Ok(true) => {
                                last_notif = tokio::time::Instant::now();
                            }
                        }
                    }
                }
            }

            // 断开时把还按着的键交代清楚:只发过按下、没发过松开的话,
            // 消费方会一直以为那个键按着(语音键卡住尤其难受——录音停不下来)。
            // 与 USB 监控线程退出时的收尾对齐。
            state.release_all(&event_tx);
            pending.lock().unwrap().clear();
            running.store(false, Ordering::SeqCst);
            log::info!(target: "gatt", "通知监听已退出");
        });
        // 保存 JoinHandle,disconnect 时 await —— 防止重连时旧 task 的 pending.clear()
        // 覆盖新 task 刚注册的 pending response(参见 disconnect 内注释)。
        if let Ok(mut slot) = self.notify_handle.lock() {
            *slot = Some(notify_handle);
        }
    }

    /// 断开连接(含 3s 超时保护,避免 disconnect().await 卡死)
    pub async fn disconnect(&self) -> Result<()> {
        self.running.store(false, Ordering::SeqCst);

        let p = self.peripheral.lock().unwrap().take();
        let c = self.chars.lock().unwrap().take();
        self.pending_responses.lock().unwrap().clear();

        if let (Some(p), Some(c)) = (p, c) {
            let _ = tokio::time::timeout(std::time::Duration::from_secs(3), async {
                let _ = p.unsubscribe(&c.event).await;
                let _ = p.unsubscribe(&c.audio).await;
                let _ = p.disconnect().await;
            })
            .await;
            log::info!(target: "gatt", "disconnect 完成(含超时保护)");
        }

        // 等通知 task 真正退出,避免重连场景下两个 task 短时并存:
        // 旧 task 的尾部 `pending.clear()` 会清掉新 task 刚注册的 pending response,
        // 导致新会话的命令交互超时。task 自身已有 heartbeat + is_connected 兜底,
        // 但设备断开不一定让 `notifications().next()` 立即返回 None,这里加 3s 超时保护。
        let notify_handle = self.notify_handle.lock().unwrap().take();
        if let Some(h) = notify_handle {
            let _ = tokio::time::timeout(std::time::Duration::from_secs(3), h).await;
        }

        // join 音频线程(它在 running=false 后最多 1s 退出,join 同样加超时保护)。
        // 不 join 会悬挂线程到 recv_timeout 超时;join 让资源释放可预测。
        let audio_handle = self.audio_thread_handle.lock().unwrap().take();
        if let Some(h) = audio_handle {
            let _ = h.join();
        }

        Ok(())
    }

    #[allow(dead_code)]
    pub fn is_connected(&self) -> bool {
        let _p = self.peripheral.lock().unwrap();
        self.running.load(Ordering::SeqCst)
    }

    pub fn running(&self) -> Arc<AtomicBool> {
        self.running.clone()
    }
}

// ============ 辅助 ============

/// subscribe 失败时手动写 CCC 描述符 (0x2902) 启用通知
async fn subscribe_with_ccc_fallback(
    p: &Peripheral,
    char: &btleplug::api::Characteristic,
    label: &str,
) -> Result<()> {
    log::info!(target: "gatt", "启用 {} 通知...", label);
    match tokio::time::timeout(std::time::Duration::from_secs(3), p.subscribe(char)).await {
        Ok(Ok(())) => {
            log::info!(target: "gatt", "{} 订阅成功", label);
            Ok(())
        }
        result => {
            log::warn!(target: "gatt", "{} subscribe 失败 {:?},尝试手动写 CCC", label, result);
            let ccc_desc = char
                .descriptors
                .iter()
                .find(|d| d.uuid == CCC_DESCRIPTOR_UUID);
            match ccc_desc {
                Some(desc) => {
                    let ccc: [u8; 2] = [0x01, 0x00];
                    p.write_descriptor(desc, &ccc)
                        .await
                        .map_err(|e| anyhow::anyhow!("{} CCC 写入失败: {}", label, e))?;
                    log::info!(target: "gatt", "{} 手动 CCC 写入成功", label);
                    Ok(())
                }
                None => Err(anyhow::anyhow!(
                    "{} 特征值无 CCC 描述符 (0x2902),无法启用通知",
                    label
                )),
            }
        }
    }
}

/// 处理 Event 通知 → 命令响应/状态变更
fn handle_event(
    data: &[u8],
    state: &mut GattEventState,
    event_tx: &broadcast::Sender<BoardEvent>,
    pending: &Arc<Mutex<HashMap<u8, PendingResponse>>>,
) {
    let Some((cmd, _len, payload)) = protocol::parse_packet(data) else {
        log::warn!(target: "gatt", "[GATT] Event 解析失败: data_len={}", data.len());
        let _ = event_tx.send(BoardEvent::Error(ErrorEvent {
            message: format!("GATT Event 解析失败(len={})", data.len()),
            recoverable: true,
        }));
        return;
    };

    // 1. 命令响应(匹配 pending map)
    {
        let mut pending_guard = pending.lock().unwrap();
        if let Some(pending_resp) = pending_guard.remove(&cmd) {
            let mut response = vec![cmd, _len];
            response.extend_from_slice(payload);
            let _ = pending_resp.tx.send(response); // oneshot send
            return;
        }
    }

    // 2. 异步事件,按 CMD 分发
    match cmd {
        #[cfg(feature = "test-mode")]
        CMD_AI_FACTORY_KEY_EVENT => {
            if let Ok(event) = parse_factory_key_event_unscoped(data) {
                let _ = event_tx.send(BoardEvent::FactoryKey(event));
            }
        }
        // Consumer 键事件:[CMD=0x0C][LEN=2][key_low][key_high]
        0x0C => {
            if payload.len() >= 2 {
                let key_value = ((payload[1] as u16) << 8) | (payload[0] as u16);

                // 按下/松开事件交给账本 + 聚合器,语义与 USB 完全一致:
                // 转旋钮不会顶掉按住的键,真正的松手才清空。
                let frame = state.consumer.on_frame(key_value, Instant::now());
                let aggregator = state
                    .aggregator
                    .get_or_insert_with(|| KeyStateAggregator::new(event_tx.clone()));
                for batch in frame.batches {
                    aggregator.report_change(KeySource::Gatt, batch, None);
                }

                if is_ai_voice_consumer_code(key_value) {
                    state.ai_voice_pressed = true;
                    let _ =
                        event_tx.send(BoardEvent::AiVoiceKey(AiVoiceKeyEvent { pressed: true }));
                }

                // 模式切换拨杆
                let dial = find_key_index_by_value(key_value)
                    .and_then(|index| key_index_to_mode(index).map(|mode| (index, mode)));
                if let Some((key_index, (mode_value, mode_name))) = dial {
                    state.mode_key_pressed = key_index;
                    let _ = event_tx.send(BoardEvent::ModeChange(ModeChangeEvent {
                        mode: mode_name.to_string(),
                        mode_value,
                        source: ModeSource::Dial,
                    }));
                }

                // 下面两件事只在账本**真正清空**时做。挂在字面的 0x0000 上是不对的:
                // 紧跟旋钮那一格的收尾帧也是 0x0000,按住语音键转旋钮会因此误报一次释放。
                if frame.cleared {
                    if state.ai_voice_pressed {
                        state.ai_voice_pressed = false;
                        let _ = event_tx
                            .send(BoardEvent::AiVoiceKey(AiVoiceKeyEvent { pressed: false }));
                    }
                    // 拨杆松开 → 回到 CHAT
                    let prev_mode = state.mode_key_pressed;
                    state.mode_key_pressed = 0;
                    if prev_mode == 9 || prev_mode == 10 {
                        let _ = event_tx.send(BoardEvent::ModeChange(ModeChangeEvent {
                            mode: "CHAT".to_string(),
                            mode_value: 0,
                            source: ModeSource::Dial,
                        }));
                    }
                }
            }
        }
        CMD_AUDIO_DATA => {
            // Audio 偶尔经 Event 通道到达,实际音频走 FE63
            log::debug!(target: "gatt", "[GATT] Event 通道收到 Audio 数据");
        }
        CMD_STATUS => {
            // 模式变更:[CMD=0x12][LEN][SUB=0xC9][MODE]
            if payload.len() >= 2 && payload[0] == CMD_WORK_MODE_DATA {
                let mode_value = payload[1];
                if let Some(mode) = WorkMode::from_u8(mode_value) {
                    let _ = event_tx.send(BoardEvent::ModeChange(ModeChangeEvent {
                        mode: mode.display_name().to_string(),
                        mode_value,
                        source: ModeSource::Dial,
                    }));
                }
            }
        }
        CMD_DEVICE_DISCONNECT => {
            // 设备主动断开(关机/用户操作):当非正常断开处理,重启后自动重连。
            // 不在此置 ble_auto_connect=false —— 那会导致设备关机重启后无法自动重连。
            // hotplug 通过扫描失败上限兜底永久离开。
            log::info!(target: "gatt", "[GATT] 收到设备主动断开通知 (CMD=0x60),当异常断开处理");
        }
        _ => {
            log::debug!(target: "gatt", "[GATT] 未处理 Event CMD=0x{:02X}", cmd);
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::kernel::protocol_hid::AI_VOICE_MEDIA_CODE;

    /// 非阻塞 drain broadcast,避免 blocking_recv 阻塞测试
    fn drain(rx: &mut broadcast::Receiver<BoardEvent>) -> Vec<BoardEvent> {
        let mut v = Vec::new();
        while let Ok(e) = rx.try_recv() {
            v.push(e);
        }
        v
    }

    /// GATT 包格式 `[CMD=0x0C][LEN=2][lo][hi]`。
    fn key_frame(key_value: u16) -> Vec<u8> {
        vec![
            0x0C,
            0x02,
            (key_value & 0xFF) as u8,
            ((key_value >> 8) & 0xFF) as u8,
        ]
    }

    #[cfg(feature = "test-mode")]
    #[test]
    fn gatt_monitor_emits_factory_physical_event() {
        let (tx, mut rx) = broadcast::channel(4);
        let pending = Arc::new(Mutex::new(HashMap::new()));
        let mut state = GattEventState::default();
        let frame = [
            CMD_AI_FACTORY_KEY_EVENT,
            0x06,
            0x01,
            0x34,
            0x12,
            0x06,
            0x01,
            0x09,
        ];

        handle_event(&frame, &mut state, &tx, &pending);
        match rx.try_recv().expect("factory event") {
            BoardEvent::FactoryKey(event) => {
                assert_eq!(event.session, 0x1234);
                assert_eq!(event.input_index, 6);
                assert!(event.pressed);
            }
            other => panic!("unexpected event: {other:?}"),
        }
    }

    /// 从事件流里挑出按键事件(忽略 ComboKey 之类的旁支)。
    fn key_events(rx: &mut broadcast::Receiver<BoardEvent>) -> Vec<(usize, bool)> {
        drain(rx)
            .into_iter()
            .filter_map(|event| match event {
                BoardEvent::KeyPress(key) => Some((key.key_index, key.pressed)),
                _ => None,
            })
            .collect()
    }

    /// 验证 key_index=0(KEY0,音量A相)的释放事件不会被吞掉。
    ///
    /// 回归背景:`last_key_index` 曾用 `usize=0` 兼做"未按下"哨兵,
    /// 而 `key_index=0` 是合法键——按下时 last_key_index=0,释放时 `> 0` 判 false
    /// 导致 KEY0 永远停在 pressed:true。
    ///
    /// 现在 KEY0/KEY1 被当作旋钮脉冲:一帧之内就进出一次(按下+松开),
    /// 所以"释放不会丢"这条不变量仍然成立,只是不再依赖后续那帧 0x0000。
    #[test]
    fn key0_pulse_emits_both_press_and_release() {
        let (tx, mut rx) = broadcast::channel(64);
        let pending = Arc::new(Mutex::new(HashMap::new()));
        let mut state = GattEventState::default();

        // KEY0 = consumer 0x0F07 → key_index 0
        handle_event(&key_frame(0x0F07), &mut state, &tx, &pending);
        assert_eq!(
            key_events(&mut rx),
            vec![(0, true), (0, false)],
            "KEY0 的按下与释放都必须发出——释放丢失正是当初修的 bug"
        );

        // 紧跟其后的 0x0000 是这一格的收尾,不该再产生任何按键事件
        handle_event(&key_frame(0x0000), &mut state, &tx, &pending);
        assert!(key_events(&mut rx).is_empty(), "脉冲收尾不该重复发释放");
    }

    /// 蓝牙下「按住 Tab → 转旋钮 → 松手」必须发出 Tab 的松开事件。
    ///
    /// 回归背景(2026-07-28 真机实测):BLE 与 USB 走同一条单值 Consumer 流,
    /// 而这里原来只记得住「最后按下的那个键」。转旋钮时这个记录被旋钮顶掉,
    /// 随后的 0x0000 把释放算在了旋钮头上——**Tab 的松手事件永远不会发出**,
    /// 上层按住 Tab 打开的应用切换器就一直卡着,只能等 30 秒硬超时强制放开 Command。
    /// 日志原文:「按键编排会话超过 30s 仍未收到抬起(键=Tab)」。
    #[test]
    fn holding_tab_through_knob_turns_still_reports_the_release() {
        let (tx, mut rx) = broadcast::channel(64);
        let pending = Arc::new(Mutex::new(HashMap::new()));
        let mut state = GattEventState::default();

        // 按住 Tab(0x0F01 → key_index 3)
        handle_event(&key_frame(0x0F01), &mut state, &tx, &pending);
        assert_eq!(key_events(&mut rx), vec![(3, true)]);

        // 连转两格旋钮(0x0F08 → key_index 1),每格后面跟一帧收尾
        for _ in 0..2 {
            handle_event(&key_frame(0x0F08), &mut state, &tx, &pending);
            assert_eq!(
                key_events(&mut rx),
                vec![(1, true), (1, false)],
                "旋钮该敲一下就走"
            );
            handle_event(&key_frame(0x0000), &mut state, &tx, &pending);
            assert!(
                key_events(&mut rx).is_empty(),
                "脉冲收尾不能把按住的 Tab 判成松开"
            );
        }

        // 松手:Tab 的释放必须发出来
        handle_event(&key_frame(0x0000), &mut state, &tx, &pending);
        assert_eq!(
            key_events(&mut rx),
            vec![(3, false)],
            "松手后必须发出 Tab 的释放——缺了它,上层按住的修饰键永远放不掉"
        );
    }

    /// 按住语音键转旋钮,不能误报一次语音键释放(会把录音打断)。
    #[test]
    fn holding_voice_key_through_a_knob_turn_does_not_report_release() {
        let (tx, mut rx) = broadcast::channel(64);
        let pending = Arc::new(Mutex::new(HashMap::new()));
        let mut state = GattEventState::default();

        handle_event(&key_frame(AI_VOICE_MEDIA_CODE), &mut state, &tx, &pending);
        let _ = drain(&mut rx);

        handle_event(&key_frame(0x0F08), &mut state, &tx, &pending);
        handle_event(&key_frame(0x0000), &mut state, &tx, &pending);
        let voice_events: Vec<bool> = drain(&mut rx)
            .into_iter()
            .filter_map(|event| match event {
                BoardEvent::AiVoiceKey(voice) => Some(voice.pressed),
                _ => None,
            })
            .collect();
        assert!(
            voice_events.is_empty(),
            "转旋钮期间不该报语音键释放:{voice_events:?}"
        );
        assert!(state.ai_voice_pressed, "语音键仍应处于按住状态");

        // 真松手才释放
        handle_event(&key_frame(0x0000), &mut state, &tx, &pending);
        let released = drain(&mut rx)
            .into_iter()
            .any(|event| matches!(event, BoardEvent::AiVoiceKey(voice) if !voice.pressed));
        assert!(released, "真松手必须报语音键释放");
    }

    /// 断连时必须把还按着的键交代成松开,别让消费方一直以为它按着。
    ///
    /// 状态本身随通知任务销毁,所以「不清也不会串到下一次连接」——但消费方那边不会
    /// 自动忘记:只收到过按下、没收到松开的键会永远停在按下。语音键卡住最难受,
    /// 录音停不下来。USB 监控线程退出时有同样的收尾,这里对齐。
    #[test]
    fn disconnect_releases_everything_still_held() {
        let (tx, mut rx) = broadcast::channel(64);
        let pending = Arc::new(Mutex::new(HashMap::new()));
        let mut state = GattEventState::default();

        // 按住语音键,再按住 Tab(单值通道下第二个键是追加)
        handle_event(&key_frame(AI_VOICE_MEDIA_CODE), &mut state, &tx, &pending);
        handle_event(&key_frame(0x0F01), &mut state, &tx, &pending);
        let _ = drain(&mut rx);

        // 设备断开:通知任务退出前的收尾
        state.release_all(&tx);

        let events = drain(&mut rx);
        let released: Vec<usize> = events
            .iter()
            .filter_map(|event| match event {
                BoardEvent::KeyPress(key) if !key.pressed => Some(key.key_index),
                _ => None,
            })
            .collect();
        assert!(
            released.contains(&6) && released.contains(&3),
            "断连必须补发所有按住键的松开,实际:{released:?}"
        );
        assert!(
            events
                .iter()
                .any(|event| matches!(event, BoardEvent::AiVoiceKey(voice) if !voice.pressed)),
            "断连必须补发语音键释放,否则录音停不下来"
        );
        assert!(!state.ai_voice_pressed);
    }

    /// 锁住 CCC UUID 的字节序:必须等于标准 Bluetooth Base UUID `00002902-...-00805f9b34fb`。
    ///
    /// 回归背景:旧代码的 `|| Uuid::from_u128(0x2902)` 分支永远不匹配标准 CCC UUID,
    /// 若哪天有人把主分支删了改成 `from_u128(0x2902)`,CCC fallback 会静默失效。
    #[test]
    fn ccc_descriptor_uuid_matches_bluetooth_spec() {
        assert_eq!(
            CCC_DESCRIPTOR_UUID.to_string(),
            "00002902-0000-1000-8000-00805f9b34fb"
        );
    }
}