处理键盘ai功能

CustomKeyboard/View/KBToolBar.m

@@ -15,23 +15,25 @@
 @property (nonatomic, strong) NSArray<UIButton *> *leftButtonsInternal;
 //@property (nonatomic, strong) UIButton *settingsButtonInternal;
 @property (nonatomic, strong) UIButton *globeButtonInternal;   // 可选：系统“切换输入法”键
+@property (nonatomic, strong) UIImageView *avatarImageView;    // 右侧头像（AppGroup persona_cover.jpg）
 @property (nonatomic, strong) UIButton *undoButtonInternal;    // 右侧撤销删除
 @property (nonatomic, assign) BOOL kbNeedsInputModeSwitchKey;
 @property (nonatomic, assign) BOOL kbUndoVisible;
+@property (nonatomic, assign) BOOL kbAvatarVisible;
 @end
 
 @implementation KBToolBar
 
-static NSString * const kKBAIKeyIdentifier = @"ai";
-static NSString * const kKBUndoKeyIdentifier = @"key_revoke";
-static const CGFloat kKBAIButtonWidth = 40;
-static const CGFloat kKBAIButtonHeight = 40;
-static const NSInteger kKBVoiceButtonIndex = 1;
+	static NSString * const kKBAIKeyIdentifier = @"ai";
+	static NSString * const kKBUndoKeyIdentifier = @"key_revoke";
+	static const CGFloat kKBAIButtonWidth = 40;
+	static const CGFloat kKBAIButtonHeight = 40;
+	static const CGFloat kKBAvatarSize = 40;
 
 - (instancetype)initWithFrame:(CGRect)frame{
     if (self = [super initWithFrame:frame]) {
         self.backgroundColor = [UIColor clearColor];
-        _leftButtonTitles = @[@"AI", KBLocalized(@"语音")]; // 默认标题
+        _leftButtonTitles = @[@"AI"]; // 默认标题
         [self setupUI];
         [[NSNotificationCenter defaultCenter] addObserver:self
                                                  selector:@selector(kb_undoStateChanged)
@@ -69,7 +71,6 @@ static const NSInteger kKBVoiceButtonIndex = 1;
         }
     }];
     [self kb_updateAIButtonAppearance];
-    [self kb_updateVoiceButtonAppearance];
 }
 
 #pragma mark - 视图搭建
@@ -79,6 +80,7 @@ static const NSInteger kKBVoiceButtonIndex = 1;
 //    [self addSubview:self.settingsButtonInternal];
     [self addSubview:self.globeButtonInternal];
     [self addSubview:self.undoButtonInternal];
+    [self addSubview:self.avatarImageView];
 
     // 右侧设置按钮
 //    [self.settingsButtonInternal mas_makeConstraints:^(MASConstraintMaker *make) {
@@ -94,13 +96,7 @@ static const NSInteger kKBVoiceButtonIndex = 1;
         make.width.height.mas_equalTo(32);
     }];
 
-    // 右侧撤销按钮
-    [self.undoButtonInternal mas_makeConstraints:^(MASConstraintMaker *make) {
-        make.right.equalTo(self.mas_right).offset(-12);
-        make.centerY.equalTo(self.mas_centerY);
-        make.height.mas_equalTo(32);
-        make.width.mas_equalTo(84);
-    }];
+    [self kb_updateRightControlsConstraints];
 
     [self kb_updateLeftContainerConstraints];
 
@@ -173,8 +169,8 @@ static const NSInteger kKBVoiceButtonIndex = 1;
 
 - (void)kb_applyTheme {
     [self kb_updateAIButtonAppearance];
-    [self kb_updateVoiceButtonAppearance];
     [self kb_updateUndoButtonAppearance];
+    [self kb_updateAvatarAppearance];
 }
 
 - (void)kb_updateAIButtonAppearance {
@@ -211,16 +207,6 @@ static const NSInteger kKBVoiceButtonIndex = 1;
     }
 }
 
-- (void)kb_updateVoiceButtonAppearance {
-    UIButton *voiceButton = [self kb_voiceButton];
-    if (!voiceButton) { return; }
-
-    voiceButton.backgroundColor = [UIColor colorWithHex:0xE53935];
-    [voiceButton setTitleColor:[UIColor whiteColor] forState:UIControlStateNormal];
-    voiceButton.layer.cornerRadius = 16;
-    voiceButton.layer.masksToBounds = YES;
-}
-
 - (void)kb_updateUndoButtonAppearance {
     if (!self.undoButtonInternal) { return; }
 
@@ -241,6 +227,41 @@ static const NSInteger kKBVoiceButtonIndex = 1;
     }
 }
 
