Experimented with ImGUI tables for session view.

2025-09-07 17:18:50 +02:00
parent 87059ced4c
commit 2320b705d3
4 changed files with 323 additions and 78 deletions
--- a/src/client/views/sessions.nim
+++ b/src/client/views/sessions.nim
@@ -0,0 +1,112 @@
+import times
+import imguin/[cimgui, glfw_opengl, simple]
+import ../utils/appImGui
+import ../../common/[types]
+
+type 
+    SessionsTableComponent = ref object of RootObj
+        title: string 
+        agents: seq[Agent]
+
+
+let exampleAgents: seq[Agent] = @[
+  Agent(
+    agentId: "DEADBEEF",
+    listenerId: "L1234567",
+    username: "alice",
+    hostname: "DESKTOP-01",
+    domain: "CORP",
+    ip: "192.168.1.10",
+    os: "Windows 10",
+    process: "explorer.exe",
+    pid: 2340,
+    elevated: true,
+    sleep: 60,
+    tasks: @[],
+    firstCheckin: now() - initDuration(hours = 2),
+    latestCheckin: now(),
+    sessionKey: [byte 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31]
+  ),
+  Agent(
+    agentId: "FACEDEAD",
+    listenerId: "L7654321",
+    username: "bob",
+    hostname: "LAPTOP-02",
+    domain: "SALES",
+    ip: "10.0.0.5",
+    os: "Windows 11",
+    process: "cmd.exe",
+    pid: 4567,
+    elevated: false,
+    sleep: 120,
+    tasks: @[],
+    firstCheckin: now() - initDuration(hours = 1, minutes = 30),
+    latestCheckin: now() - initDuration(minutes = 5),
+    sessionKey: [byte 31, 30, 29, 28, 27, 26, 25, 24, 23, 22, 21, 20, 19, 18, 17, 16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0]
+  )
+]
+
+proc SessionsTable*(title: string): SessionsTableComponent = 
+    result = new SessionsTableComponent
+    result.title = title
+    result.agents = exampleAgents
+
+proc draw*(component: SessionsTableComponent, showComponent: ptr bool) = 
+    igSetNextWindowSize(vec2(800, 600), ImGuiCond_Once.int32)
+
+    igBegin(component.title, showComponent, 0)
+    defer: igEnd() 
+
+    let tableFlags = (
+        ImGuiTableFlags_Resizable.int32 or 
+        ImGuiTableFlags_Reorderable.int32 or 
+        ImGuiTableFlags_Hideable.int32 or 
+        ImGuiTableFlags_HighlightHoveredColumn.int32 or 
+        ImGuiTableFlags_ContextMenuInBody.int32 or 
+        ImGuiTableFlags_RowBg.int32 or 
+        ImGuiTableFlags_BordersV.int32 or 
+        ImGuiTableFlags_BordersH.int32 or 
+        ImGuiTableFlags_ScrollY.int32 or
+        ImGuiTableFlags_ScrollX.int32 or 
+        ImGuiTableFlags_NoBordersInBodyUntilResize.int32 or 
+        ImGui_TableFlags_SizingStretchProp.int32
+    )
+
+    let cols: int32 = 8
+    if igBeginTable("Sessions", cols, tableFlags, vec2(0.0f, 0.0f), 0.0f):
+
+        igTableSetupColumn("AgentID", ImGuiTableColumnFlags_NoReorder.int32 or ImGuiTableColumnFlags_NoHide.int32, 0.0f, 0)
+        igTableSetupColumn("Address", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+        igTableSetupColumn("Username", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+        igTableSetupColumn("Hostname", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+        igTableSetupColumn("OS", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+        igTableSetupColumn("Process", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+        igTableSetupColumn("PID", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+        igTableSetupColumn("Activity", ImGuiTableColumnFlags_None.int32, 0.0f, 0)
+
+        igTableSetupScrollFreeze(0, 1)
+        igTableHeadersRow()
+
+        for row in 0..< component.agents.len(): 
+            igTableNextRow(ImGuiTableRowFlags_None.int32, 0.0f)
+            let agent = component.agents[row]
+            if igTableSetColumnIndex(0): 
+                igText(agent.agentId)
+            if igTableSetColumnIndex(1): 
+                igText(agent.ip)
+            if igTableSetColumnIndex(2): 
+                igText(agent.username)
+            if igTableSetColumnIndex(3): 
+                igText(agent.hostname)
+            if igTableSetColumnIndex(4): 
+                igText(agent.os)
+            if igTableSetColumnIndex(5): 
+                igText(agent.process)
+            if igTableSetColumnIndex(6): 
+                igText($agent.pid)
+            if igTableSetColumnIndex(7): 
+                igText(agent.latestCheckin.format("yyyy-MM-dd HH:mm:ss"))
+
+        igEndTable()
+
+