/******************************************************************************
**
** Copyright (C) 2017 The Qt Company Ltd.
** Contact: https://www.qt.io/licensing/
**
** This file is part of the QtKnx module.
**
** $QT_BEGIN_LICENSE:GPL$
** Commercial License Usage
** Licensees holding valid commercial Qt licenses may use this file in
** accordance with the commercial license agreement provided with the
** Software or, alternatively, in accordance with the terms contained in
** a written agreement between you and The Qt Company. For licensing terms
** and conditions see https://www.qt.io/terms-conditions. For further
** information use the contact form at https://www.qt.io/contact-us.
**
** GNU General Public License Usage
** Alternatively, this file may be used under the terms of the GNU
** General Public License version 3 or (at your option) any later version
** approved by the KDE Free Qt Foundation. The licenses are as published by
** the Free Software Foundation and appearing in the file LICENSE.GPL3
** included in the packaging of this file. Please review the following
** information to ensure the GNU General Public License requirements will
** be met: https://www.gnu.org/licenses/gpl-3.0.html.
**
** $QT_END_LICENSE$
**
******************************************************************************/

#ifndef QKNXNETIPDISCOVERYAGENT_P_H
#define QKNXNETIPDISCOVERYAGENT_P_H

//
//  W A R N I N G
//  -------------
//
// This file is not part of the Qt KNX API.  It exists for the convenience
// of the Qt KNX implementation.  This header file may change from version
// to version without notice, or even be removed.
//
// We mean it.
//

#include <QtCore/qtimer.h>
#include <QtKnx/qtknxglobal.h>
#include <QtKnx/qknxnetip.h>
#include <QtKnx/qknxnetipsearchrequest.h>
#include <QtKnx/qknxnetipsearchresponse.h>
#include <QtKnx/qknxnetipserverdiscoveryagent.h>

#include <QtNetwork/qhostaddress.h>
#include <QtNetwork/qnetworkdatagram.h>
#include <QtNetwork/qnetworkinterface.h>
#include <QtNetwork/qudpsocket.h>

#include <private/qobject_p.h>

QT_BEGIN_NAMESPACE

struct Adapter
{
    QHostAddress address;
    QNetworkInterface iface;
};

struct DiscovererConfig
{
    DiscovererConfig() = default;
    ~DiscovererConfig() = default;

    DiscovererConfig(int ttl, int timeout, bool nat, bool multicast, bool coreV1, bool coreV2,
            const QVector<QKnxNetIpSrp> &srps)
        : Ttl(ttl)
        , Timeout(timeout)
        , Nat(nat)
        , Multicast(multicast)
        , DiscoveryCoreV1(coreV1)
        , DiscoveryCoreV2(coreV2)
        , ExtendedSearchParameters(srps)
    {}

    int Ttl { 64 };
    int Timeout { 3000 };

    bool Nat { false };
    bool Multicast { true };

    bool DiscoveryCoreV1 { true };
    bool DiscoveryCoreV2 { true };

    QVector<QKnxNetIpSrp> ExtendedSearchParameters;
};

class Discoverer : public QObject
{
    Q_OBJECT

public:
    Discoverer(const QHostAddress &address, const QNetworkInterface iface, const DiscovererConfig &config);
    ~Discoverer() override;

public slots:
    void start();
    void finish();

signals:
    void started();
    void timeout();
    void finished(Discoverer *iam);

    void stateChanged(QKnxNetIpServerDiscoveryAgent::State state);
    void errorOccurred(QKnxNetIpServerDiscoveryAgent::Error error, const QString &errorString);

    void deviceDiscovered(const QKnxNetIpServerInfo &discoveryInfo);

private slots:
    void onTimeout();
    void onReadyRead();
    void onError(QUdpSocket::SocketError error);

private:
    void setAndEmitStateChanged(QKnxNetIpServerDiscoveryAgent::State state);

private:
    QHostAddress m_address;
    QNetworkInterface m_iface;

    QTimer *m_timer { nullptr };
    QUdpSocket *m_socket { nullptr };

    DiscovererConfig m_config;
    QVector<QByteArray> m_devices;

    QHostAddress m_multicast { QStringLiteral("224.0.23.12") };
    QKnxNetIpServerDiscoveryAgent::State m_state { QKnxNetIpServerDiscoveryAgent::State::NotRunning };
};

class Q_KNX_EXPORT QKnxNetIpServerDiscoveryAgentPrivate final : public QObjectPrivate
{
    Q_DECLARE_PUBLIC(QKnxNetIpServerDiscoveryAgent)

public:
    QKnxNetIpServerDiscoveryAgentPrivate(const QHostAddress &addr, quint16 prt);
    ~QKnxNetIpServerDiscoveryAgentPrivate() override = default;

    void setupSocket();

    void setupAndStartReceiveTimer();
    void setupAndStartFrequencyTimer();

    void setAndEmitStateChanged(QKnxNetIpServerDiscoveryAgent::State newState);
    void setAndEmitDeviceDiscovered(const QKnxNetIpServerInfo &discoveryInfo);
    void setAndEmitErrorOccurred(QKnxNetIpServerDiscoveryAgent::Error newError, const QString &message);

    void start();
    void start(const QVector<Adapter> &adapters);
    void start(const QVector<QHostAddress> addresses);
    void start(QKnxNetIpServerDiscoveryAgent::InterfaceTypes types);
    void stop();

private:
    QUdpSocket *socket { nullptr };
    QTimer *receiveTimer { nullptr };
    QTimer *frequencyTimer { nullptr };

    quint16 port { 0 }, usedPort { 0 };
    QHostAddress address { QHostAddress::Null }, usedAddress;

    const quint16 multicastPort { QKnxNetIp::Constants::DefaultPort };
    const QHostAddress multicastAddress { QLatin1String(QKnxNetIp::Constants::MulticastAddress) };

    quint8 ttl { 64 };
    bool nat { false };
    int frequency { 0 };
    int timeout { QKnxNetIp::Timeout::SearchTimeout };

    QString errorString;
    QVector<QKnxNetIpServerInfo> servers;

    QKnxNetIpServerDiscoveryAgent::Error error { QKnxNetIpServerDiscoveryAgent::Error::None };
    QKnxNetIpServerDiscoveryAgent::State state { QKnxNetIpServerDiscoveryAgent::State::NotRunning };
    QKnxNetIpServerDiscoveryAgent::ResponseType
        type { QKnxNetIpServerDiscoveryAgent::ResponseType::Multicast };

    QKnxNetIpServerDiscoveryAgent::DiscoveryModes mode
        { QKnxNetIpServerDiscoveryAgent::DiscoveryMode::CoreV1 };
    QVector<QKnxNetIpSrp> srps;

    Adapter adapter;
    QVector<Discoverer*> discoveries;
};

QT_END_NAMESPACE

#endif