1 files changed, 8 insertions, 2 deletions

diff --git a/src/Network/Tox/ContactInfo.hs b/src/Network/Tox/ContactInfo.hs
index f4c4bf12..0d8a9f8a 100644
--- a/src/Network/Tox/ContactInfo.hs
+++ b/src/Network/Tox/ContactInfo.hs
@@ -1,12 +1,16 @@
 {-# LANGUAGE NamedFieldPuns #-}
 module Network.Tox.ContactInfo where
 
+import ConnectionKey
+
 import Control.Concurrent.STM
 import Control.Monad
 import Crypto.PubKey.Curve25519
 import qualified Data.HashMap.Strict as HashMap
          ;import Data.HashMap.Strict (HashMap)
 import Data.Maybe
+import qualified Data.Set            as Set
+         ;import Data.Set            (Set)
 import Network.Tox.DHT.Transport     as DHT
 import Network.Tox.NodeId
 import Network.Tox.Onion.Transport   as Onion
@@ -15,8 +19,9 @@ import System.IO
 newtype ContactInfo = ContactInfo { accounts :: TVar (HashMap NodeId Account) }
 
 data Account = Account
-    { userSecret :: SecretKey                   -- local secret key
-    , contacts :: TVar (HashMap NodeId Contact) -- received contact info
+    { userSecret :: SecretKey                     -- local secret key
+    , contacts   :: TVar (HashMap NodeId Contact) -- received contact info
+    , clientRefs :: TVar (Set ConnectionKey)
     }
 
 data Contact = Contact
@@ -41,6 +46,7 @@ newContactInfo = atomically $ ContactInfo <$> newTVar HashMap.empty
 
 newAccount :: SecretKey -> STM Account
 newAccount sk = Account sk <$> newTVar HashMap.empty
+                           <*> newTVar Set.empty
 
 addContactInfo :: ContactInfo -> SecretKey -> STM ()
 addContactInfo (ContactInfo as) sk = do