fix: set edit block immediately after ui refresh

src/main/frontend/handler/editor.cljs

@@ -832,24 +832,25 @@
                                                   (not (:logseq.property/created-from-property block)))]
                       (if delete-prev-block?
                         (p/do!
+                         (state/set-state! :editor/edit-block-fn
+                                           #(edit-block! (assoc block :block/title new-content) (count (:block/title prev-block))))
                          (ui-outliner-tx/transact!
                           transact-opts
                           (delete-block-aux! prev-block)
-                          (save-block! repo block new-content {}))
-                         (edit-block! (assoc block :block/title new-content) (count (:block/title prev-block))))
+                          (save-block! repo block new-content {})))
                         (p/do!
+                         (state/set-state! :editor/edit-block-fn edit-block-f)
                          (ui-outliner-tx/transact!
                           transact-opts
                           (when (seq children)
                             (outliner-op/move-blocks! children prev-block false))
                           (delete-block-aux! block)
-                          (save-block! repo prev-block new-content {}))
-                         (when edit-block-f (edit-block-f)))))
+                          (save-block! repo prev-block new-content {})))))
 
                     :else
                     (p/do!
-                     (delete-block-aux! block)
-                     (when edit-block-f (edit-block-f)))))))))))))
+                     (state/set-state! :editor/edit-block-fn edit-block-f)
+                     (delete-block-aux! block))))))))))))
 
 (defn delete-block!
   [repo]
@@ -864,14 +865,14 @@
           sibling-block (when block-parent (util/get-prev-block-non-collapsed-non-embed block-parent))
           blocks' (block-handler/get-top-level-blocks blocks)]
       (p/do!
-       (ui-outliner-tx/transact!
-        {:outliner-op :delete-blocks}
-        (outliner-op/delete-blocks! blocks' nil))
        (when sibling-block
          (let [{:keys [edit-block-f]} (move-to-prev-block repo sibling-block
                                                           (get block :block/format :markdown)
                                                           "")]
-           (when edit-block-f (edit-block-f))))))))
+           (state/set-state! :editor/edit-block-fn edit-block-f)))
+       (ui-outliner-tx/transact!
+        {:outliner-op :delete-blocks}
+        (outliner-op/delete-blocks! blocks' nil))))))
 
 (defn set-block-timestamp!
   [block-id key value]

src/main/frontend/modules/outliner/pipeline.cljs

@@ -78,6 +78,10 @@
             (state/set-state! :editor/start-pos nil)
 
             (when-not (:graph/importing @state/state)
+              (when-let [editor-block-f @(:editor/edit-block-fn @state/state)]
+                (state/set-state! :editor/edit-block-fn nil)
+                (editor-block-f))
+
               (react/refresh! repo affected-keys)
 
               (when (and state/lsp-enabled?

src/main/frontend/state.cljs

@@ -186,6 +186,7 @@
       :editor/next-edit-block                (atom nil)
       :editor/raw-mode-block                 (atom nil)
       :editor/virtualized-scroll-fn          nil
+      :editor/edit-block-fn                  (atom nil)
 
       ;; Warning: blocks order is determined when setting this attribute
       :selection/blocks                      (atom [])