plomlombot.py

   1 import socket
   2 import datetime
   3 import select
   4 import time
   5 import re
   6 import urllib.request
   7 import html
   8
   9 servernet = "irc.freenode.net"
  10 port = 6667
  11 servername = ""
  12 timeout = 240
  13 username = "plomlombot"
  14 nickname = username
  15 channel = "#zrolaps-test"
  16
  17 class IO:
  18
  19     def __init__(self, server, port):
  20         self.socket = socket.socket()
  21         self.socket.connect((server, port))
  22         self.socket.setblocking(0)
  23         self.line_buffer = []
  24         self.rune_buffer = ""
  25         self.last_pong = time.time()
  26
  27     def _pingtest(self):
  28         if self.last_pong + timeout < time.time():
  29             raise RuntimeError("server not answering")
  30         self.send_line("PING " + nickname + " " + servername)
  31
  32     def send_line(self, msg):
  33         msg = msg.replace("\r", " ")
  34         msg = msg.replace("\n", " ")
  35         if len(msg.encode("utf-8")) > 510:
  36             print("NOT SENT LINE TO SERVER (too long): " + msg)
  37         print("LINE TO SERVER: "
  38             + str(datetime.datetime.now()) + ": " + msg)
  39         msg = msg + "\r\n"
  40         msg_len = len(msg)
  41         total_sent_len = 0
  42         while total_sent_len < msg_len:
  43             sent_len = self.socket.send(bytes(msg[total_sent_len:], "UTF-8"))
  44             if sent_len == 0:
  45                 raise RuntimeError("socket connection broken")
  46             total_sent_len += sent_len
  47
  48     def recv_line_wrapped(self):
  49         if len(self.line_buffer) > 0:
  50             return self.line_buffer.pop(0)
  51         while True:
  52             ready = select.select([self.socket], [], [], int(timeout / 2))
  53             if not ready[0]:
  54                 self._pingtest()
  55                 return None
  56             self.last_pong = time.time()
  57             received_runes = self.socket.recv(1024).decode("UTF-8")
  58             if len(received_runes) == 0:
  59                 raise RuntimeError("socket connection broken")
  60             self.rune_buffer += received_runes
  61             lines_split = str.split(self.rune_buffer, "\r\n")
  62             self.line_buffer += lines_split[:-1]
  63             self.rune_buffer = lines_split[-1]
  64             if len(self.line_buffer) > 0:
  65                 return self.line_buffer.pop(0)
  66
  67     def recv_line(self):
  68         line = self.recv_line_wrapped()
  69         if line:
  70             print("LINE FROM SERVER " + str(datetime.datetime.now()) + ": " +
  71             line)
  72         return line
  73
  74 def lineparser_loop():
  75
  76     def act_on_privmsg(tokens):
  77
  78         def url_check(msg):
  79             matches = re.findall("(https?://[^\s]+)", msg)
  80             for i in range(len(matches)):
  81                 url = matches[i]
  82                 try:
  83                     webpage = urllib.request.urlopen(url, timeout=15)
  84                 except urllib.error.HTTPError as error:
  85                     print("TROUBLE FOLLOWING URL: " + str(error))
  86                     continue
  87                 charset = webpage.info().get_content_charset()
  88                 if not charset:
  89                     charset="utf-8"
  90                 content_type = webpage.info().get_content_type()
  91                 if not content_type in ('text/html', 'text/xml',
  92                         'application/xhtml+xml'):
  93                     print("TROUBLE INTERPRETING URL: bad content type "
  94                             + content_type)
  95                     continue
  96                 content = webpage.read().decode(charset)
  97                 title = str(content).split('<title>')[1].split('</title>')[0]
  98                 title = html.unescape(title)
  99                 io.send_line("PRIVMSG " + target + " :page title for url: "
 100                     + title)
 101
 102         sender = ""
 103         for rune in tokens[0]:
 104             if rune == "!":
 105                 break
 106             if rune != ":":
 107                 sender += rune
 108         receiver = ""
 109         for rune in tokens[2]:
 110             if rune == "!":
 111                 break
 112             if rune != ":":
 113                 receiver += rune
 114         target = sender
 115         if receiver != nickname:
 116             target = receiver
 117         msg = str.join(" ", tokens[3:])[1:]
 118         url_check(msg)
 119
 120     while 1:
 121         line = io.recv_line()
 122         if not line:
 123             continue
 124         tokens = line.split(" ")
 125         if len(tokens) > 1:
 126             if tokens[1] == "PRIVMSG":
 127                 act_on_privmsg(tokens)
 128             if tokens[0] == "PING":
 129                 io.send_line("PONG " + tokens[1])
 130
 131 io = IO(servernet, port)
 132 io.send_line("NICK " + nickname)
 133 io.send_line("USER " + username + " 0 * : ")
 134 io.send_line("JOIN " + channel)
 135 servername = io.recv_line().split(" ")[0][1:]
 136 lineparser_loop()