/*
 * Copyright (C) 2019-2024 Matthias Klumpp <matthias@tenstral.net>
 *
 * Licensed under the GNU Lesser General Public License Version 3
 *
 * This program 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.
 *
 * This software 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 this software.  If not, see <http://www.gnu.org/licenses/>.
 */

#pragma once

#include <QDataStream>
#include <QDateTime>
#include <QLoggingCategory>
#include <QUuid>
#include <memory>
#include <xxhash.h>

#include "syclock.h"

class QFile;

namespace Syntalos
{

Q_DECLARE_LOGGING_CATEGORY(logTSyncFile)


enum class TSyncFileMode {
    CONTINUOUS = 0, 
    SYNCPOINTS = 1  
};

enum class TSyncFileTimeUnit {
    INDEX = 0, 
    NANOSECONDS = 1,
    MICROSECONDS = 2,
    MILLISECONDS = 3,
    SECONDS = 4
};

enum class TSyncFileDataType {
    INVALID = 0,
    INT16 = 2,
    INT32 = 3,
    INT64 = 4,

    UINT16 = 6,
    UINT32 = 7,
    UINT64 = 8
};

QString tsyncFileTimeUnitToString(const TSyncFileTimeUnit &tsftunit);
QString tsyncFileDataTypeToString(const TSyncFileDataType &dtype);
QString tsyncFileModeToString(const TSyncFileMode &mode);

class TimeSyncFileWriter
{
public:
    explicit TimeSyncFileWriter();
    ~TimeSyncFileWriter();

    QString lastError() const;

    void setTimeNames(const QString &time1Name, const QString &time2Name);
    void setTimeUnits(TSyncFileTimeUnit time1Unit, TSyncFileTimeUnit time2Unit);
    void setTimeDataTypes(TSyncFileDataType time1DType, TSyncFileDataType time2DType);

    QString fileName() const;
    void setFileName(const QString &fname);

    void setSyncMode(TSyncFileMode mode);
    void setChunkSize(int size);

    void setCreationTimeOverride(const QDateTime &dt);

    bool open(const QString &modName, const QUuid &collectionId, const QVariantHash &userData = QVariantHash());
    bool open(
        const QString &modName,
        const QUuid &collectionId,
        const microseconds_t &tolerance,
        const QVariantHash &userData = QVariantHash());
    void flush();
    void close();

    void writeTimes(const microseconds_t &deviceTime, const microseconds_t &masterTime);
    void writeTimes(const long long &timeIndex, const microseconds_t &masterTime);
    void writeTimes(const long &time1, const long &time2);
    void writeTimes(const uint64_t &time1, const uint64_t &time2);

private:
    QFile *m_file;
    QDataStream m_stream;
    TSyncFileMode m_tsMode;
    int m_blockSize;
    int m_bIndex;
    XXH3_state_t *m_xxh3State;
    QString m_lastError;
    QDateTime m_creationTimeOverride;

    QPair<QString, QString> m_timeNames;
    QPair<TSyncFileTimeUnit, TSyncFileTimeUnit> m_timeUnits;
    TSyncFileDataType m_time1DType;
    TSyncFileDataType m_time2DType;

    void writeBlockTerminator(bool check = true);
    template<class T>
    void csWriteValue(const T &data);
    template<class T1, class T2>
    void writeTimeEntry(const T1 &time1, const T2 &time2);
};

class TimeSyncFileReader
{
public:
    explicit TimeSyncFileReader();

    bool open(const QString &fname);
    QString lastError() const;

    QString moduleName() const;
    QUuid collectionId() const;
    time_t creationTime() const;
    TSyncFileMode syncMode() const;

    QVariantHash userData() const;
    microseconds_t tolerance() const;

    QPair<QString, QString> timeNames() const;
    QPair<TSyncFileTimeUnit, TSyncFileTimeUnit> timeUnits() const;
    QPair<TSyncFileDataType, TSyncFileDataType> timeDTypes() const;

    std::vector<std::pair<long long, long long>> times() const;

private:
    QString m_lastError;
    QString m_moduleName;
    qint64 m_creationTime;
    QUuid m_collectionId;
    QVariantHash m_userData;

    TSyncFileMode m_tsMode;
    int m_blockSize;

    microseconds_t m_tolerance;
    std::vector<std::pair<long long, long long>> m_times;
    QPair<QString, QString> m_timeNames;
    QPair<TSyncFileTimeUnit, TSyncFileTimeUnit> m_timeUnits;
    QPair<TSyncFileDataType, TSyncFileDataType> m_timeDTypes;
};

} // namespace Syntalos