1

CustomKeyboard/KeyboardViewController.m

@@ -20,6 +20,7 @@
 #import "KBSettingView.h"
 #import "KBChatMessage.h"
 #import "KBChatPanelView.h"
+#import "KBChatLimitPopView.h"
 #import "KBSkinInstallBridge.h"
 #import "KBSkinManager.h"
 #import "KBSuggestionEngine.h"
@@ -66,7 +67,8 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
 @interface KeyboardViewController () <KBKeyBoardMainViewDelegate,
                                       KBFunctionViewDelegate,
                                       KBKeyboardSubscriptionViewDelegate,
-                                      KBChatPanelViewDelegate>
+                                      KBChatPanelViewDelegate,
+                                      KBChatLimitPopViewDelegate>
 @property(nonatomic, strong)
     UIButton *nextKeyboardButton; // 系统“下一个键盘”按钮（可选）
 @property(nonatomic, strong) UIView *contentView;
@@ -85,6 +87,7 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
 @property(nonatomic, strong) KBSuggestionEngine *suggestionEngine;
 @property(nonatomic, copy) NSString *currentWord;
 @property(nonatomic, assign) BOOL suppressSuggestions;
+@property(nonatomic, strong) UIControl *chatLimitMaskView;
 @property(nonatomic, strong) MASConstraint *contentWidthConstraint;
 @property(nonatomic, strong) MASConstraint *contentHeightConstraint;
 @property(nonatomic, strong) MASConstraint *keyBoardMainHeightConstraint;
@@ -867,14 +870,66 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
     [KBHUD showInfo:KBLocalized(@"请开启完全访问后使用")];
     return;
   }
-  
+
   // 添加 loading 消息
   [self.chatPanelView kb_addLoadingAssistantMessage];
-  
+
   // 调用新的聊天接口
   [self kb_requestChatMessageWithContent:text];
 }
 
+#pragma mark - Chat Limit Pop
+
+- (void)kb_showChatLimitPopWithMessage:(NSString *)message {
+  [self kb_dismissChatLimitPop];
+
+  UIControl *mask = [[UIControl alloc] init];
+  mask.backgroundColor = [[UIColor blackColor] colorWithAlphaComponent:0.4];
+  mask.alpha = 0.0;
+  [mask addTarget:self
+                action:@selector(kb_dismissChatLimitPop)
+      forControlEvents:UIControlEventTouchUpInside];
+  [self.contentView addSubview:mask];
+  [mask mas_makeConstraints:^(MASConstraintMaker *make) {
+    make.edges.equalTo(self.contentView);
+  }];
+
+  CGFloat width = 252.0;
+  CGFloat height = 252.0 + 18.0 + 53.0 + 18.0 + 28.0;
+  KBChatLimitPopView *content =
+      [[KBChatLimitPopView alloc] initWithFrame:CGRectMake(0, 0, width, height)];
+  content.message = message ?: @"";
+  content.delegate = self;
+  [mask addSubview:content];
+  [content mas_makeConstraints:^(MASConstraintMaker *make) {
+    make.center.equalTo(mask);
+    make.width.mas_equalTo(width);
+    make.height.mas_equalTo(height);
+  }];
+
+  self.chatLimitMaskView = mask;
+  [self.contentView bringSubviewToFront:mask];
+  [UIView animateWithDuration:0.18
+                   animations:^{
+                     mask.alpha = 1.0;
+                   }];
+}
+
+- (void)kb_dismissChatLimitPop {
+  if (!self.chatLimitMaskView) {
+    return;
+  }
+  UIControl *mask = self.chatLimitMaskView;
+  self.chatLimitMaskView = nil;
+  [UIView animateWithDuration:0.15
+                   animations:^{
+                     mask.alpha = 0.0;
+                   }
+                   completion:^(__unused BOOL finished) {
+                     [mask removeFromSuperview];
+                   }];
+}
+
 - (void)kb_clearHostInputForText:(NSString *)text {
   if (text.length == 0) {
     return;
@@ -1019,10 +1074,12 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
     if (!self) return;
     
     if (!response.success) {
-        if (response.code == 50030) {
-            
-            return;
-        }
+      if (response.code == 50030) {
+        NSLog(@"[KB] ⚠️ 次数用尽: %@", response.message);
+        [self.chatPanelView kb_removeLoadingAssistantMessage];
+        [self kb_showChatLimitPopWithMessage:response.message];
+        return;
+      }
       NSLog(@"[KB] ❌ 请求失败: %@", response.message);
       [self.chatPanelView kb_removeLoadingAssistantMessage];
       [KBHUD showInfo:response.message ?: KBLocalized(@"请求失败")];
@@ -1356,6 +1413,17 @@ static void KBSkinInstallNotificationCallback(CFNotificationCenterRef center,
   [self kb_openRechargeForProduct:product];
 }
 
+#pragma mark - KBChatLimitPopViewDelegate
+
+- (void)chatLimitPopViewDidTapCancel:(KBChatLimitPopView *)view {
+  [self kb_dismissChatLimitPop];
+}
+
+- (void)chatLimitPopViewDidTapRecharge:(KBChatLimitPopView *)view {
+  [self kb_dismissChatLimitPop];
+  [self showSubscriptionPanel];
+}
+
 #pragma mark - lazy
 - (KBKeyBoardMainView *)keyBoardMainView {
   if (!_keyBoardMainView) {