#include "MockPlugin.h"
#include "CosinusProvider.h"

#include <DataSource/DataSourceController.h>
#include <DataSource/DataSourceItem.h>

#include <SqpApplication.h>

Q_LOGGING_CATEGORY(LOG_MockPlugin, "MockPlugin")

namespace {

/// Name of the data source
const auto DATA_SOURCE_NAME = QStringLiteral("MMS");

/// Creates the data provider relative to the plugin
std::unique_ptr<IDataProvider> createDataProvider() noexcept
{
    return std::make_unique<CosinusProvider>();
}

/// Creates the data source item relative to the plugin
std::unique_ptr<DataSourceItem> createDataSourceItem() noexcept
{
    // Magnetic field products
    auto fgmProduct = std::make_unique<DataSourceItem>(DataSourceItemType::PRODUCT,
                                                       QVector<QVariant>{QStringLiteral("FGM")});
    auto scProduct = std::make_unique<DataSourceItem>(DataSourceItemType::PRODUCT,
                                                      QVector<QVariant>{QStringLiteral("SC")});

    auto magneticFieldFolder = std::make_unique<DataSourceItem>(
        DataSourceItemType::NODE, QVector<QVariant>{QStringLiteral("Magnetic field")});
    magneticFieldFolder->appendChild(std::move(fgmProduct));
    magneticFieldFolder->appendChild(std::move(scProduct));

    // Electric field products
    auto electricFieldFolder = std::make_unique<DataSourceItem>(
        DataSourceItemType::NODE, QVector<QVariant>{QStringLiteral("Electric field")});

    // Root
    auto root = std::make_unique<DataSourceItem>(DataSourceItemType::NODE,
                                                 QVector<QVariant>{DATA_SOURCE_NAME});
    root->appendChild(std::move(magneticFieldFolder));
    root->appendChild(std::move(electricFieldFolder));

    return root;
}

} // namespace

void MockPlugin::initialize()
{
    if (auto app = sqpApp) {
        // Registers to the data source controller
        auto &dataSourceController = app->dataSourceController();
        auto dataSourceUid = dataSourceController.registerDataSource(DATA_SOURCE_NAME);

        // Sets data source tree
        dataSourceController.setDataSourceItem(dataSourceUid, createDataSourceItem());

        // Sets data provider
        dataSourceController.setDataProvider(dataSourceUid, createDataProvider());
    }
    else {
        qCWarning(LOG_MockPlugin()) << tr("Can't access to SciQlop application");
    }
}