1 files changed, 117 insertions, 0 deletions
diff --git a/src/wm_connection.hpp b/src/wm_connection.hpp
new file mode 100644
index 0000000..91f870e
--- /dev/null
+++ b/src/wm_connection.hpp
@@ -0,0 +1,117 @@
+/*
+ * Insert Copyright if needed.
+ */
+
+#ifndef WM_CONNECTION_HPP
+#define WM_CONNECTION_HPP
+
+#include <functional>
+#include <unordered_map>
+
+struct json_object;
+
+namespace wm
+{
+
+class SocketInfo
+{
+    public:
+        SocketInfo() {};
+        SocketInfo(std::string screen_name, std::string ip, int wm_port, bool master_mode) : _screen_name(screen_name),
+            _ip(ip), _wm_port(wm_port), _master_mode(master_mode){}
+        std::string screenName() const { return _screen_name; }
+        std::string ip() const { return _ip; }
+        int wmPort() const { return _wm_port; }
+        bool masterMode() const { return _master_mode; }
+        int mySocket() const { return _my_socket; }
+        int connectedSocket() const { return _connected_socket; }
+        void setMySocket(int socket) { this->_my_socket = socket; }
+        void setConnectedSocket(int socket) { this->_connected_socket = socket; }
+
+    private:
+        std::string _screen_name;
+        std::string _ip;
+        int _wm_port;
+        bool _master_mode;
+        int _my_socket = -1;
+        int _connected_socket = -1;
+};
+
+class WMConnection
+{
+  public:
+    WMConnection();
+    ~WMConnection() = default;
+
+    using ReceivedHandler = std::function<void(json_object* j_out)>;
+
+    enum ModeType
+    {
+        Mode_Standalone = 0,
+        Mode_Connection,
+    };
+
+    int initialize();
+    void registerCallback(ReceivedHandler on_received);
+    int sendRequest(char const *req, char const *appid,
+                    char const *drawing_name, char const *drawing_area);
+    bool isRemoteArea(const char* area);
+    bool isConnecting(std::string ecu_name);
+    bool isConnectionMode();
+    std::string parseMasterArea(const char* area);
+    bool isSyncDrawingForRemote(const char* role);
+    void startSyncDrawForRemote(const char* role);
+    void finishSyncDrawForRemote(const char* role);
+    int getMySocket();
+    int getConnectedSocket();
+    void setConnectedSocket(int connected_socket);
+    std::string getAreaToEcuName(const char* area);
+    std::string getSocketToEcuName(int socket);
+    std::string getMySocketToEcuName(int socket);
+    std::string getAppIdToEcuName(std::string appid);
+    std::string getIpToEcuName(std::string ip);
+    std::string getEcuName();
+    bool getEndInit();
+    void setAppIdToEcuName(std::string appid, std::string ecu_name);
+    void setAppIdToReceivedEcuName(std::string appid);
+    void callOnReceivedHandler(json_object *j_out);
+    int connectToEcu();
+    int connectToServer(std::string ecu_name);
+    int connectedToServer(int socket);
+    void connectionTimeLimit();
+
+    int serverAccept(int socket);
+    int receive(json_object** j_out, int *j_cnt, int socket);
+
+  private:
+    std::string mode;
+    std::string received_ecu_name;
+    ReceivedHandler onReceived;
+    std::string syndDrawingAppId;
+
+    std::string ecu_name;
+    std::string received_ecu;
+
+    uint64_t times;
+    uint64_t sleep;
+
+    bool end_init;
+    ModeType wm_mode;
+
+    std::unordered_map<std::string, SocketInfo> wm_socket;
+    std::unordered_map<std::string, std::string> appid2ecuName;
+
+    int initializeServer();
+    int initializeClient(std::string ecu_name);
+    int loadTimeoutConfigFile();
+    int loadConnectionConfigFile();
+
+    int send(json_object* j_in, std::string ecu_name);
+
+    uint64_t getNextTimerTime(uint64_t msec);
+};
+
+} // namespace wm
+
+#endif // WM_CONNECTION_HPP
+