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)
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/' \
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 notice("PAGE TITLE: " + title.string.strip())
141 notice("PAGE HAS NO TITLE TAG")
143 matches = re.findall("(https?://[^\s>]+)", msg)
144 for i in range(len(matches)):
145 handle_url(matches[i])
147 def command_check(msg):
150 tokens = msg[1:].split()
151 hash_string = hashlib.md5(target.encode("utf-8")).hexdigest()
152 quotesfile_name = "quotes_" + hash_string
153 if tokens[0] == "addquote":
154 if not os.access(quotesfile_name, os.F_OK):
155 quotesfile = open(quotesfile_name, "w")
156 quotesfile.write("QUOTES FOR " + target + ":\n")
158 quotesfile = open(quotesfile_name, "a")
159 quotesfile.write(str.join(" ", tokens[1:]) + "\n")
161 quotesfile = open(quotesfile_name, "r")
162 lines = quotesfile.readlines()
164 notice("ADDED QUOTE #" + str(len(lines) - 1))
165 elif tokens[0] == "quote":
166 if len(tokens) > 2 or \
167 (len(tokens) == 2 and not tokens[1].isdigit()):
168 notice("SYNTAX: !quote [int]")
170 if not os.access(quotesfile_name, os.F_OK):
171 notice("NO QUOTES AVAILABLE")
173 quotesfile = open(quotesfile_name, "r")
174 lines = quotesfile.readlines()
179 if i == 0 or i > len(lines):
180 notice("THERE'S NO QUOTE OF THAT INDEX")
184 i = random.randrange(len(lines))
185 notice("QUOTE #" + str(i + 1) + ": " + lines[i])
188 for rune in tokens[0]:
194 for rune in tokens[2]:
200 if receiver != nickname:
202 msg = str.join(" ", tokens[3:])[1:]
207 line = io.recv_line()
210 tokens = line.split(" ")
212 if tokens[1] == "PRIVMSG":
213 act_on_privmsg(tokens)
214 if tokens[0] == "PING":
215 io.send_line("PONG " + tokens[1])
218 def parse_command_line_arguments():
219 parser = argparse.ArgumentParser()
220 parser.add_argument("-s, --server", action="store", dest="server",
222 help="server or server net to connect to (default: "
224 parser.add_argument("-p, --port", action="store", dest="port", type=int,
225 default=PORT, help="port to connect to (default : "
227 parser.add_argument("-t, --timeout", action="store", dest="timeout",
228 type=int, default=TIMEOUT,
229 help="timeout in seconds after which to attempt " +
230 "reconnect (default: " + str(TIMEOUT) + ")")
231 parser.add_argument("-u, --username", action="store", dest="username",
232 default=USERNAME, help="username to use (default: "
234 parser.add_argument("-n, --nickname", action="store", dest="nickname",
235 default=NICKNAME, help="nickname to use (default: "
237 parser.add_argument("CHANNEL", action="store", help="channel to join")
238 opts, unknown = parser.parse_known_args()
241 opts = parse_command_line_arguments()
244 io = init_session(opts.server, opts.port, opts.timeout, opts.nickname,
245 opts.username, opts.CHANNEL)
246 lineparser_loop(io, opts.nickname)
247 except ExceptionForRestart: