10 # Defaults, may be overwritten by command line arguments.
11 SERVER = "irc.freenode.net"
14 USERNAME = "plomlombot"
17 class ExceptionForRestart(Exception):
22 def __init__(self, server, port, timeout):
23 self.timeout = timeout
24 self.socket = socket.socket()
25 self.socket.connect((server, port))
26 self.socket.setblocking(0)
29 self.last_pong = time.time()
30 self.servername = self.recv_line(send_ping=False).split(" ")[0][1:]
32 def _pingtest(self, send_ping=True):
33 if self.last_pong + self.timeout < time.time():
34 print("SERVER NOT ANSWERING")
35 raise ExceptionForRestart
37 self.send_line("PING " + self.servername)
39 def send_line(self, msg):
40 msg = msg.replace("\r", " ")
41 msg = msg.replace("\n", " ")
42 if len(msg.encode("utf-8")) > 510:
43 print("NOT SENT LINE TO SERVER (too long): " + msg)
44 print("LINE TO SERVER: "
45 + str(datetime.datetime.now()) + ": " + msg)
49 while total_sent_len < msg_len:
50 sent_len = self.socket.send(bytes(msg[total_sent_len:], "UTF-8"))
52 print("SOCKET CONNECTION BROKEN")
53 raise ExceptionForRestart
54 total_sent_len += sent_len
56 def _recv_line_wrapped(self, send_ping=True):
57 if len(self.line_buffer) > 0:
58 return self.line_buffer.pop(0)
60 ready = select.select([self.socket], [], [], int(self.timeout / 2))
62 self._pingtest(send_ping)
64 self.last_pong = time.time()
65 received_runes = self.socket.recv(1024).decode("UTF-8")
66 if len(received_runes) == 0:
67 print("SOCKET CONNECTION BROKEN")
68 raise ExceptionForRestart
69 self.rune_buffer += received_runes
70 lines_split = str.split(self.rune_buffer, "\r\n")
71 self.line_buffer += lines_split[:-1]
72 self.rune_buffer = lines_split[-1]
73 if len(self.line_buffer) > 0:
74 return self.line_buffer.pop(0)
76 def recv_line(self, send_ping=True):
77 line = self._recv_line_wrapped(send_ping)
79 print("LINE FROM SERVER " + str(datetime.datetime.now()) + ": " +
83 def init_session(server, port, timeout, nickname, username, channel):
84 print("CONNECTING TO " + server)
85 io = IO(server, port, timeout)
86 io.send_line("NICK " + nickname)
87 io.send_line("USER " + username + " 0 * : ")
88 io.send_line("JOIN " + channel)
91 def lineparser_loop(io, nickname):
93 def act_on_privmsg(tokens):
96 matches = re.findall("(https?://[^\s]+)", msg)
97 for i in range(len(matches)):
100 webpage = urllib.request.urlopen(url, timeout=15)
101 except urllib.error.HTTPError as error:
102 print("TROUBLE FOLLOWING URL: " + str(error))
104 charset = webpage.info().get_content_charset()
107 content_type = webpage.info().get_content_type()
108 if not content_type in ('text/html', 'text/xml',
109 'application/xhtml+xml'):
110 print("TROUBLE INTERPRETING URL: bad content type "
113 content = webpage.read().decode(charset)
114 title = str(content).split('<title>')[1].split('</title>')[0]
115 title = html.unescape(title)
116 io.send_line("PRIVMSG " + target + " :page title for url: "
120 for rune in tokens[0]:
126 for rune in tokens[2]:
132 if receiver != nickname:
134 msg = str.join(" ", tokens[3:])[1:]
138 line = io.recv_line()
141 tokens = line.split(" ")
143 if tokens[1] == "PRIVMSG":
144 act_on_privmsg(tokens)
145 if tokens[0] == "PING":
146 io.send_line("PONG " + tokens[1])
148 def parse_command_line_arguments():
149 parser = argparse.ArgumentParser()
150 parser.add_argument("-s, --server", action="store", dest="server",
152 help="server or server net to connect to (default: " + SERVER +
154 parser.add_argument("-p, --port", action="store", dest="port", type=int,
155 default=PORT, help="port to connect to (default : " + str(PORT) +
157 parser.add_argument("-t, --timeout", action="store", dest="timeout",
158 type=int, default=TIMEOUT,
159 help="timeout in seconds after which to attempt reconnect " +
160 "(default: " + str(TIMEOUT) + ")")
161 parser.add_argument("-u, --username", action="store", dest="username",
162 default=USERNAME, help="username to use (default: " + USERNAME +
164 parser.add_argument("-n, --nickname", action="store", dest="nickname",
165 default=NICKNAME, help="nickname to use (default: " + NICKNAME +
167 parser.add_argument("CHANNEL", action="store", help="channel to join")
168 opts, unknown = parser.parse_known_args()
171 opts = parse_command_line_arguments()
174 io = init_session(opts.server, opts.port, opts.timeout, opts.nickname,
175 opts.username, opts.CHANNEL)
176 lineparser_loop(io, opts.nickname)
177 except ExceptionForRestart: