Keep track of a user's away message, not just whether they're away or not

1 files changed, 2 insertions, 0 deletions

diff --git a/modules/line_handler/user.py b/modules/line_handler/user.py
index f9f14af6..cd941d8e 100644
--- a/modules/line_handler/user.py
+++ b/modules/line_handler/user.py
@@ -51,10 +51,12 @@ def away(events, event):
     message = event["args"].get(0)
     if message:
         user.away = True
+        user.away_message = message
         events.on("received.away.on").call(user=user, server=event["server"],
             message=message)
     else:
         user.away = False
+        user.away_message = None
         events.on("received.away.off").call(user=user, server=event["server"])
 
 def chghost(event):