// For license of this file, see <project-root-folder>/LICENSE.md.

#ifndef ICALPARSER_H
#define ICALPARSER_H

#include "src/parsers/feedparser.h"

#include <QTimeZone>

class IcalendarComponent {
  public:
    QString uid() const;

    QVariantMap properties() const;
    void setProperties(const QVariantMap& properties);

  protected:
    QVariant getPropertyValue(const QString& property_name) const;
    QVariant getPropertyValue(const QString& property_name, QString& property_modifier) const;

    QDateTime fixupDate(QDateTime dat,
                        const QString& dt_format,
                        const QMap<QString, QTimeZone>& time_zones,
                        const QString& modifiers,
                        bool* has_dt = nullptr) const;

    QVariantMap m_properties;
};

Q_DECLARE_METATYPE(IcalendarComponent)

class EventComponent : public IcalendarComponent {
  public:
    QDateTime startsOn(const QMap<QString, QTimeZone>& time_zones = {},
                       bool* had_dt = nullptr,
                       QString* dt_format = nullptr) const;
    QDateTime endsOn(const QMap<QString, QTimeZone>& time_zones = {},
                     bool* had_dt = nullptr,
                     QString* dt_format = nullptr) const;
    QString title() const;
    QString url() const;
    QString organizer() const;
    QString location() const;
    QString description() const;
};

Q_DECLARE_METATYPE(EventComponent)

class Icalendar : public FeedParser {
    friend class IcalParser;

  public:
    explicit Icalendar(const QByteArray& data = {});

    QString title() const;
    void setTitle(const QString& title);

  private:
    void processLines(const QString& data);
    void processComponentCalendar(const QString& body);
    void processComponentEvent(const QString& body);
    void processComponentTimezone(const QString& body);

    QDateTime parseDateTime(const QString& date_time) const;
    QVariantMap tokenizeBody(const QString& body) const;

  private:
    QString m_title;
    QMap<QString, QTimeZone> m_tzs;
    QList<IcalendarComponent> m_components;
};

class IcalParser : public FeedParser {
  public:
    explicit IcalParser(const QString& data);
    virtual ~IcalParser();

    virtual QList<StandardFeed*> discoverFeeds(ServiceRoot* root, const QUrl& url, bool greedy) const;

    virtual QPair<StandardFeed*, QList<IconLocation>> guessFeed(const QByteArray& content,
                                                                const NetworkResult& network_res) const;

    virtual QVariantList objMessageElements();
    virtual QString objMessageTitle(const QVariant& msg_element) const;
    virtual QString objMessageUrl(const QVariant& msg_element) const;
    virtual QString objMessageDescription(const QVariant& msg_element);
    virtual QString objMessageAuthor(const QVariant& msg_element) const;
    virtual QDateTime objMessageDateCreated(const QVariant& msg_element);
    virtual QString objMessageId(const QVariant& msg_element) const;
    virtual QString objMessageRawContents(const QVariant& msg_element) const;

  private:
    Icalendar m_iCalendar;
};

#endif // ICALPARSER_H
