3ece5e30b7
Where possible Task-number: QTBUG-108873 Change-Id: I8125ffd63cd0ad1970575fb1a6b85021c03c38d5 Reviewed-by: Timur Pocheptsov <timur.pocheptsov@qt.io> Reviewed-by: Konrad Kujawa <konrad.kujawa@qt.io>
101 lines
2.8 KiB
C++
101 lines
2.8 KiB
C++
// Copyright (C) 2016 The Qt Company Ltd.
|
|
// SPDX-License-Identifier: LicenseRef-Qt-Commercial OR BSD-3-Clause
|
|
|
|
#include "client.h"
|
|
#include "connection.h"
|
|
#include "peermanager.h"
|
|
|
|
#include <QHostInfo>
|
|
|
|
Client::Client()
|
|
: peerManager(new PeerManager(this))
|
|
{
|
|
peerManager->setServerPort(server.serverPort());
|
|
peerManager->startBroadcasting();
|
|
|
|
connect(peerManager, &PeerManager::newConnection,
|
|
this, &Client::newConnection);
|
|
connect(&server, &Server::newConnection,
|
|
this, &Client::newConnection);
|
|
}
|
|
|
|
void Client::sendMessage(const QString &message)
|
|
{
|
|
if (message.isEmpty())
|
|
return;
|
|
|
|
for (Connection *connection : std::as_const(peers))
|
|
connection->sendMessage(message);
|
|
}
|
|
|
|
QString Client::nickName() const
|
|
{
|
|
return 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;
|
|
|
|
const QList<Connection *> connections = peers.values(senderIp);
|
|
for (const Connection *connection : connections) {
|
|
if (connection->peerPort() == senderPort)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void Client::newConnection(Connection *connection)
|
|
{
|
|
connection->setGreetingMessage(peerManager->userName());
|
|
|
|
connect(connection, &Connection::errorOccurred, this, &Client::connectionError);
|
|
connect(connection, &Connection::disconnected, this, &Client::disconnected);
|
|
connect(connection, &Connection::readyForUse, this, &Client::readyForUse);
|
|
}
|
|
|
|
void Client::readyForUse()
|
|
{
|
|
Connection *connection = qobject_cast<Connection *>(sender());
|
|
if (!connection || hasConnection(connection->peerAddress(),
|
|
connection->peerPort()))
|
|
return;
|
|
|
|
connect(connection, &Connection::newMessage,
|
|
this, &Client::newMessage);
|
|
|
|
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();
|
|
}
|