VncClientWidget.h

#pragma once
#include "../Headers/VncDefines.h"
#include <NDEVR/NDEVRMouseManager.h>
#include <NDEVR/Device.h>
#include <NDEVR/BaseValues.h>
#include <NDEVR/String.h>
#include <NDEVR/Time.h>
#include <QWidget>
#include <QKeyEvent>
#include <QTcpSocket>
namespace NDEVR
{
    class TcpServer;
    struct VncScreen
    {
        uint04 id;    
        QRect rect;   
        uint04 flags; 
    };
    class VncClientWidget;
    class VncDevice : public Device
    {
        Q_OBJECT
    public:
        explicit VncDevice(VncClientWidget* parent);
        ~VncDevice();
        void update(Time time) override;
        [[nodiscard]] virtual TimeSpan updateInterval() const override;
        bool connectToVncServer(const StringView& ip, const StringView& password, int port = 5900);
        bool connectToVncServer();
        bool isConnectedToServer();
        void disconnectFromVncServer();
        void requestFullRefresh();
        
        
        bool postEvent(QEvent* event);
        void postKeyEvent(QKeyEvent* event, bool is_down);
        
    protected:
        void waitToRead(uint04 size) const;
        bool sendCommand(const char* const command, uint04 min_response_size, uint04 retry_count);
        
        void sendEncodings();
        void sendContinuousUpdateRequest(bool enable);
        void sendUpdateFrameBufferRequest(bool force_refresh_all);
        void sendRequestDesktopSize(QSize size, fltp08 scale_factor = 1.0);
        void sendMouseEvent(const MouseEvent& event);
        void sendKeyEvent(bool is_pressed, QKeyEvent* event);
        void sendKeyEvent(bool is_pressed, const KeyEvent& event);
        bool setupVncConnection(const String& password);
        void tearDownTcp();
        
        QByteArray desHash(QByteArray challenge, QString passStr);
        uint02 makeU16(quint8 l, quint8 h);
        uint04 makeU32(quint16 l, quint16 h);
        uint04 makeU32(quint8 lowest, quint8 low, quint8 high, quint8 highest);
        static quint32 TranslateRfbKey(int key, bool modifier);
        void readFrameBufferMessage();
        void setFrameBufferSize(const QSize& size);
    private slots:
        void onServerMessage();
    protected:
        NDEVRMouseManager m_mouse_manager;         
        VncClientWidget* m_widget;                 
        QTcpSocket* m_socket;                      
        QSize m_frame_size;                        
        Time m_last_full_refresh_time;             
        Time m_last_refresh_time;                  
        String m_password;                         
        String m_connection;                       
        Buffer<VncScreen> m_screens;               
        PixelFormat m_pixel_format;                
        Buffer<KeyEvent> m_key_events;             
        Buffer<bool> m_key_down_events;            
        BitFlag m_mouse_button_flag = BitFlag(0);  
        QImage m_copy_image;                       
        int m_port;                                
 
        fltp04 m_jpg_quality;                      
        bool m_has_sent_frame_request = false;     
        bool m_setup_connection = false;           
        bool m_request_full_refresh = false;       
        bool m_is_continuous_mode = false;         
        bool m_supports_continuous_updates = false; 
        bool m_supports_extended_size = false;     
        bool m_request_disconnect = false;         
        bool m_request_client_resize = true;       
        bool m_is_ready = false;                   
 
    };

    class VncClientWidget : public QWidget
    {
        static constexpr uint01 s_screen_count = 2; 
        Q_OBJECT
    public:
        explicit VncClientWidget(QWidget* parent = 0);
        ~VncClientWidget();
        void mousePressEvent(QMouseEvent* event) override;
        bool connectToVncServer(const StringView& ip, const StringView& password, int port = 5900);
        bool isConnectedToServer();
        void disconnectFromVncServer();
        void tryRefreshScreen(bool force_refresh_all);
        void setResizeTcpServer(TcpServer* server, fltp08 size_multiplier);
        void setMouse(const Vector<2, uint04>& hotspot, QImage mouse);
        QImage* imageToRender();
        void refreshTcpSize();
        void commit(const QRegion& region);
    protected:
        bool event(QEvent* event) override;
        void paintEvent(QPaintEvent*) override;
        void keyPressEvent(QKeyEvent* event) override;
        void keyReleaseEvent(QKeyEvent* event) override;
        void resizeEvent(QResizeEvent* event) override;
    private:
        VncDevice* m_device;                       
        TcpServer* m_tcp_server;                   
        fltp08 m_size_multiplier;                  
        QImage m_screen_buffer[s_screen_count];    
        uint01 m_screen_index = 0;                 
        bool m_is_painting;                        
    };
}