Add event listener for auto-refresh and restore new tab button functionality

Co-authored-by: xusheng6 <[email protected]>

Author: Copilot

ui/ttdeventswidget.cpp

@@ -886,13 +886,17 @@ bool TTDEventsQueryWidget::isUnused() const
 
 // TTDEventsWidget implementation
 TTDEventsWidget::TTDEventsWidget(QWidget* parent, BinaryViewRef data)
-	: QWidget(parent), m_data(data)
+	: QWidget(parent), m_data(data), m_isPopulated(false)
 {
 	m_controller = DebuggerController::GetController(data);
 	setupUI();
 
-	// Automatically load events for all tabs
-	loadAllEvents();
+	// Only automatically load events if the debugger is running
+	if (m_controller && m_controller->IsConnectedToDebugServer())
+	{
+		loadAllEvents();
+		m_isPopulated = true;
+	}
 }
 
 TTDEventsWidget::~TTDEventsWidget()
@@ -906,8 +910,20 @@ void TTDEventsWidget::setupUI()
 
 	// Create tab widget
 	m_tabWidget = new QTabWidget();
+	m_tabWidget->setTabsClosable(true);
+	connect(m_tabWidget, &QTabWidget::tabCloseRequested, this, &TTDEventsWidget::closeTab);
 	layout->addWidget(m_tabWidget);
 
+	// Create "+" button as corner widget
+	m_newTabButton = new QToolButton(m_tabWidget);
+	m_newTabButton->setText("+");
+	m_newTabButton->setAutoRaise(true);
+	m_newTabButton->setToolTip("New tab");
+	connect(m_newTabButton, &QToolButton::clicked, this, &TTDEventsWidget::createNewTab);
+	
+	// Set the button as corner widget
+	m_tabWidget->setCornerWidget(m_newTabButton, Qt::TopRightCorner);
+
 	// Create the 3 specialized tabs
 	m_moduleEventsWidget = new TTDEventsQueryWidget(this, m_data, TTDEventsQueryWidget::ModuleEvents);
 	m_threadEventsWidget = new TTDEventsQueryWidget(this, m_data, TTDEventsQueryWidget::ThreadEvents);
@@ -929,6 +945,73 @@ void TTDEventsWidget::loadAllEvents()
 		m_exceptionEventsWidget->performInitialQuery();
 }
 
+void TTDEventsWidget::refreshAllTabs()
+{
+	// Only refresh if data is available and we haven't already populated
+	if (m_controller && m_controller->IsConnectedToDebugServer() && !m_isPopulated)
+	{
+		loadAllEvents();
+		m_isPopulated = true;
+	}
+}
+
+void TTDEventsWidget::clearAllTabs()
+{
+	if (m_moduleEventsWidget)
+		m_moduleEventsWidget->clearResults();
+	if (m_threadEventsWidget)
+		m_threadEventsWidget->clearResults();
+	if (m_exceptionEventsWidget)
+		m_exceptionEventsWidget->clearResults();
+	
+	// Clear all custom tabs as well
+	for (int i = 3; i < m_tabWidget->count(); i++)
+	{
+		TTDEventsQueryWidget* widget = qobject_cast<TTDEventsQueryWidget*>(m_tabWidget->widget(i));
+		if (widget)
+			widget->clearResults();
+	}
+	
+	m_isPopulated = false;
+}
+
+TTDEventsQueryWidget* TTDEventsWidget::getCurrentOrNewQueryWidget()
+{
+	// Get current tab widget
+	TTDEventsQueryWidget* currentWidget = qobject_cast<TTDEventsQueryWidget*>(m_tabWidget->currentWidget());
+	if (currentWidget)
+		return currentWidget;
+	
+	// If no current widget or cast failed, create a new tab
+	createNewTab();
+	return qobject_cast<TTDEventsQueryWidget*>(m_tabWidget->currentWidget());
+}
+
+void TTDEventsWidget::createNewTab()
+{
+	// Create new tab with AllEvents type (has filtering controls)
+	TTDEventsQueryWidget* queryWidget = new TTDEventsQueryWidget(this, m_data, TTDEventsQueryWidget::AllEvents);
+	int tabIndex = m_tabWidget->addTab(queryWidget, QString("Query %1").arg(m_tabWidget->count() - 2)); // -2 because we have 3 fixed tabs
+	m_tabWidget->setCurrentIndex(tabIndex);
+	
+	// If we've already populated the specialized tabs, also populate this new tab
+	if (m_isPopulated)
+	{
+		queryWidget->performInitialQuery();
+	}
+}
+
+void TTDEventsWidget::closeTab(int index)
+{
+	// Don't allow closing the first 3 specialized tabs
+	if (index >= 3 && m_tabWidget->count() > 3)
+	{
+		QWidget* widget = m_tabWidget->widget(index);
+		m_tabWidget->removeTab(index);
+		widget->deleteLater();
+	}
+}
+
 // TTDEventsSidebarWidget implementation
 TTDEventsSidebarWidget::TTDEventsSidebarWidget(BinaryViewRef data)
 	: SidebarWidget("TTD Events"), m_data(data)
