1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
#include "websocketconnection.h"
#include "aoapplication.h"
#include <QNetworkRequest>
#include <QUrl>
WebSocketConnection::WebSocketConnection(AOApplication *ao_app, QObject *parent)
: QObject(parent)
, ao_app(ao_app)
, m_socket(new QWebSocket(QString(), QWebSocketProtocol::VersionLatest, this))
, m_last_state(QAbstractSocket::UnconnectedState)
{
connect(m_socket, &QWebSocket::errorOccurred, this, &WebSocketConnection::onError);
connect(m_socket, &QWebSocket::stateChanged, this, &WebSocketConnection::onStateChanged);
connect(m_socket, &QWebSocket::textMessageReceived, this, &WebSocketConnection::onTextMessageReceived);
}
WebSocketConnection::~WebSocketConnection()
{
m_socket->disconnect(this);
disconnectFromServer();
}
bool WebSocketConnection::isConnected()
{
return m_last_state == QAbstractSocket::ConnectedState;
}
void WebSocketConnection::connectToServer(const ServerInfo &server)
{
disconnectFromServer();
QUrl url;
url.setScheme(server.protocol);
url.setHost(server.address);
url.setPort(server.port);
QNetworkRequest req(url);
req.setHeader(QNetworkRequest::UserAgentHeader, QStringLiteral("AttorneyOnline/%1 (Desktop)").arg(ao_app->get_version_string()));
m_socket->open(req);
}
void WebSocketConnection::disconnectFromServer()
{
if (isConnected())
{
m_socket->close(QWebSocketProtocol::CloseCodeGoingAway);
}
}
void WebSocketConnection::sendPacket(AOPacket packet)
{
m_socket->sendTextMessage(packet.toString(true));
}
void WebSocketConnection::onError()
{
Q_EMIT errorOccurred(m_socket->errorString());
}
void WebSocketConnection::onStateChanged(QAbstractSocket::SocketState state)
{
m_last_state = state;
switch (state)
{
default:
break;
case QAbstractSocket::ConnectedState:
Q_EMIT connectedToServer();
break;
case QAbstractSocket::UnconnectedState:
Q_EMIT disconnectedFromServer();
break;
}
}
void WebSocketConnection::onTextMessageReceived(QString message)
{
if (!message.endsWith("#%"))
{
return;
}
message.chop(2);
QStringList raw_content = message.split('#');
const QString header = raw_content.takeFirst();
for (QString &data : raw_content)
{
data = AOPacket::decode(data);
}
Q_EMIT receivedPacket(AOPacket(header, raw_content));
}
|
