Add API for adding a new type of file open button on the new tab page

2 files changed, 5 insertions, 1 deletions

diff --git a/python/examples/triage/files.py b/python/examples/triage/files.py
index 333bcf5a..95de6a31 100644
--- a/python/examples/triage/files.py
+++ b/python/examples/triage/files.py
@@ -2,7 +2,7 @@ import os
 from PySide2.QtWidgets import QWidget, QTreeView, QFileSystemModel, QVBoxLayout, QMessageBox, QAbstractItemView
 from PySide2.QtGui import QKeySequence
 from PySide2.QtCore import QSettings
-from binaryninjaui import UIActionHandler, UIAction, Menu, FileContext, ContextMenuManager
+from binaryninjaui import UIActionHandler, UIAction, Menu, FileContext, ContextMenuManager, UIContext
 from binaryninja.settings import Settings
 
 
@@ -138,3 +138,5 @@ UIAction.registerAction("Open Selected Files")
 
 UIActionHandler.globalActions().bindAction("Open for Triage...", UIAction(openForTriage))
 Menu.mainMenu("File").addAction("Open for Triage...", "Open")
+
+UIContext.registerFileOpenMode("Triage...", "Open file(s) for quick analysis in the Triage Summary view.", "Open for Triage...")
diff --git a/ui/uicontext.h b/ui/uicontext.h
index e65a603f..e2a6e0a5 100644
--- a/ui/uicontext.h
+++ b/ui/uicontext.h
@@ -23,6 +23,7 @@ public:
 	virtual ~UIContextHandler();
 	virtual void updateStatus(bool updateInfo) = 0;
 	virtual void notifyThemeChanged() = 0;
+	virtual void registerFileOpenMode(const QString& buttonName, const QString& description, const QString& action);
 };
 
 class BINARYNINJAUIAPI UIContext
@@ -59,6 +60,7 @@ public:
 	static void closePreview();
 	static bool sendPreviewWheelEvent(QWheelEvent* event);
 	static void closeOtherActiveModalWidget(QWidget* current);
+	static void registerFileOpenMode(const QString& buttonName, const QString& description, const QString& action);
 
 	static UIContext* contextForWidget(QWidget* widget);
 	static UIContext* activeContext();