@@ -942,10 +1025,44 @@ TTDEventsSidebarWidget::TTDEventsSidebarWidget(BinaryViewRef data)
 	layout->addWidget(m_eventsWidget);
 
 	setLayout(layout);
+
+	// Register for debugger events
+	if (m_controller)
+	{
+		connect(this, &TTDEventsSidebarWidget::debuggerEvent, this, &TTDEventsSidebarWidget::onDebuggerEvent);
+		
+		m_debuggerEventCallback = m_controller->RegisterEventCallback(
+			[&](const DebuggerEvent& event) {
+				emit debuggerEvent(event);
+			},
+			"TTD Events Widget");
+	}
 }
 
 TTDEventsSidebarWidget::~TTDEventsSidebarWidget()
 {
+	if (m_controller)
+		m_controller->RemoveEventCallback(m_debuggerEventCallback);
+}
+
+void TTDEventsSidebarWidget::onDebuggerEvent(const DebuggerEvent& event)
+{
+	switch (event.type)
+	{
+		case TargetStoppedEventType:
+			// When target stops, refresh all tabs if not already populated
+			if (m_eventsWidget)
+				m_eventsWidget->refreshAllTabs();
+			break;
+		case TargetExitedEventType:
+		case DetachedEventType:
+			// When target exits or is detached, clear all contents
+			if (m_eventsWidget)
+				m_eventsWidget->clearAllTabs();
+			break;
+		default:
+			break;
+	}
 }
 
 // TTDEventsWidgetType implementation

ui/ttdeventswidget.h

@@ -170,6 +170,7 @@ class TTDEventsWidget : public QWidget
 	QTabWidget* m_tabWidget;
 	QToolButton* m_newTabButton;
 	TTDEventsQueryWidget *m_moduleEventsWidget, *m_threadEventsWidget, *m_exceptionEventsWidget;
+	bool m_isPopulated;
 
 	void setupUI();
 	void loadAllEvents();
@@ -180,6 +181,14 @@ class TTDEventsWidget : public QWidget
 
 	// Method to get current query widget or create new tab
 	TTDEventsQueryWidget* getCurrentOrNewQueryWidget();
+	
+	// Methods for event handling
+	void refreshAllTabs();
+	void clearAllTabs();
+
+private slots:
+	void createNewTab();
+	void closeTab(int index);
 };
 
 
@@ -191,10 +200,17 @@ class TTDEventsSidebarWidget : public SidebarWidget
 	TTDEventsWidget* m_eventsWidget;
 	BinaryViewRef m_data;
 	DbgRef<DebuggerController> m_controller;
+	size_t m_debuggerEventCallback;
 
 public:
 	TTDEventsSidebarWidget(BinaryViewRef data);
 	~TTDEventsSidebarWidget();
+
+signals:
+	void debuggerEvent(const DebuggerEvent& event);
+
+private slots:
+	void onDebuggerEvent(const DebuggerEvent& event);
 };