#pragma once

#include <platform_shared/message.pb.h>
#include <eventbus.hpp>
#include <filesystem.h>
#include <map>
#include <string>
#include <functional>
#include <cstdio>

#define FS_MAX_CHUNK_SIZE 16384
#define FS_TRANSFER_TIMEOUT_MS 30000

namespace FileSystemWS {

struct DownloadState {
    std::string path;
    FILE* file;
    uint32_t fileSize;
    uint32_t chunkSize;
    uint32_t totalChunks;
    uint32_t chunksSent;
    uint32_t lastActivityTime;
    int clientId;
};

struct UploadState {
    std::string path;
    FILE* file;
    uint32_t fileSize;
    uint32_t totalChunks;
    uint32_t chunksReceived;
    uint32_t bytesReceived;
    uint32_t lastActivityTime;
    int clientId;
    bool hasError;
    std::string errorMessage;
};

using SendMetadataCallback = std::function<void(const socket_message_FSDownloadMetadata&, int clientId)>;
using SendCallback = std::function<void(const socket_message_FSDownloadData&, int clientId)>;
using SendCompleteCallback = std::function<void(const socket_message_FSDownloadComplete&, int clientId)>;
using SendUploadCompleteCallback = std::function<void(const socket_message_FSUploadComplete&, int clientId)>;

class FileSystemHandler {
  public:
    FileSystemHandler();

    void begin();
    void setSendCallbacks(SendMetadataCallback sendMetadata, SendCallback sendData, SendCompleteCallback sendComplete,
                          SendUploadCompleteCallback sendUploadComplete);

    socket_message_FSDeleteResponse handleDelete(const socket_message_FSDeleteRequest& req);
    socket_message_FSMkdirResponse handleMkdir(const socket_message_FSMkdirRequest& req);
    socket_message_FSListResponse handleList(const socket_message_FSListRequest& req);
    void handleDownloadRequest(const socket_message_FSDownloadRequest& req, int clientId);
    socket_message_FSUploadStartResponse handleUploadStart(const socket_message_FSUploadStart& req, int clientId);
    void handleUploadData(const socket_message_FSUploadData& req);
    socket_message_FSCancelTransferResponse handleCancelTransfer(const socket_message_FSCancelTransfer& req);
    void cleanupExpiredTransfers();
    void processPendingDownloads();

  private:
    std::map<uint32_t, DownloadState> downloads_;
    std::map<uint32_t, UploadState> uploads_;
    uint32_t transferIdCounter_;

    inline uint32_t generateTransferId() { return ++transferIdCounter_; }

    SendMetadataCallback sendMetadataCallback_;
    SendCallback sendDataCallback_;
    SendCompleteCallback sendCompleteCallback_;
    SendUploadCompleteCallback sendUploadCompleteCallback_;

    void listDirectory(const std::string& path, socket_message_FSListResponse& response);
    bool deleteRecursive(const std::string& path);
    bool sendNextDownloadChunk(uint32_t transferId);
    void finalizeUpload(uint32_t transferId, bool success, const std::string& error = "");

    SubscriptionHandle uploadHandle_;
};

extern FileSystemHandler fsHandler;

} // namespace FileSystemWS