#include "AssetsManagement.hpp"

#include "assets/AssetsLoader.hpp"
#include "coders/GLSLExtension.hpp"
#include "content/Content.hpp"
#include "debug/Logger.hpp"
#include "engine/Engine.hpp"
#include "EnginePaths.hpp"
#include "graphics/core/Shader.hpp"
#include "graphics/render/ModelsGenerator.hpp"
#include "graphics/ui/GUI.hpp"

static debug::Logger logger("assets-management");

AssetsManagement::AssetsManagement(Engine& engine)
    : engine(engine), settings(engine.getSettings()) {
}

AssetsManagement::~AssetsManagement() {
    finishBackgroundLoader();
}

const Assets* AssetsManagement::getStorage() const {
    return assets.get();
}

Assets* AssetsManagement::getStorage() {
    return assets.get();
}

AssetsLoader& AssetsManagement::acquireBackgroundLoader() {
    if (backgroundLoader) {
        return *backgroundLoader;
    }
    if (assets == nullptr) {
        throw std::runtime_error("no assets storage available");
    }
    backgroundLoader = std::make_unique<AssetsLoader>(
        engine, *assets, engine.getResPaths()
    );
    backgroundLoaderTask = backgroundLoader->startTask(
        nullptr, settings.system.maxBgAssetLoaders.get()
    );
    return *backgroundLoader;
}

void AssetsManagement::loadAssets(Content* content) {
    finishBackgroundLoader();

    const auto& paths = engine.getPaths();
    logger.info() << "loading assets";
    Shader::preprocessor->setPaths(&paths.resPaths);

    auto new_assets = std::make_unique<Assets>();
    AssetsLoader loader(engine, *new_assets, paths.resPaths);
    AssetsLoader::addDefaults(loader, content);

    // no need
    // correct log messages order is more useful
    // todo: before setting to true, check if GLSLExtension thread safe 
    bool threading = false; // look at three upper lines
    if (threading) {
        auto task = loader.startTask(
            [=]() {}, 0
        );
        task->waitForEnd();
    } else {
        while (loader.hasNext()) {
            loader.loadNext();
        }
    }
    assets = std::move(new_assets);
    if (content) {
        ModelsGenerator::prepare(*content, *assets);
    }
    assets->setup();
    engine.getGUI().onAssetsLoad(assets.get());
}

void AssetsManagement::update() {
    if (backgroundLoaderTask) {
        backgroundLoaderTask->update();
    }
}

void AssetsManagement::finishBackgroundLoader() {
    if (backgroundLoaderTask == nullptr) {
        return;
    }
    backgroundLoaderTask.reset();
    backgroundLoader.reset();
}