2 files changed, 49 insertions, 7 deletions

diff --git a/src/ui/message.c b/src/ui/message.c
index f89f5ef..def4524 100644
--- a/src/ui/message.c
+++ b/src/ui/message.c
@@ -28,14 +28,28 @@
 
 UI_MESSAGE_Handle*
 ui_message_new(MESSENGER_Application *app,
-               bool sent)
+               UI_MESSAGE_Type type)
 {
   UI_MESSAGE_Handle* handle = g_malloc(sizeof(UI_MESSAGE_Handle));
 
-  if (sent)
-    handle->builder = gtk_builder_new_from_file("resources/ui/message-sent.ui");
-  else
-    handle->builder = gtk_builder_new_from_file("resources/ui/message.ui");
+  handle->type = type;
+
+  const char *ui_builder_file;
+
+  switch (handle->type)
+  {
+    case UI_MESSAGE_SENT:
+      ui_builder_file = "resources/ui/message-sent.ui";
+      break;
+    case UI_MESSAGE_STATUS:
+      ui_builder_file = "resources/ui/message-status.ui";
+      break;
+    default:
+      ui_builder_file = "resources/ui/message.ui";
+      break;
+  }
+
+  handle->builder = gtk_builder_new_from_file(ui_builder_file);
 
   handle->message_box = GTK_WIDGET(
       gtk_builder_get_object(handle->builder, "message_box")
@@ -49,7 +63,7 @@ ui_message_new(MESSENGER_Application *app,
       gtk_builder_get_object(handle->builder, "sender_label")
   );
 
-  if (sent)
+  if (UI_MESSAGE_SENT == handle->type)
   {
     const char *sender = GNUNET_CHAT_get_name(app->chat.messenger.handle);
 
@@ -61,6 +75,22 @@ ui_message_new(MESSENGER_Application *app,
       gtk_builder_get_object(handle->builder, "text_label")
   );
 
+  if (UI_MESSAGE_STATUS == handle->type)
+  {
+    handle->deny_button = GTK_BUTTON(
+        gtk_builder_get_object(handle->builder, "deny_button")
+    );
+
+    handle->accept_button = GTK_BUTTON(
+        gtk_builder_get_object(handle->builder, "accept_button")
+    );
+  }
+  else
+  {
+    handle->deny_button = NULL;
+    handle->accept_button = NULL;
+  }
+
   handle->timestamp_label = GTK_LABEL(
       gtk_builder_get_object(handle->builder, "timestamp_label")
   );
diff --git a/src/ui/message.h b/src/ui/message.h
index 1b3a80c..333ef75 100644
--- a/src/ui/message.h
+++ b/src/ui/message.h
@@ -32,8 +32,17 @@
 
 typedef struct MESSENGER_Application MESSENGER_Application;
 
+typedef enum UI_MESSAGE_Type
+{
+  UI_MESSAGE_DEFAULT    = 0,
+  UI_MESSAGE_SENT       = 1,
+  UI_MESSAGE_STATUS     = 2
+} UI_MESSAGE_Type;
+
 typedef struct UI_MESSAGE_Handle
 {
+  UI_MESSAGE_Type type;
+
   GtkBuilder *builder;
   GtkWidget *message_box;
 
@@ -42,13 +51,16 @@ typedef struct UI_MESSAGE_Handle
 
   GtkLabel *text_label;
 
+  GtkButton *deny_button;
+  GtkButton *accept_button;
+
   GtkLabel *timestamp_label;
   GtkImage *read_receipt_image;
 } UI_MESSAGE_Handle;
 
 UI_MESSAGE_Handle*
 ui_message_new(MESSENGER_Application *app,
-               bool sent);
+               UI_MESSAGE_Type type);
 
 void
 ui_message_delete(UI_MESSAGE_Handle *handle);