13 # Defaults, may be overwritten by command line arguments.
14 SERVER = "irc.freenode.net"
17 USERNAME = "plomlombot"
21 class HTMLParser(html.parser.HTMLParser):
22 def __init__(self, html, tag):
24 self._tag_to_check = tag
28 def handle_starttag(self, tag, attrs):
29 if self.data == "" and tag == self._tag_to_check:
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)):
132 r = requests.get(url, timeout=15)
133 except (requests.exceptions.TooManyRedirects,
134 requests.exceptions.ConnectionError,
135 requests.exceptions.InvalidSchema) as error:
136 notice("TROUBLE FOLLOWING URL: " + str(error))
139 title = HTMLParser(content, "title").data
140 title = html.unescape(title)
141 notice("PAGE TITLE FOR URL: " + title)
144 for rune in tokens[0]:
150 for rune in tokens[2]:
156 if receiver != nickname:
158 msg = str.join(" ", tokens[3:])[1:]
162 line = io.recv_line()
165 tokens = line.split(" ")
167 if tokens[1] == "PRIVMSG":
168 act_on_privmsg(tokens)
169 if tokens[0] == "PING":
170 io.send_line("PONG " + tokens[1])
173 def parse_command_line_arguments():
174 parser = argparse.ArgumentParser()
175 parser.add_argument("-s, --server", action="store", dest="server",
177 help="server or server net to connect to (default: "
179 parser.add_argument("-p, --port", action="store", dest="port", type=int,
180 default=PORT, help="port to connect to (default : "
182 parser.add_argument("-t, --timeout", action="store", dest="timeout",
183 type=int, default=TIMEOUT,
184 help="timeout in seconds after which to attempt " +
185 "reconnect (default: " + str(TIMEOUT) + ")")
186 parser.add_argument("-u, --username", action="store", dest="username",
187 default=USERNAME, help="username to use (default: "
189 parser.add_argument("-n, --nickname", action="store", dest="nickname",
190 default=NICKNAME, help="nickname to use (default: "
192 parser.add_argument("CHANNEL", action="store", help="channel to join")
193 opts, unknown = parser.parse_known_args()
196 opts = parse_command_line_arguments()
199 io = init_session(opts.server, opts.port, opts.timeout, opts.nickname,
200 opts.username, opts.CHANNEL)
201 lineparser_loop(io, opts.nickname)
202 except ExceptionForRestart: