examples/network/network-chat/client.cpp
changeset 0 1918ee327afb
child 3 41300fa6a67c
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/examples/network/network-chat/client.cpp	Mon Jan 11 14:00:40 2010 +0000
@@ -0,0 +1,140 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
+** All rights reserved.
+** Contact: Nokia Corporation (qt-info@nokia.com)
+**
+** This file is part of the examples of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL$
+** No Commercial Usage
+** This file contains pre-release code and may not be distributed.
+** You may use this file in accordance with the terms and conditions
+** contained in the Technology Preview License Agreement accompanying
+** this package.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU Lesser General Public License version 2.1 requirements
+** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Nokia gives you certain additional
+** rights.  These rights are described in the Nokia Qt LGPL Exception
+** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
+**
+** If you have questions regarding the use of this file, please contact
+** Nokia at qt-info@nokia.com.
+**
+**
+**
+**
+**
+**
+**
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include <QtNetwork>
+
+#include "client.h"
+#include "connection.h"
+#include "peermanager.h"
+
+Client::Client()
+{
+    peerManager = new PeerManager(this);
+    peerManager->setServerPort(server.serverPort());
+    peerManager->startBroadcasting();
+
+    QObject::connect(peerManager, SIGNAL(newConnection(Connection *)),
+                     this, SLOT(newConnection(Connection *)));
+    QObject::connect(&server, SIGNAL(newConnection(Connection *)),
+                     this, SLOT(newConnection(Connection *)));
+}
+
+void Client::sendMessage(const QString &message)
+{
+    if (message.isEmpty())
+        return;
+
+    QList<Connection *> connections = peers.values();
+    foreach (Connection *connection, connections)
+        connection->sendMessage(message);
+}
+
+QString Client::nickName() const
+{
+    return QString(peerManager->userName()) + "@" + QHostInfo::localHostName()
+           + ":" + QString::number(server.serverPort());
+}
+
+bool Client::hasConnection(const QHostAddress &senderIp, int senderPort) const
+{
+    if (senderPort == -1)
+        return peers.contains(senderIp);
+
+    if (!peers.contains(senderIp))
+        return false;
+
+    QList<Connection *> connections = peers.values(senderIp);
+    foreach (Connection *connection, connections) {
+        if (connection->peerPort() == senderPort)
+            return true;
+    }
+
+    return false;
+}
+
+void Client::newConnection(Connection *connection)
+{
+    connection->setGreetingMessage(peerManager->userName());
+
+    connect(connection, SIGNAL(error(QAbstractSocket::SocketError)),
+            this, SLOT(connectionError(QAbstractSocket::SocketError)));
+    connect(connection, SIGNAL(disconnected()), this, SLOT(disconnected()));
+    connect(connection, SIGNAL(readyForUse()), this, SLOT(readyForUse()));
+}
+
+void Client::readyForUse()
+{
+    Connection *connection = qobject_cast<Connection *>(sender());
+    if (!connection || hasConnection(connection->peerAddress(),
+                                     connection->peerPort()))
+        return;
+
+    connect(connection, SIGNAL(newMessage(const QString &, const QString &)),
+            this, SIGNAL(newMessage(const QString &, const QString &)));
+
+    peers.insert(connection->peerAddress(), connection);
+    QString nick = connection->name();
+    if (!nick.isEmpty())
+        emit newParticipant(nick);
+}
+
+void Client::disconnected()
+{
+    if (Connection *connection = qobject_cast<Connection *>(sender()))
+        removeConnection(connection);
+}
+
+void Client::connectionError(QAbstractSocket::SocketError /* socketError */)
+{
+    if (Connection *connection = qobject_cast<Connection *>(sender()))
+        removeConnection(connection);
+}
+
+void Client::removeConnection(Connection *connection)
+{
+    if (peers.contains(connection->peerAddress())) {
+        peers.remove(connection->peerAddress());
+        QString nick = connection->name();
+        if (!nick.isEmpty())
+            emit participantLeft(nick);
+    }
+    connection->deleteLater();
+}