Generic connection manager for Tox and XMPP.

1 files changed, 7 insertions, 4 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index 4c0cd114..abf0c297 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -1353,8 +1353,9 @@ main = do
         return ()
 
     -- XMPP initialization
-    cw    <- newConsoleWriter
-    state <- newPresenceState cw
+    cw        <- newConsoleWriter
+    serverVar <- atomically $ newEmptyTMVar
+    state     <- newPresenceState cw serverVar
 
     -- XMPP stanza handling
     runResourceT $ do
@@ -1363,8 +1364,10 @@ main = do
         -- we put it into the 'server' field of our /state/ record.
 
         liftIO $ do
-        atomically $ putTMVar (server state) sv -- Okay, now it's ready. :)
-                                                -- FIXME: This is error prone.
+        conns <- xmppConnections sv
+        atomically $ do
+            putTMVar serverVar  (sv,conns) -- Okay, now it's ready. :)
+                                           -- FIXME: This is error prone.
 
         forkIO $ do
             myThreadId >>= flip labelThread "XMPP.stanzas"