+#pragma mark - Avatar
+
+- (void)kb_updateAvatarAppearance {
+    UIImage *img = [self kb_personaCoverImageFromAppGroup];
+    BOOL shouldShow = (img != nil);
+    self.avatarImageView.image = img;
+    if (self.kbAvatarVisible == shouldShow) {
+        self.avatarImageView.hidden = !shouldShow;
+        return;
+    }
+    self.kbAvatarVisible = shouldShow;
+    self.avatarImageView.hidden = !shouldShow;
+    [self kb_updateRightControlsConstraints];
+    [self kb_updateLeftContainerConstraints];
+    [self setNeedsLayout];
+    [self layoutIfNeeded];
+}
+
+- (nullable UIImage *)kb_personaCoverImageFromAppGroup {
+    NSURL *containerURL = [[NSFileManager defaultManager]
+        containerURLForSecurityApplicationGroupIdentifier:AppGroup];
+    if (!containerURL) {
+        return nil;
+    }
+
+    NSString *imagePath =
+        [[containerURL path] stringByAppendingPathComponent:@"persona_cover.jpg"];
+    if (imagePath.length == 0 ||
+        ![[NSFileManager defaultManager] fileExistsAtPath:imagePath]) {
+        return nil;
+    }
+
+    return [UIImage imageWithContentsOfFile:imagePath];
+}
+
 #pragma mark - Actions
 
 - (void)onLeftAction:(UIButton *)sender {
@@ -261,6 +282,16 @@ static const NSInteger kKBVoiceButtonIndex = 1;
     }
 }
 
+- (void)onAvatarTap {
+    if (!self.kbAvatarVisible || self.avatarImageView.hidden) {
+        return;
+    }
+    // 复用原“语音”入口的 index=1 逻辑（外部会按 index 做面板切换）
+    if ([self.delegate respondsToSelector:@selector(toolBar:didTapActionAtIndex:)]) {
+        [self.delegate toolBar:self didTapActionAtIndex:1];
+    }
+}
+
 #pragma mark - Lazy
 
 - (UIView *)leftContainer {
@@ -271,6 +302,23 @@ static const NSInteger kKBVoiceButtonIndex = 1;
     return _leftContainer;
 }
 
+- (UIImageView *)avatarImageView {
+    if (!_avatarImageView) {
+        _avatarImageView = [[UIImageView alloc] init];
+        _avatarImageView.hidden = YES;
+        _avatarImageView.backgroundColor = [UIColor colorWithWhite:1 alpha:0.9];
+        _avatarImageView.contentMode = UIViewContentModeScaleAspectFill;
+        _avatarImageView.layer.cornerRadius = kKBAvatarSize * 0.5;
+        _avatarImageView.layer.masksToBounds = YES;
+        _avatarImageView.userInteractionEnabled = YES;
+        UITapGestureRecognizer *tap = [[UITapGestureRecognizer alloc]
+            initWithTarget:self
+                    action:@selector(onAvatarTap)];
+        [_avatarImageView addGestureRecognizer:tap];
+    }
+    return _avatarImageView;
+}
+
 //- (UIButton *)settingsButtonInternal {
 //    if (!_settingsButtonInternal) {
 //        _settingsButtonInternal = [UIButton buttonWithType:UIButtonTypeSystem];
@@ -319,11 +367,6 @@ static const NSInteger kKBVoiceButtonIndex = 1;
     return self.leftButtonsInternal[0];
 }
 
-- (UIButton *)kb_voiceButton {
-    if (self.leftButtonsInternal.count <= kKBVoiceButtonIndex) { return nil; }
-    return self.leftButtonsInternal[kKBVoiceButtonIndex];
-}
-
 #pragma mark - Globe (Input Mode Switch)
 
 // 根据宿主是否已提供系统切换键，决定是否显示地球按钮；并绑定系统事件。
@@ -362,6 +405,8 @@ static const NSInteger kKBVoiceButtonIndex = 1;
         }
         if (self.kbUndoVisible) {
             make.right.equalTo(self.undoButtonInternal.mas_left).offset(-8);
+        } else if (self.kbAvatarVisible) {
+            make.right.equalTo(self.avatarImageView.mas_left).offset(-8);
         } else {
             make.right.equalTo(self).offset(-12);
         }
@@ -371,6 +416,24 @@ static const NSInteger kKBVoiceButtonIndex = 1;
     }];
 }
 
+- (void)kb_updateRightControlsConstraints {
+    [self.avatarImageView mas_remakeConstraints:^(MASConstraintMaker *make) {
+        make.right.equalTo(self).offset(-12);
+        make.centerY.equalTo(self).offset(5);
+        make.width.height.mas_equalTo(kKBAvatarSize);
+    }];
+    [self.undoButtonInternal mas_remakeConstraints:^(MASConstraintMaker *make) {
+        if (self.kbAvatarVisible) {
+            make.right.equalTo(self.avatarImageView.mas_left).offset(-8);
+        } else {
+            make.right.equalTo(self).offset(-12);
+        }
+        make.centerY.equalTo(self.mas_centerY);
+        make.height.mas_equalTo(32);
+        make.width.mas_equalTo(84);
+    }];
+}
+
 - (void)kb_undoStateChanged {
     [self kb_updateUndoVisibilityAnimated:YES];
 }
@@ -405,6 +468,7 @@ static const NSInteger kKBVoiceButtonIndex = 1;
 - (void)didMoveToWindow {
     [super didMoveToWindow];
     [self kb_refreshGlobeVisibility];
+    [self kb_updateAvatarAppearance];
 }
 
 @end