From 66c099f5ff4d41715c5210d5a7e715636f880d09 Mon Sep 17 00:00:00 2001
From: Alex-Emmet <alexemmet07@gmail.com>
Date: Sun, 8 Feb 2026 20:13:50 +0100
Subject: [PATCH] [Add] Notification handling

---
 src/data/communication.rs     | 6 ++++++
 src/omega/omega_connection.rs | 2 +-
 src/rho/client_connection.rs  | 2 ++
 src/rho/iota_connection.rs    | 2 +-
 4 files changed, 10 insertions(+), 2 deletions(-)
 mode change 100644 => 100755 src/data/communication.rs

diff --git a/src/data/communication.rs b/src/data/communication.rs
old mode 100644
new mode 100755
index f6b2fe3..99229f5
--- a/src/data/communication.rs
+++ b/src/data/communication.rs
@@ -35,6 +35,7 @@ pub enum DataTypes {
     denied_profiles,
     content,
     messages,
+    notifications,
     send_time,
     get_time,
     get_variant,
@@ -142,6 +143,11 @@ pub enum CommunicationType {
     message_other_iota,
     message_chunk,
     messages_get,
+
+    push_notification,
+    read_notification,
+    get_notifications,
+
     change_confirm,
     confirm_receive,
     confirm_read,
diff --git a/src/omega/omega_connection.rs b/src/omega/omega_connection.rs
index 30b6f03..e42d5ed 100755
--- a/src/omega/omega_connection.rs
+++ b/src/omega/omega_connection.rs
@@ -126,7 +126,7 @@ impl OmegaConnection {
         }
 
         loop {
-            if retry > 5 {
+            if retry > 500 {
                 log_err!(
                     0,
                     PrintType::Omega,
diff --git a/src/rho/client_connection.rs b/src/rho/client_connection.rs
index 79417e9..9e54f1f 100644
--- a/src/rho/client_connection.rs
+++ b/src/rho/client_connection.rs
@@ -361,6 +361,8 @@ impl ClientConnection {
             }
 
             if cv.is_type(CommunicationType::change_user_data)
+                || cv.is_type(CommunicationType::read_notification)
+                || cv.is_type(CommunicationType::get_notifications)
                 || cv.is_type(CommunicationType::get_user_data)
                 || cv.is_type(CommunicationType::get_iota_data)
                 || cv.is_type(CommunicationType::delete_user)
diff --git a/src/rho/iota_connection.rs b/src/rho/iota_connection.rs
index 3f753a9..72ed224 100755
--- a/src/rho/iota_connection.rs
+++ b/src/rho/iota_connection.rs
@@ -32,7 +32,6 @@ use x448::PublicKey;
 
 use super::{rho_connection::RhoConnection, rho_manager};
 use crate::{
-    // calls::call_manager::CallManager,
     data::communication::{CommunicationType, CommunicationValue, DataTypes},
     omega::omega_connection::OmegaConnection,
 };
@@ -434,6 +433,7 @@ impl IotaConnection {
         }
 
         if cv.is_type(CommunicationType::change_iota_data)
+            || cv.is_type(CommunicationType::push_notification)
             || cv.is_type(CommunicationType::get_user_data)
             || cv.is_type(CommunicationType::get_iota_data)
             || cv.is_type(CommunicationType::get_register)