1

CustomKeyboard/KeyboardViewController.m

@@ -18,11 +18,15 @@
 #import "KBKeyboardSubscriptionProduct.h"
 #import "KBKeyboardSubscriptionView.h"
 #import "KBSettingView.h"
+#import "KBChatMessage.h"
+#import "KBChatPanelView.h"
 #import "KBSkinInstallBridge.h"
 #import "KBSkinManager.h"
 #import "KBSuggestionEngine.h"
+#import "KBNetworkManager.h"
 #import "Masonry.h"
 #import "UIImage+KBColor.h"
+#import <AVFoundation/AVFoundation.h>
 
 // #import "KBLog.h"
 
@@ -34,6 +38,8 @@
 
 // 以 375 宽设计稿为基准的键盘总高度
 static const CGFloat kKBKeyboardBaseHeight = 250.0f;
+static const CGFloat kKBChatPanelHeight = 180;
+static const NSUInteger kKBChatMessageLimit = 6;
 static NSString *const kKBDefaultSkinIdLight = @"normal_them";
 static NSString *const kKBDefaultSkinZipNameLight = @"normal_them";
 static NSString *const kKBDefaultSkinIdDark = @"normal_hei_them";
@@ -57,7 +63,8 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
 
 @interface KeyboardViewController () <KBKeyBoardMainViewDelegate,
                                       KBFunctionViewDelegate,
-                                      KBKeyboardSubscriptionViewDelegate>
+                                      KBKeyboardSubscriptionViewDelegate,
+                                      KBChatPanelViewDelegate>
 @property(nonatomic, strong)
     UIButton *nextKeyboardButton; // 系统“下一个键盘”按钮（可选）
 @property(nonatomic, strong) UIView *contentView;
@@ -67,16 +74,23 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
     KBFunctionView *functionView; // 功能面板视图（点击工具栏第0个时显示）
 @property(nonatomic, strong) KBSettingView *settingView; // 设置页
 @property(nonatomic, strong) UIImageView *bgImageView;   // 背景图（在底层）
+@property(nonatomic, strong) KBChatPanelView *chatPanelView;
 @property(nonatomic, strong) KBKeyboardSubscriptionView *subscriptionView;
 @property(nonatomic, strong) KBSuggestionEngine *suggestionEngine;
 @property(nonatomic, copy) NSString *currentWord;
 @property(nonatomic, assign) BOOL suppressSuggestions;
 @property(nonatomic, strong) MASConstraint *contentWidthConstraint;
 @property(nonatomic, strong) MASConstraint *contentHeightConstraint;
+@property(nonatomic, strong) MASConstraint *keyBoardMainHeightConstraint;
+@property(nonatomic, strong) MASConstraint *chatPanelHeightConstraint;
 @property(nonatomic, strong) NSLayoutConstraint *kb_heightConstraint;
 @property(nonatomic, strong) NSLayoutConstraint *kb_widthConstraint;
 @property(nonatomic, assign) CGFloat kb_lastPortraitWidth;
 @property(nonatomic, assign) CGFloat kb_lastKeyboardHeight;
+@property(nonatomic, strong) NSMutableArray<KBChatMessage *> *chatMessages;
+@property(nonatomic, strong) AVAudioPlayer *chatAudioPlayer;
+@property(nonatomic, strong) NSCache<NSString *, UIImage *> *chatAvatarCache;
+@property(nonatomic, assign) BOOL chatPanelVisible;
 @end
 
 @implementation KeyboardViewController
@@ -167,6 +181,8 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   // 按“短边”宽度等比缩放，横屏保持竖屏布局比例
   CGFloat portraitWidth = [self kb_portraitWidth];
   CGFloat keyboardHeight = [self kb_keyboardHeightForWidth:portraitWidth];
+  CGFloat keyboardBaseHeight = [self kb_keyboardBaseHeightForWidth:portraitWidth];
+  CGFloat chatPanelHeight = [self kb_chatPanelHeightForWidth:portraitWidth];
   CGFloat screenWidth = CGRectGetWidth([UIScreen mainScreen].bounds);
   CGFloat outerVerticalInset = KBFit(4.0f);
 
@@ -210,8 +226,20 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
 
   [self.contentView addSubview:self.keyBoardMainView];
   [self.keyBoardMainView mas_makeConstraints:^(MASConstraintMaker *make) {
-    make.edges.equalTo(self.contentView);
+    make.left.right.equalTo(self.contentView);
+    make.bottom.equalTo(self.contentView);
+    self.keyBoardMainHeightConstraint =
+        make.height.mas_equalTo(keyboardBaseHeight);
   }];
+
+  [self.contentView addSubview:self.chatPanelView];
+  [self.chatPanelView mas_makeConstraints:^(MASConstraintMaker *make) {
+    make.left.right.equalTo(self.contentView);
+    make.bottom.equalTo(self.keyBoardMainView.mas_top);
+    self.chatPanelHeightConstraint =
+        make.height.mas_equalTo(chatPanelHeight);
+  }];
+  self.chatPanelView.hidden = YES;
 }
 
 #pragma mark - Private
