Refresh tab when information element is added or removed

Signed-off-by: Maxime “pep” Buquet <pep@bouah.net>

3 files changed, 6 insertions, 0 deletions

diff --git a/poezio/tabs/conversationtab.py b/poezio/tabs/conversationtab.py
index d1ad608c..6ce5c72c 100644
--- a/poezio/tabs/conversationtab.py
+++ b/poezio/tabs/conversationtab.py
@@ -89,6 +89,7 @@ class ConversationTab(OneToOneTab):
         raise NotImplementedError
 
     @staticmethod
+    @refresh_wrapper.always
     def add_information_element(plugin_name, callback):
         """
         Lets a plugin add its own information to the ConversationInfoWin
@@ -96,6 +97,7 @@ class ConversationTab(OneToOneTab):
         ConversationTab.additional_information[plugin_name] = callback
 
     @staticmethod
+    @refresh_wrapper.always
     def remove_information_element(plugin_name):
         del ConversationTab.additional_information[plugin_name]
 
diff --git a/poezio/tabs/muctab.py b/poezio/tabs/muctab.py
index 17abb369..1c752fa9 100644
--- a/poezio/tabs/muctab.py
+++ b/poezio/tabs/muctab.py
@@ -109,6 +109,7 @@ class MucTab(ChatTab):
         return None
 
     @staticmethod
+    @refresh_wrapper.always
     def add_information_element(plugin_name: str, callback: Callable[[str], str]) -> None:
         """
         Lets a plugin add its own information to the MucInfoWin
@@ -116,6 +117,7 @@ class MucTab(ChatTab):
         MucTab.additional_information[plugin_name] = callback
 
     @staticmethod
+    @refresh_wrapper.always
     def remove_information_element(plugin_name: str) -> None:
         """
         Lets a plugin add its own information to the MucInfoWin
diff --git a/poezio/tabs/privatetab.py b/poezio/tabs/privatetab.py
index cec68ac5..a1b9a87a 100644
--- a/poezio/tabs/privatetab.py
+++ b/poezio/tabs/privatetab.py
@@ -94,6 +94,7 @@ class PrivateTab(OneToOneTab):
         super().ack_message(msg_id, msg_jid)
 
     @staticmethod
+    @refresh_wrapper.always
     def add_information_element(plugin_name, callback):
         """
         Lets a plugin add its own information to the PrivateInfoWin
@@ -101,6 +102,7 @@ class PrivateTab(OneToOneTab):
         PrivateTab.additional_information[plugin_name] = callback
 
     @staticmethod
+    @refresh_wrapper.always
     def remove_information_element(plugin_name):
         del PrivateTab.additional_information[plugin_name]