From b18509e936c3afcada122b0da9e5a39f9b619aa1 Mon Sep 17 00:00:00 2001
From: Mike Perez <thingee@gmail.com>
Date: Mon, 12 Jun 2017 10:00:50 -0700
Subject: [PATCH] Implement #thanks command in statusbot

Add simple unprivileged "#thanks" command that adds thanks snippets to
a celebration wiki page.

NB: this adds a couple of configuration options, but fails gracefully if
they are not present.

Change-Id: I995c3189f0f109df35c56b63b3e19427f60f3750
---
 statusbot/bot.py | 47 +++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 45 insertions(+), 2 deletions(-)

diff --git a/statusbot/bot.py b/statusbot/bot.py
index 68bd36d..ad2f8dc 100644
--- a/statusbot/bot.py
+++ b/statusbot/bot.py
@@ -31,6 +31,7 @@ password=password
 url=https://wiki.example.com/w/api.php
 pageid=1781
 successpageid=2434
+thankspageid=37700
 
 [irclogs]
 url=http://eavesdrop.example.com/irclogs/%(chan)s/%(chan)s.%(date)s.log.html
@@ -141,6 +142,36 @@ class SuccessPage(WikiPage):
             self.save(newtext)
 
 
+class ThanksPage(WikiPage):
+    def __init__(self, config):
+        super(ThanksPage, self).__init__(config)
+        if config.has_option('wiki', 'thankspageid'):
+            self.thankspageid = config.get('wiki', 'thankspageid')
+        else:
+            self.thankspageid = None
+        if config.has_option('irclogs', 'url'):
+            self.irclogs_url = config.get('irclogs', 'url')
+        else:
+            self.irclogs_url = None
+
+    def log(self, channel, nick, msg):
+        if self.thankspageid:
+            self.login()
+            ts = self.timestamp()
+            if self.irclogs_url:
+                url = self.irclogs_url % {
+                    'chan': urllib.quote(channel),
+                    'date': ts[0:10]}
+                onchan = "[%s %s]" % (url, channel)
+            else:
+                onchan = channel
+            data = self.load()
+            current = data.split("\n")
+            newtext = "%s\n|-\n| %s || %s (on %s) || %s\n%s" % (
+                current[0], ts, nick, onchan, msg, '\n'.join(current[1:]))
+            self.save(newtext)
+
+
 class UpdateInterface(object):
     def alert(self, msg=None):
         pass
@@ -283,7 +314,7 @@ class AlertFile(UpdateInterface):
 class StatusBot(irc.bot.SingleServerIRCBot):
     log = logging.getLogger("statusbot.bot")
 
-    def __init__(self, channels, nicks, publishers, successlog,
+    def __init__(self, channels, nicks, publishers, successlog, thankslog,
                  nickname, password, server, port=6667):
         if port == 6697:
             factory = irc.connection.Factory(wrapper=ssl.wrap_socket)
@@ -304,6 +335,7 @@ class StatusBot(irc.bot.SingleServerIRCBot):
         self.current_topic = None
         self.publishers = publishers
         self.successlog = successlog
+        self.thankslog = thankslog
 
     def on_nicknameinuse(self, c, e):
         self.log.debug("Nickname in use, releasing")
@@ -344,6 +376,9 @@ class StatusBot(irc.bot.SingleServerIRCBot):
         if msg.startswith('#success'):
             self.handle_success_command(e.target, nick, msg)
             return
+        if msg.startswith('#thanks'):
+            self.handle_thanks_command(e.target, nick, msg)
+            return
         # Privileged commands
         if not msg.startswith('#status'):
             return
@@ -366,6 +401,13 @@ class StatusBot(irc.bot.SingleServerIRCBot):
         self.successlog.log(channel, nick, text)
         self.send(channel, "%s: Added success to Success page" % (nick,))
 
+    def handle_thanks(self, channel, nick, msg):
+        parts = msg.split()
+        text = ' '.join(parts[1:])
+        self.log.info("Processing thanks from %s: %s" % (nick, text))
+        self.thankslog.log(channel, nick, text)
+        self.send(channel, "%s: Added thanks to Thanks page" % (nick,))
+
     def handle_status_command(self, channel, nick, msg):
         parts = msg.split()
         command = parts[1].lower()
@@ -457,10 +499,11 @@ def _main(configpath):
     publishers = [StatusPage(config),
                   AlertFile(config)]
     successlog = SuccessPage(config)
+    thankslog = ThanksPage(config)
     if config.has_section('twitter'):
         publishers.append(Tweet(config))
 
-    bot = StatusBot(channels, nicks, publishers, successlog,
+    bot = StatusBot(channels, nicks, publishers, successlog, thankslog,
                     config.get('ircbot', 'nick'),
                     config.get('ircbot', 'pass'),
                     config.get('ircbot', 'server'),