From 207f3a524818aeccd1d8d4e8259a685e06e296cb Mon Sep 17 00:00:00 2001
From: Aaron Bedra <aaron@aaronbedra.com>
Date: Sat, 13 Jan 2024 12:57:50 -0600
Subject: [PATCH] Fix broken state

---
 simple_websocket.hpp | 35 +++++++++++++++++++++++++++++------
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/simple_websocket.hpp b/simple_websocket.hpp
index 0afac91..45558ce 100644
--- a/simple_websocket.hpp
+++ b/simple_websocket.hpp
@@ -5,6 +5,7 @@
 #include <vector>
 #include <variant>
 #include <memory>
+#include <functional>
 
 namespace SimpleWebSocket {
     template<class... As>
@@ -287,6 +288,32 @@ namespace SimpleWebSocket {
         std::variant<Failure, std::monostate> value_;
     };
 
+    struct Workflow final {
+        explicit Workflow(std::function<WorkflowResult()> runFn,
+                          std::function<void(const std::monostate &)> successFn,
+                          std::function<void(const Failure &)> recoveryFn)
+                : runFn_(std::move(runFn))
+                , successFn_(std::move(successFn))
+                , recoveryFn_(std::move(recoveryFn))
+        { }
+
+        void runUntilCancelled() {
+            WorkflowResult workflowResult = runFn_();
+            workflowResult.template match<void>(recoveryFn_, successFn_);
+
+            while (!workflowResult.complete()) {
+                workflowResult = runFn_();
+                workflowResult.template match<void>(recoveryFn_, successFn_);
+            }
+        }
+
+    private:
+        const std::function<WorkflowResult()> runFn_;
+        const std::function<void(const std::monostate &)> successFn_;
+        const std::function<void(const Failure &)> recoveryFn_;
+    };
+
+
     struct ExecutionContext final {
         ExecutionContext(std::string host, uint16_t port, std::string uri)
                 : host_(std::move(host)), port_(port), uri_(std::move(uri)) {}
@@ -316,7 +343,8 @@ namespace SimpleWebSocket {
 #include <Poco/Net/HTTPClientSession.h>
 #include <Poco/Net/HTTPRequest.h>
 #include <Poco/Net/HTTPResponse.h>
-#include <Poco/Net/WebSocket.h>
+#include <Poco/Net/NetSSL.h>
+#include <Poco/Net/HTTPSClientSession.h>
 
 namespace SimpleWebSocket::Poco {
     constexpr int PING_FRAME   = static_cast<int>(::Poco::Net::WebSocket::FRAME_FLAG_FIN) |
@@ -384,10 +412,6 @@ namespace SimpleWebSocket::Poco {
       return Wrapper<SIZE>{::Poco::Net::WebSocket{session, request, response}};
     }
 
-#if __has_include(<Poco/Net/SSLManager.h>)
-#include <Poco/Net/SSLManager.h>
-#include <Poco/Net/HTTPSClientSession.h>
-
     template<int SIZE>
     inline Wrapper<SIZE> tls_wrapper(const std::string& host,
                                      ::Poco::UInt16 port,
@@ -399,6 +423,5 @@ namespace SimpleWebSocket::Poco {
 
       return Wrapper<SIZE>{::Poco::Net::WebSocket{session, request, response}};
     }
-#endif
 }
 #endif
\ No newline at end of file