13 # Defaults, may be overwritten by command line arguments.
14 SERVER = "irc.freenode.net"
17 USERNAME = "plomlombot"
21 class ExceptionForRestart(Exception):
27 def __init__(self, server, port, timeout):
28 self.timeout = timeout
29 self.socket = socket.socket()
30 self.socket.connect((server, port))
31 self.socket.setblocking(0)
34 self.last_pong = time.time()
35 self.servername = self.recv_line(send_ping=False).split(" ")[0][1:]
37 def _pingtest(self, send_ping=True):
38 if self.last_pong + self.timeout < time.time():
39 print("SERVER NOT ANSWERING")
40 raise ExceptionForRestart
42 self.send_line("PING " + self.servername)
44 def send_line(self, msg):
45 msg = msg.replace("\r", " ")
46 msg = msg.replace("\n", " ")
47 if len(msg.encode("utf-8")) > 510:
48 print("NOT SENT LINE TO SERVER (too long): " + msg)
49 print("LINE TO SERVER: "
50 + str(datetime.datetime.now()) + ": " + msg)
54 while total_sent_len < msg_len:
55 sent_len = self.socket.send(bytes(msg[total_sent_len:], "UTF-8"))
57 print("SOCKET CONNECTION BROKEN")
58 raise ExceptionForRestart
59 total_sent_len += sent_len
61 def _recv_line_wrapped(self, send_ping=True):
62 if len(self.line_buffer) > 0:
63 return self.line_buffer.pop(0)
65 ready = select.select([self.socket], [], [], int(self.timeout / 2))
67 self._pingtest(send_ping)
69 self.last_pong = time.time()
70 received_bytes = self.socket.recv(1024)
72 received_runes = received_bytes.decode("UTF-8")
73 except UnicodeDecodeError:
74 received_runes = received_bytes.decode("latin1")
75 if len(received_runes) == 0:
76 print("SOCKET CONNECTION BROKEN")
77 raise ExceptionForRestart
78 self.rune_buffer += received_runes
79 lines_split = str.split(self.rune_buffer, "\r\n")
80 self.line_buffer += lines_split[:-1]
81 self.rune_buffer = lines_split[-1]
82 if len(self.line_buffer) > 0:
83 return self.line_buffer.pop(0)
85 def recv_line(self, send_ping=True):
86 line = self._recv_line_wrapped(send_ping)
88 print("LINE FROM SERVER " + str(datetime.datetime.now()) + ": " +
93 def init_session(server, port, timeout, nickname, username, channel):
94 print("CONNECTING TO " + server)
95 io = IO(server, port, timeout)
96 io.send_line("NICK " + nickname)
97 io.send_line("USER " + username + " 0 * : ")
98 io.send_line("JOIN " + channel)
102 def lineparser_loop(io, nickname):
104 def act_on_privmsg(tokens):
109 io.send_line("NOTICE " + target + " :" + msg)
111 matches = re.findall("(https?://[^\s>]+)", msg)
112 for i in range(len(matches)):
114 request = urllib.request.Request(url, headers={
115 "User-Agent": "plomlombot"
118 webpage = urllib.request.urlopen(request, timeout=15)
119 except (urllib.error.HTTPError, urllib.error.URLError,
120 UnicodeError, http.client.BadStatusLine) as error:
121 notice("TROUBLE FOLLOWING URL: " + str(error))
123 charset = webpage.info().get_content_charset()
126 content_type = webpage.info().get_content_type()
127 if content_type not in ('text/html', 'text/xml',
128 'application/xhtml+xml'):
129 notice("TROUBLE READING PAGE TITLE: bad content type "
132 content = webpage.read().decode(charset)
133 title = str(content).split('<title>')[1].split('</title>')[0]
134 title = html.unescape(title)
135 notice("PAGE TITLE FOR URL: " + title)
138 for rune in tokens[0]:
144 for rune in tokens[2]:
150 if receiver != nickname:
152 msg = str.join(" ", tokens[3:])[1:]
156 line = io.recv_line()
159 tokens = line.split(" ")
161 if tokens[1] == "PRIVMSG":
162 act_on_privmsg(tokens)
163 if tokens[0] == "PING":
164 io.send_line("PONG " + tokens[1])
167 def parse_command_line_arguments():
168 parser = argparse.ArgumentParser()
169 parser.add_argument("-s, --server", action="store", dest="server",
171 help="server or server net to connect to (default: "
173 parser.add_argument("-p, --port", action="store", dest="port", type=int,
174 default=PORT, help="port to connect to (default : "
176 parser.add_argument("-t, --timeout", action="store", dest="timeout",
177 type=int, default=TIMEOUT,
178 help="timeout in seconds after which to attempt " +
179 "reconnect (default: " + str(TIMEOUT) + ")")
180 parser.add_argument("-u, --username", action="store", dest="username",
181 default=USERNAME, help="username to use (default: "
183 parser.add_argument("-n, --nickname", action="store", dest="nickname",
184 default=NICKNAME, help="nickname to use (default: "
186 parser.add_argument("CHANNEL", action="store", help="channel to join")
187 opts, unknown = parser.parse_known_args()
190 opts = parse_command_line_arguments()
193 io = init_session(opts.server, opts.port, opts.timeout, opts.nickname,
194 opts.username, opts.CHANNEL)
195 lineparser_loop(io, opts.nickname)
196 except ExceptionForRestart: