/*
    SPDX-FileCopyrightText: 2019 Volker Krause <vkrause@kde.org>

    SPDX-License-Identifier: LGPL-2.0-or-later
*/

#ifndef TIMELINEDELEGATECONTROLLER_H
#define TIMELINEDELEGATECONTROLLER_H

#include <KPublicTransport/Journey>
#include <KPublicTransport/JourneyRequest>
#include <KPublicTransport/Location>
#include <KPublicTransport/Stopover>

#include <KCalendarCore/Calendar>

#include <QObject>
#include <QVariant>
#include <qqmlintegration.h>

#include <chrono>

class QDateTime;
class QJSValue;
class QTimer;

class LiveDataManager;
class ReservationManager;
class TransferManager;
class TripGroupManager;
class DocumentManager;

/** C++ side logic for timeline delegates. */
class TimelineDelegateController : public QObject
{
    Q_OBJECT
    QML_ELEMENT

    Q_PROPERTY(ReservationManager *reservationManager READ reservationManager WRITE setReservationManager NOTIFY setupChanged)
    Q_PROPERTY(LiveDataManager *liveDataManager READ liveDataManager WRITE setLiveDataManager NOTIFY setupChanged)
    Q_PROPERTY(TransferManager *transferManager READ transferManager WRITE setTransferManager NOTIFY setupChanged)
    Q_PROPERTY(DocumentManager *documentManager READ documentManager WRITE setDocumentManager NOTIFY setupChanged)
    Q_PROPERTY(TripGroupManager *tripGroupManager MEMBER m_tripGroupMgr NOTIFY setupChanged)
    Q_PROPERTY(QString batchId READ batchId WRITE setBatchId NOTIFY batchIdChanged)

    Q_PROPERTY(bool isCurrent READ isCurrent NOTIFY currentChanged)
    Q_PROPERTY(float progress READ progress NOTIFY progressChanged)

    /** The location we are in before this element begins.
     *  This is only relevant for future elements, past elements, or elements without a non-current predecessor return nothing here.
     */
    Q_PROPERTY(QVariant previousLocation READ previousLocation NOTIFY previousLocationChanged)

    Q_PROPERTY(KPublicTransport::Stopover arrival READ arrival NOTIFY journeyChanged)
    Q_PROPERTY(KPublicTransport::Stopover departure READ departure NOTIFY journeyChanged)
    Q_PROPERTY(KPublicTransport::JourneySection journey READ journey NOTIFY journeyChanged)

    Q_PROPERTY(KPublicTransport::JourneySection trip READ trip NOTIFY journeyChanged)
    Q_PROPERTY(qsizetype tripDepartureIndex READ tripDepartureIndex NOTIFY journeyChanged)
    Q_PROPERTY(qsizetype tripArrivalIndex READ tripArrivalIndex NOTIFY journeyChanged)

    /** Planned start time of this element. */
    Q_PROPERTY(QDateTime startTime READ startTime NOTIFY journeyChanged)
    /** Effective end time, ie. our best knowledge of arriving at the destination (for transit elements), or
     *  ending of the associated event for non-transit elements.
     */
    Q_PROPERTY(QDateTime effectiveEndTime READ effectiveEndTime NOTIFY journeyChanged)

    /** Returns whether this is a location changing element. */
    Q_PROPERTY(bool isLocationChange READ isLocationChange NOTIFY contentChanged)
    /** Returns whether the current element is a public transport transit element.
     *  That is, a location change where you don't have to navigate to the destination yourself.
     */
    Q_PROPERTY(bool isPublicTransport READ isPublicTransport NOTIFY contentChanged)

    /** A KPublicTransport::JourneyRequest for the current journey.
     *  This has no destination on its own, @see journeyDestinations.
     */
    Q_PROPERTY(KPublicTransport::JourneyRequest journeyRequest READ journeyRequest NOTIFY contentChanged)
    /** Destinations for journeyRequest for this and all immediately following connected
     *  destinations.
     */
    Q_PROPERTY(QList<KPublicTransport::Location> journeyDestinations READ journeyDestinations NOTIFY contentChanged) // TODO technically notification also depends on other elements, so similar to previousLocationChanged

    /** Inbound connection is unlikely to work. */
    Q_PROPERTY(bool connectionWarning READ connectionWarning NOTIFY connectionWarningChanged)

    /** Reservation has been canceled (by user or provider, we usually don't know which). */
    Q_PROPERTY(bool isCanceled READ isCanceled NOTIFY contentChanged)

    /** Platform sections to the extend known. */
    Q_PROPERTY(QString departurePlatformSections READ departurePlatformSections NOTIFY layoutChanged)
    Q_PROPERTY(QString arrivalPlatformSections READ arrivalPlatformSections NOTIFY layoutChanged)

