def send_message(self, dest, message, master): msg = NetworkMessage() msg.buffer = "PRIVMSG %s :%s" % (dest, message) self.send(msg) if master != "": self.send_message(master, "Message '%s' sent to '%s', master!" % (message, dest), "") SendPrivmsgEvent(self.username, dest, message, master).post(self.ed)
def part_channel(self, event): msg = NetworkMessage() channel = self.make_channel(event.channel) msg.buffer = "PART " + channel self.send(msg) if event.master != "": self.send_message(event.master, "Parting channel '%s', master!" % (channel), "") PartEvent(self.username, channel, event.master).post(self.ed)
def join_channel(self, event): msg = NetworkMessage() channel = self.make_channel(event.channel) msg.buffer = "JOIN " + channel self.send(msg) if event.master != "": self.send_message(event.master, "Joining channel '%s', master!" % (channel), "") JoinEvent(self.username, channel, event.master).post(self.ed)
def connect(self, event): msg = NetworkMessage() msg.buffer = "USER " + event.username + " * 8 :" + event.username + " sloffson" self.send(msg) msg = NetworkMessage() msg.buffer = "NICK " + event.username self.username = event.username self.send(msg)
def reconnect(self, event): msg = NetworkMessage(True, "reconnect") if self.connection is not False: if event.master != "": self.send_message(event.master, "Reconnecting with message '%s', master!" % (event.message), "") msg.buffer = "QUIT " + event.message self.send(msg) ReconnectEvent(event.message, event.master).post(self.ed)
def send_command(self, event): msg = NetworkMessage() msg.buffer = "%s " % event.type if not isinstance(event.message, basestring): for word in event.message: msg.buffer += "%s " % word msg.buffer = msg.buffer.strip() else: msg.buffer = "%s %s" % (msg.buffer, event.message) self.send(msg) if event.master != "": self.send_message(event.master, "Command '%s' sent with parameters '%s', master!" % (event.type, event.message), "") SendCommandEvent(event.type, event.message, event.master).post(self.ed)
def __init__(self, host, port, ed):
self.host = (host, port)
self.readingError = self.writingError = False
self.msg = NetworkMessage()
self.ed = ed
self.netview = NetworkView(self, self.ed)
self.netcontrol = NetworkController(self.ed)
self.connectionLock = thread.allocate_lock()
self.ssl = None
def pong(self, event): msg = NetworkMessage() msg.buffer = "PONG :" + event.message msg.silent = True self.send(msg) PongEvent(event.message).post(self.ed)
class Connection:
def __init__(self, host, port, ed):
self.host = (host, port)
self.readingError = self.writingError = False
self.msg = NetworkMessage()
self.ed = ed
self.netview = NetworkView(self, self.ed)
self.netcontrol = NetworkController(self.ed)
self.connectionLock = thread.allocate_lock()
self.ssl = None
def connect(self, use_ssl=False):
#print self.host
self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
if use_ssl:
self.ssl = ssl.wrap_socket(self.socket, ssl_version=ssl.PROTOCOL_SSLv23)
self.ssl.connect(self.host)
else:
self.socket.connect(self.host)
thread.start_new_thread(self.parse_packet, ())
#self.ed.post(ConnectedEvent())
ConnectedEvent().post(self.ed)
return
def check_state(self):
if self.writingError is True or self.readingError is True:
self.close_connection()
return False
return True
def parse_packet(self):
while self.check_state():
self.msg.buffer = self.safe_recv(2048) #ACQUIRE PACKET BRO
if isinstance(self.msg.buffer, basestring) and self.msg.buffer != "":
self.netcontrol.on_recv_message(self.msg)
self.msg.reset()
return
def send(self, msg): # Can be called from any thread
if not self.check_state():
return
self.connectionLock.acquire()
self.netview.on_send_message(msg)
self.internal_send(msg)
self.connectionLock.release()
def internal_send(self, msg):
self.safe_send(msg)
self.on_write_operation(msg)
def on_write_operation(self, msg):
msg.reset()
def close_connection(self, type="server"):
self.connectionLock.acquire()
#self.netview.disconnect(DisconnectEvent("Hey man, is this a dream?"))
try:
if self.ssl:
self.ssl.shutdown(socket.SHUT_RDWR)
self.ssl.close()
else:
self.socket.close()
except socket.error, msg:
#self.ed.post(OutputEvent("Internal", msg))
OutputEvent("Internal", msg).post(self.ed)
#self.ed.post(ConnectionClosedEvent(type))
ConnectionClosedEvent(type).post(self.ed)
self.connectionLock.release()
return True