def _initMessageModel(self):
self._model = ChatMessagesModel(self, self)
class ChatWidget(QWidget):
PREFERRED_WIDTH = 400
_URI_REGEX="""
(
(
[A-Za-z][A-Za-z0-9\+\.\-]*:\/\/
[A-Za-z0-9\.\-]+
|
www\.
[A-Za-z0-9\.\-]+
\.[A-Za-z]+
)
(?::\d+)?
(
(?:\/[\+~%\/\.\w\-=]*)
?\??(?:[\-\+=&;%@\.\w]*)
#?(?:[\.\!\/\\w]*)
)?
)
""".replace("\n", "").replace(" ", "")
_MAIL_REGEX="""
(
(
[\-;:&=\+\$,\w\.]+@
[A-Za-z0-9\.\-]+
\.[A-Za-z]+
)
)
""".replace("\n", "").replace(" ", "")
_MD_LINK_REGEX="\[[^]]+\]\([^)]+\)"
_URI_MATCHER = QRegExp(_URI_REGEX)
_MAIL_MATCHER = QRegExp(_MAIL_REGEX)
_MD_LINK_MATCHER = QRegExp(_MD_LINK_REGEX)
_TIME_ROW_INTERVAL = 10*60 # once every 10 minutes
sendMessage = pyqtSignal(object, object) # peer ID, message HTML
typing = pyqtSignal()
cleared = pyqtSignal()
def __init__(self, parent, logger, ownName, otherName, ownPicFile, otherPicFile, otherID, sendAction):
super(ChatWidget, self).__init__(parent)
self.logger = logger
self._firstShowEvent = True
self._offline = False
if sendAction is not None:
self._blocked = sendAction.getPeerState(otherID) == PrivacySettings.STATE_BLOCKED
else:
self._blocked = False
self._delivering = False
self._lastTimeRow = 0
self._textChanged = False
self._keepEntryText = False
self._markdownEnabled = False
self._md = None
self._typingTimer = QTimer(self)
self._typingTimer.timeout.connect(self._checkTyping)
self._typingTimer.start(1000)
self._entryWasEmpty = True
self._selfWasTyping = False
self._otherWasTyping = False
self._lastTimeSelfTyped = None
self._lastTimePartnerTyped = None
self._otherID = otherID
self._otherName = otherName
self._ownName = ownName
self._sendAction = sendAction
self._errIcon = None
self._warnIcon = None
try:
from PyQt4.QtGui import QCommonStyle, QStyle
style = QCommonStyle()
self._errIcon = style.standardIcon(QStyle.SP_MessageBoxCritical)
if self._errIcon.isNull():
self._errIcon = None
self._warnIcon = style.standardIcon(QStyle.SP_MessageBoxWarning)
if self._warnIcon.isNull():
self._warnIcon = None
except:
pass
if self._errIcon is None:
self._errIcon = QIcon(get_settings().get_resource("images", "error.png"))
if self._warnIcon is None:
self._warnIcon = QIcon(get_settings().get_resource("images", "warning.png"))
self._initMessageModel()
self._initMessageTable()
self._initTextEntry()
mainLayout = QVBoxLayout(self)
mainLayout.setSpacing(0)
self._addTopLayout(mainLayout)
mainLayout.addWidget(self.table)
mainLayout.addWidget(self.entry)
# initialize GUI
self._updateOtherName()
self._updateOwnName()
self.setOwnIconPath(ownPicFile)
self.setOtherIconPath(otherPicFile)
# TODO option to change behavior
self.entry.returnPressed.connect(self.eventTriggered)
self.setSizePolicy(QSizePolicy.Minimum, QSizePolicy.MinimumExpanding)
get_notification_center().connectPeerAppended(self._peerUpdated)
get_notification_center().connectPeerUpdated(self._peerUpdated)
get_notification_center().connectPeerRemoved(self._peerRemoved)
get_notification_center().connectAvatarChanged(self._avatarChanged)
get_notification_center().connectDisplayedPeerNameChanged(self._displayedPeerNameChanged)
get_notification_center().connectPrivacySettingsChanged(self._privacySettingsChanged)
if get_peers() != None:
self._setOffline(not get_peers().isPeerID(pID=self._otherID))
else:
self._setOffline(True)
self.setAcceptDrops(True)
def setMarkdownEnabled(self, en):
self._markdownEnabled = en
def _canSendFilesToOther(self):
peerInfo = get_peers().getPeerInfo(pID=self._otherID)
if peerInfo is None:
return False
action = PeerActions.get().getPeerActionByName(u"hannesrauhe.lunchinator.file_transfer", u"Send File")
if action is None or not action.appliesToPeer(self._otherID, peerInfo):
return False
return True
def dragEnterEvent(self, event):
if not self._canSendFilesToOther():
event.ignore()
return
if event.mimeData().hasUrls() and int(event.possibleActions()) & Qt.CopyAction:
for url in event.mimeData().urls():
if url.scheme() == u"file":
event.setDropAction(Qt.CopyAction)
event.accept()
return
event.ignore()
def dropEvent(self, event):
if not self._canSendFilesToOther():
event.ignore()
return
if event.mimeData().hasUrls() and int(event.possibleActions()) & Qt.CopyAction:
files = []
for url in event.mimeData().urls():
if url.scheme() == u"file":
files.append(convert_string(url.toLocalFile()))
if len(files) > 0:
action = PeerActions.get().getPeerActionByName(u"hannesrauhe.lunchinator.file_transfer", u"Send File")
action.sendFilesToPeer(files, self._otherID)
event.accept()
return
event.ignore()
def _setOffline(self, offline):
if self._offline == offline:
return
self._offline = offline
self._updateOtherName()
def _updateOtherName(self):
if self._offline:
title = self._otherName + " (Offline)"
elif self._blocked:
title = self._otherName + " (Blocked)"
else:
title = self._otherName
self._otherPicLabel.setEnabled(not self._offline and not self._blocked)
self._otherNameLabel.setText(title)
self.parent().setWindowTitle(title)
self._checkEntryState()
def _updateOwnName(self):
self._ownNameLabel.setText(self._ownName)
@loggingSlot(object, object)
def _peerUpdated(self, peerID, peerInfo):
peerID = convert_string(peerID)
if peerID == self._otherID:
self._setOffline(u"PM_v" not in peerInfo)
@loggingSlot(object)
def _peerRemoved(self, peerID):
peerID = convert_string(peerID)
if peerID == self._otherID:
self._setOffline(True)
@loggingSlot(object, object)
def _avatarChanged(self, peerID, newFile):
peerID = convert_string(peerID)
newFile = convert_string(newFile)
if peerID == self._otherID:
self.setOtherIconPath(get_peers().getPeerAvatarFile(pID=peerID))
if peerID == get_settings().get_ID():
self.setOwnIconPath(get_peers().getPeerAvatarFile(pID=peerID))
@loggingSlot(object, object, object)
def _displayedPeerNameChanged(self, pID, newName, _infoDict):
pID = convert_string(pID)
newName = convert_string(newName)
if pID == self._otherID:
self._otherName = newName
self._updateOtherName()
if pID == get_settings().get_ID():
self._ownName = newName
self._updateOwnName()
@loggingSlot(object, object)
def _privacySettingsChanged(self, pluginName, actionName):
if self._sendAction is None:
return
if pluginName == self._sendAction.getPluginName() and \
actionName == self._sendAction.getName():
blocked = self._sendAction.getPeerState(self._otherID) == PrivacySettings.STATE_BLOCKED
if blocked != self._blocked:
self._blocked = blocked
self._updateOtherName()
def _clearEntry(self):
self.entry.clear()
self.entry.setCurrentCharFormat(QTextCharFormat())
def _checkEntryState(self):
self.entry.setEnabled(not self._offline and not self._delivering and not self._blocked)
if self._offline or self._blocked:
if self.entry.document().isEmpty():
self._clearEntry()
if self._offline:
self.entry.setText(u"Partner is offline")
else:
self.entry.setText(u"Partner is blocked")
else:
self._keepEntryText = True
elif self._delivering:
self._clearEntry()
self.entry.setText(u"Delivering...")
elif not self._keepEntryText:
self._clearEntry()
if self._keepEntryText and (not self._offline and not self._blocked):
# reset if not offline any more
self._keepEntryText = False
def nextInFocusChain(self):
return self.entry
def previousInFocusChain(self):
return self.entry
def focusInEvent(self, _event):
self.entry.setFocus(Qt.OtherFocusReason)
def _filterPeerAction(self, pluginName, action):
return pluginName != u"hannesrauhe.lunchinator.private_messages" or action.getName() != "Open Chat"
def _addTopLayout(self, mainLayout):
topWidget = QWidget(self)
topLayout = QHBoxLayout(topWidget)
topLayout.setSpacing(0)
topLayout.setContentsMargins(0, 0, 0, 0)
self._otherNameLabel = QToolButton(topWidget)
self._otherNameLabel.setToolButtonStyle(Qt.ToolButtonTextOnly)
self._otherNameLabel.setStyleSheet("QToolButton { text-align: left; font-size: 13pt; border: none; margin-left: -5px; padding-right:5px; padding-bottom: -2px;}")
self._otherNameLabel.setContextMenuPolicy(Qt.CustomContextMenu)
self._otherNameLabel.customContextMenuRequested.connect(partial(showPeerActionsPopup, self._otherID, self._filterPeerAction, self._otherNameLabel))
self._otherNameLabel.setPopupMode(QToolButton.InstantPopup)
menu = QMenu(self._otherNameLabel)
menu.aboutToShow.connect(partial(initializePeerActionsMenu, menu, self._otherID, self._filterPeerAction, self))
self._otherNameLabel.setMenu(menu)
self._otherPicLabel = QLabel(topWidget)
topLayout.addWidget(self._otherPicLabel, 0, Qt.AlignLeft)
topLayout.addSpacing(5)
topLayout.addWidget(self._otherNameLabel, 0, Qt.AlignLeft)
self._otherStatusLabel = QLabel(topWidget)
topLayout.addSpacing(2)
topLayout.addWidget(self._otherStatusLabel, 1, Qt.AlignLeft)
self._ownNameLabel = QToolButton(topWidget)
self._ownNameLabel.setStyleSheet("QToolButton { text-align: left; font-size: 13pt; border: none; margin-right: -5px;}")
self._ownNameLabel.setToolButtonStyle(Qt.ToolButtonTextOnly)
self._ownPicLabel = QLabel(topWidget)
topLayout.addWidget(self._ownNameLabel, 1, Qt.AlignRight)
topLayout.addSpacing(5)
topLayout.addWidget(self._ownPicLabel, 0, Qt.AlignRight)
mainLayout.addWidget(topWidget)
separator = QFrame(self)
separator.setFrameShape(QFrame.HLine)
separator.setFrameShadow(QFrame.Sunken)
mainLayout.addSpacing(5)
mainLayout.addWidget(separator)
mainLayout.addSpacing(5)
def _initTextEntry(self):
self.entry = HistoryTextEdit(self, True)
self.entry.textChanged.connect(self._textChangedSlot)
def _initMessageModel(self):
self._model = ChatMessagesModel(self, self)
def _initMessageTable(self):
self.table = ChatMessagesView(self._model, self, self.logger)
self.table.setContextMenuPolicy(Qt.CustomContextMenu)
self.table.customContextMenuRequested.connect(self._showContextMenu)
@loggingSlot(QPoint)
def _showContextMenu(self, point):
index = self.table.indexAt(point)
if index != None:
if index.column() != ChatMessagesModel.MESSAGE_COLUMN:
return
isOwn = index.data(ChatMessagesModel.OWN_MESSAGE_ROLE).toBool()
menu = QMenu(self)
if isOwn:
menu.addAction(u"Send again", partial(self._sendAgain, index))
menu.addAction(u"Copy", partial(self._copy, index))
menu.exec_(QCursor.pos())
menu.deleteLater()
@loggingSlot()
def _sendAgain(self, index):
self.sendMessage.emit(self._otherID, convert_string(index.data(Qt.DisplayRole).toString()))
@loggingSlot()
def _copy(self, index):
doc = QTextDocument()
doc.setHtml(index.data(Qt.DisplayRole).toString())
clipboard = QApplication.clipboard()
richTextData = QMimeData()
richTextData.setHtml(index.data(Qt.DisplayRole).toString())
richTextData.setText(doc.toPlainText())
clipboard.setMimeData(richTextData)
@loggingSlot()
def _textChangedSlot(self):
if not self.entry.isEnabled() or self.entry.document().isEmpty():
self._textChanged = False
if not self._entryWasEmpty:
self._entryWasEmpty = True
self._selfWasTyping = False
self._informCleared()
elif not self._selfWasTyping:
self._entryWasEmpty = False
self._informTyping()
self._selfWasTyping = True
self._lastTimeSelfTyped = time()
else:
self._textChanged = True
@loggingSlot()
def _checkTyping(self):
curTime = time()
if self._textChanged:
self._informTyping()
# TODO do we really need thread safety here?
self._textChanged = False
self._lastTimeSelfTyped = curTime
elif self._selfWasTyping and curTime - self._lastTimeSelfTyped > 3:
self._selfWasTyping = False
if self._otherWasTyping and curTime - self._lastTimePartnerTyped > 3:
self.setStatus("paused typing")
self._otherWasTyping = False
def _informTyping(self):
if not self._offline and not self._blocked:
self.typing.emit()
def _informCleared(self):
if not self._offline:
self.cleared.emit()
@loggingSlot()
def otherIsTyping(self):
if not self._otherWasTyping:
self._otherWasTyping = True
self.setStatus("typing...")
self._lastTimePartnerTyped = time()
@loggingSlot()
def otherCleared(self):
self._otherWasTyping = False
self.setStatus(None)
def showEvent(self, event):
QWidget.showEvent(self, event)
if self._firstShowEvent:
self._firstShowEvent = False
# relayout (i don't know why this is necessary)
self.table.reset()
def addTimeRow(self, rtime):
self._model.addTimeRow(rtime)
def _checkTime(self, msgTime):
if msgTime - self._lastTimeRow > self._TIME_ROW_INTERVAL:
self.addTimeRow(msgTime)
self._lastTimeRow = msgTime
@loggingFunc
def addOwnMessage(self, msgID, recvTime, msg, msgTime, messageState=None, toolTip=None):
self._checkTime(msgTime)
self._model.addOwnMessage(msgID, recvTime, msg, msgTime, messageState, toolTip)
self.entry.clear()
self._delivering = False
self._checkEntryState()
self.entry.setFocus(Qt.OtherFocusReason)
self.table.setScrollToEnd(True)
def addOtherMessage(self, msg, msgTime, recvTime):
self._checkTime(msgTime)
self._model.addOtherMessage(msg, msgTime, recvTime)
def delayedDelivery(self, msgID, recvTime, error, errorMessage):
return self._model.messageDelivered(msgID, recvTime, error, errorMessage)
def messageIDChanged(self, oldID, newID):
self._model.messageIDChanged(oldID, newID)
def canClose(self):
return not self._delivering
def finish(self):
get_notification_center().disconnectPeerAppended(self._peerUpdated)
get_notification_center().disconnectPeerUpdated(self._peerUpdated)
get_notification_center().disconnectPeerRemoved(self._peerRemoved)
get_notification_center().disconnectAvatarChanged(self._avatarChanged)
get_notification_center().disconnectDisplayedPeerNameChanged(self._displayedPeerNameChanged)
if self._typingTimer is not None:
self._typingTimer.stop()
self._typingTimer.deleteLater()
self._typingTimer = None
def getOwnIcon(self):
return self._ownIcon
def getOwnIconPath(self):
return self._ownIconPath
def setOwnIconPath(self, iconPath):
if not iconPath:
iconPath = get_settings().get_resource("images", "me.png")
self._ownIconPath = iconPath
self._ownIcon = QIcon(iconPath)
self._ownPicLabel.setPixmap(QPixmap(self._ownIconPath).scaled(24,24, Qt.KeepAspectRatio, Qt.SmoothTransformation))
self._model.setOwnIcon(self._ownIcon)
def getOtherIcon(self):
return self._otherIcon
def getOtherIconPath(self):
return self._otherIconPath
def setOtherIconPath(self, iconPath):
if not iconPath:
iconPath = get_settings().get_resource("images", "lunchinator.png")
self._otherIconPath = iconPath
self._otherIcon = QIcon(iconPath)
self._otherPicLabel.setPixmap(QPixmap(self._otherIconPath).scaled(24,24, Qt.KeepAspectRatio, Qt.SmoothTransformation))
self._model.setOtherIcon(self._otherIcon)
def getOtherName(self):
return self._otherName
def getWarnIcon(self):
return self._warnIcon
def getErrorIcon(self):
return self._errIcon
def setStatus(self, statusText):
if statusText:
title = u"(%s)" % (statusText)
else:
title = u""
self._otherStatusLabel.setText(title)
def _getMD(self):
if self._md is None:
try:
from markdown import Markdown
if getPlatform()==PLATFORM_WINDOWS and isPyinstallerBuild():
self._md = Markdown()
else:
self._md = Markdown(extensions=['extra'])
except ImportError:
self.logger.error("Cannot enable Markdown (%s)", formatException())
raise
return self._md
@loggingSlot()
def eventTriggered(self):
if self.entry.toPlainText().trimmed().length() is 0:
return
text = None
if self._markdownEnabled:
try:
md = self._getMD()
text = self._detectHyperlinks(True)
text = md.convert(text)
except ImportError:
pass
if text is None:
# fallback to default method
self._detectHyperlinks()
text = self._cleanHTML(convert_string(self.entry.toHtml()))
self.sendMessage.emit(self._otherID, text)
self._delivering = True
self._checkEntryState()
def _cleanHTML(self, html):
# only body, no paragraph attributes
html = html.encode("utf-8")
cleaned = u""
e = ElementTree.fromstring(html)
body = e.getiterator("html")[0].getiterator("body")[0]
for p in body.getiterator("p"):
p.attrib = {}
sio = StringIO()
ElementTree.ElementTree(p).write(sio, "utf-8")
cleaned += sio.getvalue().replace('<br />', '').decode("utf-8")
return cleaned
def _iterMatchedRanges(self, matcher, text):
pos = 0
while pos != -1:
pos = matcher.indexIn(text, pos)
if pos == -1:
break
yield pos, pos + matcher.matchedLength(), matcher
pos += matcher.matchedLength()
def _insertAnchors(self, cursor, plainText, matcher, hrefFunc):
for start, end, matcher in self._iterMatchedRanges(matcher, plainText):
cursor.setPosition(start);
cursor.setPosition(end, QTextCursor.KeepAnchor)
fmt = QTextCharFormat()
fmt.setAnchor(True)
fmt.setAnchorHref(hrefFunc(matcher.cap()))
cursor.mergeCharFormat(fmt)
def _findPlainLinks(self, plainText, matcher, hrefFunc, mdLinks):
newLinks = []
for start, end, matcher in self._iterMatchedRanges(matcher, plainText):
if start in mdLinks:
# is already a markdown link
continue
newLinks.append((start, end, hrefFunc(matcher.cap())))
return newLinks
def _detectHyperlinks(self, toMarkdown=False):
cursor = QTextCursor(self.entry.document())
plainText = self.entry.toPlainText()
makeURL = lambda uri : u"http://" + convert_string(uri) if uri.startsWith(u"www.") else uri
makeMail = lambda mail : u"mailto:" + convert_string(mail)
if toMarkdown:
# find markdown links
mdLinks = IndexSet()
for start, end, _matcher in self._iterMatchedRanges(self._MD_LINK_MATCHER, plainText):
mdLinks.addRange(start, end)
# replace plain links with markdown links
plainLinks = self._findPlainLinks(plainText,
self._URI_MATCHER,
makeURL,
mdLinks)
plainLinks += self._findPlainLinks(plainText,
self._MAIL_MATCHER,
makeMail,
mdLinks)
plainLinks = sorted(plainLinks, key=lambda aTup : aTup[0])
plainText = convert_string(plainText)
for start, end, linkTarget in reversed(plainLinks):
mdLink = "[%s](%s)" % (plainText[start:end], linkTarget)
plainText = plainText[:start] + mdLink + plainText[end:]
return plainText
else:
self._insertAnchors(cursor,
plainText,
self._URI_MATCHER,
makeURL)
self._insertAnchors(cursor,
plainText,
self._MAIL_MATCHER,
makeMail)
def sizeHint(self):
sizeHint = QWidget.sizeHint(self)
return QSize(self.PREFERRED_WIDTH, sizeHint.height())