Merge pull request #346 from V13Axel/status_as_dm_topic

Add emojis to status as a DM topic

1 files changed, 35 insertions, 6 deletions

diff --git a/wee_slack.py b/wee_slack.py
index 6b69096..97a602f 100644
--- a/wee_slack.py
+++ b/wee_slack.py
@@ -1060,6 +1060,12 @@ class SlackTeam(object):
     def buffer_prnt(self, data):
         w.prnt_date_tags(self.channel_buffer, SlackTS().major, tag("team"), data)
 
+    def find_channel_by_members(self, members, channel_type=None):
+        for channel in self.channels.itervalues():
+            if channel.get_members() == members and (
+                    channel_type is None or channel.type == channel_type):
+                return channel
+
     def get_channel_map(self):
         return {v.slack_name: k for k, v in self.channels.iteritems()}
 
@@ -1633,6 +1639,7 @@ class SlackDMChannel(SlackChannel):
         self.type = 'im'
         self.update_color()
         self.set_name(self.slack_name)
+        self.topic = create_user_status_string(users[dmuser].profile)
 
     def set_name(self, slack_name):
         self.name = slack_name
@@ -1942,6 +1949,10 @@ class SlackUser(object):
         self.color_name = get_nick_color_name(self.name)
         self.color = w.color(self.color_name)
 
+    def update_status(self, status_emoji, status_text):
+        self.profile["status_emoji"] = status_emoji
+        self.profile["status_text"] = status_text
+
     def formatted_name(self, prepend="", enable_color=True):
         if enable_color:
             return self.color + prepend + self.name
@@ -2295,6 +2306,19 @@ def process_pref_change(message_json, eventrouter, **kwargs):
         dbg("Preference change not implemented: {}\n".format(message_json['name']))
 
 
+def process_user_change(message_json, eventrouter, **kwargs):
+    """
+    Currently only used to update status, but lots here we could do.
+    """
+    user = message_json['user']
+    profile = user.get("profile")
+    team = kwargs["team"]
+    team.users[user["id"]].update_status(profile.get("status_emoji"), profile.get("status_text"))
+    dmchannel = team.find_channel_by_members({user["id"]}, channel_type='im')
+    if dmchannel:
+        dmchannel.set_topic(create_user_status_string(profile))
+
+
 def process_user_typing(message_json, eventrouter, **kwargs):
     channel = kwargs["channel"]
     team = kwargs["team"]
@@ -2821,6 +2845,16 @@ def resolve_ref(ref):
     return ref
 
 
+def create_user_status_string(profile):
+    real_name = profile.get("real_name")
+    status_emoji = profile.get("status_emoji")
+    status_text = profile.get("status_text")
+    if status_emoji or status_text:
+        return "{} | {} {}".format(real_name, status_emoji, status_text)
+    else:
+        return real_name
+
+
 def create_reaction_string(reactions):
     count = 0
     if not isinstance(reactions, list):
@@ -3193,12 +3227,7 @@ def command_talk(data, current_buffer, args):
             else:
                 channel_type = 'im'
 
-            # Try finding the channel by type and members
-            for channel in team.channels.itervalues():
-                if (channel.type == channel_type and
-                        channel.get_members() == users):
-                    chan = channel
-                    break
+            chan = team.find_channel_by_members(users, channel_type=channel_type)
 
             # If the DM or MPDM doesn't exist, create it
             if not chan: