From 0580a0c804252d6318fcaa3d29696be306acb315 Mon Sep 17 00:00:00 2001
From: Vanessa <lly219@gmail.com>
Date: Thu, 5 Oct 2023 22:08:56 +0800
Subject: [PATCH] :bug: focus av

---
 app/src/protyle/util/selection.ts | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/app/src/protyle/util/selection.ts b/app/src/protyle/util/selection.ts
index b162a4d10..ae83923c4 100644
--- a/app/src/protyle/util/selection.ts
+++ b/app/src/protyle/util/selection.ts
@@ -356,7 +356,7 @@ export const focusByOffset = (container: Element, start: number, end: number) =>
     const editElement = getContenteditableElement(container);
     if (editElement) {
         container = editElement;
-    } else if (isNotEditBlock(container)) {
+    } else if (isNotEditBlock(container) || container.classList.contains("av")) {
         return focusBlock(container);
     }
     let startNode;
@@ -533,6 +533,17 @@ export const focusBlock = (element: Element, parentElement?: HTMLElement, toStar
             focusSideBlock(element);
             return false;
         }
+    } else if (element.classList.contains("av")) {
+        const avTitleElement = element.querySelector(".av__title")
+        if (avTitleElement) {
+            const range = document.createRange();
+            range.selectNodeContents(avTitleElement);
+            range.collapse();
+            focusByRange(range);
+            return range;
+        } else {
+            return false;
+        }
     }
     let cursorElement;
     if (toStart) {