150 lines
5.4 KiB
Objective-C
150 lines
5.4 KiB
Objective-C
//
|
||
// 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
|