Started work on websocket communication: Parsing/Serialization of WebSocket packets.

src/client/views/console.nim

@@ -1,3 +1,4 @@
+import whisky
 import strformat, strutils, times
 import imguin/[cimgui, glfw_opengl, simple]
 import ../utils/[appImGui, colors]
@@ -113,11 +114,11 @@ proc callback(data: ptr ImGuiInputTextCallbackData): cint {.cdecl.} =
 #[
     API to add new console item
 ]#
-proc addItem*(component: ConsoleComponent, itemType: LogType, data: string) = 
+proc addItem*(component: ConsoleComponent, itemType: LogType, data: string, timestamp: int64 = now().toTime().toUnix()) = 
 
     for line in data.split("\n"): 
         component.console.items.add(ConsoleItem(
-            timestamp: if itemType == LOG_OUTPUT: 0 else: now().toTime().toUnix(),
+            timestamp: if itemType == LOG_OUTPUT: 0 else: timestamp,
             itemType: itemType,
             text: line
         ))
@@ -148,7 +149,7 @@ proc print(item: ConsoleItem) =
     igSameLine(0.0f, 0.0f)
     igTextUnformatted(item.text.cstring, nil)
 
-proc draw*(component: ConsoleComponent) =
+proc draw*(component: ConsoleComponent, ws: WebSocket) =
     igBegin(fmt"[{component.agent.agentId}] {component.agent.username}@{component.agent.hostname}", addr component.showConsole, 0)
     defer: igEnd()
     
@@ -251,22 +252,25 @@ proc draw*(component: ConsoleComponent) =
     let inputFlags = ImGuiInputTextFlags_EnterReturnsTrue.int32 or ImGuiInputTextFlags_EscapeClearsAll.int32 or ImGuiInputTextFlags_CallbackHistory.int32 or ImGuiInputTextFlags_CallbackCompletion.int32
     if igInputText("##Input", addr component.inputBuffer[0], MAX_INPUT_LENGTH, inputFlags, callback, cast[pointer](component)):
 
-        let command = $(addr component.inputBuffer[0]).cstring
-        component.addItem(LOG_COMMAND, command)
+        let command = ($(addr component.inputBuffer[0])).strip()
+        if not command.isEmptyOrWhitespace(): 
+        
+            component.addItem(LOG_COMMAND, command)
 
-        # For testing
-        component.addItem(LOG_ERROR, "error message")
-        component.addItem(LOG_SUCCESS, "success message")
-        component.addItem(LOG_INFO, "info message")
-        component.addItem(LOG_WARNING, "warning message")
-        component.addItem(LOG_OUTPUT, "error message\nLong output\n\tindented output\nasdasd")
+            # For testing
+            # component.addItem(LOG_ERROR, "error message")
+            # component.addItem(LOG_SUCCESS, "success message")
+            # component.addItem(LOG_INFO, "info message")
+            # component.addItem(LOG_WARNING, "warning message")
+            # component.addItem(LOG_OUTPUT, "error message\nLong output\n\tindented output\nasdasd")
 
-        # TODO: Handle command execution
-        # console.handleCommand(command)
+            # TODO: Handle command execution
+            # console.handleCommand(command)
+            ws.send("CMD:" & component.agent.agentId & ":" & command)
 
-        # Add command to console history
-        component.history.add(command)
-        component.historyPosition = -1 
+            # Add command to console history
+            component.history.add(command)
+            component.historyPosition = -1 
 
         zeroMem(addr component.inputBuffer[0], MAX_INPUT_LENGTH)
         focusInput = true

src/client/views/eventlog.nim

@@ -4,7 +4,7 @@ import ../utils/[appImGui, colors]
 import ../../common/types
 
 type 
-    EventlogComponent = ref object of RootObj
+    EventlogComponent* = ref object of RootObj
         title: string 
         log*: ConsoleItems
         textSelect: ptr TextSelect
@@ -41,11 +41,11 @@ proc Eventlog*(title: string): EventlogComponent =
 #[
     API to add new log entry
 ]#
-proc addItem*(component: EventlogComponent, itemType: LogType, data: string) = 
+proc addItem*(component: EventlogComponent, itemType: LogType, data: string, timestamp: int64 = now().toTime().toUnix()) = 
 
     for line in data.split("\n"): 
         component.log.items.add(ConsoleItem(
-            timestamp: if itemType == LOG_OUTPUT: 0 else: now().toTime().toUnix(),
+            timestamp: if itemType == LOG_OUTPUT: 0 else: timestamp,
             itemType: itemType,
             text: line
         ))

src/client/views/listeners.nim

@@ -3,11 +3,12 @@ import imguin/[cimgui, glfw_opengl, simple]
 import ../utils/appImGui
 import ../../common/[types, utils]
 import ./modals/startListener
+import whisky
 
 type 
-    ListenersTableComponent = ref object of RootObj
+    ListenersTableComponent* = ref object of RootObj
         title: string 
-        listeners: seq[Listener]
+        listeners*: seq[Listener]
         selection: ptr ImGuiSelectionBasicStorage
         startListenerModal: ListenerModalComponent
 
@@ -33,7 +34,7 @@ proc ListenersTable*(title: string): ListenersTableComponent =
     result.selection = ImGuiSelectionBasicStorage_ImGuiSelectionBasicStorage()
     result.startListenerModal = ListenerModal()
 
-proc draw*(component: ListenersTableComponent, showComponent: ptr bool) = 
+proc draw*(component: ListenersTableComponent, showComponent: ptr bool, ws: WebSocket) = 
     igBegin(component.title, showComponent, 0)
     defer: igEnd() 
 
@@ -46,7 +47,7 @@ proc draw*(component: ListenersTableComponent, showComponent: ptr bool) =
     let listener = component.startListenerModal.draw()
     if listener != nil: 
         # TODO: Start listener
-        
+        ws.send("Starting listener: " & listener.listenerId)
         component.listeners.add(listener)    
 
     #[

src/client/views/sessions.nim

@@ -6,9 +6,9 @@ import ../utils/appImGui
 import ../../common/[types, utils]
 
 type 
-    SessionsTableComponent = ref object of RootObj
+    SessionsTableComponent* = ref object of RootObj
         title: string 
-        agents: seq[Agent]
+        agents*: seq[Agent]
         selection: ptr ImGuiSelectionBasicStorage
         consoles: ptr Table[string, ConsoleComponent]