Add an empty weapon to avoid selection of weapons which aren't yet ready. Might all be useful to switch to amNothing in certain situations, like after using up all ropes, instead of bazooka.
/*
* Hedgewars, a free turn based strategy game
* Copyright (c) 2006-2008 Ulyanov Igor <iulyanov@gmail.com>
* Copyright (c) 2009 Andrey Korotaev <unC0Rr@gmail.com>
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; version 2 of the License
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
*/
#include <QDebug>
#include <QInputDialog>
#include <QCryptographicHash>
#include "hwconsts.h"
#include "newnetclient.h"
#include "proto.h"
#include "gameuiconfig.h"
#include "game.h"
#include "gamecfgwidget.h"
#include "teamselect.h"
#include "misc.h"
char delimeter='\n';
HWNewNet::HWNewNet(GameUIConfig * config, GameCFGWidget* pGameCFGWidget, TeamSelWidget* pTeamSelWidget) :
config(config),
m_pGameCFGWidget(pGameCFGWidget),
m_pTeamSelWidget(pTeamSelWidget),
isChief(false),
m_game_connected(false),
loginStep(0),
netClientState(0)
{
// socket stuff
connect(&NetSocket, SIGNAL(readyRead()), this, SLOT(ClientRead()));
connect(&NetSocket, SIGNAL(connected()), this, SLOT(OnConnect()));
connect(&NetSocket, SIGNAL(disconnected()), this, SLOT(OnDisconnect()));
connect(&NetSocket, SIGNAL(error(QAbstractSocket::SocketError)), this,
SLOT(displayError(QAbstractSocket::SocketError)));
// config stuff
connect(this, SIGNAL(paramChanged(const QString &, const QStringList &)), pGameCFGWidget, SLOT(setParam(const QString &, const QStringList &)));
connect(pGameCFGWidget, SIGNAL(paramChanged(const QString &, const QStringList &)), this, SLOT(onParamChanged(const QString &, const QStringList &)));
connect(this, SIGNAL(configAsked()), pGameCFGWidget, SLOT(fullNetConfig()));
}
HWNewNet::~HWNewNet()
{
if (m_game_connected)
{
RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));
emit Disconnected();
}
NetSocket.flush();
}
void HWNewNet::Connect(const QString & hostName, quint16 port, const QString & nick)
{
mynick = nick.isEmpty() ? QLineEdit::tr("unnamed") : nick;
NetSocket.connectToHost(hostName, port);
}
void HWNewNet::Disconnect()
{
if (m_game_connected)
RawSendNet(QString("QUIT%1%2").arg(delimeter).arg("User quit"));
m_game_connected = false;
NetSocket.disconnectFromHost();
}
void HWNewNet::CreateRoom(const QString & room)
{
if(netClientState != 2)
{
qWarning("Illegal try to create room!");
return;
}
RawSendNet(QString("CREATE_ROOM%1%2").arg(delimeter).arg(room));
isChief = true;
}
void HWNewNet::JoinRoom(const QString & room)
{
if(netClientState != 2)
{
qWarning("Illegal try to join room!");
return;
}
RawSendNet(QString("JOIN_ROOM%1%2").arg(delimeter).arg(room));
isChief = false;
}
void HWNewNet::AddTeam(const HWTeam & team)
{
QString cmd = QString("ADD_TEAM") + delimeter +
team.TeamName + delimeter +
team.teamColor.name() + delimeter +
team.Grave + delimeter +
team.Fort + delimeter +
team.Voicepack + delimeter +
QString::number(team.difficulty);
for(int i = 0; i < 8; ++i)
{
cmd.append(delimeter);
cmd.append(team.HHName[i]);
cmd.append(delimeter);
cmd.append(team.HHHat[i]);
}
RawSendNet(cmd);
}
void HWNewNet::RemoveTeam(const HWTeam & team)
{
RawSendNet(QString("REMOVE_TEAM") + delimeter + team.TeamName);
}
void HWNewNet::ToggleReady()
{
RawSendNet(QString("TOGGLE_READY"));
}
void HWNewNet::SendNet(const QByteArray & buf)
{
QString msg = QString(buf.toBase64());
RawSendNet(QString("EM%1%2").arg(delimeter).arg(msg));
}
void HWNewNet::RawSendNet(const QString & str)
{
RawSendNet(str.toUtf8());
}
void HWNewNet::RawSendNet(const QByteArray & buf)
{
qDebug() << "Client: " << QString(buf).split("\n");
NetSocket.write(buf);
NetSocket.write("\n\n", 2);
}
void HWNewNet::ClientRead()
{
while (NetSocket.canReadLine()) {
QString s = QString::fromUtf8(NetSocket.readLine());
if (s.endsWith('\n')) s.chop(1);
if (s.size() == 0) {
ParseCmd(cmdbuf);
cmdbuf.clear();
} else
cmdbuf << s;
}
}
void HWNewNet::OnConnect()
{
}
void HWNewNet::OnDisconnect()
{
if(m_game_connected) emit Disconnected();
m_game_connected = false;
}
void HWNewNet::displayError(QAbstractSocket::SocketError socketError)
{
emit Disconnected();
switch (socketError) {
case QAbstractSocket::RemoteHostClosedError:
break;
case QAbstractSocket::HostNotFoundError:
emit showMessage(tr("The host was not found. Please check the host name and port settings."));
break;
case QAbstractSocket::ConnectionRefusedError:
emit showMessage(tr("Connection refused"));
break;
default:
emit showMessage(NetSocket.errorString());
}
}
void HWNewNet::ParseCmd(const QStringList & lst)
{
qDebug() << "Server: " << lst;
if(!lst.size())
{
qWarning("Net client: Bad message");
return;
}
if (lst[0] == "NICK")
{
mynick = lst[1];
return ;
}
if (lst[0] == "PROTO")
return ;
if (lst[0] == "ERROR") {
if (lst.size() == 2)
emit showMessage("Error: " + lst[1]);
else
emit showMessage("Unknown error");
return;
}
if (lst[0] == "WARNING") {
if (lst.size() == 2)
emit showMessage("Warning: " + lst[1]);
else
emit showMessage("Unknown warning");
return;
}
if (lst[0] == "CONNECTED") {
RawSendNet(QString("NICK%1%2").arg(delimeter).arg(mynick));
RawSendNet(QString("PROTO%1%2").arg(delimeter).arg(*cProtoVer));
netClientState = 1;
m_game_connected = true;
emit adminAccess(false);
return;
}
if (lst[0] == "PING") {
if (lst.size() > 1)
RawSendNet(QString("PONG%1%2").arg(delimeter).arg(lst[1]));
else
RawSendNet(QString("PONG"));
return;
}
if (lst[0] == "ROOMS") {
QStringList tmp = lst;
tmp.removeFirst();
emit roomsList(tmp);
return;
}
if (lst[0] == "SERVER_MESSAGE") {
if(lst.size() < 2)
{
qWarning("Net: Empty SERVERMESSAGE message");
return;
}
emit serverMessage(lst[1]);
return;
}
if (lst[0] == "CHAT") {
if(lst.size() < 3)
{
qWarning("Net: Empty CHAT message");
return;
}
if (netClientState == 2)
emit chatStringLobby(formatChatMsg(lst[1], lst[2]));
else
emit chatStringFromNet(formatChatMsg(lst[1], lst[2]));
return;
}
if (lst[0] == "INFO") {
if(lst.size() < 5)
{
qWarning("Net: Malformed INFO message");
return;
}
QStringList tmp = lst;
tmp.removeFirst();
if (netClientState == 2)
emit chatStringLobby(tmp.join("\n"));
else
emit chatStringFromNet(tmp.join("\n"));
return;
}
if (lst[0] == "READY") {
if(lst.size() < 2)
{
qWarning("Net: Malformed READY message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
if (lst[i] == mynick)
emit setMyReadyStatus(true);
emit setReadyStatus(lst[i], true);
}
return;
}
if (lst[0] == "NOT_READY") {
if(lst.size() < 2)
{
qWarning("Net: Malformed NOT_READY message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
if (lst[i] == mynick)
emit setMyReadyStatus(false);
emit setReadyStatus(lst[i], false);
}
return;
}
if (lst[0] == "ADD_TEAM") {
if(lst.size() != 23)
{
qWarning("Net: Bad ADDTEAM message");
return;
}
QStringList tmp = lst;
tmp.removeFirst();
emit AddNetTeam(tmp);
return;
}
if (lst[0] == "REMOVE_TEAM") {
if(lst.size() != 2)
{
qWarning("Net: Bad REMOVETEAM message");
return;
}
m_pTeamSelWidget->removeNetTeam(HWTeam(lst[1]));
return;
}
if(lst[0] == "ROOMABANDONED") {
netClientState = 2;
emit showMessage(HWNewNet::tr("Room destroyed"));
emit LeftRoom();
return;
}
if(lst[0] == "KICKED") {
netClientState = 2;
emit showMessage(HWNewNet::tr("You got kicked"));
emit LeftRoom();
return;
}
if(lst[0] == "JOINED") {
if(lst.size() < 2)
{
qWarning("Net: Bad JOINED message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
if (lst[i] == mynick)
{
netClientState = 3;
emit EnteredGame();
emit roomMaster(isChief);
if (isChief)
emit configAsked();
}
emit nickAdded(lst[i]);
emit chatStringFromNet(QString(tr("[ %1 has joined the room ]")).arg(lst[i]));
}
return;
}
if(lst[0] == "LOBBY:JOINED") {
if(lst.size() < 2)
{
qWarning("Net: Bad JOINED message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
if (lst[i] == mynick)
{
netClientState = 2;
RawSendNet(QString("LIST"));
emit Connected();
}
emit nickAddedLobby(lst[i]);
emit chatStringLobby(QString(tr("[ %1 has joined ]")).arg(lst[i]));
}
return;
}
if(lst[0] == "LEFT") {
if(lst.size() < 2)
{
qWarning("Net: Bad LEFT message");
return;
}
emit nickRemoved(lst[1]);
if (lst.size() < 3)
emit chatStringFromNet(QString(tr("[ %1 has left ]")).arg(lst[1]));
else
emit chatStringFromNet(QString(tr("[ %1 has left (%2) ]")).arg(lst[1], lst[2]));
return;
}
if(lst[0] == "ROOM") {
if(lst.size() < 2)
{
qWarning("Net: Bad ROOM message");
return;
}
RawSendNet(QString("LIST"));
return;
}
if(lst[0] == "LOBBY:LEFT") {
if(lst.size() < 2)
{
qWarning("Net: Bad LOBBY:LEFT message");
return;
}
emit nickRemovedLobby(lst[1]);
if (lst.size() < 3)
emit chatStringLobby(QString(tr("[ %1 has left ]")).arg(lst[1]));
else
emit chatStringLobby(QString(tr("[ %1 has left (%2) ]")).arg(lst[1], lst[2]));
return;
}
if (lst[0] == "RUN_GAME") {
netClientState = 5;
emit AskForRunGame();
return;
}
if (lst[0] == "ASKPASSWORD") {
int passLength = config->value("net/passwordlength", 0).toInt();
QString hash = config->value("net/passwordhash", "").toString();
QString password = QInputDialog::getText(0, tr("Password"), tr("Your nickname %1 is\nregistered on Hedgewars.org\nPlease provide your password\nor pick another nickname:").arg(mynick), QLineEdit::Password, passLength==0?NULL:QString(passLength,'\0'));
if (!passLength || password!=QString(passLength, '\0')) {
hash = QCryptographicHash::hash(password.toLatin1(), QCryptographicHash::Md5).toHex();
config->setValue("net/passwordhash", hash);
config->setValue("net/passwordlength", password.size());
}
RawSendNet(QString("PASSWORD%1%2").arg(delimeter).arg(hash));
return;
}
if (lst[0] == "TEAM_ACCEPTED") {
if (lst.size() != 2)
{
qWarning("Net: Bad TEAM_ACCEPTED message");
return;
}
m_pTeamSelWidget->changeTeamStatus(lst[1]);
return;
}
if (lst[0] == "CFG") {
if(lst.size() < 3)
{
qWarning("Net: Bad CFG message");
return;
}
QStringList tmp = lst;
tmp.removeFirst();
tmp.removeFirst();
if (lst[1] == "SCHEME")
emit netSchemeConfig(tmp);
else
emit paramChanged(lst[1], tmp);
return;
}
if (lst[0] == "HH_NUM") {
if (lst.size() != 3)
{
qWarning("Net: Bad TEAM_ACCEPTED message");
return;
}
HWTeam tmptm(lst[1]);
tmptm.numHedgehogs = lst[2].toUInt();
emit hhnumChanged(tmptm);
return;
}
if (lst[0] == "TEAM_COLOR") {
if (lst.size() != 3)
{
qWarning("Net: Bad TEAM_COLOR message");
return;
}
HWTeam tmptm(lst[1]);
tmptm.teamColor = QColor(lst[2]);
emit teamColorChanged(tmptm);
return;
}
if (lst[0] == "EM") {
if(lst.size() < 2)
{
qWarning("Net: Bad EM message");
return;
}
for(int i = 1; i < lst.size(); ++i)
{
QByteArray em = QByteArray::fromBase64(lst[i].toAscii());
emit FromNet(em);
}
return;
}
if (lst[0] == "BYE") {
if (lst.size() < 2)
{
qWarning("Net: Bad BYE message");
return;
}
emit showMessage(HWNewNet::tr("Quit reason: ") + lst[1]);
return;
}
if (lst[0] == "ADMIN_ACCESS") {
emit adminAccess(true);
return;
}
if (lst[0] == "ROOM_CONTROL_ACCESS") {
if (lst.size() < 2)
{
qWarning("Net: Bad BYE message");
return;
}
bool b = lst[1] != "0";
m_pGameCFGWidget->setEnabled(b);
m_pTeamSelWidget->setInteractivity(b);
isChief = b;
emit roomMaster(isChief);
return;
}
qWarning() << "Net: Unknown message:" << lst;
}
void HWNewNet::onHedgehogsNumChanged(const HWTeam& team)
{
if (isChief)
RawSendNet(QString("HH_NUM%1%2%1%3")
.arg(delimeter)
.arg(team.TeamName)
.arg(team.numHedgehogs));
}
void HWNewNet::onTeamColorChanged(const HWTeam& team)
{
if (isChief)
RawSendNet(QString("TEAM_COLOR%1%2%1%3")
.arg(delimeter)
.arg(team.TeamName)
.arg(team.teamColor.name()));
}
void HWNewNet::onParamChanged(const QString & param, const QStringList & value)
{
if (isChief)
RawSendNet(
QString("CFG%1%2%1%3")
.arg(delimeter)
.arg(param)
.arg(value.join(QString(delimeter)))
);
}
void HWNewNet::chatLineToNet(const QString& str)
{
if(str != "") {
RawSendNet(QString("CHAT") + delimeter + str);
emit(chatStringFromMe(formatChatMsg(mynick, str)));
}
}
void HWNewNet::chatLineToLobby(const QString& str)
{
if(str != "") {
RawSendNet(QString("CHAT") + delimeter + str);
emit(chatStringFromMeLobby(formatChatMsg(mynick, str)));
}
}
void HWNewNet::askRoomsList()
{
if(netClientState != 2)
{
qWarning("Illegal try to get rooms list!");
return;
}
RawSendNet(QString("LIST"));
}
bool HWNewNet::isRoomChief()
{
return isChief;
}
void HWNewNet::gameFinished()
{
if (netClientState == 5) netClientState = 3;
RawSendNet(QString("ROUNDFINISHED"));
}
QString HWNewNet::formatChatMsg(const QString & nick, const QString & msg)
{
if(msg.left(4) == "/me ")
return QString("* %1 %2").arg(nick).arg(msg.mid(4));
else
return QString("%1: %2").arg(nick).arg(msg);
}
void HWNewNet::banPlayer(const QString & nick)
{
RawSendNet(QString("BAN%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::kickPlayer(const QString & nick)
{
RawSendNet(QString("KICK%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::infoPlayer(const QString & nick)
{
RawSendNet(QString("INFO%1%2").arg(delimeter).arg(nick));
}
void HWNewNet::startGame()
{
RawSendNet(QString("START_GAME"));
}
void HWNewNet::toggleRestrictJoins()
{
RawSendNet(QString("TOGGLE_RESTRICT_JOINS"));
}
void HWNewNet::toggleRestrictTeamAdds()
{
RawSendNet(QString("TOGGLE_RESTRICT_TEAMS"));
}
void HWNewNet::clearAccountsCache()
{
RawSendNet(QString("CLEAR_ACCOUNTS_CACHE"));
}
void HWNewNet::partRoom()
{
netClientState = 2;
RawSendNet(QString("PART"));
}
bool HWNewNet::isInRoom()
{
return netClientState > 2;
}
void HWNewNet::newServerMessage(const QString & msg)
{
RawSendNet(QString("SET_SERVER_MESSAGE%1%2").arg(delimeter).arg(msg));
}