From 85f6fa053dce2cdb41757e038c6e842ccbe2a6e8 Mon Sep 17 00:00:00 2001
From: aclist <92275929+aclist@users.noreply.github.com>
Date: Sat, 13 Jun 2026 02:10:41 +0900
Subject: [PATCH] fix: disable context menu on custom mod tree

---
 dzgui/const/enum.py            | 1 +
 dzgui/views/pages/offline.py   | 3 ++-
 dzgui/views/trees/tree_mods.py | 6 ++++--
 3 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/dzgui/const/enum.py b/dzgui/const/enum.py
index c14f6fe..9e84cc5 100644
--- a/dzgui/const/enum.py
+++ b/dzgui/const/enum.py
@@ -185,6 +185,7 @@ class ContextMenuGroup(Enum):
 
     SERVER_MOD = (ContextMenu.OPEN_WORKSHOP,)
     MOD = (ContextMenu.OPEN_WORKSHOP, ContextMenu.DELETE_MOD)
+    MOD_OFFLINE = (None,)
     LOG = (ContextMenu.COPY_LOG_CLIPBOARD,)
     SERVER_BROWSER = (
         ContextMenu.CONNECT,
diff --git a/dzgui/views/pages/offline.py b/dzgui/views/pages/offline.py
index 9a3d3bc..ad9642c 100644
--- a/dzgui/views/pages/offline.py
+++ b/dzgui/views/pages/offline.py
@@ -12,7 +12,7 @@ from dzgui.const.constants import (
     ERROR,
     FOLDER,
 )
-from dzgui.const.enum import NotebookPage
+from dzgui.const.enum import ContextMenuGroup, NotebookPage
 from dzgui.managers.offline import OfflineManager
 from dzgui.strings import generic, offline
 from dzgui.views.components.buttons import Icon, IconTextButton
@@ -259,6 +259,7 @@ class CustomModFrame(ModFrame):
     ) -> None:
         super().__init__(parent, controller, heading)
 
+        self.tree.set_menu(ContextMenuGroup.MOD_OFFLINE)
         self.parent = parent
 
         self.controller = controller
diff --git a/dzgui/views/trees/tree_mods.py b/dzgui/views/trees/tree_mods.py
index 45ee6ef..64dd55c 100644
--- a/dzgui/views/trees/tree_mods.py
+++ b/dzgui/views/trees/tree_mods.py
@@ -23,8 +23,10 @@ logger = logging.getLogger(APP_NAME)
 
 
 class ModTreeView(ModsMixin, ContextMixin, TreeView):  # type: ignore
-    def __init__(self, controller: "Controller") -> None:
-        super().__init__(controller, menu=ContextMenuGroup.MOD)
+    def __init__(
+        self, controller: "Controller", menu: ContextMenuGroup = ContextMenuGroup.MOD
+    ) -> None:
+        super().__init__(controller, menu=menu)
 
         self.controller = controller
         self.mod_man = ModManager(self, controller)