/**
 * SPDX-FileCopyrightText: 2023 Albert Vaca Cintora <albertvaka@gmail.com>
 *
 * SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
 */

#ifndef KDECONNECT_MDNS_WRAPPER_H
#define KDECONNECT_MDNS_WRAPPER_H

#include <QHostAddress>
#include <QMap>
#include <QSocketNotifier>
#include <QString>
#include <QVector>

/*
 * A Qt wrapper for the mdns.h header-only library
 * from https://github.com/mjansson/mdns
 */
class MdnsWrapper : public QObject
{
    Q_OBJECT

public:
    struct MdnsService {
        QString name;
        int port;
        QHostAddress address;
        QMap<QString, QString> txtRecords;
    };

    // serviceType should be of the form _kdeconnect._udp.local
    void startDiscovering(const QString &serviceType);
    void stopDiscovering();

    void sendQuery(const QString &serviceName);

Q_SIGNALS:
    void serviceFound(const MdnsService &service);

private:
    int listenForQueryResponses();
    void stopListeningForQueryResponses();

    QVector<QSocketNotifier *> responseSocketNotifiers;
};

class MdnsServiceAnnouncer : public QObject
{
    Q_OBJECT

public:
    struct AnnouncedInfo {
        QString serviceType; // ie: "<_service-type>._tcp.local."
        QString serviceInstance; // ie: "<service-name>.<_service-type>._tcp.local."
        QString hostname; // ie: "<hostname>.local."
        int port;
        QMap<QString, QString> txtRecords;
    };

    // serviceType should be of the form _kdeconnect._udp.local
    MdnsServiceAnnouncer(const QString &name, const QString &serviceType, int port);

    void putTxtRecord(const QString &key, const QString &value)
    {
        myself.txtRecords[key] = value;
    }

    void startAnnouncing();
    void stopAnnouncing();

    void sendMulticastAnnounce(bool isGoodbye);

private:
    int listenForQueries();
    void stopListeningForQueries();

    AnnouncedInfo myself;

    QSocketNotifier *socketNotifier = nullptr;
    QSocketNotifier *socketNotifierV6 = nullptr;
};

#endif