15 # Defaults, may be overwritten by command line arguments.
16 SERVER = "irc.freenode.net"
19 USERNAME = "plomlombot"
23 class ExceptionForRestart(Exception):
29 def __init__(self, server, port, timeout):
30 self.timeout = timeout
31 self.socket = socket.socket()
32 self.socket.connect((server, port))
33 self.socket.setblocking(0)
36 self.last_pong = time.time()
37 self.servername = self.recv_line(send_ping=False).split(" ")[0][1:]
39 def _pingtest(self, send_ping=True):
40 if self.last_pong + self.timeout < time.time():
41 print("SERVER NOT ANSWERING")
42 raise ExceptionForRestart
44 self.send_line("PING " + self.servername)
46 def send_line(self, msg):
47 msg = msg.replace("\r", " ")
48 msg = msg.replace("\n", " ")
49 if len(msg.encode("utf-8")) > 510:
50 print("NOT SENT LINE TO SERVER (too long): " + msg)
51 print("LINE TO SERVER: "
52 + str(datetime.datetime.now()) + ": " + msg)
56 while total_sent_len < msg_len:
57 sent_len = self.socket.send(bytes(msg[total_sent_len:], "UTF-8"))
59 print("SOCKET CONNECTION BROKEN")
60 raise ExceptionForRestart
61 total_sent_len += sent_len
63 def _recv_line_wrapped(self, send_ping=True):
64 if len(self.line_buffer) > 0:
65 return self.line_buffer.pop(0)
67 ready = select.select([self.socket], [], [], int(self.timeout / 2))
69 self._pingtest(send_ping)
71 self.last_pong = time.time()
72 received_bytes = self.socket.recv(1024)
74 received_runes = received_bytes.decode("UTF-8")
75 except UnicodeDecodeError:
76 received_runes = received_bytes.decode("latin1")
77 if len(received_runes) == 0:
78 print("SOCKET CONNECTION BROKEN")
79 raise ExceptionForRestart
80 self.rune_buffer += received_runes
81 lines_split = str.split(self.rune_buffer, "\r\n")
82 self.line_buffer += lines_split[:-1]
83 self.rune_buffer = lines_split[-1]
84 if len(self.line_buffer) > 0:
85 return self.line_buffer.pop(0)
87 def recv_line(self, send_ping=True):
88 line = self._recv_line_wrapped(send_ping)
90 print("LINE FROM SERVER " + str(datetime.datetime.now()) + ": " +
95 def init_session(server, port, timeout, nickname, username, channel):
96 print("CONNECTING TO " + server)
97 io = IO(server, port, timeout)
98 io.send_line("NICK " + nickname)
99 io.send_line("USER " + username + " 0 * : ")
100 io.send_line("JOIN " + channel)
104 def lineparser_loop(io, nickname):
106 def act_on_privmsg(tokens):
109 io.send_line("NOTICE " + target + " :" + msg)
113 def handle_url(url, show_url=False):
115 def mobile_twitter_hack(url):
116 re1 = 'https?://(mobile.twitter.com/)[^/]+(/status/)'
117 re2 = 'https?://mobile.twitter.com/([^/]+)/status/([^\?]+)'
118 m = re.search(re1, url)
119 if m and m.group(1) == 'mobile.twitter.com/' \
120 and m.group(2) == '/status/':
121 m = re.search(re2, url)
122 url = 'https://twitter.com/' + m.group(1) + '/status/' \
124 handle_url(url, True)
128 r = requests.get(url, timeout=15)
129 except (requests.exceptions.TooManyRedirects,
130 requests.exceptions.ConnectionError,
131 requests.exceptions.InvalidURL,
132 requests.exceptions.InvalidSchema) as error:
133 notice("TROUBLE FOLLOWING URL: " + str(error))
135 if mobile_twitter_hack(url):
137 title = bs4.BeautifulSoup(r.text).title
139 prefix = "PAGE TITLE: "
141 prefix = "PAGE TITLE FOR <" + url + ">: "
142 notice(prefix + title.string.strip())
144 notice("PAGE HAS NO TITLE TAG")
146 matches = re.findall("(https?://[^\s>]+)", msg)
147 for i in range(len(matches)):
148 handle_url(matches[i])
150 def command_check(msg):
153 tokens = msg[1:].split()
154 hash_string = hashlib.md5(target.encode("utf-8")).hexdigest()
155 quotesfile_name = "quotes_" + hash_string
156 if tokens[0] == "addquote":
157 if not os.access(quotesfile_name, os.F_OK):
158 quotesfile = open(quotesfile_name, "w")
159 quotesfile.write("QUOTES FOR " + target + ":\n")
161 quotesfile = open(quotesfile_name, "a")
162 quotesfile.write(str.join(" ", tokens[1:]) + "\n")
164 quotesfile = open(quotesfile_name, "r")
165 lines = quotesfile.readlines()
167 notice("ADDED QUOTE #" + str(len(lines) - 1))
168 elif tokens[0] == "quote":
169 if len(tokens) > 2 or \
170 (len(tokens) == 2 and not tokens[1].isdigit()):
171 notice("SYNTAX: !quote [int]")
173 if not os.access(quotesfile_name, os.F_OK):
174 notice("NO QUOTES AVAILABLE")
176 quotesfile = open(quotesfile_name, "r")
177 lines = quotesfile.readlines()
182 if i == 0 or i > len(lines):
183 notice("THERE'S NO QUOTE OF THAT INDEX")
187 i = random.randrange(len(lines))
188 notice("QUOTE #" + str(i + 1) + ": " + lines[i])
191 for rune in tokens[0]:
197 for rune in tokens[2]:
203 if receiver != nickname:
205 msg = str.join(" ", tokens[3:])[1:]
210 line = io.recv_line()
213 tokens = line.split(" ")
215 if tokens[1] == "PRIVMSG":
216 act_on_privmsg(tokens)
217 if tokens[0] == "PING":
218 io.send_line("PONG " + tokens[1])
221 def parse_command_line_arguments():
222 parser = argparse.ArgumentParser()
223 parser.add_argument("-s, --server", action="store", dest="server",
225 help="server or server net to connect to (default: "
227 parser.add_argument("-p, --port", action="store", dest="port", type=int,
228 default=PORT, help="port to connect to (default : "
230 parser.add_argument("-t, --timeout", action="store", dest="timeout",
231 type=int, default=TIMEOUT,
232 help="timeout in seconds after which to attempt " +
233 "reconnect (default: " + str(TIMEOUT) + ")")
234 parser.add_argument("-u, --username", action="store", dest="username",
235 default=USERNAME, help="username to use (default: "
237 parser.add_argument("-n, --nickname", action="store", dest="nickname",
238 default=NICKNAME, help="nickname to use (default: "
240 parser.add_argument("CHANNEL", action="store", help="channel to join")
241 opts, unknown = parser.parse_known_args()
244 opts = parse_command_line_arguments()
247 io = init_session(opts.server, opts.port, opts.timeout, opts.nickname,
248 opts.username, opts.CHANNEL)
249 lineparser_loop(io, opts.nickname)
250 except ExceptionForRestart: