fi(wysiwyg): fix navigation

ts_web/elements/wysiwyg/blocks/text/heading.block.ts

@@ -280,6 +280,22 @@ export class HeadingBlockHandler extends BaseBlockHandler {
     }
   }
   
+  /**
+   * Helper to get the last text node in an element
+   */
+  private getLastTextNode(element: Node): Text | null {
+    if (element.nodeType === Node.TEXT_NODE) {
+      return element as Text;
+    }
+    
+    for (let i = element.childNodes.length - 1; i >= 0; i--) {
+      const lastText = this.getLastTextNode(element.childNodes[i]);
+      if (lastText) return lastText;
+    }
+    
+    return null;
+  }
+  
   // Helper methods for heading functionality (mostly the same as paragraph)
   
   getCursorPosition(element: HTMLElement, context?: any): number | null {
@@ -404,19 +420,40 @@ export class HeadingBlockHandler extends BaseBlockHandler {
     const headingBlock = element.querySelector(`.block.heading-${this.level}`) as HTMLDivElement;
     if (!headingBlock) return;
     
+    
     // Ensure element is focusable first
     if (!headingBlock.hasAttribute('contenteditable')) {
       headingBlock.setAttribute('contenteditable', 'true');
     }
     
-    // Focus the element
+    // For 'end' position, we need to set up selection before focus to prevent browser default
+    if (position === 'end' && headingBlock.textContent && headingBlock.textContent.length > 0) {
+      // Set up the selection first
+      const sel = window.getSelection();
+      if (sel) {
+        const range = document.createRange();
+        const lastNode = this.getLastTextNode(headingBlock) || headingBlock;
+        if (lastNode.nodeType === Node.TEXT_NODE) {
+          range.setStart(lastNode, lastNode.textContent?.length || 0);
+          range.setEnd(lastNode, lastNode.textContent?.length || 0);
+        } else {
+          range.selectNodeContents(lastNode);
+          range.collapse(false);
+        }
+        sel.removeAllRanges();
+        sel.addRange(range);
+      }
+    }
+    
+    // Now focus the element
     headingBlock.focus();
     
-    // Set cursor position after focus is established
+    // Set cursor position after focus is established (for non-end positions)
     const setCursor = () => {
       if (position === 'start') {
         this.setCursorToStart(element, context);
-      } else if (position === 'end') {
+      } else if (position === 'end' && (!headingBlock.textContent || headingBlock.textContent.length === 0)) {
+        // Only call setCursorToEnd for empty blocks
         this.setCursorToEnd(element, context);
       } else if (typeof position === 'number') {
         // Use the selection utility to set cursor position
@@ -432,6 +469,13 @@ export class HeadingBlockHandler extends BaseBlockHandler {
       Promise.resolve().then(() => {
         if (document.activeElement === headingBlock || element.shadowRoot?.activeElement === headingBlock) {
           setCursor();
+        } else {
+          // Try again with a small delay - sometimes focus needs more time
+          setTimeout(() => {
+            if (document.activeElement === headingBlock || element.shadowRoot?.activeElement === headingBlock) {
+              setCursor();
+            }
+          }, 10);
         }
       });
     }