Fix issue #8 (#91)

FluidGroup · Oct 6, 2020 · e275c45 · e275c45
1 parent c01f4ba
commit e275c45
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/NextGrowingTextView/NextGrowingTextView.swift b/NextGrowingTextView/NextGrowingTextView.swift
@@ -165,6 +165,7 @@ open class NextGrowingTextView: UIScrollView {
 
   private func setup() {
 
+    NotificationCenter.default.addObserver(self, selector: #selector(willShowMenu(_:)), name: UIMenuController.willShowMenuNotification, object: nil)
     _textView.textContainerInset = .init(top: 4, left: 0, bottom: 4, right: 0)
     _textView.isScrollEnabled = false
     _textView.font = UIFont.systemFont(ofSize: 16)
@@ -181,6 +182,21 @@ open class NextGrowingTextView: UIScrollView {
     }
   }
 
+  private var isFixingMenuPosition: Bool = false
+  @objc private func willShowMenu(_ notification: Notification) {
+    guard let menuController = notification.object as? UIMenuController,
+          let superview = superview,
+          isFixingMenuPosition == false,
+          _textView.isFirstResponder,
+          !menuController.menuFrame.intersects(superview.convert(frame, to: nil))
+    else { return }
+    menuController.setMenuVisible(false, animated: false)
+    menuController.setTargetRect(frame, in: superview)
+    isFixingMenuPosition = true
+    menuController.setMenuVisible(true, animated: true)
+    isFixingMenuPosition = false
+  }
+
   private func measureTextViewSize() -> CGSize {
     let size = _textView.sizeThatFits(CGSize(width: self.bounds.width, height: CGFloat.infinity))
     return .init(width: size.width, height: max(size.height, _minHeight))