From 053a84483cac1aa8589121ae76dd39524099f4c3 Mon Sep 17 00:00:00 2001
From: Adorian Doran <adorian@esevo.ro>
Date: Sat, 20 Sep 2025 03:25:16 +0300
Subject: [PATCH] client/menus: improve the multicolumn breaking strategy

---
 apps/client/src/menus/context_menu.ts | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/apps/client/src/menus/context_menu.ts b/apps/client/src/menus/context_menu.ts
index faef59ad0..1c4da4c23 100644
--- a/apps/client/src/menus/context_menu.ts
+++ b/apps/client/src/menus/context_menu.ts
@@ -2,6 +2,7 @@ import { KeyboardActionNames } from "@triliumnext/commons";
 import keyboardActionService, { getActionSync } from "../services/keyboard_actions.js";
 import note_tooltip from "../services/note_tooltip.js";
 import utils from "../services/utils.js";
+import { should } from "vitest";
 
 export interface ContextMenuOptions<T> {
     x: number;
@@ -170,9 +171,13 @@ class ContextMenu {
             // This is a workaround for Firefox not supporting break-after: avoid on columns.
             const nextItem = (index < items.length - 1) ? items[index + 1] : null;
             if (multicolumn && nextItem && "kind" in nextItem) {
-                if (!shouldResetGroup && (nextItem.kind === "separator" || nextItem.kind === "header")) {
-                    $group = $("<div class='dropdown-no-break'>");
-                    $parent.append($group);
+                if (nextItem.kind === "separator" || nextItem.kind === "header") {
+                    if (!shouldResetGroup) {
+                        $group = $("<div class='dropdown-no-break'>");
+                        $parent.append($group);
+                    } else {
+                        shouldResetGroup = false; // Continue the current group
+                    }
                 }
             }