#include "engine_paths.h"

#include <filesystem>
#include <sstream>
#include "../typedefs.h"

#define SCREENSHOTS_FOLDER "screenshots"

namespace fs = std::filesystem;
using std::string;
using std::vector;
using fs::path;

path EnginePaths::getUserfiles() const {
	return userfiles;
}

path EnginePaths::getResources() const {
	return resources;
}

path EnginePaths::getScreenshotFile(string ext) {
	path folder = userfiles/path(SCREENSHOTS_FOLDER);
	if (!fs::is_directory(folder)) {
		fs::create_directory(folder);
	}

	auto t = std::time(nullptr);
    auto tm = *std::localtime(&t);

	const char* format = "%Y-%m-%d_%H-%M-%S";
	std::stringstream ss;
	ss << std::put_time(&tm, format);
	string datetimestr = ss.str();

	path filename = folder/path("screenshot-"+datetimestr+"."+ext);
	uint index = 0;
	while (fs::exists(filename)) {
		filename = folder/path("screenshot-"+datetimestr+"-"+std::to_string(index)+"."+ext);
		index++;
	}
	return filename;
}

path EnginePaths::getWorldsFolder() {
    return userfiles/path("worlds");
}

bool EnginePaths::isWorldNameUsed(string name) {
	return fs::exists(EnginePaths::getWorldsFolder()/fs::u8path(name));
}

void EnginePaths::setUserfiles(path folder) {
	this->userfiles = folder;
}

void EnginePaths::setResources(path folder) {
	this->resources = folder;
}

ResPaths::ResPaths(path mainRoot, vector<path> roots) 
    : mainRoot(mainRoot), roots(roots) {
}

path ResPaths::find(const string& filename) const {
    for (auto& root : roots) {
        path file = root / path(filename);
        if (fs::exists(file)) {
            return file;
        }
    }
    return mainRoot / path(filename);
}

vector<path> ResPaths::listdir(const string& folderName) const {
    vector<path> entries;
    for (auto& root : roots) {
        path folder = root / path(folderName);
        if (!fs::is_directory(folder))
            continue;
        for (const auto& entry : fs::directory_iterator(folder)) {
            entries.push_back(entry.path());
        }
    }
    {
        path folder = mainRoot / path(folderName);
        if (!fs::is_directory(folder))
            return entries;
        for (const auto& entry : fs::directory_iterator(folder)) {
            entries.push_back(entry.path());
        }
    }
    return entries;
}