Add late-bound probe manager access for TUI

2 files changed, 28 insertions, 0 deletions

diff --git a/internal/ior.go b/internal/ior.go
index d78c351..61c5e10 100644
--- a/internal/ior.go
+++ b/internal/ior.go
@@ -231,6 +231,8 @@ func runTraceWithContext(parentCtx context.Context, started chan<- struct{}, con
 	if err := mgr.AttachAll(flags.Get().ShouldIAttachTracepoint, tracepoints.List); err != nil {
 		return err
 	}
+	tui.SetProbeManager(mgr)
+	defer tui.SetProbeManager(nil)
 
 	// 4096 channel size, minimises event drops
 	ch := make(chan []byte, 4096)
diff --git a/internal/tui/tui.go b/internal/tui/tui.go
index 5aff6a2..ea35b04 100644
--- a/internal/tui/tui.go
+++ b/internal/tui/tui.go
@@ -6,6 +6,7 @@ import (
 	"errors"
 	"fmt"
 	"ior/internal/flags"
+	"ior/internal/probemanager"
 	"ior/internal/statsengine"
 	common "ior/internal/tui/common"
 	dashboardui "ior/internal/tui/dashboard"
@@ -41,6 +42,13 @@ type snapshotSource interface {
 	Snapshot() *statsengine.Snapshot
 }
 
+// ProbeManager exposes runtime probe controls to TUI layers.
+type ProbeManager interface {
+	States() []probemanager.ProbeState
+	Toggle(syscall string) error
+	ActiveCount() (int, int)
+}
+
 var dashboardSourceState struct {
 	mu     sync.RWMutex
 	source snapshotSource
@@ -51,6 +59,11 @@ var eventStreamSourceState struct {
 	source *eventstream.RingBuffer
 }
 
+var probeManagerState struct {
+	mu      sync.RWMutex
+	manager ProbeManager
+}
+
 // SetDashboardSnapshotSource sets the snapshot source used by dashboard mode.
 func SetDashboardSnapshotSource(source snapshotSource) {
 	dashboardSourceState.mu.Lock()
@@ -65,6 +78,13 @@ func SetEventStreamSource(source *eventstream.RingBuffer) {
 	eventStreamSourceState.mu.Unlock()
 }
 
+// SetProbeManager sets the probe manager used by TUI probe controls.
+func SetProbeManager(manager ProbeManager) {
+	probeManagerState.mu.Lock()
+	probeManagerState.manager = manager
+	probeManagerState.mu.Unlock()
+}
+
 func getDashboardSnapshotSource() snapshotSource {
 	dashboardSourceState.mu.RLock()
 	defer dashboardSourceState.mu.RUnlock()
@@ -77,6 +97,12 @@ func getEventStreamSource() *eventstream.RingBuffer {
 	return eventStreamSourceState.source
 }
 
+func getProbeManager() ProbeManager {
+	probeManagerState.mu.RLock()
+	defer probeManagerState.mu.RUnlock()
+	return probeManagerState.manager
+}
+
 // Run starts the TUI program in alternate screen mode.
 func Run() error {
 	return RunWithTraceStarter(defaultTraceStarter)