496 lines
19 KiB
Objective-C
496 lines
19 KiB
Objective-C
//
|
||
// KBChatMessageCell.m
|
||
// CustomKeyboard
|
||
//
|
||
|
||
#import "KBChatMessageCell.h"
|
||
#import "KBChatMessage.h"
|
||
#import "Masonry.h"
|
||
|
||
@interface KBChatMessageCell ()
|
||
|
||
@property (nonatomic, strong) UIImageView *avatarView;
|
||
@property (nonatomic, strong) UILabel *nameLabel;
|
||
@property (nonatomic, strong) UIView *bubbleView;
|
||
@property (nonatomic, strong) UILabel *messageLabel;
|
||
@property (nonatomic, strong) UIImageView *audioIconView;
|
||
@property (nonatomic, strong) UILabel *audioLabel;
|
||
|
||
/// 语音播放按钮
|
||
@property (nonatomic, strong) UIButton *voiceButton;
|
||
/// 语音时长标签
|
||
@property (nonatomic, strong) UILabel *durationLabel;
|
||
/// 语音加载指示器
|
||
@property (nonatomic, strong) UIActivityIndicatorView *voiceLoadingIndicator;
|
||
/// 消息加载指示器（AI 回复 loading）
|
||
@property (nonatomic, strong) UIActivityIndicatorView *messageLoadingIndicator;
|
||
|
||
/// 当前消息
|
||
@property (nonatomic, strong) KBChatMessage *currentMessage;
|
||
|
||
/// 打字机效果
|
||
@property (nonatomic, strong) NSTimer *typewriterTimer;
|
||
@property (nonatomic, copy) NSString *fullText;
|
||
@property (nonatomic, assign) NSInteger currentCharIndex;
|
||
|
||
@end
|
||
|
||
@implementation KBChatMessageCell
|
||
|
||
- (instancetype)initWithStyle:(UITableViewCellStyle)style reuseIdentifier:(NSString *)reuseIdentifier {
|
||
if (self = [super initWithStyle:style reuseIdentifier:reuseIdentifier]) {
|
||
self.backgroundColor = [UIColor clearColor];
|
||
self.contentView.backgroundColor = [UIColor clearColor];
|
||
self.selectionStyle = UITableViewCellSelectionStyleNone;
|
||
|
||
[self.contentView addSubview:self.avatarView];
|
||
[self.contentView addSubview:self.nameLabel];
|
||
[self.contentView addSubview:self.voiceButton];
|
||
[self.contentView addSubview:self.durationLabel];
|
||
[self.contentView addSubview:self.voiceLoadingIndicator];
|
||
[self.contentView addSubview:self.messageLoadingIndicator];
|
||
[self.contentView addSubview:self.bubbleView];
|
||
[self.bubbleView addSubview:self.messageLabel];
|
||
[self.bubbleView addSubview:self.audioIconView];
|
||
[self.bubbleView addSubview:self.audioLabel];
|
||
}
|
||
return self;
|
||
}
|
||
|
||
- (void)kb_configureWithMessage:(KBChatMessage *)message {
|
||
// 先停止之前的打字机效果
|
||
[self kb_stopTypewriterEffect];
|
||
|
||
self.currentMessage = message;
|
||
|
||
BOOL outgoing = message.outgoing;
|
||
BOOL audioMessage = (!outgoing && message.audioFilePath.length > 0);
|
||
UIColor *bubbleColor = outgoing ? [UIColor colorWithHex:0x02BEAC] : [UIColor colorWithWhite:1 alpha:0.95];
|
||
UIColor *incomingTextColor =
|
||
[UIColor kb_dynamicColorWithLightColor:[UIColor colorWithHex:0x1B1F1A]
|
||
darkColor:[UIColor whiteColor]];
|
||
UIColor *textColor = outgoing ? [UIColor whiteColor] : incomingTextColor;
|
||
UIColor *nameColor =
|
||
[UIColor kb_dynamicColorWithLightColor:[UIColor colorWithHex:0x6B6F7A]
|
||
darkColor:[UIColor colorWithHex:0xC7CBD4]];
|
||
|
||
self.bubbleView.backgroundColor = bubbleColor;
|
||
self.messageLabel.textColor = textColor;
|
||
self.audioLabel.textColor = textColor;
|
||
self.audioIconView.tintColor = textColor;
|
||
self.audioLabel.text =
|
||
(message.text.length > 0) ? message.text : KBLocalized(@"语音回复");
|
||
self.messageLabel.hidden = audioMessage;
|
||
self.audioIconView.hidden = !audioMessage;
|
||
self.audioLabel.hidden = !audioMessage;
|
||
|
||
UIImage *avatarImage = message.avatarImage;
|
||
if (!avatarImage) {
|
||
avatarImage = [self kb_defaultAvatarImage];
|
||
}
|
||
self.avatarView.image = avatarImage;
|
||
self.avatarView.backgroundColor =
|
||
avatarImage ? [UIColor clearColor] : [UIColor colorWithWhite:0.9 alpha:1.0];
|
||
self.nameLabel.hidden = outgoing;
|
||
self.nameLabel.textColor = nameColor;
|
||
self.nameLabel.text =
|
||
(message.displayName.length > 0) ? message.displayName : KBLocalized(@"AI助手");
|
||
|
||
// 处理 loading 状态
|
||
if (message.isLoading && !outgoing) {
|
||
self.bubbleView.hidden = YES;
|
||
self.voiceButton.hidden = YES;
|
||
self.durationLabel.hidden = YES;
|
||
[self.messageLoadingIndicator startAnimating];
|
||
[self kb_layoutForOutgoing:outgoing audioMessage:NO];
|
||
return;
|
||
}
|
||
|
||
// 非 loading 状态
|
||
[self.messageLoadingIndicator stopAnimating];
|
||
self.bubbleView.hidden = NO;
|
||
|
||
// 语音按钮显示逻辑（仅 AI 消息且有 audioId 或 audioData）
|
||
BOOL hasAudio = (!outgoing) && (message.audioId.length > 0 || message.audioData.length > 0);
|
||
self.voiceButton.hidden = !hasAudio;
|
||
self.durationLabel.hidden = !hasAudio;
|
||
if (hasAudio && message.audioDuration > 0) {
|
||
NSInteger seconds = (NSInteger)ceil(message.audioDuration);
|
||
self.durationLabel.text = [NSString stringWithFormat:@"%ld\"", (long)seconds];
|
||
} else {
|
||
self.durationLabel.text = @"";
|
||
}
|
||
|
||
// 打字机效果
|
||
if (!outgoing && message.needsTypewriterEffect && !message.isComplete && message.text.length > 0) {
|
||
[self kb_startTypewriterEffectWithText:message.text];
|
||
} else {
|
||
self.messageLabel.attributedText = nil;
|
||
self.messageLabel.text = message.text ?: @"";
|
||
}
|
||
|
||
[self kb_layoutForOutgoing:outgoing audioMessage:audioMessage];
|
||
}
|
||
|
||
- (void)kb_layoutForOutgoing:(BOOL)outgoing audioMessage:(BOOL)audioMessage {
|
||
CGFloat avatarSize = 28.0;
|
||
[self.avatarView mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.height.mas_equalTo(avatarSize);
|
||
make.top.equalTo(self.contentView.mas_top).offset(6);
|
||
if (outgoing) {
|
||
make.right.equalTo(self.contentView.mas_right).offset(-8);
|
||
} else {
|
||
make.left.equalTo(self.contentView.mas_left).offset(8);
|
||
}
|
||
}];
|
||
|
||
if (outgoing) {
|
||
[self.nameLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.top.equalTo(self.contentView.mas_top).offset(0);
|
||
make.left.equalTo(self.contentView.mas_left);
|
||
}];
|
||
// 用户消息不显示语音按钮
|
||
[self.voiceButton mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.height.mas_equalTo(0);
|
||
make.left.top.equalTo(self.contentView);
|
||
}];
|
||
[self.durationLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.height.mas_equalTo(0);
|
||
make.left.top.equalTo(self.contentView);
|
||
}];
|
||
} else {
|
||
[self.nameLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.left.equalTo(self.avatarView.mas_right).offset(6);
|
||
make.top.equalTo(self.contentView.mas_top).offset(2);
|
||
make.right.lessThanOrEqualTo(self.contentView.mas_right).offset(-12);
|
||
}];
|
||
// AI 消息语音按钮
|
||
[self.voiceButton mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.left.equalTo(self.avatarView.mas_right).offset(6);
|
||
make.top.equalTo(self.nameLabel.mas_bottom).offset(4);
|
||
make.width.height.mas_equalTo(20);
|
||
}];
|
||
[self.durationLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.left.equalTo(self.voiceButton.mas_right).offset(4);
|
||
make.centerY.equalTo(self.voiceButton);
|
||
}];
|
||
[self.voiceLoadingIndicator mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.center.equalTo(self.voiceButton);
|
||
}];
|
||
}
|
||
|
||
// 消息加载指示器
|
||
[self.messageLoadingIndicator mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
if (outgoing) {
|
||
make.right.equalTo(self.avatarView.mas_left).offset(-10);
|
||
} else {
|
||
make.left.equalTo(self.avatarView.mas_right).offset(10);
|
||
}
|
||
make.top.equalTo(self.nameLabel.mas_bottom).offset(8);
|
||
}];
|
||
|
||
[self.bubbleView mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.lessThanOrEqualTo(self.contentView.mas_width).multipliedBy(0.65);
|
||
if (outgoing) {
|
||
make.top.equalTo(self.contentView.mas_top).offset(6);
|
||
make.bottom.equalTo(self.contentView.mas_bottom).offset(-6);
|
||
make.right.equalTo(self.avatarView.mas_left).offset(-6);
|
||
} else {
|
||
// AI 消息：气泡在语音按钮下方
|
||
make.top.equalTo(self.voiceButton.mas_bottom).offset(4);
|
||
make.bottom.equalTo(self.contentView.mas_bottom).offset(-6);
|
||
make.left.equalTo(self.avatarView.mas_right).offset(6);
|
||
make.right.lessThanOrEqualTo(self.contentView.mas_right).offset(-12);
|
||
}
|
||
}];
|
||
|
||
if (audioMessage) {
|
||
[self.messageLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.height.mas_equalTo(0);
|
||
make.left.equalTo(self.bubbleView.mas_left);
|
||
make.top.equalTo(self.bubbleView.mas_top);
|
||
}];
|
||
[self.audioIconView mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.left.equalTo(self.bubbleView.mas_left).offset(10);
|
||
make.centerY.equalTo(self.bubbleView);
|
||
make.width.height.mas_equalTo(16);
|
||
}];
|
||
[self.audioLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.left.equalTo(self.audioIconView.mas_right).offset(6);
|
||
make.centerY.equalTo(self.bubbleView);
|
||
make.right.equalTo(self.bubbleView.mas_right).offset(-10);
|
||
make.top.greaterThanOrEqualTo(self.bubbleView.mas_top).offset(8);
|
||
make.bottom.lessThanOrEqualTo(self.bubbleView.mas_bottom).offset(-8);
|
||
}];
|
||
} else {
|
||
[self.audioIconView mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.height.mas_equalTo(0);
|
||
make.left.equalTo(self.bubbleView.mas_left);
|
||
make.top.equalTo(self.bubbleView.mas_top);
|
||
}];
|
||
[self.audioLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.width.height.mas_equalTo(0);
|
||
make.left.equalTo(self.audioIconView.mas_right);
|
||
make.top.equalTo(self.bubbleView.mas_top);
|
||
}];
|
||
[self.messageLabel mas_remakeConstraints:^(MASConstraintMaker *make) {
|
||
make.edges.equalTo(self.bubbleView).insets(UIEdgeInsetsMake(8, 10, 8, 10));
|
||
}];
|
||
}
|
||
}
|
||
|
||
#pragma mark - Typewriter Effect
|
||
|
||
- (void)kb_startTypewriterEffectWithText:(NSString *)text {
|
||
if (text.length == 0) return;
|
||
|
||
self.fullText = text;
|
||
self.currentCharIndex = 0;
|
||
|
||
// 先设置完整文本让布局计算正确高度
|
||
self.messageLabel.text = text;
|
||
[self.contentView setNeedsLayout];
|
||
[self.contentView layoutIfNeeded];
|
||
|
||
// 应用打字机效果
|
||
dispatch_async(dispatch_get_main_queue(), ^{
|
||
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;
|
||
|
||
self.typewriterTimer = [NSTimer scheduledTimerWithTimeInterval:0.03
|
||
target:self
|
||
selector:@selector(kb_typewriterTick)
|
||
userInfo:nil
|
||
repeats:YES];
|
||
[[NSRunLoop currentRunLoop] addTimer:self.typewriterTimer forMode:NSRunLoopCommonModes];
|
||
[self kb_typewriterTick];
|
||
});
|
||
}
|
||
|
||
- (void)kb_typewriterTick {
|
||
NSString *text = self.fullText;
|
||
if (!text || text.length == 0) {
|
||
[self kb_stopTypewriterEffect];
|
||
return;
|
||
}
|
||
|
||
if (self.currentCharIndex < text.length) {
|
||
self.currentCharIndex++;
|
||
|
||
NSMutableAttributedString *attributedText = [[NSMutableAttributedString alloc] initWithString:text];
|
||
UIColor *textColor = self.messageLabel.textColor ?: [UIColor blackColor];
|
||
|
||
if (self.currentCharIndex > 0) {
|
||
[attributedText addAttribute:NSForegroundColorAttributeName
|
||
value:textColor
|
||
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;
|
||
} else {
|
||
[self kb_stopTypewriterEffect];
|
||
|
||
// 显示完整文本
|
||
NSMutableAttributedString *attributedText = [[NSMutableAttributedString alloc] initWithString:text];
|
||
UIColor *textColor = self.messageLabel.textColor ?: [UIColor blackColor];
|
||
[attributedText addAttribute:NSForegroundColorAttributeName
|
||
value:textColor
|
||
range:NSMakeRange(0, text.length)];
|
||
[attributedText addAttribute:NSFontAttributeName
|
||
value:self.messageLabel.font
|
||
range:NSMakeRange(0, text.length)];
|
||
self.messageLabel.attributedText = attributedText;
|
||
|
||
// 标记完成
|
||
if (self.currentMessage) {
|
||
self.currentMessage.isComplete = YES;
|
||
self.currentMessage.needsTypewriterEffect = NO;
|
||
}
|
||
}
|
||
}
|
||
|
||
- (void)kb_stopTypewriterEffect {
|
||
if (self.typewriterTimer && self.typewriterTimer.isValid) {
|
||
[self.typewriterTimer invalidate];
|
||
}
|
||
self.typewriterTimer = nil;
|
||
self.currentCharIndex = 0;
|
||
self.fullText = nil;
|
||
}
|
||
|
||
#pragma mark - Voice Button
|
||
|
||
- (void)kb_updateVoicePlayingState:(BOOL)isPlaying {
|
||
UIImage *icon = nil;
|
||
if (@available(iOS 13.0, *)) {
|
||
icon = isPlaying ? [UIImage systemImageNamed:@"pause.circle.fill"] : [UIImage systemImageNamed:@"play.circle.fill"];
|
||
}
|
||
[self.voiceButton setImage:icon forState:UIControlStateNormal];
|
||
}
|
||
|
||
- (void)kb_showVoiceLoadingAnimation {
|
||
[self.voiceButton setImage:nil forState:UIControlStateNormal];
|
||
[self.voiceLoadingIndicator startAnimating];
|
||
}
|
||
|
||
- (void)kb_hideVoiceLoadingAnimation {
|
||
[self.voiceLoadingIndicator stopAnimating];
|
||
UIImage *icon = nil;
|
||
if (@available(iOS 13.0, *)) {
|
||
icon = [UIImage systemImageNamed:@"play.circle.fill"];
|
||
}
|
||
[self.voiceButton setImage:icon forState:UIControlStateNormal];
|
||
}
|
||
|
||
- (void)kb_onVoiceButtonTapped {
|
||
if ([self.delegate respondsToSelector:@selector(chatMessageCell:didTapVoiceButtonForMessage:)]) {
|
||
[self.delegate chatMessageCell:self didTapVoiceButtonForMessage:self.currentMessage];
|
||
}
|
||
}
|
||
|
||
#pragma mark - Reuse
|
||
|
||
- (void)prepareForReuse {
|
||
[super prepareForReuse];
|
||
[self kb_stopTypewriterEffect];
|
||
self.messageLabel.text = @"";
|
||
self.messageLabel.attributedText = nil;
|
||
[self.messageLoadingIndicator stopAnimating];
|
||
[self.voiceLoadingIndicator stopAnimating];
|
||
}
|
||
|
||
- (void)dealloc {
|
||
[self kb_stopTypewriterEffect];
|
||
}
|
||
|
||
#pragma mark - Lazy
|
||
|
||
- (UIImageView *)avatarView {
|
||
if (!_avatarView) {
|
||
_avatarView = [[UIImageView alloc] init];
|
||
_avatarView.contentMode = UIViewContentModeScaleAspectFill;
|
||
_avatarView.layer.cornerRadius = 14;
|
||
_avatarView.layer.masksToBounds = YES;
|
||
_avatarView.backgroundColor = [UIColor colorWithWhite:0.9 alpha:1.0];
|
||
_avatarView.tintColor =
|
||
[UIColor kb_dynamicColorWithLightColor:[UIColor colorWithHex:0xB9BDC8]
|
||
darkColor:[UIColor colorWithHex:0x6B6F7A]];
|
||
}
|
||
return _avatarView;
|
||
}
|
||
|
||
- (UILabel *)nameLabel {
|
||
if (!_nameLabel) {
|
||
_nameLabel = [[UILabel alloc] init];
|
||
_nameLabel.font = [UIFont systemFontOfSize:11];
|
||
_nameLabel.textColor = [UIColor colorWithHex:0x6B6F7A];
|
||
_nameLabel.numberOfLines = 1;
|
||
}
|
||
return _nameLabel;
|
||
}
|
||
|
||
- (UIView *)bubbleView {
|
||
if (!_bubbleView) {
|
||
_bubbleView = [[UIView alloc] init];
|
||
_bubbleView.layer.cornerRadius = 12;
|
||
_bubbleView.layer.masksToBounds = YES;
|
||
}
|
||
return _bubbleView;
|
||
}
|
||
|
||
- (UILabel *)messageLabel {
|
||
if (!_messageLabel) {
|
||
_messageLabel = [[UILabel alloc] init];
|
||
_messageLabel.font = [UIFont systemFontOfSize:14];
|
||
_messageLabel.numberOfLines = 0;
|
||
}
|
||
return _messageLabel;
|
||
}
|
||
|
||
- (UIImageView *)audioIconView {
|
||
if (!_audioIconView) {
|
||
_audioIconView = [[UIImageView alloc] init];
|
||
_audioIconView.contentMode = UIViewContentModeScaleAspectFit;
|
||
_audioIconView.tintColor = [UIColor colorWithHex:0x1B1F1A];
|
||
UIImage *icon = nil;
|
||
if (@available(iOS 13.0, *)) {
|
||
icon = [UIImage systemImageNamed:@"waveform"];
|
||
}
|
||
_audioIconView.image = icon;
|
||
}
|
||
return _audioIconView;
|
||
}
|
||
|
||
- (UILabel *)audioLabel {
|
||
if (!_audioLabel) {
|
||
_audioLabel = [[UILabel alloc] init];
|
||
_audioLabel.font = [UIFont systemFontOfSize:14 weight:UIFontWeightMedium];
|
||
_audioLabel.numberOfLines = 1;
|
||
}
|
||
return _audioLabel;
|
||
}
|
||
|
||
- (UIButton *)voiceButton {
|
||
if (!_voiceButton) {
|
||
_voiceButton = [UIButton buttonWithType:UIButtonTypeCustom];
|
||
UIImage *icon = nil;
|
||
if (@available(iOS 13.0, *)) {
|
||
icon = [UIImage systemImageNamed:@"play.circle.fill"];
|
||
}
|
||
[_voiceButton setImage:icon forState:UIControlStateNormal];
|
||
_voiceButton.tintColor = [UIColor whiteColor];
|
||
[_voiceButton addTarget:self action:@selector(kb_onVoiceButtonTapped) forControlEvents:UIControlEventTouchUpInside];
|
||
}
|
||
return _voiceButton;
|
||
}
|
||
|
||
- (UILabel *)durationLabel {
|
||
if (!_durationLabel) {
|
||
_durationLabel = [[UILabel alloc] init];
|
||
_durationLabel.font = [UIFont systemFontOfSize:11];
|
||
_durationLabel.textColor = [UIColor whiteColor];
|
||
}
|
||
return _durationLabel;
|
||
}
|
||
|
||
- (UIActivityIndicatorView *)voiceLoadingIndicator {
|
||
if (!_voiceLoadingIndicator) {
|
||
_voiceLoadingIndicator = [[UIActivityIndicatorView alloc] initWithActivityIndicatorStyle:UIActivityIndicatorViewStyleMedium];
|
||
_voiceLoadingIndicator.color = [UIColor whiteColor];
|
||
_voiceLoadingIndicator.hidesWhenStopped = YES;
|
||
}
|
||
return _voiceLoadingIndicator;
|
||
}
|
||
|
||
- (UIActivityIndicatorView *)messageLoadingIndicator {
|
||
if (!_messageLoadingIndicator) {
|
||
_messageLoadingIndicator = [[UIActivityIndicatorView alloc] initWithActivityIndicatorStyle:UIActivityIndicatorViewStyleMedium];
|
||
_messageLoadingIndicator.color = [UIColor whiteColor];
|
||
_messageLoadingIndicator.hidesWhenStopped = YES;
|
||
}
|
||
return _messageLoadingIndicator;
|
||
}
|
||
|
||
- (UIImage *)kb_defaultAvatarImage {
|
||
if (@available(iOS 13.0, *)) {
|
||
return [UIImage systemImageNamed:@"person.circle.fill"];
|
||
}
|
||
return nil;
|
||
}
|
||
|
||
@end
|