键盘添加撤销操作

CustomKeyboard/Utils/KBBackspaceLongPressHandler.m

@@ -6,6 +6,7 @@
 #import "KBBackspaceLongPressHandler.h"
 #import "KBResponderUtils.h"
 #import "KBSkinManager.h"
+#import "KBBackspaceUndoManager.h"
 
 static const NSTimeInterval kKBBackspaceLongPressMinDuration = 0.35;
 static const NSTimeInterval kKBBackspaceRepeatInterval = 0.06;
@@ -96,6 +97,7 @@ typedef NS_ENUM(NSInteger, KBBackspaceChunkClass) {
     }
     switch (gr.state) {
         case UIGestureRecognizerStateBegan: {
+            [[KBBackspaceUndoManager shared] registerNonClearAction];
             self.backspaceHoldToken += 1;
             NSUInteger token = self.backspaceHoldToken;
             self.backspaceHoldActive = YES;
@@ -406,6 +408,12 @@ typedef NS_ENUM(NSInteger, KBBackspaceChunkClass) {
 #pragma mark - Clear
 
 - (void)kb_clearAllInput {
+    UIResponder *start = (UIResponder *)([self kb_hostView] ?: self.backspaceButton);
+    UIInputViewController *ivc = KBFindInputViewController(start);
+    if (ivc) {
+        NSString *before = ivc.textDocumentProxy.documentContextBeforeInput ?: @"";
+        [[KBBackspaceUndoManager shared] recordClearWithContext:before];
+    }
     self.backspaceClearToken += 1;
     NSUInteger token = self.backspaceClearToken;
     [self kb_clearAllInputStepForToken:token guard:0 emptyRounds:0];