// integration_tests/failures/error_handling_test.cpp:58
// TODO: Fix root cause in io_context lifecycle management (Issue #348)

// integration_tests/scenarios/connection_lifecycle_test.cpp:109
// TODO: Fix root cause in io_context lifecycle management (Issue #348)

// integration_tests/scenarios/connection_lifecycle_test.cpp:224
// TODO: Fix root cause in io_context lifecycle management (Issue #315)

class connection_context {
public:
    explicit connection_context()
        : io_context_(std::make_unique<asio::io_context>())
        , work_guard_(std::make_unique<work_guard_t>(io_context_->get_executor()))
    {}
    
    ~connection_context() {
        shutdown();
    }
    
    void shutdown() {
        // 1. Stop accepting new work
        work_guard_.reset();
        
        // 2. Cancel all outstanding operations
        io_context_->stop();
        
        // 3. Wait for io thread to complete
        if (io_thread_.joinable()) {
            io_thread_.join();
        }
    }
    
private:
    std::unique_ptr<asio::io_context> io_context_;
    std::unique_ptr<work_guard_t> work_guard_;
    std::thread io_thread_;
};

template<typename Handler>
auto make_safe_handler(std::weak_ptr<connection> weak_conn, Handler&& handler) {
    return [weak_conn, handler = std::forward<Handler>(handler)]
           (auto&&... args) {
        if (auto conn = weak_conn.lock()) {
            handler(std::forward<decltype(args)>(args)...);
        }
        // Silently drop callback if connection is gone
    };
}

auto connection::stop() -> VoidResult {
    // 1. Set stopping flag
    stopping_.store(true);
    
    // 2. Cancel all pending operations
    socket_->cancel();
    
    // 3. Close socket
    socket_->close();
    
    // 4. Wait for callbacks to drain
    drain_pending_callbacks();
    
    // 5. Stop io_context
    context_->shutdown();
    
    return common::ok();
}