ImageServer.hpp

Go to the documentation of this file.

#pragma once
 
#include <fineftp/server.h>
 
#include <atomic>
#include <condition_variable>
#include <cstdint>
#include <deque>
#include <functional>
#include <mutex>
#include <string>
#include <thread>
 
#include "OpenSRX/Image.hpp"
 
namespace OpenSRX {

class ImageServer {
   public:
    ImageServer(uint16_t port = 0, const std::string& username = "opensrx",
                const std::string& password = "opensrx");
 
    ~ImageServer();
 
    // Non-copyable / non-movable
    ImageServer(const ImageServer&) = delete;
    ImageServer& operator=(const ImageServer&) = delete;

    void start();

    void stop();

    bool isRunning() const { return running.load(); }

    uint16_t getPort() const;

    std::string getUsername() const { return username; }

    std::string getPassword() const { return password; }

    std::string getRootPath() const { return rootPath; }

    Image waitForImage();

    bool tryGetImage(Image& image);

    std::deque<Image> getImages();

    void setImageCallback(std::function<void(const Image&)> cb);
 
   private:
    void watchDirectory();
 
    uint16_t port;
    std::string username;
    std::string password;
    std::string rootPath;
 
    std::unique_ptr<fineftp::FtpServer> ftpServer;
 
    std::atomic<bool> running{false};
    std::thread watcherThread;
 
    std::mutex queueMutex;
    std::condition_variable queueCV;
    std::deque<Image> imageQueue;
 
    std::function<void(const Image&)> imageCallback;
};
 
}  // namespace OpenSRX