    /** Attached documents of this element. */
    Q_PROPERTY(QStringList documentIds READ documentIds NOTIFY contentChanged)

    /** Reservation object for the current batch id. */
    Q_PROPERTY(QVariant reservation READ reservation NOTIFY contentChanged)

public:
    TimelineDelegateController(QObject *parent = nullptr);
    ~TimelineDelegateController() override;

    [[nodiscard]] ReservationManager *reservationManager() const;
    void setReservationManager(ReservationManager *resMgr);
    [[nodiscard]] LiveDataManager *liveDataManager() const;
    void setLiveDataManager(LiveDataManager *liveDataMgr);
    [[nodiscard]] TransferManager *transferManager() const;
    void setTransferManager(TransferManager *transferMgr);
    [[nodiscard]] DocumentManager *documentManager() const;
    void setDocumentManager(DocumentManager *documentMgr);

    [[nodiscard]] QString batchId() const;
    void setBatchId(const QString &batchId);

    [[nodiscard]] bool isCurrent() const;
    [[nodiscard]] float progress() const;

    [[nodiscard]] QVariant previousLocation() const;

    [[nodiscard]] KPublicTransport::Stopover arrival() const;
    [[nodiscard]] KPublicTransport::Stopover departure() const;
    [[nodiscard]] KPublicTransport::JourneySection journey() const;

    [[nodiscard]] KPublicTransport::JourneySection trip() const;
    [[nodiscard]] qsizetype tripDepartureIndex() const;
    [[nodiscard]] qsizetype tripArrivalIndex() const;

    [[nodiscard]] QDateTime startTime() const;
    [[nodiscard]] QDateTime effectiveEndTime() const;

    [[nodiscard]] bool isLocationChange() const;
    [[nodiscard]] bool isPublicTransport() const;

    [[nodiscard]] KPublicTransport::JourneyRequest journeyRequest() const;
    [[nodiscard]] QList<KPublicTransport::Location> journeyDestinations() const;
    Q_INVOKABLE void applyJourney(const QVariant &journey, int includeFollowing);

    [[nodiscard]] bool connectionWarning() const;
    [[nodiscard]] bool isCanceled() const;

    /** Map page arguments for the arrival side, if this is a location change element. */
    Q_INVOKABLE [[nodiscard]] QJSValue arrivalMapArguments() const;
    /** Map page arguments for the departure side, if this is a location change element. */
    Q_INVOKABLE [[nodiscard]] QJSValue departureMapArguments() const;
    /** Map page arguments for an element that isn't a location change. */
    Q_INVOKABLE [[nodiscard]] QJSValue mapArguments() const;

    /** Add the current reservation to the calendar @p cal. */
    Q_INVOKABLE void addToCalendar(KCalendarCore::Calendar *cal);

    /** Store vehicle layout data we got from the vehicle layout page. */
    Q_INVOKABLE void setVehicleLayout(const KPublicTransport::Stopover &stopover, bool arrival);

    /** Platform section information. */
    [[nodiscard]] QString departurePlatformSections() const;
    [[nodiscard]] QString arrivalPlatformSections() const;

    /** Ids of attached documents. */
    [[nodiscard]] QStringList documentIds() const;

Q_SIGNALS:
    void setupChanged();
    void batchIdChanged();
    void contentChanged();
    void currentChanged();
    void progressChanged();
    void journeyChanged();
    void previousLocationChanged();
    void connectionWarningChanged();
    void layoutChanged();

private:
    void setCurrent(bool current, const QVariant &res = {});
    void checkForUpdate(const QString &batchId);
    /** Time at which we consider @p res "current". */
    [[nodiscard]] static QDateTime relevantStartDateTime(const QVariant &res);
    /** Time at which the event starts/stops based on realtime data. */
    [[nodiscard]] QDateTime liveStartDateTime(const QVariant &res) const;
    [[nodiscard]] QDateTime liveEndDateTime(const QVariant &res) const;

    [[nodiscard]] QVariant reservation() const;

    void batchChanged(const QString &batchId);

    ReservationManager *m_resMgr = nullptr; // ### should this be static?
    LiveDataManager *m_liveDataMgr = nullptr;
    TransferManager *m_transferMgr = nullptr;
    DocumentManager *m_documentMgr = nullptr;
    TripGroupManager *m_tripGroupMgr = nullptr;
    QString m_batchId;
    bool m_isCurrent = false;

    static void scheduleNextUpdate(std::chrono::milliseconds ms);
    static QTimer *s_currentTimer;
    static int s_progressRefCount;
    static QTimer *s_progressTimer;
};

#endif // TIMELINEDELEGATECONTROLLER_H
