#include "story_project.h"

#include <fstream>
#include <iostream>
#include <queue>
#include <filesystem>
#include "json.hpp"

void StoryProject::New(const std::string &uuid, const std::string &file_path)
{
    m_uuid = uuid;
    Initialize(file_path);
}

#define TLV_ARRAY_TYPE  0xAB
#define TLV_OBJECT_TYPE  0xE7
#define TLV_STRING_TYPE  0x3D


class Tlv
{
public:
    explicit Tlv(const std::string &filename)
    {
        m_file = std::ofstream(filename, std::ios::out | std::ios::binary);
    }

    ~Tlv() {
        m_file.close();
    }

    void add_array(uint16_t size)
    {
        m_file.write(reinterpret_cast<const char*>(&m_objectType), sizeof(m_objectType));
        m_file.write(reinterpret_cast<const char*>(&size), sizeof(size));
    }

    void add_string(const char *s, uint16_t size)
    {
        m_file.write(reinterpret_cast<const char*>(&m_stringType), sizeof(m_stringType));
        m_file.write(reinterpret_cast<const char*>(&size), sizeof(size));
        m_file.write(s, size);
    }

    void add_object(uint16_t entries)
    {
        m_file.write(reinterpret_cast<const char*>(&m_arrayType), sizeof(m_arrayType));
        m_file.write(reinterpret_cast<const char*>(&entries), sizeof(entries));
    }

private:
    std::ofstream m_file;

    uint8_t m_arrayType = TLV_ARRAY_TYPE;
    uint8_t m_objectType = TLV_OBJECT_TYPE;
    uint8_t m_stringType = TLV_STRING_TYPE;

};



void StoryProject::SaveStory(const std::vector<uint8_t> &m_program)
{
    std::ofstream o(m_working_dir + std::filesystem::path::preferred_separator + "story.c32", std::ios::out | std::ios::binary);
    o.write(reinterpret_cast<const char*>(m_program.data()), m_program.size());
    o.close();


    Tlv tlv(m_working_dir + std::filesystem::path::preferred_separator + "index.ost");

    tlv.add_array(1);

    tlv.add_object(3);
    tlv.add_string(m_uuid.c_str(), m_uuid.size()); // uuid

    // Title image
    std::string image =  RemoveFileExtension(m_titleImage) + ".qoi";
    tlv.add_string(image.c_str(), image.size());

    std::string sound =  RemoveFileExtension(m_titleSound) + ".wav";
    tlv.add_string(sound.c_str(), sound.size()); // title sound
}

void StoryProject::Initialize(const std::string &file_path)
{
    m_project_file_path = file_path;
    std::filesystem::path p(file_path);
    m_working_dir= p.parent_path().generic_string();

    // Frist try to create the working directory
    if (!std::filesystem::is_directory(m_working_dir))
    {
        std::filesystem::create_directories(m_working_dir);
    }
    m_assetsPath = std::filesystem::path(m_working_dir) /  "assets";

    std::filesystem::create_directories(m_assetsPath);

    m_initialized = true;
}

bool StoryProject::Load(const std::string &file_path, nlohmann::json &model)
{

    std::ifstream f(file_path);
    bool success = false;

    std::filesystem::path p(file_path);
    m_working_dir= p.parent_path().generic_string();

    std::cout << "Working dir is: " << m_working_dir << std::endl;

    try {

        nlohmann::json j = nlohmann::json::parse(f);

        m_nodes.clear();


        if (j.contains("project"))
        {
            nlohmann::json projectData = j["project"];

            m_name = projectData["name"].get<std::string>();
            m_uuid = projectData["uuid"].get<std::string>();
            m_titleImage = projectData.value("title_image", "");
            m_titleSound = projectData.value("title_sound", "");

            if (j.contains("resources"))
            {
                nlohmann::json resourcesData = j["resources"];

                for (const auto &obj : resourcesData)
                {
                    Resource rData;

                    rData.type = obj["type"].get<std::string>();
                    rData.format = obj["format"].get<std::string>();
                    rData.description = obj["description"].get<std::string>();
                    rData.file = obj["file"].get<std::string>();
                    m_resources.push_back(rData);
                }

                if (j.contains("nodegraph"))
                {
                    model = j["nodegraph"];
                    success = true;
                }
            }
        }

 /*

        if (j.contains("nodes"))
        {
            for (auto& element : j["nodes"])
            {
                StoryNode n;

                n.auto_jump = element["auto_jump"];

                for (auto& jump : element["jumps"])
                {
                    n.jumps.push_back(jump.get<int>());
                }

                n.id = element["id"];
                n.image = element["image"];
                n.sound = element["sound"];

                m_nodes.push_back(n);
            }
        }

        m_images.clear();
        if (j.contains("images"))
        {
            for (auto& element : j["images"])
            {
                Resource r;

                r.file = element["file"];
                r.description = element["description"];
                r.format = element["format"];

                m_images.push_back(r);
            }
        }

        m_sounds.clear();
        if (j.contains("sounds"))
        {
            for (auto& element : j["sounds"])
            {
                Resource r;

                r.file = element["file"];
                r.description = element["description"];
                r.format = element["format"];

                m_sounds.push_back(r);
            }
        }

        m_type = j["type"];
        m_code = j["code"];
        m_name = j["name"];

        success = true;

*/
    }
    catch(std::exception &e)
    {
        std::cout << e.what() << std::endl;
    }

    return success;
}

