1

2026-01-23 21:51:37 +08:00
parent 6ad9783bcb
commit 77fd46aa34
26 changed files with 3681 additions and 199 deletions
--- a/keyBoard/Class/AiTalk/VC/KBChatTestVC.m
+++ b/keyBoard/Class/AiTalk/VC/KBChatTestVC.m
@@ -0,0 +1,149 @@
+//
+//  KBChatTestVC.m
+//  keyBoard
+//
+//  Created by Kiro on 2026/1/23.
+//
+
+#import "KBChatTestVC.h"
+#import "KBChatTableView.h"
+#import <Masonry/Masonry.h>
+
+@interface KBChatTestVC ()
+
+@property (nonatomic, strong) KBChatTableView *chatView;
+@property (nonatomic, strong) UIButton *addUserMessageButton;
+@property (nonatomic, strong) UIButton *addAIMessageButton;
+@property (nonatomic, strong) UIButton *clearButton;
+
+@end
+
+@implementation KBChatTestVC
+
+- (void)viewDidLoad {
+    [super viewDidLoad];
+    
+    self.title = @"聊天 UI 测试";
+    self.view.backgroundColor = [UIColor whiteColor];
+    
+    [self setupUI];
+    [self loadMockData];
+}
+
+- (void)setupUI {
+    // 聊天视图
+    self.chatView = [[KBChatTableView alloc] init];
+    self.chatView.backgroundColor = [UIColor colorWithWhite:0.95 alpha:1.0];
+    [self.view addSubview:self.chatView];
+    
+    // 按钮容器
+    UIView *buttonContainer = [[UIView alloc] init];
+    buttonContainer.backgroundColor = [UIColor whiteColor];
+    [self.view addSubview:buttonContainer];
+    
+    // 添加用户消息按钮
+    self.addUserMessageButton = [UIButton buttonWithType:UIButtonTypeSystem];
+    [self.addUserMessageButton setTitle:@"添加用户消息" forState:UIControlStateNormal];
+    [self.addUserMessageButton addTarget:self
+                                   action:@selector(addUserMessage)
+                         forControlEvents:UIControlEventTouchUpInside];
+    [buttonContainer addSubview:self.addUserMessageButton];
+    
+    // 添加 AI 消息按钮
+    self.addAIMessageButton = [UIButton buttonWithType:UIButtonTypeSystem];
+    [self.addAIMessageButton setTitle:@"添加 AI 消息" forState:UIControlStateNormal];
+    [self.addAIMessageButton addTarget:self
+                                 action:@selector(addAIMessage)
+                       forControlEvents:UIControlEventTouchUpInside];
+    [buttonContainer addSubview:self.addAIMessageButton];
+    
+    // 清空按钮
+    self.clearButton = [UIButton buttonWithType:UIButtonTypeSystem];
+    [self.clearButton setTitle:@"清空" forState:UIControlStateNormal];
+    [self.clearButton addTarget:self
+                          action:@selector(clearMessages)
+                forControlEvents:UIControlEventTouchUpInside];
+    [buttonContainer addSubview:self.clearButton];
+    
+    // 布局
+    [self.chatView mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.top.equalTo(self.view.mas_safeAreaLayoutGuideTop);
+        make.left.right.equalTo(self.view);
+        make.bottom.equalTo(buttonContainer.mas_top);
+    }];
+    
+    [buttonContainer mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.left.right.equalTo(self.view);
+        make.bottom.equalTo(self.view.mas_safeAreaLayoutGuideBottom);
+        make.height.mas_equalTo(60);
+    }];
+    
+    [self.addUserMessageButton mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.left.equalTo(buttonContainer).offset(16);
+        make.centerY.equalTo(buttonContainer);
+    }];
+    
+    [self.addAIMessageButton mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.center.equalTo(buttonContainer);
+    }];
+    
+    [self.clearButton mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.right.equalTo(buttonContainer).offset(-16);
+        make.centerY.equalTo(buttonContainer);
+    }];
+}
+
+- (void)loadMockData {
+    // 模拟对话数据
+    [self.chatView addUserMessage:@"你好"];
+    
+    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, 0.5 * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
+        [self.chatView addAssistantMessage:@"你好！很高兴见到你。"
+                             audioDuration:3.0
+                                 audioData:[self generateMockAudioData:3.0]];
+    });
+    
+    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, 1.0 * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
+        [self.chatView addUserMessage:@"今天天气怎么样？"];
+    });
+    
+    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, 1.5 * NSEC_PER_SEC), dispatch_get_main_queue(), ^{
+        [self.chatView addAssistantMessage:@"今天天气不错，阳光明媚，温度适宜，非常适合外出活动。"
+                             audioDuration:8.0
+                                 audioData:[self generateMockAudioData:8.0]];
+    });
+}
+
+- (void)addUserMessage {
+    static NSInteger userMessageCount = 0;
+    userMessageCount++;
+    NSString *text = [NSString stringWithFormat:@"这是用户消息 %ld", (long)userMessageCount];
+    [self.chatView addUserMessage:text];
+}
+
+- (void)addAIMessage {
+    static NSInteger aiMessageCount = 0;
+    aiMessageCount++;
+    NSString *text = [NSString stringWithFormat:@"这是 AI 回复消息 %ld，包含一些较长的文本内容，用于测试气泡的自适应高度和换行效果。", (long)aiMessageCount];
+    NSTimeInterval duration = 5.0 + (aiMessageCount % 10);
+    [self.chatView addAssistantMessage:text
+                         audioDuration:duration
+                             audioData:[self generateMockAudioData:duration]];
+}
+
+- (void)clearMessages {
+    [self.chatView clearMessages];
+}
+
+/// 生成模拟音频数据（实际项目中应该从 TTS 获取）
+- (NSData *)generateMockAudioData:(NSTimeInterval)duration {
+    // 这里返回 nil，实际使用时应该传入真实的音频数据
+    // 为了测试，可以加载一个本地音频文件
+    NSString *audioPath = [[NSBundle mainBundle] pathForResource:@"ai_test" ofType:@"m4a"];
+    if (audioPath) {
+        return [NSData dataWithContentsOfFile:audioPath];
+    }
+    return nil;
+}
+
+@end