488 lines
19 KiB
Python
488 lines
19 KiB
Python
from PyQt4 import QtGui, QtCore
|
|
from oyoyo.client import IRCClient
|
|
from oyoyo.cmdhandler import DefaultCommandHandler
|
|
from oyoyo import helpers
|
|
import logging
|
|
import random
|
|
import socket
|
|
|
|
from dataobjs import Mood, PesterProfile
|
|
from generic import PesterList
|
|
from version import _pcVersion
|
|
|
|
logging.basicConfig(level=logging.INFO)
|
|
|
|
class PesterIRC(QtCore.QThread):
|
|
def __init__(self, config, window):
|
|
QtCore.QThread.__init__(self)
|
|
self.mainwindow = window
|
|
self.config = config
|
|
self.registeredIRC = False
|
|
self.stopIRC = None
|
|
def IRCConnect(self):
|
|
server = self.config.server()
|
|
port = self.config.port()
|
|
self.cli = IRCClient(PesterHandler, host=server, port=int(port), nick=self.mainwindow.profile().handle, real_name='pcc30', blocking=True, timeout=15)
|
|
self.cli.command_handler.parent = self
|
|
self.cli.command_handler.mainwindow = self.mainwindow
|
|
self.cli.connect()
|
|
self.conn = self.cli.conn()
|
|
def run(self):
|
|
try:
|
|
self.IRCConnect()
|
|
except socket.error, se:
|
|
self.stopIRC = se
|
|
return
|
|
while 1:
|
|
res = True
|
|
try:
|
|
logging.debug("updateIRC()")
|
|
res = self.updateIRC()
|
|
except socket.timeout, se:
|
|
logging.debug("timeout in thread %s" % (self))
|
|
self.cli.close()
|
|
self.stopIRC = se
|
|
return
|
|
except socket.error, se:
|
|
if self.registeredIRC:
|
|
self.stopIRC = None
|
|
else:
|
|
self.stopIRC = se
|
|
logging.debug("socket error, exiting thread")
|
|
return
|
|
else:
|
|
if not res:
|
|
logging.debug("false Yield: %s, returning" % res)
|
|
return
|
|
|
|
def setConnected(self):
|
|
self.registeredIRC = True
|
|
self.connected.emit()
|
|
def setConnectionBroken(self):
|
|
logging.debug("setconnection broken")
|
|
self.reconnectIRC()
|
|
#self.brokenConnection = True
|
|
@QtCore.pyqtSlot()
|
|
def updateIRC(self):
|
|
try:
|
|
res = self.conn.next()
|
|
except socket.timeout, se:
|
|
if self.registeredIRC:
|
|
return True
|
|
else:
|
|
raise se
|
|
except socket.error, se:
|
|
raise se
|
|
except StopIteration:
|
|
self.conn = self.cli.conn()
|
|
return True
|
|
else:
|
|
return res
|
|
@QtCore.pyqtSlot()
|
|
def reconnectIRC(self):
|
|
logging.debug("reconnectIRC() from thread %s" % (self))
|
|
self.cli.close()
|
|
|
|
@QtCore.pyqtSlot(PesterProfile)
|
|
def getMood(self, *chums):
|
|
self.cli.command_handler.getMood(*chums)
|
|
@QtCore.pyqtSlot(PesterList)
|
|
def getMoods(self, chums):
|
|
self.cli.command_handler.getMood(*chums)
|
|
@QtCore.pyqtSlot(QtCore.QString, QtCore.QString)
|
|
def sendNotice(self, text, handle):
|
|
h = unicode(handle)
|
|
t = unicode(text)
|
|
try:
|
|
helpers.notice(self.cli, h, t)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString, QtCore.QString)
|
|
def sendMessage(self, text, handle):
|
|
h = unicode(handle)
|
|
textl = [unicode(text)]
|
|
def splittext(l):
|
|
if len(l[0]) > 450:
|
|
space = l[0].rfind(" ", 0,430)
|
|
if space == -1:
|
|
space = 450
|
|
elif l[0][space+1:space+5] == "</c>":
|
|
space = space+4
|
|
a = l[0][0:space+1]
|
|
b = l[0][space+1:]
|
|
if a.count("<c") > a.count("</c>"):
|
|
# oh god ctags will break!! D=
|
|
hanging = []
|
|
usedends = []
|
|
c = a.rfind("<c")
|
|
while c != -1:
|
|
d = a.find("</c>", c)
|
|
while d in usedends:
|
|
d = a.find("</c>", d+1)
|
|
if d != -1: usedends.append(d)
|
|
else:
|
|
f = a.find(">", c)+1
|
|
hanging.append(a[c:f])
|
|
c = a.rfind("<c",0,c)
|
|
|
|
# end all ctags in first part
|
|
for i in range(a.count("<c")-a.count("</c>")):
|
|
a = a + "</c>"
|
|
#start them up again in the second part
|
|
for c in hanging:
|
|
b = c + b
|
|
if len(b) > 0:
|
|
return [a] + splittext([b])
|
|
else:
|
|
return [a]
|
|
else:
|
|
return l
|
|
textl = splittext(textl)
|
|
try:
|
|
for t in textl:
|
|
helpers.msg(self.cli, h, t)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString, bool)
|
|
def startConvo(self, handle, initiated):
|
|
h = unicode(handle)
|
|
try:
|
|
if initiated:
|
|
helpers.msg(self.cli, h, "PESTERCHUM:BEGIN")
|
|
helpers.msg(self.cli, h, "COLOR >%s" % (self.mainwindow.profile().colorcmd()))
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def endConvo(self, handle):
|
|
h = unicode(handle)
|
|
try:
|
|
helpers.msg(self.cli, h, "PESTERCHUM:CEASE")
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot()
|
|
def updateProfile(self):
|
|
me = self.mainwindow.profile()
|
|
handle = me.handle
|
|
try:
|
|
helpers.nick(self.cli, handle)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
self.mainwindow.closeConversations(True)
|
|
self.updateMood()
|
|
@QtCore.pyqtSlot()
|
|
def updateMood(self):
|
|
me = self.mainwindow.profile()
|
|
try:
|
|
helpers.msg(self.cli, "#pesterchum", "MOOD >%d" % (me.mood.value()))
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot()
|
|
def updateColor(self):
|
|
me = self.mainwindow.profile()
|
|
for h in self.mainwindow.convos.keys():
|
|
try:
|
|
helpers.msg(self.cli, h, "COLOR >%s" % (self.mainwindow.profile().colorcmd()))
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def blockedChum(self, handle):
|
|
h = unicode(handle)
|
|
try:
|
|
helpers.msg(self.cli, h, "PESTERCHUM:BLOCK")
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def unblockedChum(self, handle):
|
|
h = unicode(handle)
|
|
try:
|
|
helpers.msg(self.cli, h, "PESTERCHUM:UNBLOCK")
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def requestNames(self, channel):
|
|
c = unicode(channel)
|
|
try:
|
|
helpers.names(self.cli, c)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot()
|
|
def requestChannelList(self):
|
|
try:
|
|
helpers.channel_list(self.cli)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def joinChannel(self, channel):
|
|
c = unicode(channel)
|
|
try:
|
|
helpers.join(self.cli, c)
|
|
helpers.mode(self.cli, c, "", None)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def leftChannel(self, channel):
|
|
c = unicode(channel)
|
|
try:
|
|
helpers.part(self.cli, c)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString, QtCore.QString)
|
|
def kickUser(self, handle, channel):
|
|
l = handle.split(":")
|
|
c = unicode(channel)
|
|
h = unicode(l[0])
|
|
if len(l) > 1:
|
|
reason = unicode(l[1])
|
|
else:
|
|
reason = ""
|
|
try:
|
|
helpers.kick(self.cli, h, c, reason)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString, QtCore.QString, QtCore.QString)
|
|
def setChannelMode(self, channel, mode, command):
|
|
c = unicode(channel)
|
|
m = unicode(mode)
|
|
cmd = unicode(command)
|
|
if cmd == "":
|
|
cmd = None
|
|
try:
|
|
helpers.mode(self.cli, c, m, cmd)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString)
|
|
def channelNames(self, channel):
|
|
c = unicode(channel)
|
|
try:
|
|
helpers.names(self.cli, c)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
@QtCore.pyqtSlot(QtCore.QString, QtCore.QString)
|
|
def inviteChum(self, handle, channel):
|
|
h = unicode(handle)
|
|
c = unicode(channel)
|
|
try:
|
|
helpers.invite(self.cli, h, c)
|
|
except socket.error:
|
|
self.setConnectionBroken()
|
|
|
|
moodUpdated = QtCore.pyqtSignal(QtCore.QString, Mood)
|
|
colorUpdated = QtCore.pyqtSignal(QtCore.QString, QtGui.QColor)
|
|
messageReceived = QtCore.pyqtSignal(QtCore.QString, QtCore.QString)
|
|
memoReceived = QtCore.pyqtSignal(QtCore.QString, QtCore.QString, QtCore.QString)
|
|
noticeReceived = QtCore.pyqtSignal(QtCore.QString, QtCore.QString)
|
|
inviteReceived = QtCore.pyqtSignal(QtCore.QString, QtCore.QString)
|
|
timeCommand = QtCore.pyqtSignal(QtCore.QString, QtCore.QString, QtCore.QString)
|
|
namesReceived = QtCore.pyqtSignal(QtCore.QString, PesterList)
|
|
channelListReceived = QtCore.pyqtSignal(PesterList)
|
|
nickCollision = QtCore.pyqtSignal(QtCore.QString, QtCore.QString)
|
|
myHandleChanged = QtCore.pyqtSignal(QtCore.QString)
|
|
chanInviteOnly = QtCore.pyqtSignal(QtCore.QString)
|
|
modesUpdated = QtCore.pyqtSignal(QtCore.QString, QtCore.QString)
|
|
connected = QtCore.pyqtSignal()
|
|
userPresentUpdate = QtCore.pyqtSignal(QtCore.QString, QtCore.QString,
|
|
QtCore.QString)
|
|
cannotSendToChan = QtCore.pyqtSignal(QtCore.QString, QtCore.QString)
|
|
|
|
class PesterHandler(DefaultCommandHandler):
|
|
def notice(self, nick, chan, msg):
|
|
try:
|
|
msg = msg.decode('utf-8')
|
|
except UnicodeDecodeError:
|
|
msg = msg.decode('iso-8859-1', 'ignore')
|
|
handle = nick[0:nick.find("!")]
|
|
logging.info("---> recv \"NOTICE %s :%s\"" % (handle, msg))
|
|
if handle == "ChanServ" and chan == self.parent.mainwindow.profile().handle and msg[0:2] == "[#":
|
|
self.parent.memoReceived.emit(msg[1:msg.index("]")], handle, msg)
|
|
else:
|
|
self.parent.noticeReceived.emit(handle, msg)
|
|
def privmsg(self, nick, chan, msg):
|
|
try:
|
|
msg = msg.decode('utf-8')
|
|
except UnicodeDecodeError:
|
|
msg = msg.decode('iso-8859-1', 'ignore')
|
|
# display msg, do other stuff
|
|
if len(msg) == 0:
|
|
return
|
|
# silently ignore CTCP
|
|
if msg[0] == '\x01':
|
|
handle = nick[0:nick.find("!")]
|
|
logging.info("---> recv \"CTCP %s :%s\"" % (handle, msg[1:-1]))
|
|
if msg[1:-1] == "VERSION":
|
|
helpers.notice(self.parent.cli, handle, "\x01VERSION Pesterchum %s\x01" % (_pcVersion))
|
|
return
|
|
handle = nick[0:nick.find("!")]
|
|
logging.info("---> recv \"PRIVMSG %s :%s\"" % (handle, msg))
|
|
if chan == "#pesterchum":
|
|
# follow instructions
|
|
if msg[0:6] == "MOOD >":
|
|
try:
|
|
mood = Mood(int(msg[6:]))
|
|
except ValueError:
|
|
mood = Mood(0)
|
|
self.parent.moodUpdated.emit(handle, mood)
|
|
elif msg[0:7] == "GETMOOD":
|
|
mychumhandle = self.mainwindow.profile().handle
|
|
mymood = self.mainwindow.profile().mood.value()
|
|
if msg.find(mychumhandle, 8) != -1:
|
|
helpers.msg(self.client, "#pesterchum",
|
|
"MOOD >%d" % (mymood))
|
|
elif chan[0] == '#':
|
|
if msg[0:16] == "PESTERCHUM:TIME>":
|
|
self.parent.timeCommand.emit(chan, handle, msg[16:])
|
|
else:
|
|
self.parent.memoReceived.emit(chan, handle, msg)
|
|
else:
|
|
# private message
|
|
# silently ignore messages to yourself.
|
|
if handle == self.mainwindow.profile().handle:
|
|
return
|
|
if msg[0:7] == "COLOR >":
|
|
colors = msg[7:].split(",")
|
|
try:
|
|
colors = [int(d) for d in colors]
|
|
except ValueError:
|
|
colors = [0,0,0]
|
|
color = QtGui.QColor(*colors)
|
|
self.parent.colorUpdated.emit(handle, color)
|
|
else:
|
|
self.parent.messageReceived.emit(handle, msg)
|
|
|
|
|
|
def welcome(self, server, nick, msg):
|
|
self.parent.setConnected()
|
|
helpers.join(self.client, "#pesterchum")
|
|
mychumhandle = self.mainwindow.profile().handle
|
|
mymood = self.mainwindow.profile().mood.value()
|
|
helpers.msg(self.client, "#pesterchum", "MOOD >%d" % (mymood))
|
|
|
|
def nicknameinuse(self, server, cmd, nick, msg):
|
|
newnick = "pesterClient%d" % (random.randint(100,999))
|
|
helpers.nick(self.client, newnick)
|
|
self.parent.nickCollision.emit(nick, newnick)
|
|
def quit(self, nick, reason):
|
|
handle = nick[0:nick.find("!")]
|
|
self.parent.userPresentUpdate.emit(handle, "", "quit")
|
|
self.parent.moodUpdated.emit(handle, Mood("offline"))
|
|
def kick(self, opnick, channel, handle, reason):
|
|
op = opnick[0:opnick.find("!")]
|
|
self.parent.userPresentUpdate.emit(handle, channel, "kick:%s:%s" % (op, reason))
|
|
# ok i shouldnt be overloading that but am lazy
|
|
def part(self, nick, channel, reason="nanchos"):
|
|
handle = nick[0:nick.find("!")]
|
|
self.parent.userPresentUpdate.emit(handle, channel, "left")
|
|
if channel == "#pesterchum":
|
|
self.parent.moodUpdated.emit(handle, Mood("offline"))
|
|
def join(self, nick, channel):
|
|
handle = nick[0:nick.find("!")]
|
|
self.parent.userPresentUpdate.emit(handle, channel, "join")
|
|
if channel == "#pesterchum":
|
|
self.parent.moodUpdated.emit(handle, Mood("chummy"))
|
|
def mode(self, op, channel, mode, *handles):
|
|
if len(handles) <= 0: handles = [""]
|
|
opnick = op[0:op.find("!")]
|
|
if op == channel or channel == self.parent.mainwindow.profile().handle:
|
|
modes = list(self.parent.mainwindow.modes)
|
|
if modes and modes[0] == "+": modes = modes[1:]
|
|
if mode[0] == "+":
|
|
for m in mode[1:]:
|
|
if m not in modes:
|
|
modes.extend(m)
|
|
elif mode[0] == "-":
|
|
for i in mode[1:]:
|
|
try:
|
|
modes.remove(i)
|
|
except ValueError:
|
|
pass
|
|
modes.sort()
|
|
self.parent.mainwindow.modes = "+" + "".join(modes)
|
|
modes = []
|
|
cur = "+"
|
|
for l in mode:
|
|
if l in ["+","-"]: cur = l
|
|
else:
|
|
modes.append("%s%s" % (cur, l))
|
|
for (i,m) in enumerate(modes):
|
|
try:
|
|
self.parent.userPresentUpdate.emit(handles[i], channel, m+":%s" % (op))
|
|
except IndexError:
|
|
self.parent.userPresentUpdate.emit("", channel, m+":%s" % (op))
|
|
def nick(self, oldnick, newnick):
|
|
oldhandle = oldnick[0:oldnick.find("!")]
|
|
if oldhandle == self.mainwindow.profile().handle:
|
|
self.parent.myHandleChanged.emit(newnick)
|
|
newchum = PesterProfile(newnick, chumdb=self.mainwindow.chumdb)
|
|
self.parent.moodUpdated.emit(oldhandle, Mood("offline"))
|
|
self.parent.userPresentUpdate.emit("%s:%s" % (oldhandle, newnick), "", "nick")
|
|
if newnick in self.mainwindow.chumList.chums:
|
|
self.getMood(newchum)
|
|
def namreply(self, server, nick, op, channel, names):
|
|
namelist = names.split(" ")
|
|
logging.info("---> recv \"NAMES %s: %d names\"" % (channel, len(namelist)))
|
|
if not hasattr(self, 'channelnames'):
|
|
self.channelnames = {}
|
|
if channel not in self.channelnames:
|
|
self.channelnames[channel] = []
|
|
self.channelnames[channel].extend(namelist)
|
|
def endofnames(self, server, nick, channel, msg):
|
|
namelist = self.channelnames[channel]
|
|
pl = PesterList(namelist)
|
|
del self.channelnames[channel]
|
|
self.parent.namesReceived.emit(channel, pl)
|
|
if channel == "#pesterchum" and not hasattr(self, "joined"):
|
|
self.joined = True
|
|
chums = self.mainwindow.chumList.chums
|
|
lesschums = []
|
|
for c in chums:
|
|
chandle = c.handle
|
|
if chandle in namelist:
|
|
lesschums.append(c)
|
|
self.getMood(*lesschums)
|
|
|
|
def liststart(self, server, handle, *info):
|
|
self.channel_list = []
|
|
info = list(info)
|
|
self.channel_field = info.index("Channel") # dunno if this is protocol
|
|
logging.info("---> recv \"CHANNELS: %s " % (self.channel_field))
|
|
def list(self, server, handle, *info):
|
|
channel = info[self.channel_field]
|
|
usercount = info[1]
|
|
if channel not in self.channel_list and channel != "#pesterchum":
|
|
self.channel_list.append((channel, usercount))
|
|
logging.info("---> recv \"CHANNELS: %s " % (channel))
|
|
def listend(self, server, handle, msg):
|
|
pl = PesterList(self.channel_list)
|
|
logging.info("---> recv \"CHANNELS END\"")
|
|
self.parent.channelListReceived.emit(pl)
|
|
self.channel_list = []
|
|
|
|
def umodeis(self, server, handle, modes):
|
|
self.parent.mainwindow.modes = modes
|
|
def invite(self, sender, you, channel):
|
|
handle = sender.split('!')[0]
|
|
self.parent.inviteReceived.emit(handle, channel)
|
|
def inviteonlychan(self, server, handle, channel, msg):
|
|
self.parent.chanInviteOnly.emit(channel)
|
|
def channelmodeis(self, server, handle, channel, modes):
|
|
self.parent.modesUpdated.emit(channel, modes)
|
|
def cannotsendtochan(self, server, handle, channel, msg):
|
|
self.parent.cannotSendToChan.emit(channel, msg)
|
|
|
|
def getMood(self, *chums):
|
|
chumglub = "GETMOOD "
|
|
for c in chums:
|
|
chandle = c.handle
|
|
if len(chumglub+chandle) >= 350:
|
|
try:
|
|
helpers.msg(self.client, "#pesterchum", chumglub)
|
|
except socket.error:
|
|
self.parent.setConnectionBroken()
|
|
chumglub = "GETMOOD "
|
|
chumglub += chandle
|
|
if chumglub != "GETMOOD ":
|
|
try:
|
|
helpers.msg(self.client, "#pesterchum", chumglub)
|
|
except socket.error:
|
|
self.parent.setConnectionBroken()
|
|
|