void StoryProject::Save(const nlohmann::json &model)
{
    nlohmann::json j;
    j["project"] = { {"name", m_name}, {"uuid", m_uuid}, { "title_image", m_titleImage }, { "title_sound", m_titleSound } };

    {
        nlohmann::json resourcesData;

        for (auto &r : m_resources)
        {
            nlohmann::json obj = {{"type", r.type},
                                  {"format", r.format},
                                  {"description", r.description},
                                  {"file", r.file}};

            resourcesData.push_back(obj);
        }
        j["resources"] = resourcesData;
    }

    j["nodegraph"] = model;

    std::ofstream o(m_project_file_path);
    o << std::setw(4) << j << std::endl;
}

void StoryProject::CreateTree()
{
    // Algorithm: level order traversal of N-ary tree
    std::queue<StoryNode *> nlist;

    m_tree = &m_nodes[0];
    nlist.push(m_tree);

    while (!nlist.empty())
    {
        StoryNode *p = nlist.front();
        std::cout << "Node: " << p->id << std::endl;

        for (int i = 0; i < p->jumps.size(); i++)
        {
            int jump = p->jumps[i];

            if (jump < m_nodes.size())
            {
                StoryNode *child = &m_nodes[jump];
                nlist.push(child);
                p->children.push_back(child);
            }
            else
            {
                std::cout << "End node" << std::endl;
            }
        }

        nlist.pop();
    }
}

void StoryProject::EraseString(std::string &theString, const std::string &toErase)
{
    std::size_t found;
    found = theString.find(toErase);
    if (found != std::string::npos)
    {
        theString.erase(found, toErase.size());
    }
}

std::string StoryProject::ToUpper(const std::string &input)
{
    std::string str = input;
    std::transform(str.begin(), str.end(), str.begin(), ::toupper);
    return str;
}

std::string StoryProject::GetFileName(const std::string &path)
{
    auto found = path.find_last_of("/\\");
    return path.substr(found+1);
}


void StoryProject::ReplaceCharacter(std::string &theString, const std::string &toFind, const std::string &toReplace)
{
    std::size_t found;
    do
    {
        found = theString.find(toFind);
        if (found != std::string::npos)
        {
            theString.replace(found, 1, toReplace);
        }
    }
    while (found != std::string::npos);
}

std::string StoryProject::RemoveFileExtension(const std::string &FileName)
{
    std::string f = GetFileName(FileName);
    std::string ext = GetFileExtension(f);
    EraseString(f, "." + ext);
    return f;
}

std::string StoryProject::FileToConstant(const std::string &FileName, const std::string &extension)
{
    std::string f = RemoveFileExtension(FileName);
    return "$" + f + " DC8 \"" + f + extension + "\", 8\r\n";
}

void StoryProject::SetTitleImage(const std::string &titleImage)
{
    m_titleImage = titleImage;
}

void StoryProject::SetTitleSound(const std::string &titleSound)
{
    m_titleSound = titleSound;
}

void StoryProject::AppendResource(const Resource &res)
{
    m_resources.push_back(res);
}

bool StoryProject::GetResourceAt(int index, Resource &resOut)
{
    bool success = false;
    if ((index >= 0) && (index < m_resources.size()))
    {
        resOut = m_resources[index];
        success = true;
    }
    return success;
}

void StoryProject::ClearResources()
{
    m_resources.clear();
}

void StoryProject::DeleteResourceAt(int index)
{
    if ((index >= 0) && (index < m_resources.size()))
    {
        m_resources.erase(m_resources.begin() + index);
    }
}

std::string StoryProject::GetFileExtension(const std::string &fileName)
{
    if(fileName.find_last_of(".") != std::string::npos)
        return fileName.substr(fileName.find_last_of(".")+1);
    return "";
}


void StoryProject::SetImageFormat(ImageFormat format)
{
    m_imageFormat = format;
}

void StoryProject::SetSoundFormat(SoundFormat format)
{
    m_soundFormat = format;
}

void StoryProject::SetDisplayFormat(int w, int h)
{
    m_display_w = w;
    m_display_h = h;
}

std::string StoryProject::GetProjectFilePath() const
{
    return m_project_file_path;
}

std::string StoryProject::GetWorkingDir() const
{
    return m_working_dir;
}