/* This file is part of the KDE project
 *
 * Copyright (C) 2008 Jakub Stachowski <qbast@go2.pl>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Library General Public
 * License as published by the Free Software Foundation; either
 * version 2 of the License, or (at your option) any later version.
 *
 * This library 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
 * Library General Public License for more details.
 *
 * You should have received a copy of the GNU Library General Public License
 * along with this library; see the file COPYING.LIB.  If not, write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
 * Boston, MA 02110-1301, USA.
 */

#include "servicemodel.h"
#include "servicebrowser.h"
#include <klocale.h>

namespace DNSSD
{

struct ServiceModelPrivate
{
        ServiceBrowser* m_browser;
};


ServiceModel::ServiceModel(ServiceBrowser* browser, QObject* parent) 
    :  QAbstractItemModel(parent), d(new ServiceModelPrivate)
{
    d->m_browser=browser;
    browser->setParent(this);
    connect(browser, SIGNAL(serviceAdded(DNSSD::RemoteService::Ptr)), this,
	SIGNAL(layoutChanged()));
    connect(browser, SIGNAL(serviceRemoved(DNSSD::RemoteService::Ptr)), this,
	SIGNAL(layoutChanged()));
    browser->startBrowse();
}

ServiceModel::~ServiceModel()
{
    delete d;
}

int ServiceModel::columnCount(const QModelIndex&) const
{
    return d->m_browser->isAutoResolving() ? 3 : 1;
}
int ServiceModel::rowCount(const QModelIndex& parent ) const
{
    return (parent.isValid()) ? 0 : d->m_browser->services().size();
}

QModelIndex ServiceModel::parent(const QModelIndex&) const
{
    return QModelIndex();
}

QModelIndex ServiceModel::index(int row, int column, const QModelIndex& parent  ) const
{
    return hasIndex(row, column, parent) ? createIndex(row, column) : QModelIndex();
}

bool ServiceModel::hasIndex(int row, int column, const QModelIndex &parent) const
{
    if (parent.isValid()) return false;
    if (column<0 || column>=columnCount()) return false;
    if (row<0 || row>=rowCount(parent)) return false;
    return true;
}

QVariant ServiceModel::data(const QModelIndex& index, int role  ) const
{
    if (!index.isValid()) return QVariant();
    if (!hasIndex(index.row(), index.column(), index.parent())) return QVariant();
    const QList<RemoteService::Ptr> srv=d->m_browser->services();
    switch (role) {
	case Qt::DisplayRole: 
	    switch (index.column()) {
		case ServiceName: return srv[index.row()]->serviceName();
		case Host: return srv[index.row()]->hostName();
		case Port: return srv[index.row()]->port();
	    }
	case ServicePtrRole: QVariant ret;
			     ret.setValue(srv[index.row()]);
			    return ret;
    }
    return QVariant();
}

QVariant ServiceModel::headerData(int section, Qt::Orientation orientation, int role) const
{
    if (orientation!=Qt::Horizontal || role!=Qt::DisplayRole) return QVariant();
    switch (section) {
	case ServiceName: return i18n("Name");
	case Host: return i18n("Host");
	case Port: return i18n("Port");
    }
    return QVariant();
}

}