702 lines
26 KiB
Objective-C
702 lines
26 KiB
Objective-C
//
|
||
// KBChatTableView.m
|
||
// keyBoard
|
||
//
|
||
// Created by Kiro on 2026/1/23.
|
||
//
|
||
|
||
#import "KBChatTableView.h"
|
||
#import "KBAiChatMessage.h"
|
||
#import "KBChatUserMessageCell.h"
|
||
#import "KBChatAssistantMessageCell.h"
|
||
#import "KBChatTimeCell.h"
|
||
#import "AiVM.h"
|
||
#import <MJRefresh/MJRefresh.h>
|
||
#import <Masonry/Masonry.h>
|
||
#import <AVFoundation/AVFoundation.h>
|
||
|
||
static NSString * const kUserCellIdentifier = @"KBChatUserMessageCell";
|
||
static NSString * const kAssistantCellIdentifier = @"KBChatAssistantMessageCell";
|
||
static NSString * const kTimeCellIdentifier = @"KBChatTimeCell";
|
||
|
||
/// 时间戳显示间隔（秒）
|
||
static const NSTimeInterval kTimestampInterval = 5 * 60; // 5 分钟
|
||
|
||
@interface KBChatTableView () <UITableViewDataSource, UITableViewDelegate, KBChatAssistantMessageCellDelegate, AVAudioPlayerDelegate>
|
||
|
||
@property (nonatomic, strong) BaseTableView *tableView;
|
||
@property (nonatomic, strong) NSMutableArray<KBAiChatMessage *> *messages;
|
||
@property (nonatomic, strong) AVAudioPlayer *audioPlayer;
|
||
@property (nonatomic, strong) NSIndexPath *playingCellIndexPath;
|
||
@property (nonatomic, strong) AiVM *aiVM;
|
||
@property (nonatomic, assign) BOOL hasMoreData;
|
||
@property (nonatomic, assign) CGFloat contentBottomInset;
|
||
|
||
@end
|
||
|
||
@implementation KBChatTableView
|
||
|
||
- (instancetype)initWithFrame:(CGRect)frame {
|
||
self = [super initWithFrame:frame];
|
||
if (self) {
|
||
[self setup];
|
||
}
|
||
return self;
|
||
}
|
||
|
||
- (instancetype)initWithCoder:(NSCoder *)coder {
|
||
self = [super initWithCoder:coder];
|
||
if (self) {
|
||
[self setup];
|
||
}
|
||
return self;
|
||
}
|
||
|
||
- (void)setup {
|
||
self.messages = [[NSMutableArray alloc] init];
|
||
self.aiVM = [[AiVM alloc] init];
|
||
self.hasMoreData = YES;
|
||
|
||
// 创建 TableView
|
||
self.tableView = [[BaseTableView alloc] initWithFrame:self.bounds
|
||
style:UITableViewStylePlain];
|
||
self.tableView.useEmptyDataSet = false;
|
||
self.tableView.dataSource = self;
|
||
self.tableView.delegate = self;
|
||
self.tableView.separatorStyle = UITableViewCellSeparatorStyleNone;
|
||
self.tableView.backgroundColor = [UIColor clearColor];
|
||
// 关键修复：使用合理的估算高度（避免抖动，但不能为0）
|
||
self.tableView.estimatedRowHeight = 80;
|
||
self.tableView.estimatedSectionHeaderHeight = 0;
|
||
self.tableView.estimatedSectionFooterHeight = 0;
|
||
self.tableView.rowHeight = UITableViewAutomaticDimension;
|
||
self.tableView.showsVerticalScrollIndicator = YES;
|
||
// 关键修复：禁用内容自动调整，防止与外层 CollectionView 冲突
|
||
if (@available(iOS 11.0, *)) {
|
||
self.tableView.contentInsetAdjustmentBehavior = UIScrollViewContentInsetAdjustmentNever;
|
||
}
|
||
[self addSubview:self.tableView];
|
||
|
||
// 注册 Cell
|
||
[self.tableView registerClass:[KBChatUserMessageCell class]
|
||
forCellReuseIdentifier:kUserCellIdentifier];
|
||
[self.tableView registerClass:[KBChatAssistantMessageCell class]
|
||
forCellReuseIdentifier:kAssistantCellIdentifier];
|
||
[self.tableView registerClass:[KBChatTimeCell class]
|
||
forCellReuseIdentifier:kTimeCellIdentifier];
|
||
|
||
// 布局
|
||
[self.tableView mas_makeConstraints:^(MASConstraintMaker *make) {
|
||
make.edges.equalTo(self);
|
||
}];
|
||
|
||
// 关键修复：减少初始 contentBottomInset，因为 chatView 已经通过约束避开了底部
|
||
self.contentBottomInset = 20; // 简单的缓冲空间
|
||
[self updateContentBottomInset:self.contentBottomInset];
|
||
|
||
__weak typeof(self) weakSelf = self;
|
||
self.tableView.mj_footer = [MJRefreshAutoNormalFooter footerWithRefreshingBlock:^{
|
||
__strong typeof(weakSelf) strongSelf = weakSelf;
|
||
if (!strongSelf) {
|
||
return;
|
||
}
|
||
|
||
if (!strongSelf.hasMoreData) {
|
||
[strongSelf.tableView.mj_footer endRefreshingWithNoMoreData];
|
||
return;
|
||
}
|
||
|
||
if ([strongSelf.delegate respondsToSelector:@selector(chatTableViewDidTriggerLoadMore:)]) {
|
||
[strongSelf.delegate chatTableViewDidTriggerLoadMore:strongSelf];
|
||
} else {
|
||
[strongSelf.tableView.mj_footer endRefreshing];
|
||
}
|
||
}];
|
||
|
||
// 隐藏"已经全部加载完毕"的提示文字
|
||
MJRefreshAutoNormalFooter *footer = (MJRefreshAutoNormalFooter *)self.tableView.mj_footer;
|
||
footer.stateLabel.hidden = YES; // 隐藏状态文字
|
||
footer.refreshingBlock = footer.refreshingBlock; // 保持刷新逻辑
|
||
|
||
self.tableView.mj_footer.hidden = YES;
|
||
}
|
||
|
||
#pragma mark - Public Methods
|
||
|
||
- (void)addUserMessage:(NSString *)text {
|
||
KBAiChatMessage *message = [KBAiChatMessage userMessageWithText:text];
|
||
[self addMessage:message autoScroll:YES];
|
||
}
|
||
|
||
- (void)addAssistantMessage:(NSString *)text
|
||
audioDuration:(NSTimeInterval)duration
|
||
audioData:(NSData *)audioData {
|
||
KBAiChatMessage *message = [KBAiChatMessage assistantMessageWithText:text
|
||
audioDuration:duration
|
||
audioData:audioData];
|
||
[self addMessage:message autoScroll:YES];
|
||
}
|
||
|
||
- (void)addAssistantMessage:(NSString *)text
|
||
audioId:(NSString *)audioId {
|
||
KBAiChatMessage *message = [KBAiChatMessage assistantMessageWithText:text
|
||
audioId:audioId];
|
||
message.needsTypewriterEffect = YES; // 新消息需要打字机效果
|
||
[self addMessage:message autoScroll:YES];
|
||
}
|
||
|
||
- (void)updateLastAssistantMessage:(NSString *)text {
|
||
// 查找最后一条未完成的 AI 消息
|
||
for (NSInteger i = self.messages.count - 1; i >= 0; i--) {
|
||
KBAiChatMessage *message = self.messages[i];
|
||
if (message.type == KBAiChatMessageTypeAssistant && !message.isComplete) {
|
||
NSLog(@"[KBChatTableView] 更新最后一条 AI 消息 - 索引: %ld, 文本长度: %lu, needsTypewriter: %d",
|
||
(long)i, (unsigned long)text.length, message.needsTypewriterEffect);
|
||
message.text = text;
|
||
|
||
// 直接更新 Cell 的文本，不刷新整个 Cell（避免打断打字机效果）
|
||
NSIndexPath *indexPath = [NSIndexPath indexPathForRow:i inSection:0];
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
NSLog(@"[KBChatTableView] 找到 Cell，直接配置消息");
|
||
// 直接调用 configureWithMessage，让 Cell 自己决定是否使用打字机效果
|
||
[cell configureWithMessage:message];
|
||
} else {
|
||
NSLog(@"[KBChatTableView] 未找到 Cell 或类型不匹配");
|
||
}
|
||
return;
|
||
}
|
||
}
|
||
|
||
// 如果没找到，添加新消息
|
||
NSLog(@"[KBChatTableView] 未找到未完成的 AI 消息，添加新消息");
|
||
[self addAssistantMessage:text audioDuration:0 audioData:nil];
|
||
}
|
||
|
||
- (void)markLastAssistantMessageComplete {
|
||
for (NSInteger i = self.messages.count - 1; i >= 0; i--) {
|
||
KBAiChatMessage *message = self.messages[i];
|
||
if (message.type == KBAiChatMessageTypeAssistant) {
|
||
NSLog(@"[KBChatTableView] 标记消息完成 - 索引: %ld, 文本: %@", (long)i, message.text);
|
||
message.isComplete = YES;
|
||
message.needsTypewriterEffect = NO; // 完成后不再需要打字机效果
|
||
|
||
// 刷新 Cell 以显示完整文本
|
||
NSIndexPath *indexPath = [NSIndexPath indexPathForRow:i inSection:0];
|
||
[self.tableView reloadRowsAtIndexPaths:@[indexPath]
|
||
withRowAnimation:UITableViewRowAnimationNone];
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
- (void)clearMessages {
|
||
[self.messages removeAllObjects];
|
||
[self.tableView reloadData];
|
||
[self updateFooterVisibility];
|
||
}
|
||
|
||
- (void)scrollToBottom {
|
||
[self scrollToBottomAnimated:YES];
|
||
}
|
||
|
||
- (void)scrollToBottomAnimated:(BOOL)animated {
|
||
if (self.messages.count == 0) return;
|
||
|
||
// 关键修复：使用 layoutIfNeeded 确保布局完成后再滚动
|
||
[self.tableView layoutIfNeeded];
|
||
|
||
NSIndexPath *lastIndexPath = [NSIndexPath indexPathForRow:self.messages.count - 1
|
||
inSection:0];
|
||
|
||
// 直接滚动到最后一条消息，不检查是否可见（确保新消息能被看到）
|
||
[self.tableView scrollToRowAtIndexPath:lastIndexPath
|
||
atScrollPosition:UITableViewScrollPositionBottom
|
||
animated:animated];
|
||
}
|
||
|
||
#pragma mark - Public Helpers
|
||
|
||
- (void)endLoadMoreWithHasMoreData:(BOOL)hasMoreData {
|
||
self.hasMoreData = hasMoreData;
|
||
if (hasMoreData) {
|
||
[self.tableView.mj_footer endRefreshing];
|
||
} else {
|
||
[self.tableView.mj_footer endRefreshingWithNoMoreData];
|
||
}
|
||
[self updateFooterVisibility];
|
||
}
|
||
|
||
- (void)resetNoMoreData {
|
||
self.hasMoreData = YES;
|
||
[self.tableView.mj_footer resetNoMoreData];
|
||
[self updateFooterVisibility];
|
||
}
|
||
|
||
- (void)updateContentBottomInset:(CGFloat)bottomInset {
|
||
self.contentBottomInset = bottomInset;
|
||
UIEdgeInsets insets = self.tableView.contentInset;
|
||
insets.bottom = bottomInset;
|
||
self.tableView.contentInset = insets;
|
||
self.tableView.scrollIndicatorInsets = insets;
|
||
}
|
||
|
||
- (void)addMessage:(KBAiChatMessage *)message
|
||
autoScroll:(BOOL)autoScroll {
|
||
if (!message) {
|
||
return;
|
||
}
|
||
|
||
if (message.type == KBAiChatMessageTypeAssistant &&
|
||
message.needsTypewriterEffect &&
|
||
!message.isComplete) {
|
||
[self stopPreviousIncompleteAssistantMessageIfNeeded];
|
||
}
|
||
|
||
NSInteger oldCount = self.messages.count;
|
||
[self insertMessageWithTimestamp:message];
|
||
|
||
NSInteger newCount = self.messages.count;
|
||
NSInteger insertedCount = newCount - oldCount;
|
||
|
||
if (insertedCount > 0) {
|
||
NSMutableArray *indexPaths = [NSMutableArray array];
|
||
for (NSInteger i = oldCount; i < newCount; i++) {
|
||
[indexPaths addObject:[NSIndexPath indexPathForRow:i inSection:0]];
|
||
}
|
||
|
||
// 关键修复：批量插入前先布局，避免高度计算不准确
|
||
[self.tableView layoutIfNeeded];
|
||
|
||
[self.tableView insertRowsAtIndexPaths:indexPaths
|
||
withRowAnimation:UITableViewRowAnimationNone];
|
||
}
|
||
|
||
[self updateFooterVisibility];
|
||
|
||
if (autoScroll) {
|
||
// 关键修复：插入完成后立即滚动，使用 dispatch_async 确保插入动画完成
|
||
dispatch_async(dispatch_get_main_queue(), ^{
|
||
[self.tableView layoutIfNeeded]; // 再次确保布局完成
|
||
[self scrollToBottomAnimated:YES];
|
||
});
|
||
}
|
||
}
|
||
|
||
- (void)reloadWithMessages:(NSArray<KBAiChatMessage *> *)messages
|
||
keepOffset:(BOOL)keepOffset
|
||
scrollToBottom:(BOOL)scrollToBottom {
|
||
CGFloat oldContentHeight = self.tableView.contentSize.height;
|
||
CGFloat oldOffsetY = self.tableView.contentOffset.y;
|
||
|
||
[self.messages removeAllObjects];
|
||
if (messages.count > 0) {
|
||
for (KBAiChatMessage *message in messages) {
|
||
[self insertMessageWithTimestamp:message];
|
||
}
|
||
}
|
||
|
||
[self.tableView reloadData];
|
||
[self.tableView layoutIfNeeded];
|
||
[self updateFooterVisibility];
|
||
|
||
if (keepOffset) {
|
||
CGFloat newContentHeight = self.tableView.contentSize.height;
|
||
CGFloat delta = newContentHeight - oldContentHeight;
|
||
CGFloat offsetY = oldOffsetY + delta;
|
||
// 关键修复：使用非动画方式设置 offset，避免抖动
|
||
[self.tableView setContentOffset:CGPointMake(0, offsetY) animated:NO];
|
||
return;
|
||
}
|
||
|
||
if (scrollToBottom) {
|
||
// 关键修复：直接滚动，不使用延迟
|
||
[self scrollToBottomAnimated:NO];
|
||
}
|
||
}
|
||
|
||
#pragma mark - Private Methods
|
||
|
||
/// 插入消息并自动添加时间戳
|
||
- (void)insertMessageWithTimestamp:(KBAiChatMessage *)message {
|
||
// 判断是否需要插入时间戳
|
||
if ([self shouldInsertTimestampForMessage:message]) {
|
||
KBAiChatMessage *timeMessage = [KBAiChatMessage timeMessageWithTimestamp:message.timestamp];
|
||
[self.messages addObject:timeMessage];
|
||
}
|
||
|
||
[self.messages addObject:message];
|
||
}
|
||
|
||
/// 判断是否需要插入时间戳
|
||
- (BOOL)shouldInsertTimestampForMessage:(KBAiChatMessage *)message {
|
||
// 第一条消息总是显示时间
|
||
if (self.messages.count == 0) {
|
||
return YES;
|
||
}
|
||
|
||
// 查找最后一条非时间戳消息
|
||
KBAiChatMessage *lastMessage = nil;
|
||
for (NSInteger i = self.messages.count - 1; i >= 0; i--) {
|
||
KBAiChatMessage *msg = self.messages[i];
|
||
if (msg.type != KBAiChatMessageTypeTime) {
|
||
lastMessage = msg;
|
||
break;
|
||
}
|
||
}
|
||
|
||
if (!lastMessage) {
|
||
return YES;
|
||
}
|
||
|
||
// 计算时间间隔
|
||
NSTimeInterval interval = [message.timestamp timeIntervalSinceDate:lastMessage.timestamp];
|
||
|
||
// 超过 5 分钟或跨天则显示时间
|
||
if (interval >= kTimestampInterval) {
|
||
return YES;
|
||
}
|
||
|
||
NSCalendar *calendar = [NSCalendar currentCalendar];
|
||
NSDateComponents *lastComponents = [calendar components:NSCalendarUnitDay | NSCalendarUnitMonth | NSCalendarUnitYear
|
||
fromDate:lastMessage.timestamp];
|
||
NSDateComponents *currentComponents = [calendar components:NSCalendarUnitDay | NSCalendarUnitMonth | NSCalendarUnitYear
|
||
fromDate:message.timestamp];
|
||
|
||
return ![lastComponents isEqual:currentComponents];
|
||
}
|
||
|
||
/// 刷新并滚动到底部
|
||
- (void)reloadAndScroll {
|
||
// 使用 insert 而不是 reload，避免刷新已有的 Cell
|
||
NSInteger lastIndex = self.messages.count - 1;
|
||
if (lastIndex >= 0) {
|
||
NSIndexPath *indexPath = [NSIndexPath indexPathForRow:lastIndex inSection:0];
|
||
[self.tableView insertRowsAtIndexPaths:@[indexPath]
|
||
withRowAnimation:UITableViewRowAnimationNone];
|
||
}
|
||
|
||
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.1 * NSEC_PER_SEC)), dispatch_get_main_queue(), ^{
|
||
[self scrollToBottom];
|
||
});
|
||
}
|
||
|
||
- (void)stopPreviousIncompleteAssistantMessageIfNeeded {
|
||
for (NSInteger i = self.messages.count - 1; i >= 0; i--) {
|
||
KBAiChatMessage *msg = self.messages[i];
|
||
if (msg.type == KBAiChatMessageTypeAssistant && !msg.isComplete) {
|
||
msg.isComplete = YES;
|
||
msg.needsTypewriterEffect = NO;
|
||
|
||
NSIndexPath *oldIndexPath = [NSIndexPath indexPathForRow:i inSection:0];
|
||
KBChatAssistantMessageCell *oldCell = [self.tableView cellForRowAtIndexPath:oldIndexPath];
|
||
if ([oldCell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[oldCell stopTypewriterEffect];
|
||
oldCell.messageLabel.text = msg.text;
|
||
}
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
|
||
- (void)updateFooterVisibility {
|
||
BOOL canLoadMore = (self.delegate &&
|
||
[self.delegate respondsToSelector:@selector(chatTableViewDidTriggerLoadMore:)]);
|
||
self.tableView.mj_footer.hidden = !canLoadMore || self.messages.count == 0;
|
||
}
|
||
|
||
#pragma mark - UITableViewDataSource
|
||
|
||
- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {
|
||
return self.messages.count;
|
||
}
|
||
|
||
- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {
|
||
KBAiChatMessage *message = self.messages[indexPath.row];
|
||
|
||
NSLog(@"[KBChatTableView] cellForRow: %ld, 消息类型: %ld, needsTypewriter: %d, isComplete: %d",
|
||
(long)indexPath.row, (long)message.type, message.needsTypewriterEffect, message.isComplete);
|
||
|
||
switch (message.type) {
|
||
case KBAiChatMessageTypeUser: {
|
||
KBChatUserMessageCell *cell = [tableView dequeueReusableCellWithIdentifier:kUserCellIdentifier
|
||
forIndexPath:indexPath];
|
||
[cell configureWithMessage:message];
|
||
return cell;
|
||
}
|
||
|
||
case KBAiChatMessageTypeAssistant: {
|
||
KBChatAssistantMessageCell *cell = [tableView dequeueReusableCellWithIdentifier:kAssistantCellIdentifier
|
||
forIndexPath:indexPath];
|
||
cell.delegate = self;
|
||
[cell configureWithMessage:message];
|
||
|
||
// 更新播放状态
|
||
BOOL isPlaying = [indexPath isEqual:self.playingCellIndexPath];
|
||
[cell updateVoicePlayingState:isPlaying];
|
||
|
||
return cell;
|
||
}
|
||
|
||
case KBAiChatMessageTypeTime: {
|
||
KBChatTimeCell *cell = [tableView dequeueReusableCellWithIdentifier:kTimeCellIdentifier
|
||
forIndexPath:indexPath];
|
||
[cell configureWithMessage:message];
|
||
return cell;
|
||
}
|
||
}
|
||
}
|
||
|
||
- (void)setDelegate:(id<KBChatTableViewDelegate>)delegate {
|
||
_delegate = delegate;
|
||
[self updateFooterVisibility];
|
||
}
|
||
|
||
#pragma mark - UIScrollViewDelegate
|
||
|
||
- (void)scrollViewDidScroll:(UIScrollView *)scrollView {
|
||
if ([self.delegate respondsToSelector:@selector(chatTableViewDidScroll:scrollView:)]) {
|
||
[self.delegate chatTableViewDidScroll:self scrollView:scrollView];
|
||
}
|
||
}
|
||
|
||
/// 关键修复：优化嵌套滚动体验，减少边界弹簧效果导致的抖动
|
||
- (void)scrollViewWillEndDragging:(UIScrollView *)scrollView
|
||
withVelocity:(CGPoint)velocity
|
||
targetContentOffset:(inout CGPoint *)targetContentOffset {
|
||
|
||
CGFloat offsetY = scrollView.contentOffset.y;
|
||
CGFloat contentHeight = scrollView.contentSize.height;
|
||
CGFloat scrollViewHeight = scrollView.bounds.size.height;
|
||
|
||
// 如果内容不够长，禁用弹簧效果
|
||
if (contentHeight <= scrollViewHeight) {
|
||
scrollView.bounces = NO;
|
||
} else {
|
||
scrollView.bounces = YES;
|
||
|
||
// 在快速滑动到底部时，避免过度弹簧导致抖动
|
||
if (velocity.y < 0) { // 向上滑动（到底部）
|
||
CGFloat maxOffset = contentHeight - scrollViewHeight + scrollView.contentInset.bottom;
|
||
if (targetContentOffset->y > maxOffset) {
|
||
targetContentOffset->y = maxOffset;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
#pragma mark - KBChatAssistantMessageCellDelegate
|
||
|
||
- (void)assistantMessageCell:(KBChatAssistantMessageCell *)cell
|
||
didTapVoiceButtonForMessage:(KBAiChatMessage *)message {
|
||
|
||
NSIndexPath *indexPath = [self.tableView indexPathForCell:cell];
|
||
if (!indexPath) return;
|
||
|
||
// 如果正在播放同一条消息，则暂停
|
||
if ([indexPath isEqual:self.playingCellIndexPath]) {
|
||
[self stopPlayingAudio];
|
||
return;
|
||
}
|
||
|
||
// 停止之前的播放
|
||
[self stopPlayingAudio];
|
||
|
||
// 如果有 audioData，直接播放
|
||
if (message.audioData && message.audioData.length > 0) {
|
||
[self playAudioForMessage:message atIndexPath:indexPath];
|
||
return;
|
||
}
|
||
|
||
// 如果有 audioId，异步加载音频
|
||
if (message.audioId.length > 0) {
|
||
[self loadAndPlayAudioForMessage:message atIndexPath:indexPath];
|
||
return;
|
||
}
|
||
|
||
NSLog(@"[KBChatTableView] 没有音频数据或 audioId");
|
||
}
|
||
|
||
#pragma mark - Audio Playback
|
||
|
||
- (void)loadAndPlayAudioForMessage:(KBAiChatMessage *)message atIndexPath:(NSIndexPath *)indexPath {
|
||
// 显示加载动画
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[cell showLoadingAnimation];
|
||
}
|
||
|
||
// 开始轮询请求（最多5次，每次间隔0.5秒）
|
||
[self pollAudioForMessage:message atIndexPath:indexPath retryCount:0 maxRetries:5];
|
||
}
|
||
|
||
- (void)pollAudioForMessage:(KBAiChatMessage *)message
|
||
atIndexPath:(NSIndexPath *)indexPath
|
||
retryCount:(NSInteger)retryCount
|
||
maxRetries:(NSInteger)maxRetries {
|
||
|
||
__weak typeof(self) weakSelf = self;
|
||
[self.aiVM requestAudioWithAudioId:message.audioId
|
||
completion:^(NSString *_Nullable audioURL, NSError *_Nullable error) {
|
||
__strong typeof(weakSelf) strongSelf = weakSelf;
|
||
if (!strongSelf) return;
|
||
|
||
dispatch_async(dispatch_get_main_queue(), ^{
|
||
// 如果成功获取到 audioURL
|
||
if (!error && audioURL.length > 0) {
|
||
// 下载并播放音频
|
||
[strongSelf downloadAndPlayAudioFromURL:audioURL
|
||
forMessage:message
|
||
atIndexPath:indexPath];
|
||
return;
|
||
}
|
||
|
||
// 如果还没达到最大重试次数，继续轮询
|
||
if (retryCount < maxRetries - 1) {
|
||
NSLog(@"[KBChatTableView] 音频未就绪，0.5秒后重试 (%ld/%ld)",
|
||
(long)(retryCount + 1), (long)maxRetries);
|
||
|
||
// 0.5秒后重试
|
||
dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.5 * NSEC_PER_SEC)),
|
||
dispatch_get_main_queue(), ^{
|
||
[strongSelf pollAudioForMessage:message
|
||
atIndexPath:indexPath
|
||
retryCount:retryCount + 1
|
||
maxRetries:maxRetries];
|
||
});
|
||
} else {
|
||
// 达到最大重试次数，隐藏加载动画
|
||
KBChatAssistantMessageCell *cell = [strongSelf.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[cell hideLoadingAnimation];
|
||
}
|
||
NSLog(@"[KBChatTableView] 音频加载失败，已重试 %ld 次", (long)maxRetries);
|
||
}
|
||
});
|
||
}];
|
||
}
|
||
|
||
- (void)downloadAndPlayAudioFromURL:(NSString *)urlString
|
||
forMessage:(KBAiChatMessage *)message
|
||
atIndexPath:(NSIndexPath *)indexPath {
|
||
NSURL *url = [NSURL URLWithString:urlString];
|
||
if (!url) {
|
||
NSLog(@"[KBChatTableView] 无效的音频 URL: %@", urlString);
|
||
// 隐藏加载动画
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[cell hideLoadingAnimation];
|
||
}
|
||
return;
|
||
}
|
||
|
||
NSURLSessionConfiguration *config = [NSURLSessionConfiguration defaultSessionConfiguration];
|
||
NSURLSession *session = [NSURLSession sessionWithConfiguration:config];
|
||
|
||
NSURLSessionDataTask *task = [session dataTaskWithURL:url
|
||
completionHandler:^(NSData *_Nullable data,
|
||
NSURLResponse *_Nullable response,
|
||
NSError *_Nullable error) {
|
||
if (error || !data || data.length == 0) {
|
||
NSLog(@"[KBChatTableView] 下载音频失败: %@", error.localizedDescription ?: @"");
|
||
dispatch_async(dispatch_get_main_queue(), ^{
|
||
// 隐藏加载动画
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[cell hideLoadingAnimation];
|
||
}
|
||
});
|
||
return;
|
||
}
|
||
|
||
dispatch_async(dispatch_get_main_queue(), ^{
|
||
// 隐藏加载动画
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[cell hideLoadingAnimation];
|
||
}
|
||
|
||
// 缓存音频数据到消息对象
|
||
message.audioData = data;
|
||
|
||
// 计算音频时长
|
||
NSError *playerError = nil;
|
||
AVAudioPlayer *player = [[AVAudioPlayer alloc] initWithData:data error:&playerError];
|
||
if (!playerError && player) {
|
||
message.audioDuration = player.duration;
|
||
}
|
||
|
||
// 不刷新 Cell，避免触发打字机效果
|
||
// 直接播放音频
|
||
[self playAudioForMessage:message atIndexPath:indexPath];
|
||
});
|
||
}];
|
||
|
||
[task resume];
|
||
}
|
||
|
||
- (void)playAudioForMessage:(KBAiChatMessage *)message atIndexPath:(NSIndexPath *)indexPath {
|
||
if (!message.audioData || message.audioData.length == 0) {
|
||
NSLog(@"[KBChatTableView] 没有音频数据");
|
||
return;
|
||
}
|
||
|
||
// 配置音频会话为播放模式
|
||
NSError *sessionError = nil;
|
||
AVAudioSession *audioSession = [AVAudioSession sharedInstance];
|
||
[audioSession setCategory:AVAudioSessionCategoryPlayback error:&sessionError];
|
||
[audioSession setActive:YES error:&sessionError];
|
||
|
||
if (sessionError) {
|
||
NSLog(@"[KBChatTableView] 音频会话配置失败: %@", sessionError.localizedDescription);
|
||
}
|
||
|
||
NSError *error = nil;
|
||
self.audioPlayer = [[AVAudioPlayer alloc] initWithData:message.audioData error:&error];
|
||
|
||
if (error || !self.audioPlayer) {
|
||
NSLog(@"[KBChatTableView] 音频播放器初始化失败: %@", error.localizedDescription);
|
||
return;
|
||
}
|
||
|
||
self.audioPlayer.delegate = self;
|
||
self.audioPlayer.volume = 1.0; // 设置音量为最大
|
||
[self.audioPlayer prepareToPlay];
|
||
[self.audioPlayer play];
|
||
|
||
self.playingCellIndexPath = indexPath;
|
||
|
||
// 更新 Cell 状态，禁用动画避免 TableView 跳动
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:indexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
[UIView performWithoutAnimation:^{
|
||
[cell updateVoicePlayingState:YES];
|
||
}];
|
||
}
|
||
}
|
||
|
||
- (void)stopPlayingAudio {
|
||
if (self.audioPlayer && self.audioPlayer.isPlaying) {
|
||
[self.audioPlayer stop];
|
||
}
|
||
|
||
if (self.playingCellIndexPath) {
|
||
KBChatAssistantMessageCell *cell = [self.tableView cellForRowAtIndexPath:self.playingCellIndexPath];
|
||
if ([cell isKindOfClass:[KBChatAssistantMessageCell class]]) {
|
||
// 禁用动画，避免 TableView 跳动
|
||
[UIView performWithoutAnimation:^{
|
||
[cell updateVoicePlayingState:NO];
|
||
}];
|
||
}
|
||
self.playingCellIndexPath = nil;
|
||
}
|
||
}
|
||
|
||
#pragma mark - AVAudioPlayerDelegate
|
||
|
||
- (void)audioPlayerDidFinishPlaying:(AVAudioPlayer *)player successfully:(BOOL)flag {
|
||
[self stopPlayingAudio];
|
||
}
|
||
|
||
@end
|