// SPDX-License-Identifier: LGPL-3.0-or-later

/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
*
* Copyright (C) 2013 - 2024, nymea GmbH
* Copyright (C) 2024 - 2025, chargebyte austria GmbH
*
* This file is part of libnymea-app.
*
* libnymea-app is free software: you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation, either version 3
* of the License, or (at your option) any later version.
*
* libnymea-app 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 Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with libnymea-app. If not, see <https://www.gnu.org/licenses/>.
*
* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */

#include "bluetoothdeviceinfos.h"

#include <QDebug>
#include <QBluetoothUuid>

BluetoothDeviceInfos::BluetoothDeviceInfos(QObject *parent) : QAbstractListModel(parent)
{

}

QList<BluetoothDeviceInfo *> BluetoothDeviceInfos::deviceInfos()
{
    return m_deviceInfos;
}

int BluetoothDeviceInfos::rowCount(const QModelIndex &parent) const
{
    Q_UNUSED(parent)
    return m_deviceInfos.count();
}

QVariant BluetoothDeviceInfos::data(const QModelIndex &index, int role) const
{
    if (index.row() < 0 || index.row() >= m_deviceInfos.count())
        return QVariant();

    BluetoothDeviceInfo *deviceInfo = m_deviceInfos.at(index.row());
    if (role == BluetoothDeviceInfoRoleName) {
        return deviceInfo->name();
    } else if (role == BluetoothDeviceInfoRoleAddress) {
        return deviceInfo->address();
    } else if (role == BluetoothDeviceInfoRoleLe) {
        return deviceInfo->isLowEnergy();
    } else if (role == BluetoothDeviceInfoRoleSignalStrength) {
        return deviceInfo->signalStrength();
    }

    return QVariant();
}

int BluetoothDeviceInfos::count() const
{
    return m_deviceInfos.count();
}

BluetoothDeviceInfo *BluetoothDeviceInfos::get(int index) const
{
    if (index >= m_deviceInfos.count() || index < 0)
        return Q_NULLPTR;

    return m_deviceInfos.at(index);
}

void BluetoothDeviceInfos::addBluetoothDeviceInfo(BluetoothDeviceInfo *deviceInfo)
{
    qDebug() << "Adding device" << deviceInfo->name();
    deviceInfo->setParent(this);
    beginInsertRows(QModelIndex(), m_deviceInfos.count(), m_deviceInfos.count());
    m_deviceInfos.append(deviceInfo);
    connect(deviceInfo, &BluetoothDeviceInfo::deviceChanged, this, [=]{
        int idx = m_deviceInfos.indexOf(deviceInfo);
        QModelIndex index = this->index(idx);
        emit dataChanged(index, index);
    });
    endInsertRows();
    emit countChanged();
}

void BluetoothDeviceInfos::clearModel()
{
    beginResetModel();
    qDeleteAll(m_deviceInfos);
    m_deviceInfos.clear();
    endResetModel();
    emit countChanged();
}

QHash<int, QByteArray> BluetoothDeviceInfos::roleNames() const
{
    QHash<int, QByteArray> roles;
    roles[BluetoothDeviceInfoRoleName] = "name";
    roles[BluetoothDeviceInfoRoleAddress] = "address";
    roles[BluetoothDeviceInfoRoleLe] = "lowEnergy";
    roles[BluetoothDeviceInfoRoleSignalStrength] = "signalStrength";
    return roles;
}

BluetoothDeviceInfosProxy::BluetoothDeviceInfosProxy(QObject *parent): QSortFilterProxyModel(parent)
{

}

BluetoothDeviceInfos *BluetoothDeviceInfosProxy::model() const
{
    return m_model;
}

void BluetoothDeviceInfosProxy::setModel(BluetoothDeviceInfos *model)
{
    if (m_model == model) {
        return;
    }

    if (m_model) {
        disconnect(m_model, &BluetoothDeviceInfos::countChanged, this, &BluetoothDeviceInfosProxy::countChanged);
    }

    m_model = model;
    setSourceModel(model);
    emit modelChanged();
    emit countChanged();

    if (m_model) {
        connect(m_model, &BluetoothDeviceInfos::countChanged, this, &BluetoothDeviceInfosProxy::countChanged);
    }
}

QStringList BluetoothDeviceInfosProxy::nameWhitelist() const
{
    return m_nameWhitelist;
}

void BluetoothDeviceInfosProxy::setNameWhitelist(const QStringList &nameWhitelist)
{
    if (m_nameWhitelist != nameWhitelist) {
        m_nameWhitelist = nameWhitelist;
        emit nameWhitelistChanged();
        invalidateFilter();
        emit countChanged();
    }
}

bool BluetoothDeviceInfosProxy::filterForLowEnergy() const
{
    return m_filterForLowEnergy;
}

void BluetoothDeviceInfosProxy::setFilterForLowEnergy(bool filterForLowEnergy)
{
    if (m_filterForLowEnergy != filterForLowEnergy) {
        m_filterForLowEnergy = filterForLowEnergy;
        emit filterForLowEnergyChanged();
        invalidateFilter();
        emit countChanged();
    }
}

QString BluetoothDeviceInfosProxy::filterForServiceUUID() const
{
    return m_filterForServiceUUID.toString();
}

void BluetoothDeviceInfosProxy::setFilterForServiceUUID(const QString &filterForServiceUUID)
{
    if (m_filterForServiceUUID != filterForServiceUUID) {
        m_filterForServiceUUID = filterForServiceUUID;
        emit filterForServiceUUIDChanged();
        invalidateFilter();
        emit countChanged();
    }
}

QString BluetoothDeviceInfosProxy::filterForName() const
{
    return m_filterForName;
}

void BluetoothDeviceInfosProxy::setFilterForName(const QString &name)
{
    if (m_filterForName != name) {
        m_filterForName = name;
        emit filterForNameChanged();

        invalidateFilter();
        emit countChanged();
    }
}

BluetoothDeviceInfo *BluetoothDeviceInfosProxy::get(int index) const
{
    return m_model->get(mapToSource(this->index(index, 0)).row());
}

bool BluetoothDeviceInfosProxy::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const
{
    Q_UNUSED(source_parent)

    BluetoothDeviceInfo *info = m_model->get(source_row);

    if (!m_nameWhitelist.isEmpty()) {
        if (m_nameWhitelist.contains(info->name())) {
            return true;
        }
    }

    if (m_filterForLowEnergy && !info->bluetoothDeviceInfo().coreConfigurations().testFlag(QBluetoothDeviceInfo::LowEnergyCoreConfiguration)) {
        return false;
    }

    if (!m_filterForServiceUUID.isNull() && !info->bluetoothDeviceInfo().serviceUuids().contains(QBluetoothUuid(m_filterForServiceUUID))) {
        return false;
    }

    if (!m_filterForName.isEmpty() && info->name() != m_filterForName) {
        return false;
    }

    return true;
}
