14 # Defaults, may be overwritten by command line arguments.
15 SERVER = "irc.freenode.net"
18 USERNAME = "plomlombot"
22 class HTMLParser(html.parser.HTMLParser):
23 def __init__(self, html, tag):
28 def handle_starttag(self, tag, attrs):
31 def handle_endtag(self, tag):
33 def handle_data(self, data):
38 class ExceptionForRestart(Exception):
44 def __init__(self, server, port, timeout):
45 self.timeout = timeout
46 self.socket = socket.socket()
47 self.socket.connect((server, port))
48 self.socket.setblocking(0)
51 self.last_pong = time.time()
52 self.servername = self.recv_line(send_ping=False).split(" ")[0][1:]
54 def _pingtest(self, send_ping=True):
55 if self.last_pong + self.timeout < time.time():
56 print("SERVER NOT ANSWERING")
57 raise ExceptionForRestart
59 self.send_line("PING " + self.servername)
61 def send_line(self, msg):
62 msg = msg.replace("\r", " ")
63 msg = msg.replace("\n", " ")
64 if len(msg.encode("utf-8")) > 510:
65 print("NOT SENT LINE TO SERVER (too long): " + msg)
66 print("LINE TO SERVER: "
67 + str(datetime.datetime.now()) + ": " + msg)
71 while total_sent_len < msg_len:
72 sent_len = self.socket.send(bytes(msg[total_sent_len:], "UTF-8"))
74 print("SOCKET CONNECTION BROKEN")
75 raise ExceptionForRestart
76 total_sent_len += sent_len
78 def _recv_line_wrapped(self, send_ping=True):
79 if len(self.line_buffer) > 0:
80 return self.line_buffer.pop(0)
82 ready = select.select([self.socket], [], [], int(self.timeout / 2))
84 self._pingtest(send_ping)
86 self.last_pong = time.time()
87 received_bytes = self.socket.recv(1024)
89 received_runes = received_bytes.decode("UTF-8")
90 except UnicodeDecodeError:
91 received_runes = received_bytes.decode("latin1")
92 if len(received_runes) == 0:
93 print("SOCKET CONNECTION BROKEN")
94 raise ExceptionForRestart
95 self.rune_buffer += received_runes
96 lines_split = str.split(self.rune_buffer, "\r\n")
97 self.line_buffer += lines_split[:-1]
98 self.rune_buffer = lines_split[-1]
99 if len(self.line_buffer) > 0:
100 return self.line_buffer.pop(0)
102 def recv_line(self, send_ping=True):
103 line = self._recv_line_wrapped(send_ping)
105 print("LINE FROM SERVER " + str(datetime.datetime.now()) + ": " +
110 def init_session(server, port, timeout, nickname, username, channel):
111 print("CONNECTING TO " + server)
112 io = IO(server, port, timeout)
113 io.send_line("NICK " + nickname)
114 io.send_line("USER " + username + " 0 * : ")
115 io.send_line("JOIN " + channel)
119 def lineparser_loop(io, nickname):
121 def act_on_privmsg(tokens):
126 io.send_line("NOTICE " + target + " :" + msg)
128 matches = re.findall("(https?://[^\s>]+)", msg)
129 for i in range(len(matches)):
131 request = urllib.request.Request(url, headers={
132 "User-Agent": "plomlombot"
135 webpage = urllib.request.urlopen(request, timeout=15)
136 except (urllib.error.HTTPError, urllib.error.URLError,
137 UnicodeError, http.client.BadStatusLine) as error:
138 notice("TROUBLE FOLLOWING URL: " + str(error))
140 charset = webpage.info().get_content_charset()
142 notice("TROUBLE READING PAGE TITLE: no charset in header")
144 content_type = webpage.info().get_content_type()
145 if content_type not in ('text/html', 'text/xml',
146 'application/xhtml+xml'):
147 notice("TROUBLE READING PAGE TITLE: bad content type "
150 content = webpage.read().decode(charset)
151 title = HTMLParser(content, "title").data
152 title = html.unescape(title)
153 notice("PAGE TITLE FOR URL: " + title)
156 for rune in tokens[0]:
162 for rune in tokens[2]:
168 if receiver != nickname:
170 msg = str.join(" ", tokens[3:])[1:]
174 line = io.recv_line()
177 tokens = line.split(" ")
179 if tokens[1] == "PRIVMSG":
180 act_on_privmsg(tokens)
181 if tokens[0] == "PING":
182 io.send_line("PONG " + tokens[1])
185 def parse_command_line_arguments():
186 parser = argparse.ArgumentParser()
187 parser.add_argument("-s, --server", action="store", dest="server",
189 help="server or server net to connect to (default: "
191 parser.add_argument("-p, --port", action="store", dest="port", type=int,
192 default=PORT, help="port to connect to (default : "
194 parser.add_argument("-t, --timeout", action="store", dest="timeout",
195 type=int, default=TIMEOUT,
196 help="timeout in seconds after which to attempt " +
197 "reconnect (default: " + str(TIMEOUT) + ")")
198 parser.add_argument("-u, --username", action="store", dest="username",
199 default=USERNAME, help="username to use (default: "
201 parser.add_argument("-n, --nickname", action="store", dest="nickname",
202 default=NICKNAME, help="nickname to use (default: "
204 parser.add_argument("CHANNEL", action="store", help="channel to join")
205 opts, unknown = parser.parse_known_args()
208 opts = parse_command_line_arguments()
211 io = init_session(opts.server, opts.port, opts.timeout, opts.nickname,
212 opts.username, opts.CHANNEL)
213 lineparser_loop(io, opts.nickname)
214 except ExceptionForRestart: