/*
 * Copyright (c) 2014-2024 Meltytech, LLC
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "metadatamodel.h"
#include "controllers/filtercontroller.h"
#include "mainwindow.h"
#include "qmltypes/qmlmetadata.h"
#include "settings.h"
#include <Logger.h>

#include <QGuiApplication>
#include <QClipboard>
#include <QSaveFile>

MetadataModel::MetadataModel(QObject *parent)
    : QSortFilterProxyModel(parent)
    , m_filter(FavoritesFilter)
    , m_isClipProducer(true)
    , m_filterMask(HiddenMaskBit)
{
    if (Settings.playerGPU()) {
        m_filterMask |= gpuIncompatibleMaskBit;
    } else {
        m_filterMask |= needsGPUMaskBit;
    }
    setSourceModel(new InternalMetadataModel(this));
}

int MetadataModel::rowCount(const QModelIndex &parent) const
{
    return QSortFilterProxyModel::rowCount(parent);
}

int MetadataModel::sourceRowCount(const QModelIndex &parent) const
{
    return static_cast<InternalMetadataModel *>(sourceModel())->rowCount();
}

int InternalMetadataModel::rowCount(const QModelIndex &) const
{
    return m_list.size();
}

QVariant InternalMetadataModel::data(const QModelIndex &index, int role) const
{
    QVariant result;
    QmlMetadata *meta = m_list.at(index.row());

    if (meta) {
        switch (role) {
        case Qt::DisplayRole:
        case MetadataModel::NameRole:
            result = meta->name();
            break;
        case MetadataModel::HiddenRole:
            result = meta->isHidden();
            break;
        case MetadataModel::FavoriteRole:
            result = meta->isFavorite();
            break;
        case MetadataModel::ServiceRole:
            result = meta->mlt_service();
            break;
        case MetadataModel::IsAudioRole:
            result = meta->isAudio();
            break;
        case MetadataModel::NeedsGpuRole:
            result = meta->needsGPU();
            break;
        case MetadataModel::PluginTypeRole:
            result = meta->type();
            break;
        }
    }

    return result;
}

bool InternalMetadataModel::setData(const QModelIndex &index, const QVariant &value, int role)
{
    if (!index.isValid()) return false;
    switch (role) {
    case MetadataModel::FavoriteRole: {
        QmlMetadata *meta = m_list.at(index.row());
        meta->setIsFavorite(value.value<bool>());
        emit dataChanged(index, index);
        break;
    }
    }
    return true;
}

QHash<int, QByteArray> InternalMetadataModel::roleNames() const
{
    QHash<int, QByteArray> roles = QAbstractListModel::roleNames();
    roles[MetadataModel::NameRole] = "name";
    roles[MetadataModel::HiddenRole] = "hidden";
    roles[MetadataModel::FavoriteRole] = "favorite";
    roles[MetadataModel::ServiceRole] = "service";
    roles[MetadataModel::IsAudioRole] = "isAudio";
    roles[MetadataModel::NeedsGpuRole] = "needsGpu";
    roles[MetadataModel::PluginTypeRole] = "pluginType";
    return roles;
}

Qt::ItemFlags InternalMetadataModel::flags(const QModelIndex &index) const
{
    if (!index.isValid())
        return Qt::NoItemFlags;
    return QAbstractListModel::flags(index) | Qt::ItemIsEditable | Qt::ItemIsSelectable;
}

void MetadataModel::add(QmlMetadata *data)
{
    static_cast<InternalMetadataModel *>(sourceModel())->add(data);
}

void InternalMetadataModel::add(QmlMetadata *data)
{
    int i = 0;
    for ( i = 0; i < m_list.size(); i++ ) {
        if (m_list[i]->name().toLower() > data->name().toLower() ) {
            break;
        }
    }

    data->filterMask = computeFilterMask(data);
    beginInsertRows(QModelIndex(), i, i);
    m_list.insert(i, data);
    endInsertRows();

    data->setParent(this);
}

QmlMetadata *MetadataModel::get(int row) const
{
    auto sourceIndex = mapToSource(index(row, 0));
    return getFromSource(sourceIndex.row());
}

QmlMetadata *MetadataModel::getFromSource(int index) const
{
    return static_cast<InternalMetadataModel *>(sourceModel())->get(index);
}

QmlMetadata *InternalMetadataModel::get(int index) const
{
    if (index >= 0 && index < m_list.size()) {
        return m_list[index];
    }
    return nullptr;
}

void InternalMetadataModel::remove(int index)
{
    beginRemoveRows(QModelIndex(), index, index);
    m_list.remove(index);
    endRemoveRows();
}

void MetadataModel::setFilter(MetadataFilter filter)
{
    m_filter = filter;
    emit filterChanged();
    invalidateFilter();
}

void MetadataModel::setSearch(const QString &search)
{
    m_search = search;
    emit searchChanged();
    invalidateFilter();
}

bool MetadataModel::filterAcceptsRow(int sourceRow, const QModelIndex &sourceParent) const
{
    auto meta = getFromSource(sourceRow);
    if (meta->filterMask & m_filterMask) {
        return false;
    }
    if (Settings.playerGPU() && meta->needsGPU() && MAIN.filterController()->isOutputTrackSelected()) {
        return false;
    }
    if (m_search.isEmpty()) {
        switch (m_filter) {
        case FavoritesFilter:
            if (!meta->isFavorite()) return false;
            break;
        case VideoFilter:
            if (meta->isAudio() || meta->needsGPU() || meta->type() == QmlMetadata::Link
                    || meta->type() == QmlMetadata::FilterSet) return false;
            break;
        case AudioFilter:
            if (!meta->isAudio()) return false;
            break;
        case LinkFilter:
            if (meta->type() != QmlMetadata::Link) return false;
            break;
        case FilterSetFilter:
            if (meta->type() != QmlMetadata::FilterSet) return false;
            break;
        case GPUFilter:
            if (!meta->needsGPU()) return false;
            break;
        default:
            break;
        }
    } else if (!meta->name().contains(m_search, Qt::CaseInsensitive)
               && !meta->keywords().contains(m_search, Qt::CaseInsensitive)) {
        return false;
    }
    return true;
}

void MetadataModel::updateFilterMask(bool isClipProducer, bool isChainProducer,
                                     bool isTrackProducer, bool isOutputProducer)
{
    beginResetModel();
    m_isClipProducer = isClipProducer;
    if (m_isClipProducer) {
        m_filterMask &= ~clipOnlyMaskBit;
    } else {
        m_filterMask |= clipOnlyMaskBit;
    }
    m_isChainProducer = isChainProducer;
    if (m_isChainProducer) {
        m_filterMask &= ~linkMaskBit;
    } else {
        m_filterMask |= linkMaskBit;
    }
    m_isTrackProducer = isTrackProducer;
    if (m_isTrackProducer) {
        m_filterMask &= ~trackOnlyMaskBit;
    } else {
        m_filterMask |= trackOnlyMaskBit;
    }
    m_isOutputProducer = isOutputProducer;
    if (m_isOutputProducer) {
        m_filterMask &= ~outputOnlyMaskBit;
    } else {
        m_filterMask |= outputOnlyMaskBit;
    }
    endResetModel();
}

unsigned InternalMetadataModel::computeFilterMask(const QmlMetadata *meta)
{
    unsigned mask = 0;
    if (meta->isHidden()) mask |= MetadataModel::HiddenMaskBit;
    if (meta->isClipOnly()) mask |= MetadataModel::clipOnlyMaskBit;
    if (meta->isTrackOnly()) mask |= MetadataModel::trackOnlyMaskBit;
    if (meta->isOutputOnly()) mask |= MetadataModel::outputOnlyMaskBit;
    if (!meta->isGpuCompatible()) mask |= MetadataModel::gpuIncompatibleMaskBit;
    if (meta->needsGPU()) mask |= MetadataModel::needsGPUMaskBit;
    if (meta->type() == QmlMetadata::Link) mask |= MetadataModel::linkMaskBit;
    return mask;
}

void MetadataModel::saveFilterSet(const QString &name)
{
    QDir dir(Settings.appDataLocation());
    const auto folder = QString::fromLatin1("filter-sets");

    if (!dir.exists())
        dir.mkpath(dir.path());
    if (!dir.cd(folder)) {
        if (dir.mkdir(folder))
            dir.cd(folder);
    }

    auto filename = QString::fromUtf8(QUrl::toPercentEncoding(name));
    auto exists = dir.exists(filename);
    QSaveFile file(dir.filePath(filename));
    file.setDirectWriteFallback(true);
    if (!file.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
        LOG_ERROR() << "failed to open filter set file for writing" << file.fileName();
        return;
    }
    QTextStream stream(&file);
    stream.setEncoding(QStringConverter::Utf8);
    stream << QGuiApplication::clipboard()->text();
    if (file.error() != QFileDevice::NoError) {
        LOG_ERROR() << "error while writing filter set file" << file.fileName() << ":" <<
                    file.errorString();
        return;
    }
    if (file.commit() && !exists) {
        auto meta = new QmlMetadata;
        meta->setType(QmlMetadata::FilterSet);
        meta->setName(name);
        auto model = static_cast<InternalMetadataModel *>(sourceModel());
        model->add(meta);
    }
}

void MetadataModel::deleteFilterSet(const QString &name)
{
    QDir dir(Settings.appDataLocation());
    if (!dir.cd("filter-sets"))
        return;

    auto fileName = QUrl::toPercentEncoding(name.toUtf8());
    if (QFile::remove(dir.filePath(fileName)) || QFile::remove(dir.filePath(name))) {
        auto i = 0;
        auto source = static_cast<InternalMetadataModel *>(sourceModel());
        auto list = source->list();
        for (const auto &meta : list) {
            if (meta->type() == QmlMetadata::FilterSet && meta->name() == name)
                source->remove(i);
            ++i;
        }
    }
}
