1 files changed, 25 insertions, 10 deletions

diff --git a/src/Network/Tox/ContactInfo.hs b/src/Network/Tox/ContactInfo.hs
index 153cd130..880740bf 100644
--- a/src/Network/Tox/ContactInfo.hs
+++ b/src/Network/Tox/ContactInfo.hs
@@ -27,15 +27,19 @@ data Account = Account
     { userSecret :: SecretKey                     -- local secret key
     , contacts   :: TVar (HashMap NodeId{-friend's userkey-} Contact) -- received contact info
     , clientRefs :: TVar (Set ConnectionKey) -- reference count so we know when to send offline etc
+    , eventChan  :: TChan ContactEvent
     }
 
+data ContactEvent = OnionRouted  { contact :: PublicKey, onionRouted :: OnionData }
+                  | PolicyChange { contact :: PublicKey, policyChange :: Policy }
+
 data Contact = Contact
     { contactKeyPacket     :: Maybe (DHT.DHTPublicKey)
     , contactLastSeenAddr  :: Maybe SockAddr
     , contactFriendRequest :: Maybe (DHT.FriendRequest)
     , contactPolicy        :: Maybe (Connection.Policy)
         -- Possible semantics
-        --  RefusingToConnect : rejected friend-request or blocked.
+        --  RefusingToConnect : rejected friend-request or blocked or unknown.
         --  OpenToConnect     : pending friend-request.
         --  TryingToConnect   : roster entry.
     }
@@ -43,6 +47,7 @@ data Contact = Contact
 nullContact :: Contact
 nullContact = Contact
     { contactKeyPacket     = Nothing
+    , contactLastSeenAddr  = Nothing
     , contactFriendRequest = Nothing
     , contactPolicy        = Nothing
     }
@@ -69,6 +74,7 @@ newContactInfo = atomically $ ContactInfo <$> newTVar HashMap.empty
 newAccount :: SecretKey -> STM Account
 newAccount sk = Account sk <$> newTVar HashMap.empty
                            <*> newTVar Set.empty
+                           <*> newBroadcastTChan
 
 addContactInfo :: ContactInfo -> SecretKey -> STM ()
 addContactInfo (ContactInfo as) sk = do
@@ -88,16 +94,25 @@ updateContactInfo roster Onion.AnnouncedRendezvous{remoteUserKey} (localUserKey,
               $ HashMap.lookup (key2id localUserKey) as
 
 
+onionUpdate :: OnionData -> Maybe Contact -> Maybe Contact
+onionUpdate (Onion.OnionDHTPublicKey dhtpk)
+                = mergeContact nullContact { contactKeyPacket     = Just dhtpk }
+onionUpdate (Onion.OnionFriendRequest fr)
+                = mergeContact nullContact { contactFriendRequest = Just fr    }
+
+policyUpdate :: Policy -> Maybe Contact -> Maybe Contact
+policyUpdate policy = mergeContact nullContact { contactPolicy = Just policy }
+
 updateAccount :: PublicKey -> Onion.OnionData -> Account -> STM ()
-updateAccount remoteUserKey (Onion.OnionDHTPublicKey dhtpk) acc = do
-    modifyTVar' (contacts acc)
-        $ HashMap.alter (mergeContact nullContact { contactKeyPacket = Just dhtpk })
-                        (key2id remoteUserKey)
-
-updateAccount remoteUserKey (Onion.OnionFriendRequest fr) acc = do
-    modifyTVar' (contacts acc)
-        $ HashMap.alter (mergeContact nullContact { contactFriendRequest = Just fr })
-                        (key2id remoteUserKey)
+updateAccount remoteUserKey omsg acc = do
+    modifyTVar' (contacts acc) $ HashMap.alter (onionUpdate omsg) (key2id remoteUserKey)
+    writeTChan (eventChan acc) $ OnionRouted remoteUserKey omsg
+
+setContactPolicy :: PublicKey -> Policy -> Account -> STM ()
+setContactPolicy remoteUserKey policy acc = do
+    modifyTVar' (contacts acc) $ HashMap.alter (policyUpdate policy) (key2id remoteUserKey)
+    writeTChan (eventChan acc) $ PolicyChange remoteUserKey policy
+
 
 dnsPresentation :: ContactInfo -> STM String
 dnsPresentation (ContactInfo accsvar) = do