standup with a timer

4 files changed, 115 insertions, 10 deletions

diff --git a/bin/dbolla b/bin/dbolla
index 7a1eba6..2b4fe02 100755
--- a/bin/dbolla
+++ b/bin/dbolla
@@ -1,6 +1,7 @@
 #!/usr/bin/env python3
 # -*- mode: python -*-
 import asyncio
+import datetime
 import functools
 import logging.config
 
@@ -49,7 +50,8 @@ class Bot(object):
 
         self.loaded_plugins = []
 
-        self.load_plugin('asdf')
+        self.load_plugin('warmachine.addons.giphy.GiphySearch')
+        self.load_plugin('warmachine.addons.standup.StandUpPlugin')
 
     def start(self):
         for connection in self.connections:
@@ -75,18 +77,22 @@ class Bot(object):
 
             for p in self.loaded_plugins:
                 self.log.debug('Calling {}'.format(p.__class__.__name__))
-                await p.recv_msg(connection, message)
+                try:
+                    await p.recv_msg(connection, message)
+                except Exception as e:
+                    self.log.exception(e)
+                    continue
 
             self.log.debug('MSG {}: {}'.format(
                 connection.__class__.__name__, message))
 
-    def load_plugin(self, path):
+    def load_plugin(self, class_path):
         """
         Loads plugins
         """
         from importlib import import_module
 
-        mod_path, cls_name = 'warmachine.addons.giphy.GiphySearch'.rsplit('.', 1)
+        mod_path, cls_name = class_path.rsplit('.', 1)
 
         mod = import_module(mod_path)
 
diff --git a/warmachine/addons/standup.py b/warmachine/addons/standup.py
index a21efdd..7d1ec99 100644
--- a/warmachine/addons/standup.py
+++ b/warmachine/addons/standup.py
@@ -1,3 +1,7 @@
+import asyncio
+from datetime import datetime, timedelta
+import functools
+
 from .base import WarMachinePlugin
 
 
@@ -9,6 +13,11 @@ class StandUpPlugin(WarMachinePlugin):
         !standup-add <24 hr time to kick off> <SunMTWThFSat> [channel]
         !standup-remove [channel]
     """
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        self.standup_schedules = {}
+
     async def recv_msg(self, connection, message):
         if not message['message'].startswith('!standup'):
             return
@@ -18,11 +27,52 @@ class StandUpPlugin(WarMachinePlugin):
         cmd = message['message'].split(' ')[0]
         parts = message['message'].split(' ')[1:]
 
+        self._loop = asyncio.get_event_loop()
+
         if cmd == '!standup-add':
-            await connection.say('Scheduling standup for {} on {}'.format(
-                parts[1], parts[2]))
+            pretty_next_standup, next_standup_secs = \
+                self.get_next_standup_secs(parts[0])
+
+            f = self._loop.call_later(
+                next_standup_secs, functools.partial(
+                    self.standup_schedule_func, connection, message['channel']))
+
+            self.standup_schedules[message['channel']] = {
+                'future': f,
+            }
+            await connection.say('Next standup in {}'.format(
+                pretty_next_standup), message['channel'])
+            await connection.say(str(self.standup_schedules),
+                                 message['channel'])
+
+    def standup_schedule_func(self, connection, channel):
+            asyncio.ensure_future(self.start_standup(connection, channel))
+
+    async def start_standup(self, connection, channel):
+        await connection.say('@channel Time for standup', channel)
+        connection.get_users_by_channel(channel)
+
+    @classmethod
+    def get_next_standup_secs(cls, time24h):
+        """
+        calculate the number of seconds until the next standup time
+        """
+        now = datetime.now()
+
+        # if it's friday, wait 72 hours
+        if now.isoweekday() == 5:
+            hours = 72
+        # if it's saturday, wait 48
+        elif now.isoweekday() == 6:
+            hours = 48
+        # if it's sunday-thur wait 24
+        else:
+            hours = 24
 
-        # await connection.say('{}, {}'.format(cmd, parts), message['channel'])
+        standup_hour, standup_minute = (int(s) for s in time24h.split(':'))
 
-    async def start_standup(self, connection):
-        pass
+        future = now + timedelta(hours=hours)
+        next_standup = datetime(future.year, future.month, future.day,
+                                standup_hour, standup_minute)
+        standup_in = next_standup-now
+        return standup_in, standup_in.seconds
diff --git a/warmachine/connections/base.py b/warmachine/connections/base.py
index e68e9b9..4df5996 100644
--- a/warmachine/connections/base.py
+++ b/warmachine/connections/base.py
@@ -26,6 +26,13 @@ class Connection(object):
         raise NotImplementedError('{} must implement `read` method'.format(
             self.__class__.__name__))
 
+    def get_users_by_channel(self, channel):
+        """
+        Return a list of users who are in the provided channel
+        """
+        raise NotImplementedError('{} must implement `get_users_by_channel` '
+                                  'method'.format(self.__class__.__name__))
+
     def id(self):
         """
         Unique ID for this connection. Since there can be more than one
