Servus-chiark.py

   1 # This file is part of Acrobat.
   2 #
   3 # Acrobat is free software; you can redistribute it and/or modify
   4 # it under the terms of the GNU General Public License as published
   5 # by the Free Software Foundation; either version 2 of the License,
   6 # or (at your option) any later version.
   7 #
   8 # Acrobat is distributed in the hope that it will be useful,
   9 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 # GNU General Public License for more details.
  12 #
  13 # You should have received a copy of the GNU General Public License
  14 # along with Acrobat; if not, write to the Free Software
  15 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
  16 # USA.
  17
  18 # Andrew Walkingshaw <andrew@lexical.org.uk>
  19 # Peter Corbett <ptc24@cam.ac.uk>
  20 # Matthew Vernon <matthew@debian.org>
  21 # Stephen Early <steve@greenend.org.uk>
  22 # Richard Kettlewell <rjk@greenend.org.uk
  23
  24 # Acrobat configuration file
  25
  26 # The following definitions are required to be present in this module:
  27 server = "rapun"
  28 port = 6667
  29 nickname = "Servus"
  30 channel = "#chiark"
  31 owner = "Emperor"
  32 # Also a function called "command"; see later.
  33
  34 # Everything else in this file is configuration-specific.
  35
  36 import os, time, re
  37
  38 # Most command implementations are stored in a separate module.
  39 import commands as c
  40
  41 # This fishpond is shared between trouts and flirts.  It doesn't have to be;
  42 # you can define as many ponds as you like.
  43 class fish:
  44         cur_fish=5
  45         max_fish=5
  46         nofish_time=60
  47         fish_time_inc=60
  48         fish_inc=2
  49         DoS=0
  50         Boring_Git='Nobody'
  51         quotatime=0
  52
  53 # load a file full of flirts or trouts
  54 def __load(filename):
  55     try:
  56         f = open(filename, "r")
  57         r = [l.strip() for l in f.readlines() if l.find("%s") != -1]
  58         f.close()
  59     except IOError:
  60         r = [ "doesn't know what to do about %s." ]
  61     return r
  62
  63 # (troutlist,selftroutmsg,DoSmsg,notargetmsg,nofishmsg,fishpond,selftroutprob)
  64 troutcfg = (
  65         __load("trouts"),
  66         ' (at the instigation of %s)',
  67         "Sorry, but %s is being a spoilsport.",
  68         "Who do you wish me to trout?",
  69         "Fish stocks exhausted.",
  70         fish,
  71         0.1)
  72
  73 flirtcfg = (
  74         __load("flirts"),
  75         ' (but %s is their secret admirer)',
  76         "Sorry, but %s made me take Holy Orders.",
  77         "Who do you wish me to flirt with?",
  78         "My libido is over-used!",
  79         fish,
  80         0.1)
  81
  82 slashcfg= (
  83         __load("slashes"),
  84         ' (while %s watches)',
  85         "Sorry, but %s stole my pen.",
  86         "Who do you want to slash?",
  87         "I have writer's block!",
  88         fish,
  89         0.1)
  90
  91 # Hacky command to output the current fishpond state
  92 def fishq(bot, cmd, nick, conn, public,f):
  93         from irclib import irc_lower
  94         if not public and irc_lower(nick) == irc_lower(bot.owner):
  95                 state=("Fishpond state: cur_fish=%d, max_fish=%d, nofish_time=%d, "
  96                        +"fish_time_inc=%d, fish_inc=%d, DoS=%d, Boring_Git=%s, "
  97                        +"quotatime=%d")%(f.cur_fish,f.max_fish,f.nofish_time,
  98                                          f.fish_time_inc,f.fish_inc,f.DoS,f.Boring_Git,
  99                                          f.quotatime)
 100                 bot.automsg(public,nick,state)
 101
 102 # Karma implementation
 103 import cPickle
 104 karmafilename = "chiark-karma"
 105 # load the karma db
 106 try:
 107     f = open(karmafilename, "r")
 108     karmadb = cPickle.load(f)
 109     f.close()
 110 except IOError:
 111     karmadb = {}
 112 # Modify karma
 113 def karma(cmd, amount):
 114     thing=cmd.split()[0][:-2].lower()
 115     if karmadb.has_key(thing):
 116         karmadb[thing] += amount
 117     else:
 118         karmadb[thing] = amount
 119     savekarma()
 120 def savekarma():
 121     tmp = "%s.tmp" % karmafilename
 122     try:
 123         f = open(tmp, "w")
 124         cPickle.dump(karmadb, f)
 125         f.close()
 126         os.rename(tmp, karmafilename)
 127     except IOError, e:
 128         sys.stderr.write("error writing karma: %s" % e)
 129
 130 def quit(bot,cmd,nick,conn,public):
 131     c.quitq(bot,cmd,nick,conn,public)
 132 def reload(bot,cmd,nick,conn,public):
 133     c.reloadq(bot,cmd,nick,conn,public)
 134
 135 #The game...
 136 class game:
 137         trigger="Servus"
 138         grace=time.time()
 139         minlose=24*60*60 #1 day
 140         maxlose=14*minlose #2 weeks
 141         losetime=time.time()+300000
 142
 143 def canonical_url(urlstring):
 144 # canonicalise BBC URLs
 145   if (urlstring.find("news.bbc.co.uk") != -1):
 146     for middle in ("/low/","/mobile/"):
 147       x = urlstring.find(middle)
 148       if (x != -1):
 149         urlstring.replace(middle,"/hi/")
 150   return urlstring
 151
 152 def nicetime(tempus):
 153   if (tempus<120):
 154     tm="%d seconds ago"%int(tempus)
 155   elif (tempus<7200):
 156     tm="%d minutes ago"%int(tempus/60)
 157   if (tempus>7200):
 158     tm="%d hours ago"%int(tempus/3600)
 159   return tm
 160
 161 urldb = {}
 162 urlre = re.compile("(https?://[^ ]+)( |$)")
 163 urlcomplaints = [" contemporary","n interesting"," fascinating","n overused"," vastly overused"]
 164
 165 def urlq(bot, cmd, nick, conn, public):
 166   if (not urlre.search(cmd)):
 167     bot.automsg(False,nick,"Please use 'url' only with http URLs")
 168     return
 169
 170   url="".join(cmd.split(" ")[1:])
 171
 172   url=canonical_url(url)
 173   if (url in urldb):
 174     users = urldb[url]
 175     complaint="The url %s was mentioned %s by %s"%(url,nicetime(time.time()-users[-1][1]),users[-1][0])
 176     if (public):
 177       complaint=complaint+". Furthermore it defeats the point of this command to use it other than via /msg."
 178     bot.automsg(False,nick,complaint)
 179   else:
 180     if (public):
 181       bot.automsg(False,nick,"That URL was unique. There is little point in using !url out loud; please use it via /msg")
 182     else:
 183       conn.privmsg(bot.channel,"%s would like to draw your attention to %s"%(nick,url))
 184     urldb[url]=[[nick,time.time()]]
 185
 186 def dourl(bot,conn,nick,command):
 187   urlstring=urlre.search(command).group(1)
 188   urlstring=canonical_url(urlstring)
 189
 190   if urlstring in urldb:
 191     T=urldb[urlstring]
 192     uci = len(T)
 193     if uci >= len(urlcomplaints):
 194       uci = len(urlcomplaints)
 195     message="observes a"+urlcomplaints[uci-1]+" URL: mentioned by "
 196     if (len(T)>5):
 197       cutoff=len(T)-5
 198     else:
 199       cutoff=-1
 200     for t in range(len(T)-1,cutoff,-1):
 201       tempus = time.time()-T[t][1]
 202       message += T[t][0]+" ("+nicetime(tempus)+")"
 203       if (t!=cutoff+1):
 204         message += ", "
 205     if (cutoff != -1):
 206       message += ", amongst others"
 207     conn.action(bot.channel, message)
 208     urldb[urlstring]=[[nick,time.time()]]+urldb[urlstring]
 209   else:
 210     urldb[urlstring]=[[nick,time.time()]]
 211
 212 # Command processing: whenever something is said that the bot can hear,
 213 # "command" is invoked and must decide what to do.  This configuration
 214 # defines a couple of special cases (for karma) but is otherwise driven
 215 # by a dictionary of commands.
 216
 217 commands = {"karma": (c.karmaq,karmadb),
 218             "karmalist": (c.listkeysq,karmadb),
 219             "karmadel": (c.karmadelq,karmadb),
 220             "info": (c.infoq,karmadb),
 221             "trout": (c.troutq,troutcfg),
 222             "slash": (c.slashq, slashcfg),
 223             "rot13": c.rot13q,
 224             "fish": (fishq,fish),
 225             "flirt": (c.troutq,flirtcfg),
 226             "quiet": (c.nofishq,fish),
 227             "reload": reload,
 228             "quit": quit,
 229             "die": quit,
 230             "define": c.defineq,
 231             "google": c.googleq,
 232             "url": urlq,
 233             "units": c.unitq,
 234             "help": c.helpq,
 235 #           "game": (c.gameq,game),
 236             "say": c.sayq,
 237             "do": c.doq }
 238 # disconnect and hop annoy people
 239 #            "disconnect": c.disconnq,
 240 #            "hop": c.disconnq }
 241 commands["list"]=(c.listkeysq,commands,True)
 242
 243 triggers = ("!", "~") # what character should the bot be invoked by:
 244                       # eg !trout, ~trout etc.
 245
 246 def command(bot, cmd, nick, conn, public):
 247     ours=0
 248     try:
 249             if public and cmd[0] in triggers:
 250                     ours=1
 251                     cmd=cmd[1:]
 252             if not public:
 253                     ours=1
 254             command = cmd.split()[0]
 255     except IndexError:
 256             command=""
 257
 258     if public:
 259       if urlre.search(cmd):
 260         dourl(bot,conn,nick,cmd)
 261
 262     # karma: up
 263     if command.endswith("++"):
 264         karma(cmd,1)
 265     # karma: down
 266     if command.endswith("--"):
 267         karma(cmd,-1)
 268
 269     if ours and command.lower() in commands.keys():
 270         e=commands[command.lower()]
 271         if callable(e):
 272             e(bot,cmd,nick,conn,public)
 273         else:
 274             e[0](bot,cmd,nick,conn,public,*e[1:])
 275 #    elif public:
 276 #        if cmd.find("GAME")!=-1: #someone else lost
 277 #           grace.grace=time.time()+60*20
 278 #       elif cmd.find(game.trigger)!=-1 and len(game.trigger)>2: #we lost!
 279 #           c.gameq(bot,"pad "+game.trigger,bot.owner,conn,False,game)
 280 #       elif time.time()>game.losetime: #we randomly lost, take new trigger
 281 #           c.gameq(bot,cmd,bot.owner,conn,False,game)
 282 #