#ifndef FileExplorer_h
#define FileExplorer_h

#include <ESPFS.h>
#include <PsychicHttp.h>
#include <SecurityManager.h>

#define FILE_EXPLORER_SERVICE_PATH "/api/files"
#define FILE_EXPLORER_DELETE_SERVICE_PATH "/api/files/delete"

class FileExplorer
{
  public:
    FileExplorer(PsychicHttpServer *server, SecurityManager *securityManager)
        : _server(server), _securityManager(securityManager)
    {
    }

    void begin()
    {
        _server->on(FILE_EXPLORER_SERVICE_PATH, HTTP_GET,
                    _securityManager->wrapRequest(std::bind(&FileExplorer::explore, this, std::placeholders::_1),
                                                  AuthenticationPredicates::IS_AUTHENTICATED));
        _server->on(FILE_EXPLORER_DELETE_SERVICE_PATH, HTTP_POST,
                    _securityManager->wrapCallback(std::bind(&FileExplorer::deleteFile, this, std::placeholders::_1, std::placeholders::_2),
                                                  AuthenticationPredicates::IS_AUTHENTICATED));

        ESP_LOGV("APStatus", "Registered GET endpoint: %s", FILE_EXPLORER_SERVICE_PATH);
    }

  private:
    PsychicHttpServer *_server;
    SecurityManager *_securityManager;

    esp_err_t explore(PsychicRequest *request)
    {
        return request->reply(200, "application/json", listFiles("/").c_str());
    }

    esp_err_t deleteFile(PsychicRequest *request, JsonVariant &json)
    {
        if (json.is<JsonObject>())
        {
            String filename = json["file"];
            ESP_LOGI("FileExplorer", "Deleting file: %s", filename.c_str());
            return ESPFS.remove(filename.c_str()) ? request->reply(200) : request->reply(500);
        }
        return request->reply(400);
    }

    String listFiles(const String &directory, bool isRoot = true)
    {
        File root = ESPFS.open(directory.startsWith("/") ? directory : "/" + directory);
        if (!root.isDirectory())
        {
            return "";
        }

        File file = root.openNextFile();
        String output = isRoot ? "{ \"root\": {" : "{";

        while (file)
        {
            if (file.isDirectory())
            {
                output += "\"" + String(file.name()) + "\": " + listFiles(file.name(), false) + ", ";
            }
            else
            {
                output += "\"" + String(file.name()) + "\": " + String(file.size()) + ", ";
            }
            file = root.openNextFile();
        }

        if (output.endsWith(", "))
        {
            output.remove(output.length() - 2);
        }
        output += "}";
        if (isRoot)
        {
            output += "}";
        }
        return output;
    }
};

#endif // end FileExplorer_h