1

keyBoard/Class/AiTalk/V/KBChatAssistantMessageCell.m

@@ -0,0 +1,339 @@
+//
+//  KBChatAssistantMessageCell.m
+//  keyBoard
+//
+//  Created by Kiro on 2026/1/23.
+//
+
+#import "KBChatAssistantMessageCell.h"
+#import "KBAiChatMessage.h"
+#import <Masonry/Masonry.h>
+
+@interface KBChatAssistantMessageCell ()
+
+@property (nonatomic, strong) UIButton *voiceButton;
+@property (nonatomic, strong) UILabel *durationLabel;
+@property (nonatomic, strong) UIView *bubbleView;
+@property (nonatomic, strong, readwrite) UILabel *messageLabel; // readwrite 允许内部修改
+@property (nonatomic, strong) UIActivityIndicatorView *loadingIndicator;
+@property (nonatomic, strong) KBAiChatMessage *currentMessage;
+
+// 打字机效果
+@property (nonatomic, strong) NSTimer *typewriterTimer;
+@property (nonatomic, copy) NSString *fullText;
+@property (nonatomic, assign) NSInteger currentCharIndex;
+
+@end
+
+@implementation KBChatAssistantMessageCell
+
+- (instancetype)initWithStyle:(UITableViewCellStyle)style
+              reuseIdentifier:(NSString *)reuseIdentifier {
+    self = [super initWithStyle:style reuseIdentifier:reuseIdentifier];
+    if (self) {
+        self.backgroundColor = [UIColor clearColor];
+        self.selectionStyle = UITableViewCellSelectionStyleNone;
+        [self setupUI];
+    }
+    return self;
+}
+
+- (void)setupUI {
+    // 语音按钮
+    self.voiceButton = [UIButton buttonWithType:UIButtonTypeCustom];
+    [self.voiceButton setImage:[UIImage systemImageNamed:@"play.circle.fill"]
+                      forState:UIControlStateNormal];
+    self.voiceButton.tintColor = [UIColor whiteColor];
+    [self.voiceButton addTarget:self
+                         action:@selector(voiceButtonTapped)
+               forControlEvents:UIControlEventTouchUpInside];
+    [self.contentView addSubview:self.voiceButton];
+    
+    // 语音时长标签
+    self.durationLabel = [[UILabel alloc] init];
+    self.durationLabel.font = [UIFont systemFontOfSize:14];
+    self.durationLabel.textColor = [UIColor whiteColor];
+    [self.contentView addSubview:self.durationLabel];
+    
+    // 加载指示器
+    self.loadingIndicator = [[UIActivityIndicatorView alloc] initWithActivityIndicatorStyle:UIActivityIndicatorViewStyleMedium];
+    self.loadingIndicator.color = [UIColor whiteColor];
+    self.loadingIndicator.hidesWhenStopped = YES;
+    [self.contentView addSubview:self.loadingIndicator];
+    
+    // 气泡视图
+    self.bubbleView = [[UIView alloc] init];
+    self.bubbleView.backgroundColor = [UIColor colorWithRed:0.2 green:0.2 blue:0.2 alpha:0.7];
+    self.bubbleView.layer.cornerRadius = 16;
+    self.bubbleView.layer.masksToBounds = YES;
+    [self.contentView addSubview:self.bubbleView];
+    
+    // 消息标签
+    self.messageLabel = [[UILabel alloc] init];
+    self.messageLabel.numberOfLines = 0;
+    self.messageLabel.font = [UIFont systemFontOfSize:16];
+    self.messageLabel.textColor = [UIColor whiteColor];
+    // 设置 preferredMaxLayoutWidth 让 AutoLayout 能正确计算多行高度
+    CGFloat maxWidth = [UIScreen mainScreen].bounds.size.width * 0.75 - 16 - 24;
+    self.messageLabel.preferredMaxLayoutWidth = maxWidth;
+    [self.bubbleView addSubview:self.messageLabel];
+    
+    // 布局约束
+    [self.voiceButton mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.left.equalTo(self.contentView).offset(16);
+        make.top.equalTo(self.contentView).offset(8);
+        make.width.height.mas_equalTo(24);
+    }];
+    
+    [self.durationLabel mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.left.equalTo(self.voiceButton.mas_right).offset(4);
+        make.centerY.equalTo(self.voiceButton);
+    }];
+    
+    [self.loadingIndicator mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.center.equalTo(self.voiceButton);
+    }];
+    
+    // bubbleView 约束
+    [self.bubbleView mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.top.equalTo(self.voiceButton.mas_bottom).offset(4);
+        make.bottom.equalTo(self.contentView).offset(-4);
+        make.left.equalTo(self.contentView).offset(16);
+        make.width.lessThanOrEqualTo(self.contentView).multipliedBy(0.75);
+    }];
+    
+    // messageLabel 约束
+    [self.messageLabel mas_makeConstraints:^(MASConstraintMaker *make) {
+        make.top.equalTo(self.bubbleView).offset(10);
+        make.bottom.equalTo(self.bubbleView).offset(-10);
+        make.left.equalTo(self.bubbleView).offset(12);
+        make.right.equalTo(self.bubbleView).offset(-12);
+    }];
+}
+
+- (void)configureWithMessage:(KBAiChatMessage *)message {
+    NSLog(@"[KBChatAssistantMessageCell] 配置消息 - 文本长度: %lu, isComplete: %d, needsTypewriter: %d, 打字机运行中: %d", 
+          (unsigned long)message.text.length, message.isComplete, message.needsTypewriterEffect,
+          (self.typewriterTimer && self.typewriterTimer.isValid));
+    
+    // 先停止之前的打字机效果（无论是否是同一条消息）
+    [self stopTypewriterEffect];
+    
+    self.currentMessage = message;
+    
+    // 只有明确需要打字机效果的消息才使用打字机
+    if (message.needsTypewriterEffect && !message.isComplete && message.text.length > 0) {
+        // 启动新的打字机效果
+        NSLog(@"[KBChatAssistantMessageCell] 启动新的打字机效果");
+        [self startTypewriterEffectWithText:message.text];
+    } else {
+        // 直接显示完整文本
+        NSLog(@"[KBChatAssistantMessageCell] 直接显示完整文本（needsTypewriter: %d, isComplete: %d）", 
+              message.needsTypewriterEffect, message.isComplete);
+        self.messageLabel.text = message.text;
+    }
+    
+    // 格式化语音时长（如果时长为 0，不显示）
+    if (message.audioDuration > 0) {
+        NSInteger seconds = (NSInteger)ceil(message.audioDuration);
+        self.durationLabel.text = [NSString stringWithFormat:@"%ld\"", (long)seconds];
+    } else {
+        self.durationLabel.text = @"";
+    }
+    
+    // 如果有 audioId 或 audioData，显示语音按钮
+    BOOL hasAudio = (message.audioId.length > 0) || 
+                    (message.audioData != nil && message.audioData.length > 0);
+    self.voiceButton.hidden = !hasAudio;
+    self.durationLabel.hidden = !hasAudio;
+}
+
+- (void)updateVoicePlayingState:(BOOL)isPlaying {
+    if (isPlaying) {
+        [self.voiceButton setImage:[UIImage systemImageNamed:@"pause.circle.fill"]
+                          forState:UIControlStateNormal];
+    } else {
+        [self.voiceButton setImage:[UIImage systemImageNamed:@"play.circle.fill"]
+                          forState:UIControlStateNormal];
+    }
+}
+
+- (void)showLoadingAnimation {
+    // 隐藏按钮图标，显示加载动画
+    [self.voiceButton setImage:nil forState:UIControlStateNormal];
+    [self.loadingIndicator startAnimating];
+}
+
+- (void)hideLoadingAnimation {
+    // 停止加载动画，恢复按钮图标
+    [self.loadingIndicator stopAnimating];
+    [self.voiceButton setImage:[UIImage systemImageNamed:@"play.circle.fill"]
+                      forState:UIControlStateNormal];
+}
+
+- (void)voiceButtonTapped {
+    if ([self.delegate respondsToSelector:@selector(assistantMessageCell:didTapVoiceButtonForMessage:)]) {
+        [self.delegate assistantMessageCell:self didTapVoiceButtonForMessage:self.currentMessage];
+    }
+}
+
+#pragma mark - Typewriter Effect
+
+- (void)startTypewriterEffectWithText:(NSString *)text {
+    if (text.length == 0) {
+        NSLog(@"[KBChatAssistantMessageCell] 文本为空，跳过打字机效果");
+        return;
+    }
+    
+    NSLog(@"[KBChatAssistantMessageCell] 开始打字机效果，文本长度: %lu, 文本: %@", 
+          (unsigned long)text.length, text);
+    
+    self.fullText = text;
+    self.currentCharIndex = 0;
+    
+    // 先设置完整文本，让布局系统计算出正确的高度
+    self.messageLabel.text = text;
+    
+    // 强制布局更新
+    [self.messageLabel setNeedsLayout];
+    [self.bubbleView setNeedsLayout];
+    [self.contentView setNeedsLayout];
+    [self layoutIfNeeded];
+    
+    NSLog(@"[KBChatAssistantMessageCell] 布局后 Label frame: %@", NSStringFromCGRect(self.messageLabel.frame));
+    
+    // 使用 NSAttributedString 实现打字机效果
+    // 先把所有文字设置为透明，然后逐个显示
+    NSMutableAttributedString *attributedText = [[NSMutableAttributedString alloc] initWithString:text];
+    [attributedText addAttribute:NSForegroundColorAttributeName
+                           value:[UIColor clearColor]
+                           range:NSMakeRange(0, text.length)];
+    [attributedText addAttribute:NSFontAttributeName
+                           value:self.messageLabel.font
+                           range:NSMakeRange(0, text.length)];
+    self.messageLabel.attributedText = attributedText;
+    
+    // 确保在主线程创建定时器
+    dispatch_async(dispatch_get_main_queue(), ^{
+        // 每 0.03 秒显示一个字符（更快的打字机效果）
+        self.typewriterTimer = [NSTimer scheduledTimerWithTimeInterval:0.03
+                                                                target:self
+                                                              selector:@selector(typewriterTick)
+                                                              userInfo:nil
+                                                               repeats:YES];
+        
+        // 将定时器添加到 RunLoop 的 common 模式，确保滚动时也能触发
+        [[NSRunLoop currentRunLoop] addTimer:self.typewriterTimer forMode:NSRunLoopCommonModes];
+        
+        // 立即触发一次，显示第一个字符
+        [self typewriterTick];
+        
+        NSLog(@"[KBChatAssistantMessageCell] 定时器已创建: %@", self.typewriterTimer);
+    });
+}
+
+- (void)typewriterTick {
+    // 使用局部变量保存 fullText，避免多线程问题
+    NSString *text = self.fullText;
+    
+    // 检查 fullText 是否有效
+    if (!text || text.length == 0) {
+        NSLog(@"[KBChatAssistantMessageCell] fullText 无效，停止打字机");
+        [self stopTypewriterEffect];
+        return;
+    }
+    
+    // 如果当前索引小于完整文本长度，继续显示
+    if (self.currentCharIndex < text.length) {
+        self.currentCharIndex++;
+        
+        // 使用 NSAttributedString 显示前 N 个字符，隐藏后面的字符
+        NSMutableAttributedString *attributedText = [[NSMutableAttributedString alloc] initWithString:text];
+        
+        // 已显示的部分：白色
+        if (self.currentCharIndex > 0) {
+            [attributedText addAttribute:NSForegroundColorAttributeName
+                                   value:[UIColor whiteColor]
+                                   range:NSMakeRange(0, self.currentCharIndex)];
+        }
+        
+        // 未显示的部分：透明
+        if (self.currentCharIndex < text.length) {
+            [attributedText addAttribute:NSForegroundColorAttributeName
+                                   value:[UIColor clearColor]
+                                   range:NSMakeRange(self.currentCharIndex, text.length - self.currentCharIndex)];
+        }
+        
+        // 设置字体
+        [attributedText addAttribute:NSFontAttributeName
+                               value:self.messageLabel.font
+                               range:NSMakeRange(0, text.length)];
+        
+        self.messageLabel.attributedText = attributedText;
+        
+        // 每 10 个字符打印一次日志
+        if (self.currentCharIndex % 10 == 0) {
+            NSString *displayText = [text substringToIndex:self.currentCharIndex];
+            NSLog(@"[KBChatAssistantMessageCell] 打字机进度: %ld/%lu, 当前文本: %@", 
+                  (long)self.currentCharIndex, (unsigned long)text.length, displayText);
+            NSLog(@"[KBChatAssistantMessageCell] Label 状态 - frame: %@, hidden: %d, alpha: %.2f", 
+                  NSStringFromCGRect(self.messageLabel.frame), 
+                  self.messageLabel.hidden, 
+                  self.messageLabel.alpha);
+        }
+    } else {
+        // 打字机效果已完成，停止定时器并标记消息完成
+        NSLog(@"[KBChatAssistantMessageCell] 打字机效果完成，停止定时器");
+        
+        // 检查 text 是否为空
+        if (!text || text.length == 0) {
+            NSLog(@"[KBChatAssistantMessageCell] text 为空，跳过");
+            [self stopTypewriterEffect];
+            return;
+        }
+        
+        [self stopTypewriterEffect];
+        
+        // 显示完整文本（白色），使用 attributedText 保持布局
+        NSMutableAttributedString *attributedText = [[NSMutableAttributedString alloc] initWithString:text];
+        [attributedText addAttribute:NSForegroundColorAttributeName
+                               value:[UIColor whiteColor]
+                               range:NSMakeRange(0, text.length)];
+        [attributedText addAttribute:NSFontAttributeName
+                               value:self.messageLabel.font
+                               range:NSMakeRange(0, text.length)];
+        self.messageLabel.attributedText = attributedText;
+        
+        // 标记消息为完成状态，避免 Cell 复用时重新触发打字机
+        if (self.currentMessage) {
+            self.currentMessage.isComplete = YES;
+            self.currentMessage.needsTypewriterEffect = NO;
+            NSLog(@"[KBChatAssistantMessageCell] 标记消息完成 - 文本: %@", self.currentMessage.text);
+        }
+    }
+}
+
+
+- (void)stopTypewriterEffect {
+    if (self.typewriterTimer && self.typewriterTimer.isValid) {
+        NSLog(@"[KBChatAssistantMessageCell] 停止打字机定时器");
+        [self.typewriterTimer invalidate];
+    }
+    self.typewriterTimer = nil;
+    self.currentCharIndex = 0;
+    self.fullText = nil;
+}
+
+- (void)prepareForReuse {
+    [super prepareForReuse];
+    [self stopTypewriterEffect];
+    // 清空文本，避免复用时显示旧内容
+    self.messageLabel.text = @"";
+    self.messageLabel.attributedText = nil;
+}
+
+- (void)dealloc {
+    [self stopTypewriterEffect];
+}
+
+@end