From: Christian Heller Date: Sun, 17 Jan 2016 18:07:17 +0000 (+0100) Subject: Initial commit. X-Git-Url: https://plomlompom.com/repos/%7B%7B%20web_path%20%7D%7D/decks/edit?a=commitdiff_plain;h=24cf06720f7e0c13e6075f69426529a2119085ba;p=plomlombot-irc.git Initial commit. --- diff --git a/plomlombot.py b/plomlombot.py new file mode 100644 index 0000000..747ab07 --- /dev/null +++ b/plomlombot.py @@ -0,0 +1,102 @@ +import socket +import datetime +import select +import time +import re +import urllib.request +import html + +servername = "" +timeout = 480 +username = "plomlombot" +nickname = username +channel = "#zrolaps" + +class IO: + def __init__(self, server, port): + self.socket = socket.socket() + self.socket.connect((server, port)) + self.socket.setblocking(0) + self.line_buffer = [] + self.rune_buffer = "" + self.last_pong = time.time() + def _pingtest(self): + if self.last_pong + timeout < time.time(): + raise RuntimeError("server not answering") + self.send_line("PING " + nickname + " " + servername) + def send_line(self, msg_orig): + msg = msg_orig + "\r\n" + msg_len = len(msg) + total_sent_len = 0 + while total_sent_len < msg_len: + sent_len = self.socket.send(bytes(msg[total_sent_len:], "UTF-8")) + if sent_len == 0: + raise RuntimeError("socket connection broken") + total_sent_len += sent_len + print("LINE TO SERVER: " + + str(datetime.datetime.now()) + ": " + msg_orig) + def recv_line_wrapped(self): + if len(self.line_buffer) > 0: + return self.line_buffer.pop(0) + while True: + ready = select.select([self.socket], [], [], int(timeout / 2)) + if not ready[0]: + self._pingtest() + return None + self.last_pong = time.time() + received_runes = self.socket.recv(1024).decode("UTF-8") + if len(received_runes) == 0: + raise RuntimeError("socket connection broken") + self.rune_buffer += received_runes + lines_split = str.split(self.rune_buffer, "\r\n") + self.line_buffer += lines_split[:-1] + self.rune_buffer = lines_split[-1] + if len(self.line_buffer) > 0: + return self.line_buffer.pop(0) + def recv_line(self): + line = self.recv_line_wrapped() + if line: + print("LINE FROM SERVER " + str(datetime.datetime.now()) + ": " + + line) + return line + +io = IO("irc.freenode.net", 6667) +io.send_line("NICK " + nickname) +io.send_line("USER " + username + " 0 * : ") +io.send_line("JOIN " + channel) +servername = io.recv_line().split(" ")[0][1:] +while 1: + line = io.recv_line() + if not line: + continue + tokens = line.split(" ") + if len(tokens) > 1: + if tokens[1] == "PRIVMSG": + sender = "" + for rune in tokens[0]: + if rune == "!": + break + if rune != ":": + sender += rune + receiver = "" + for rune in tokens[2]: + if rune == "!": + break + if rune != ":": + receiver += rune + target = sender + if receiver != nickname: + target = receiver + msg = str.join(" ", tokens[3:])[1:] + matches = re.findall("(https?://[^\s]+)", msg) + for i in range(len(matches)): + url = matches[i] + webpage = urllib.request.urlopen(url) + charset = webpage.info().get_content_charset() + content = webpage.read().decode(charset) + title = str(content).split('')[1].split('')[0] + title = html.unescape(title) + io.send_line("PRIVMSG " + + target + " :page title for url: " + title) + if tokens[0] == "PING": + io.send_line("PONG " + tokens[1])