#include "imageio.h"

#include "png.h"
#include "../graphics/core/ImageData.hpp"

#include <filesystem>
#include <functional>
#include <unordered_map>

namespace fs = std::filesystem;

using image_reader = std::function<ImageData*(const std::string&)>;
using image_writer = std::function<void(const std::string&, const ImageData*)>;

static std::unordered_map<std::string, image_reader> readers {
    {".png", png::load_image},
};

static std::unordered_map<std::string, image_writer> writers {
    {".png", png::write_image},
};

bool imageio::is_read_supported(const std::string& extension) {
    return readers.find(extension) != readers.end();
}

bool imageio::is_write_supported(const std::string& extension) {
    return writers.find(extension) != writers.end();
}

inline std::string extensionOf(const std::string& filename) {
    return fs::u8path(filename).extension().u8string();
}

std::unique_ptr<ImageData> imageio::read(const std::string& filename) {
    auto found = readers.find(extensionOf(filename));
    if (found == readers.end()) {
        throw std::runtime_error("file format is not supported (read): "+filename);
    }
    return std::unique_ptr<ImageData>(found->second(filename));
}

void imageio::write(const std::string& filename, const ImageData* image) {
    auto found = writers.find(extensionOf(filename));
    if (found == writers.end()) {
        throw std::runtime_error("file format is not supported (write): "+filename);
    }
    return found->second(filename, image);
}