first commit.

1 files changed, 79 insertions, 0 deletions

diff --git a/modules/in.py b/modules/in.py
new file mode 100644
index 00000000..7c543124
--- /dev/null
+++ b/modules/in.py
@@ -0,0 +1,79 @@
+import time
+import Utils
+
+SECONDS_MAX = Utils.SECONDS_WEEKS*8
+SECONDS_MAX_DESCRIPTION = "8 weeks"
+
+class Module(object):
+    def __init__(self, bot):
+        self.bot = bot
+        bot.events.on("received").on("command").on("in").hook(
+            self.in_command, min_args=2)
+        bot.events.on("received").on("numeric").on("001").hook(
+            self.on_connect)
+
+    def on_connect(self, event):
+        self.load_reminders(event["server"])
+
+    def remove_timer(self, target, due_time, server_id, nickname, message):
+        setting = "in-%s" % nickname
+        reminders = self.bot.database.get_server_setting(server_id, setting, [])
+        try:
+            reminders.remove([target, due_time, server_id, nickname, message])
+        except:
+            print("failed to remove a reminder. huh.")
+        if reminders:
+            self.bot.database.set_server_setting(server_id,
+                setting, reminders)
+        else:
+            self.bot.database.del_server_setting(server_id,
+                setting)
+
+    def load_reminders(self, server):
+        reminders = server.find_settings("in-%")
+        for user_reminders in reminders:
+            for target, due_time, server_id, nickname, message in user_reminders[1]:
+                time_left = due_time-time.time()
+                if time_left > 0:
+                    self.bot.add_timer(self.timer_due, time_left, target=target,
+                        due_time=due_time, server_id=server_id, nickname=nickname,
+                        message=message)
+                else:
+                    self.remove_timer(target, due_time, server_id, nickname,
+                        message)
+
+    def in_command(self, event):
+        seconds = Utils.from_pretty_time(event["args_split"][0])
+        message = " ".join(event["args_split"][1:])
+        if seconds:
+            if seconds <= SECONDS_MAX:
+                due_time = int(time.time())+seconds
+
+                setting = "in-%s" % event["user"].nickname
+                reminders = event["server"].get_setting(setting, [])
+                reminders.append([event["target"].name, due_time,
+                    event["server"].id, event["user"].nickname, message])
+                event["server"].set_setting(setting, reminders)
+
+                self.bot.add_timer(self.timer_due, seconds,
+                    target=event["target"].name, due_time=due_time,
+                    server_id=event["server"].id, nickname=event["user"].nickname,
+                    message=message)
+            else:
+                event["stderr"].write(
+                    "The given time is above the max (%s)" % (
+                    SECONDS_MAX_DESCRIPTION))
+        else:
+            event["stderr"].write(
+                "Please provided a valid time above 0 seconds")
+
+    def timer_due(self, timer, **kwargs):
+        for server in self.bot.servers.values():
+            if kwargs["server_id"] == server.id:
+                server.send_message(kwargs["target"],
+                    "%s, this is your reminder: %s" % (
+                    kwargs["nickname"], kwargs["message"]))
+                break
+        setting = "in-%s" % kwargs["nickname"]
+        self.remove_timer(kwargs["target"], kwargs["due_time"],
+            kwargs["server_id"], kwargs["nickname"], kwargs["message"])