diff --git a/warmachine/connections/slack.py b/warmachine/connections/slack.py
index b1a5390..90903e7 100644
--- a/warmachine/connections/slack.py
+++ b/warmachine/connections/slack.py
@@ -3,6 +3,7 @@ import json
 import logging
 from pprint import pformat
 from urllib.parse import urlencode
+import urllib.request
 
 import websockets
 
@@ -95,7 +96,6 @@ class SlackWS(Connection):
         Returns:
             str: websocket url to connect to
         """
-        import urllib.request
         url = 'https://slack.com/api/rtm.start?{}'.format(
             urlencode(
                 {'token':
@@ -121,6 +121,8 @@ class SlackWS(Connection):
         """
         if not self._info:
             return
+        with open('slack_info.json', 'w') as f:
+            f.write(pformat(self._info))
 
         self.status = CONNECTED
 
@@ -190,6 +192,18 @@ class SlackWS(Connection):
         ))
         self.user_map[msg['user']]['presence'] = msg['presence']
 
+    def get_users_by_channel(self, channel):
+        url = 'https://slack.com/api/groups.info?{}'.format(urlencode(
+            {
+                'token': self.token,
+                'channel': channel,
+            }))
+        self.log.debug(url)
+        req = urllib.request.Request(url)
+        r = urllib.request.urlopen(req).read().decode('utf-8')
+
+        self.log.debug(r)
+
     async def on_group_join(self, channel):
         """
         The group_joined event is sent to all connections for a user when that
@@ -279,3 +293,31 @@ class SlackWS(Connection):
         #         'user': 'U1U05AF5J'
         #     }
         # }
+
+
+# Invited to a public channel
+#     2016-07-29 16:23:24,817 [DEBUG] SlackWS: on_channel_joined does not exist for message: {'type': 'channel_joined', 'chan
+# nel': {'members': ['U0286NL58', 'U1U05AF5J'], 'purpose': {'last_set': 0, 'creator': '', 'value': ''}, 'topic': {'last_s
+# et': 0, 'creator': '', 'value': ''}, 'is_member': True, 'is_channel': True, 'creator': 'U0286NL58', 'is_archived': Fals
+# e, 'unread_count_display': 0, 'id': 'C1WJU3ZU0', 'name': 'wm-test2', 'is_general': False, 'created': 1469830985, 'unrea
+# d_count': 0, 'latest': {'text': '<@U0286NL58|jason> has joined the channel', 'type': 'message', 'user': 'U0286NL58', 's
+# ubtype': 'channel_join', 'ts': '1469830985.000002'}, 'last_read': '1469830985.000002'}}
+# 2016-07-29 16:23:24,878 [DEBUG] SlackWS: on_message_channel_join does not exist for message: {'channel': 'C1WJU3ZU0', '
+# text': '<@U1U05AF5J|wm-standup-test> has joined the channel', 'type': 'message', 'inviter': 'U0286NL58', 'subtype': 'ch
+# annel_join', 'user_profile': {'real_name': '', 'name': 'wm-standup-test', 'image_72': 'https://avatars.slack-edge.com/2
+# 016-07-21/62015427159_1da65a3cf7a85e85c3cb_72.png', 'first_name': None, 'avatar_hash': '1da65a3cf7a8'}, 'ts': '14698310
+# 04.000003', 'user': 'U1U05AF5J', 'team': 'T027XPE12'}
+
+# Someone else joins a public channel
+# 2016-07-29 16:26:19,966 [DEBUG] SlackWS: on_message_channel_join does not exist for message: {'type': 'message', 'invit
+# er': 'U0286NL58', 'ts': '1469831179.000004', 'team': 'T027XPE12', 'user': 'U0286167T', 'channel': 'C1WJU3ZU0', 'user_pr
+# ofile': {'name': 'synic', 'image_72': 'https://avatars.slack-edge.com/2016-06-24/54136624065_49ec8bc368966c152817_72.jp
+# g', 'real_name': 'Adam Olsen', 'first_name': 'Adam', 'avatar_hash': '49ec8bc36896'}, 'subtype': 'channel_join', 'text':
+#  '<@U0286167T|synic> has joined the channel'}
+
+# Invited to a private channel
+# 2016-07-29 16:27:29,376 [DEBUG] SlackWS: on_message_group_join does not exist for message: {'type': 'message', 'inviter
+# ': 'U0286NL58', 'ts': '1469831249.000047', 'team': 'T027XPE12', 'user': 'U0286167T', 'channel': 'G1W837CGP', 'user_prof
+# ile': {'name': 'synic', 'image_72': 'https://avatars.slack-edge.com/2016-06-24/54136624065_49ec8bc368966c152817_72.jpg'
+# , 'real_name': 'Adam Olsen', 'first_name': 'Adam', 'avatar_hash': '49ec8bc36896'}, 'subtype': 'group_join', 'text': '<@
+# U0286167T|synic> has joined the group'}