Pass down Message stanza in ChatStateListener

Fixes SMACK-704.
igniterealtime · Jan 23, 2016 · 0761fe0 · 0761fe0
1 parent 85e818c
commit 0761fe0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/smack-extensions/src/main/java/org/jivesoftware/smackx/chatstates/ChatStateListener.java b/smack-extensions/src/main/java/org/jivesoftware/smackx/chatstates/ChatStateListener.java
@@ -19,6 +19,7 @@
 
 import org.jivesoftware.smack.chat.Chat;
 import org.jivesoftware.smack.chat.ChatMessageListener;
+import org.jivesoftware.smack.packet.Message;
 
 /**
  * Events for when the state of a user in a chat changes.
@@ -32,6 +33,7 @@ public interface ChatStateListener extends ChatMessageListener {
      *
      * @param chat the chat in which the state has changed.
      * @param state the new state of the participant.
+     * @message the message carrying the chat state.
      */
-    void stateChanged(Chat chat, ChatState state);
+    void stateChanged(Chat chat, ChatState state, Message message);
 }
diff --git a/smack-extensions/src/main/java/org/jivesoftware/smackx/chatstates/ChatStateManager.java b/smack-extensions/src/main/java/org/jivesoftware/smackx/chatstates/ChatStateManager.java
@@ -142,10 +142,10 @@ private synchronized boolean updateChatState(Chat chat, ChatState newState) {
         return false;
     }
 
-    private static void fireNewChatState(Chat chat, ChatState state) {
+    private static void fireNewChatState(Chat chat, ChatState state, Message message) {
         for (ChatMessageListener listener : chat.getListeners()) {
             if (listener instanceof ChatStateListener) {
-                ((ChatStateListener) listener).stateChanged(chat, state);
+                ((ChatStateListener) listener).stateChanged(chat, state, message);
             }
         }
     }
@@ -184,7 +184,7 @@ public void processMessage(Chat chat, Message message) {
                 return;
             }
 
-            fireNewChatState(chat, state);
+            fireNewChatState(chat, state, message);
         }
     }
 }