/*
 * Hedgewars, a worms-like game
 * Copyright (c) 2007 Ulyanov Igor <iulyanov@gmail.com>
 * Copyright (c) 2007 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 <QUdpSocket>

#include "netudpwidget.h"

HWNetUdpModel::HWNetUdpModel(QObject* parent) :
  HWNetServersModel(parent)
{
	pUdpSocket = new QUdpSocket(this);

	pUdpSocket->bind();
	connect(pUdpSocket, SIGNAL(readyRead()), this, SLOT(onClientRead()));
}

void HWNetUdpModel::updateList()
{
	games.clear();

	reset();

	pUdpSocket->writeDatagram("hedgewars client", QHostAddress::Broadcast, 46631);
}

void HWNetUdpModel::onClientRead()
{
	while (pUdpSocket->hasPendingDatagrams()) {
		QByteArray datagram;
		datagram.resize(pUdpSocket->pendingDatagramSize());
		QHostAddress clientAddr;
		quint16 clientPort;

		pUdpSocket->readDatagram(datagram.data(), datagram.size(), &clientAddr, &clientPort);

		if(QString("%1").arg(datagram.data())==QString("hedgewars server")) {
			QStringList sl;
			sl << "-" << clientAddr.toString() << "46631";
			games.append(sl);
		}
	}

	reset();
}

QVariant HWNetUdpModel::data(const QModelIndex &index,
                             int role) const
{
	if (!index.isValid() || index.row() < 0
		|| index.row() >= games.size()
		|| role != Qt::DisplayRole)
	return QVariant();

	return games[index.row()][index.column()];
}

QVariant HWNetUdpModel::headerData(int section,
            Qt::Orientation orientation, int role) const
{
	if (role != Qt::DisplayRole)
		return QVariant();

	if (orientation == Qt::Horizontal)
	{
		switch (section)
		{
			case 0: return tr("Title");
			case 1: return tr("IP");
			case 2: return tr("Port");
			default: return QVariant();
		}
	} else
		return QString("%1").arg(section + 1);
}

int HWNetUdpModel::rowCount(const QModelIndex &parent) const
{
	if (parent.isValid())
		return 0;
	else
		return games.size();
}

int HWNetUdpModel::columnCount(const QModelIndex & parent) const
{
	if (parent.isValid())
		return 0;
	else
		return 3;
}