@@ -349,6 +377,9 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
 /// 切换显示功能面板/键盘主视图
 - (void)showFunctionPanel:(BOOL)show {
   // 简单显隐切换，复用相同的布局区域
+  if (show) {
+    [self showChatPanel:NO];
+  }
   self.functionView.hidden = !show;
   self.keyBoardMainView.hidden = show;
 
@@ -378,6 +409,7 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
 /// 显示/隐藏设置页（高度与 keyBoardMainView 一致），右侧滑入/滑出
 - (void)showSettingView:(BOOL)show {
   if (show) {
+    [self showChatPanel:NO];
     [[KBMaiPointReporter sharedReporter]
         reportPageExposureWithEventName:@"enter_keyboard_settings"
                                  pageId:@"keyboard_settings"
@@ -436,6 +468,40 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   }
 }
 
+/// 显示/隐藏聊天面板（覆盖整个键盘区域）
+- (void)showChatPanel:(BOOL)show {
+  if (show == self.chatPanelVisible) {
+    return;
+  }
+  self.chatPanelVisible = show;
+  if (show) {
+    self.chatPanelView.hidden = NO;
+    self.chatPanelView.alpha = 0.0;
+    [self.contentView bringSubviewToFront:self.chatPanelView];
+    self.functionView.hidden = YES;
+    [self hideSubscriptionPanel];
+    [self showSettingView:NO];
+    [UIView animateWithDuration:0.2
+                          delay:0
+                        options:UIViewAnimationOptionCurveEaseOut
+                     animations:^{
+                       self.chatPanelView.alpha = 1.0;
+                     }
+                     completion:nil];
+  } else {
+    [UIView animateWithDuration:0.18
+        delay:0
+        options:UIViewAnimationOptionCurveEaseIn
+        animations:^{
+          self.chatPanelView.alpha = 0.0;
+        }
+        completion:^(BOOL finished) {
+          self.chatPanelView.hidden = YES;
+        }];
+  }
+  [self kb_updateKeyboardLayoutIfNeeded];
+}
+
 - (void)showSubscriptionPanel {
   // 1) 先判断权限：未开启“完全访问”则走引导逻辑
   if (![[KBFullAccessManager shared] hasFullAccess]) {
@@ -544,6 +610,10 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
     [[KBInputBufferManager shared] appendText:@" "];
     break;
   case KBKeyTypeReturn:
+    if (self.chatPanelVisible) {
+      [self kb_handleChatSendAction];
+      break;
+    }
     [[KBBackspaceUndoManager shared] registerNonClearAction];
     [self.textDocumentProxy insertText:@"\n"];
     [self kb_clearCurrentWord];
@@ -575,11 +645,18 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
                          extra:extra
                     completion:nil];
   if (index == 0) {
+    [self showChatPanel:NO];
     [self showFunctionPanel:YES];
     [self kb_clearCurrentWord];
     return;
   }
+  if (index == 1) {
+    [self showFunctionPanel:NO];
+    [self showChatPanel:YES];
+    return;
+  }
   [self showFunctionPanel:NO];
+  [self showChatPanel:NO];
 }
 
 - (void)keyBoardMainViewDidTapSettings:(KBKeyBoardMainView *)keyBoardMainView {
@@ -697,6 +774,399 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   [self showSubscriptionPanel];
 }
 
+#pragma mark - KBChatPanelViewDelegate
+
+- (void)chatPanelView:(KBChatPanelView *)view didSendText:(NSString *)text {
+  NSString *trim =
+      [text stringByTrimmingCharactersInSet:
+                [NSCharacterSet whitespaceAndNewlineCharacterSet]];
+  if (trim.length == 0) {
+    return;
+  }
+  [self kb_sendChatText:trim];
+}
+
+- (void)chatPanelView:(KBChatPanelView *)view
+       didTapMessage:(KBChatMessage *)message {
+  if (message.audioFilePath.length == 0) {
+    return;
+  }
+  [self kb_playChatAudioAtPath:message.audioFilePath];
+}
+
+#pragma mark - Chat Helpers
+
+- (void)kb_handleChatSendAction {
+  if (!self.chatPanelVisible) {
+    return;
+  }
+  [[KBInputBufferManager shared] refreshFromProxyIfPossible:self.textDocumentProxy];
+  NSString *rawText = [KBInputBufferManager shared].liveText ?: @"";
+  NSString *trim =
+      [rawText stringByTrimmingCharactersInSet:
+                  [NSCharacterSet whitespaceAndNewlineCharacterSet]];
+  if (trim.length == 0) {
+    [KBHUD showInfo:KBLocalized(@"请输入内容")];
+    return;
+  }
+  [self kb_sendChatText:trim];
+  [self kb_clearHostInputForText:rawText];
+}
+
+- (void)kb_sendChatText:(NSString *)text {
+  if (text.length == 0) {
+    return;
+  }
+  KBChatMessage *outgoing =
+      [KBChatMessage messageWithText:text outgoing:YES audioFilePath:nil];
+  outgoing.avatarURL = [self kb_sharedUserAvatarURL];
+  [self kb_appendChatMessage:outgoing];
+  [self kb_prefetchAvatarForMessage:outgoing];
+
+  if (![[KBFullAccessManager shared] ensureFullAccessOrGuideInView:self.view]) {
+    [KBHUD showInfo:KBLocalized(@"请开启完全访问后使用")];
+    return;
+  }
+  [self kb_requestChatAudioForText:text];
+}
+
+- (void)kb_clearHostInputForText:(NSString *)text {
+  if (text.length == 0) {
+    return;
+  }
+  NSUInteger count = [self kb_composedCharacterCountForString:text];
+  for (NSUInteger i = 0; i < count; i++) {
+    [self.textDocumentProxy deleteBackward];
+  }
+  [[KBInputBufferManager shared] clearAllLiveText];
+  [self kb_clearCurrentWord];
+}
+
+- (NSUInteger)kb_composedCharacterCountForString:(NSString *)text {
+  if (text.length == 0) {
+    return 0;
+  }
+  __block NSUInteger count = 0;
+  [text enumerateSubstringsInRange:NSMakeRange(0, text.length)
+                           options:NSStringEnumerationByComposedCharacterSequences
+                        usingBlock:^(__unused NSString *substring,
+                                     __unused NSRange substringRange,
+                                     __unused NSRange enclosingRange,
+                                     __unused BOOL *stop) {
+                          count += 1;
+                        }];
+  return count;
+}
+
+- (NSString *)kb_sharedUserAvatarURL {
+  NSUserDefaults *ud = [[NSUserDefaults alloc] initWithSuiteName:AppGroup];
+  NSString *url = [ud stringForKey:AppGroup_UserAvatarURL];
+  return url ?: @"";
+}
+
+- (void)kb_prefetchAvatarForMessage:(KBChatMessage *)message {
+  if (!message || message.avatarImage) {
+    return;
+  }
+  NSString *urlString = message.avatarURL ?: @"";
+  if (urlString.length == 0) {
+    return;
+  }
+  UIImage *cached = [self.chatAvatarCache objectForKey:urlString];
+  if (cached) {
+    message.avatarImage = cached;
+    [self.chatPanelView kb_reloadWithMessages:self.chatMessages];
+    return;
+  }
+  if (![[KBFullAccessManager shared] hasFullAccess]) {
+    return;
+  }
+  __weak typeof(self) weakSelf = self;
+  [[KBNetworkManager shared]
+      GETData:urlString
+   parameters:nil
+      headers:nil
+   completion:^(NSData *data, NSURLResponse *response, NSError *error) {
+     if (error || data.length == 0) {
+       return;
+     }
+     UIImage *image = [UIImage imageWithData:data];
+     if (!image) {
+       return;
+     }
+     dispatch_async(dispatch_get_main_queue(), ^{
+       __strong typeof(weakSelf) self = weakSelf;
+       if (!self) {
+         return;
+       }
+       [self.chatAvatarCache setObject:image forKey:urlString];
+       message.avatarImage = image;
+       [self.chatPanelView kb_reloadWithMessages:self.chatMessages];
+     });
+   }];
+}
+
+- (void)kb_requestChatAudioForText:(NSString *)text {
+  NSString *mockPath = [self kb_mockChatAudioPath];
+  if (mockPath.length > 0) {
+    dispatch_after(dispatch_time(DISPATCH_TIME_NOW, (int64_t)(0.35 * NSEC_PER_SEC)),
+                   dispatch_get_main_queue(), ^{
+                     NSString *displayText = KBLocalized(@"语音回复");
+                     KBChatMessage *incoming =
+                         [KBChatMessage messageWithText:displayText
+                                                outgoing:NO
+                                           audioFilePath:mockPath];
+                     incoming.displayName = KBLocalized(@"AI助手");
+                     [self kb_appendChatMessage:incoming];
+                     [self kb_playChatAudioAtPath:mockPath];
+                   });
+    return;
+  }
+  NSDictionary *payload = @{@"message" : text ?: @""};
+  __weak typeof(self) weakSelf = self;
+  [[KBNetworkManager shared] POST:API_AI_TALK
+                         jsonBody:payload
+                          headers:nil
+                       completion:^(NSDictionary *json, NSURLResponse *response,
+                                    NSError *error) {
+                         dispatch_async(dispatch_get_main_queue(), ^{
+                           __strong typeof(weakSelf) self = weakSelf;
+                           if (!self) {
+                             return;
+                           }
+                           if (error) {
+                             NSString *tip = error.localizedDescription
+                                                 ?: KBLocalized(@"请求失败");
+                             [KBHUD showInfo:tip];
+                             return;
+                           }
+                           NSString *displayText =
+                               [self kb_chatTextFromJSON:json];
+                           NSString *audioURL =
+                               [self kb_chatAudioURLFromJSON:json];
+                           NSString *audioBase64 =
+                               [self kb_chatAudioBase64FromJSON:json];
+                           if (audioURL.length > 0) {
+                             [self kb_downloadChatAudioFromURL:audioURL
+                                                   displayText:displayText];
+                             return;
+                           }
+                           if (audioBase64.length > 0) {
+                             NSData *data = [[NSData alloc]
+                                 initWithBase64EncodedString:audioBase64
+                                                      options:0];
+                             if (data.length == 0) {
+                               [KBHUD showInfo:KBLocalized(@"音频数据解析失败")];
+                               return;
+                             }
+                             [self kb_handleChatAudioData:data
+                                              fileExtension:@"m4a"
+                                                 displayText:displayText];
+                             return;
+                           }
+                           [KBHUD showInfo:KBLocalized(@"未获取到音频文件")];
+                         });
+                       }];
+}
+
+- (void)kb_downloadChatAudioFromURL:(NSString *)audioURL
+                        displayText:(NSString *)displayText {
+  __weak typeof(self) weakSelf = self;
+  [[KBNetworkManager shared]
+      GETData:audioURL
+   parameters:nil
+      headers:nil
+   completion:^(NSData *data, NSURLResponse *response, NSError *error) {
+     dispatch_async(dispatch_get_main_queue(), ^{
+       __strong typeof(weakSelf) self = weakSelf;
+       if (!self) {
+         return;
+       }
+       if (error) {
+         NSString *tip = error.localizedDescription ?: KBLocalized(@"下载失败");
+         [KBHUD showInfo:tip];
+         return;
+       }
+       if (data.length == 0) {
+         [KBHUD showInfo:KBLocalized(@"未获取到音频数据")];
+         return;
+       }
+       NSString *ext = @"m4a";
+       NSURL *url = [NSURL URLWithString:audioURL];
+       if (url.pathExtension.length > 0) {
+         ext = url.pathExtension;
+       }
+       [self kb_handleChatAudioData:data
+                        fileExtension:ext
+                           displayText:displayText];
+     });
+   }];
+}
+
+- (void)kb_handleChatAudioData:(NSData *)data
+                 fileExtension:(NSString *)extension
+                    displayText:(NSString *)displayText {
+  if (data.length == 0) {
+    [KBHUD showInfo:KBLocalized(@"音频数据为空")];
+    return;
+  }
+  NSString *ext = extension.length > 0 ? extension : @"m4a";
+  NSString *fileName = [NSString
+      stringWithFormat:@"kb_chat_%@.%@",
+                       @((long long)([NSDate date].timeIntervalSince1970 *
+                                     1000)),
+                       ext];
+  NSString *filePath =
+      [NSTemporaryDirectory() stringByAppendingPathComponent:fileName];
+  if (![data writeToFile:filePath atomically:YES]) {
+    [KBHUD showInfo:KBLocalized(@"音频保存失败")];
+    return;
+  }
+  NSString *text = displayText.length > 0 ? displayText : KBLocalized(@"语音消息");
+  KBChatMessage *incoming =
+      [KBChatMessage messageWithText:text
+                             outgoing:NO
+                        audioFilePath:filePath];
+  incoming.displayName = KBLocalized(@"AI助手");
+  [self kb_appendChatMessage:incoming];
+}
+
+- (void)kb_appendChatMessage:(KBChatMessage *)message {
+  if (!message) {
+    return;
+  }
+  [self.chatMessages addObject:message];
+  if (self.chatMessages.count > kKBChatMessageLimit) {
+    NSUInteger overflow = self.chatMessages.count - kKBChatMessageLimit;
+    NSArray<KBChatMessage *> *removed =
+        [self.chatMessages subarrayWithRange:NSMakeRange(0, overflow)];
+    [self.chatMessages removeObjectsInRange:NSMakeRange(0, overflow)];
+    for (KBChatMessage *msg in removed) {
+      if (msg.audioFilePath.length > 0) {
+        NSString *tmpRoot = NSTemporaryDirectory();
+        if (tmpRoot.length > 0 &&
+            [msg.audioFilePath hasPrefix:tmpRoot]) {
+          [[NSFileManager defaultManager] removeItemAtPath:msg.audioFilePath
+                                                     error:nil];
+        }
+      }
+    }
+  }
+  [self.chatPanelView kb_reloadWithMessages:self.chatMessages];
+}
+
+- (NSString *)kb_mockChatAudioPath {
+  NSString *path = [[NSBundle mainBundle] pathForResource:@"ai_test"
+                                                   ofType:@"m4a"];
+  return path ?: @"";
+}
+
+- (NSString *)kb_chatTextFromJSON:(NSDictionary *)json {
+  NSDictionary *data = [self kb_chatDataDictionaryFromJSON:json];
+  NSString *text =
+      [self kb_stringValueInDict:data
+                             keys:@[ @"text", @"message", @"content" ]];
+  if (text.length == 0) {
+    text = [self kb_stringValueInDict:json
+                                  keys:@[ @"text", @"message", @"content" ]];
+  }
+  return text ?: @"";
+}
+
+- (NSString *)kb_chatAudioURLFromJSON:(NSDictionary *)json {
+  NSDictionary *data = [self kb_chatDataDictionaryFromJSON:json];
+  NSArray<NSString *> *keys =
+      @[ @"audioUrl", @"audioURL", @"audio_url", @"url", @"fileUrl",
+         @"file_url", @"audioFileUrl", @"audio_file_url" ];
+  NSString *url = [self kb_stringValueInDict:data keys:keys];
+  if (url.length == 0) {
+    url = [self kb_stringValueInDict:json keys:keys];
+  }
+  return url ?: @"";
+}
+
+- (NSString *)kb_chatAudioBase64FromJSON:(NSDictionary *)json {
+  NSDictionary *data = [self kb_chatDataDictionaryFromJSON:json];
+  NSArray<NSString *> *keys =
+      @[ @"audioBase64", @"audio_base64", @"audioData", @"audio_data",
+         @"base64" ];
+  NSString *b64 = [self kb_stringValueInDict:data keys:keys];
+  if (b64.length == 0) {
+    b64 = [self kb_stringValueInDict:json keys:keys];
+  }
+  return b64 ?: @"";
+}
+
+- (NSDictionary *)kb_chatDataDictionaryFromJSON:(NSDictionary *)json {
+  if (![json isKindOfClass:[NSDictionary class]]) {
+    return @{};
+  }
+  id dataObj = json[@"data"] ?: json[@"result"] ?: json[@"response"];
+  if ([dataObj isKindOfClass:[NSDictionary class]]) {
+    return (NSDictionary *)dataObj;
+  }
+  return @{};
+}
+
+- (NSString *)kb_stringValueInDict:(NSDictionary *)dict
+                               keys:(NSArray<NSString *> *)keys {
+  if (![dict isKindOfClass:[NSDictionary class]]) {
+    return @"";
+  }
+  for (NSString *key in keys) {
+    id value = dict[key];
+    if ([value isKindOfClass:[NSString class]] &&
+        ((NSString *)value).length > 0) {
+      return (NSString *)value;
+    }
+  }
+  return @"";
+}
+
+- (void)kb_playChatAudioAtPath:(NSString *)path {
+  if (path.length == 0) {
+    return;
+  }
+  NSURL *url = [NSURL fileURLWithPath:path];
+  if (![NSFileManager.defaultManager fileExistsAtPath:path]) {
+    [KBHUD showInfo:KBLocalized(@"音频文件不存在")];
+    return;
+  }
+
+  if (self.chatAudioPlayer && self.chatAudioPlayer.isPlaying) {
+    NSURL *currentURL = self.chatAudioPlayer.url;
+    if ([currentURL isEqual:url]) {
+      [self.chatAudioPlayer stop];
+      self.chatAudioPlayer = nil;
+      return;
+    }
+    [self.chatAudioPlayer stop];
+    self.chatAudioPlayer = nil;
+  }
+
+  NSError *sessionError = nil;
+  AVAudioSession *session = [AVAudioSession sharedInstance];
+  if ([session respondsToSelector:@selector(setCategory:options:error:)]) {
+    [session setCategory:AVAudioSessionCategoryPlayback
+             withOptions:AVAudioSessionCategoryOptionDuckOthers
+                   error:&sessionError];
+  } else {
+    [session setCategory:AVAudioSessionCategoryPlayback error:&sessionError];
+  }
+  [session setActive:YES error:nil];
+
+  NSError *playerError = nil;
+  AVAudioPlayer *player =
+      [[AVAudioPlayer alloc] initWithContentsOfURL:url error:&playerError];
+  if (playerError || !player) {
+    [KBHUD showInfo:KBLocalized(@"音频播放失败")];
+    return;
+  }
+  self.chatAudioPlayer = player;
+  [player prepareToPlay];
+  [player play];
+}
+
 #pragma mark - KBKeyboardSubscriptionViewDelegate
 
 - (void)subscriptionViewDidTapClose:(KBKeyboardSubscriptionView *)view {
@@ -750,6 +1220,28 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   return _settingView;
 }
 
+- (KBChatPanelView *)chatPanelView {
+  if (!_chatPanelView) {
+    _chatPanelView = [[KBChatPanelView alloc] init];
+    _chatPanelView.delegate = self;
+  }
+  return _chatPanelView;
+}
+
+- (NSMutableArray<KBChatMessage *> *)chatMessages {
+  if (!_chatMessages) {
+    _chatMessages = [NSMutableArray array];
+  }
+  return _chatMessages;
+}
+
+- (NSCache<NSString *, UIImage *> *)chatAvatarCache {
+  if (!_chatAvatarCache) {
+    _chatAvatarCache = [[NSCache alloc] init];
+  }
+  return _chatAvatarCache;
+}
+
 - (KBKeyboardSubscriptionView *)subscriptionView {
   if (!_subscriptionView) {
     _subscriptionView = [[KBKeyboardSubscriptionView alloc] init];
@@ -1150,12 +1642,36 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   if (width <= 0) {
     width = KB_DESIGN_WIDTH;
   }
-  return kKBKeyboardBaseHeight * (width / KB_DESIGN_WIDTH);
+  CGFloat scale = width / KB_DESIGN_WIDTH;
+  CGFloat baseHeight = kKBKeyboardBaseHeight * scale;
+  CGFloat chatHeight = kKBChatPanelHeight * scale;
+  if (self.chatPanelVisible) {
+    return baseHeight + chatHeight;
+  }
+  return baseHeight;
+}
+
+- (CGFloat)kb_keyboardBaseHeightForWidth:(CGFloat)width {
+  if (width <= 0) {
+    width = KB_DESIGN_WIDTH;
+  }
+  CGFloat scale = width / KB_DESIGN_WIDTH;
+  return kKBKeyboardBaseHeight * scale;
+}
+
+- (CGFloat)kb_chatPanelHeightForWidth:(CGFloat)width {
+  if (width <= 0) {
+    width = KB_DESIGN_WIDTH;
+  }
+  CGFloat scale = width / KB_DESIGN_WIDTH;
+  return kKBChatPanelHeight * scale;
 }
 
 - (void)kb_updateKeyboardLayoutIfNeeded {
   CGFloat portraitWidth = [self kb_portraitWidth];
   CGFloat keyboardHeight = [self kb_keyboardHeightForWidth:portraitWidth];
+  CGFloat keyboardBaseHeight = [self kb_keyboardBaseHeightForWidth:portraitWidth];
+  CGFloat chatPanelHeight = [self kb_chatPanelHeightForWidth:portraitWidth];
   CGFloat containerWidth = CGRectGetWidth(self.view.superview.bounds);
   if (containerWidth <= 0) {
     containerWidth = CGRectGetWidth(self.view.window.bounds);
@@ -1186,6 +1702,12 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   if (self.contentHeightConstraint) {
     [self.contentHeightConstraint setOffset:keyboardHeight];
   }
+  if (self.keyBoardMainHeightConstraint) {
+    [self.keyBoardMainHeightConstraint setOffset:keyboardBaseHeight];
+  }
+  if (self.chatPanelHeightConstraint) {
+    [self.chatPanelHeightConstraint setOffset:chatPanelHeight];
+  }
   [self.view layoutIfNeeded];
 }