#include <websocketpp/config/asio_no_tls.hpp>
#include <websocketpp/server.hpp>
#include "logger.hpp"

namespace bite_im {
typedef websocketpp::server<websocketpp::config::asio> server_t;
// 连接的类型： server_t::connection_ptr

class Connection {
    public:
        struct Client {
            Client(const std::string &u, const std::string &s):uid(u), ssid(s){}
            std::string uid;
            std::string ssid;
        };
        using ptr = std::shared_ptr<Connection>;
        Connection(){}
        ~Connection() {}
        void insert(const server_t::connection_ptr &conn, 
            const std::string &uid, const std::string &ssid) {
            std::unique_lock<std::mutex> lock(_mutex);
            _uid_connections.insert(std::make_pair(uid, conn));
            _conn_clients.insert(std::make_pair(conn, Client(uid, ssid)));
            LOG_DEBUG("新增长连接用户信息：{}-{}-{}", (size_t)conn.get(), uid, ssid);
        }
        server_t::connection_ptr connection(const std::string &uid) {
            std::unique_lock<std::mutex> lock(_mutex);
            auto it = _uid_connections.find(uid);
            if (it == _uid_connections.end()) {
                LOG_ERROR("未找到 {} 客户端的长连接！", uid);
                return server_t::connection_ptr();
            }
            LOG_DEBUG("找到 {} 客户端的长连接！", uid);
            return it->second;
        }
        bool client(const server_t::connection_ptr &conn, std::string &uid, std::string &ssid) {
            std::unique_lock<std::mutex> lock(_mutex);
            auto it = _conn_clients.find(conn);
            if (it == _conn_clients.end()) {
                LOG_ERROR("获取-未找到长连接 {} 对应的客户端信息！", (size_t)conn.get());
                return false;
            }
            uid = it->second.uid;
            ssid = it->second.ssid;
            LOG_DEBUG("获取长连接客户端信息成功！");
            return true;
        }
        void remove(const server_t::connection_ptr &conn) {
            std::unique_lock<std::mutex> lock(_mutex);
            auto it = _conn_clients.find(conn);
            if (it == _conn_clients.end()) {
                LOG_ERROR("删除-未找到长连接 {} 对应的客户端信息！", (size_t)conn.get());
                return;
            }
            _uid_connections.erase(it->second.uid);
            _conn_clients.erase(it);
            LOG_DEBUG("删除长连接信息完毕！");
        }
    private:
        std::mutex _mutex;
        std::unordered_map<std::string, server_t::connection_ptr> _uid_connections;
        std::unordered_map<server_t::connection_ptr, Client> _conn_clients;
};
}