From 82292590e6276c9435d370da2d9077bfcafdb8c4 Mon Sep 17 00:00:00 2001
From: John Rehbein <john.rehbein@icloud.com>
Date: Wed, 14 Aug 2024 22:35:38 -0700
Subject: [PATCH] Add 'Origin' column to table

---
 netarch.cpp | 39 +++++++++++++++++++++++++++++++++++++--
 1 file changed, 37 insertions(+), 2 deletions(-)

diff --git a/netarch.cpp b/netarch.cpp
index fb184b9..80d0bc3 100644
--- a/netarch.cpp
+++ b/netarch.cpp
@@ -1246,15 +1246,17 @@ void draw_imgui() {
         static bool isWindowOpen = false;
         static int selected_message_index = 0;
         if (ImGui::CollapsingHeader("Messages")) {
-            if (ImGui::BeginTable("MessagesTable", 1, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg)) {
+            if (ImGui::BeginTable("MessagesTable", 2, ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg)) {
                 ImGui::TableSetupColumn("Header", ImGuiTableColumnFlags_WidthFixed, 150.0f);
+                ImGui::TableSetupColumn("Origin", ImGuiTableColumnFlags_WidthFixed, 100.0f);
                 ImGui::TableHeadersRow();
 
                 for (int i = 0; i < g_libretro_context.message_history_length; ++i) {
                     sam2_message_u *message = &g_libretro_context.message_history[i];
                     ImGui::TableNextRow();
-                    ImGui::TableSetColumnIndex(0);
 
+                    // Header column
+                    ImGui::TableSetColumnIndex(0);
                     ImGui::Text("%.8s", (char *) message);
                     ImGui::SameLine();
                     char button_label[64] = {0};
@@ -1264,6 +1266,39 @@ void draw_imgui() {
                         selected_message_index = i;
                         isWindowOpen = true;
                     }
+
+                    // Origin column
+                    ImGui::TableSetColumnIndex(1);
+                    const char* origin = "Unknown";
+                    char header[9] = {0};
+                    strncpy(header, (char*)message, 8);
+
+                    if (header[7] == 'r') {
+                        ImGui::TextColored(GOLD, "Peer %05" PRIu16, g_ulnet_session.our_peer_id);
+                    } else {
+                        uint16_t peer_id = 0;
+                        if (strncmp(header, sam2_make_header, 4) == 0 ||
+                            strncmp(header, sam2_list_header, 4) == 0 ||
+                            strncmp(header, sam2_fail_header, 4) == 0 ||
+                            strncmp(header, sam2_conn_header, 4) == 0) {
+                            origin = "Server";
+                        } else if (strncmp(header, sam2_join_header, 4) == 0) {
+                            peer_id = ((sam2_room_join_message_t*)message)->peer_id;
+                        } else if (strncmp(header, sam2_sign_header, 4) == 0 ||
+                                   strncmp(header, sam2_sigx_header, 4) == 0) {
+                            peer_id = ((sam2_signal_message_t*)message)->peer_id;
+                        }
+
+                        if (peer_id == 0) {
+                            origin = "Server";
+                        } else {
+                            static char peer_id_str[16];
+                            snprintf(peer_id_str, sizeof(peer_id_str), "Peer %05" PRIu16, peer_id);
+                            origin = peer_id_str;
+                        }
+
+                        ImGui::Text("%s", origin);
+                    }
                 }
 
                 ImGui::EndTable();