tox: Automatically share dhtkey with roster contacts.

1 files changed, 9 insertions, 0 deletions

diff --git a/src/Network/Tox/ContactInfo.hs b/src/Network/Tox/ContactInfo.hs
index 47c07237..9f29d587 100644
--- a/src/Network/Tox/ContactInfo.hs
+++ b/src/Network/Tox/ContactInfo.hs
@@ -34,6 +34,8 @@ data Account = Account
 data ContactEvent = OnionRouted  { contact :: PublicKey, onionRouted :: OnionData }
                   | PolicyChange { contact :: PublicKey, policyChange :: Policy }
                   | AddrChange   { contact :: PublicKey, addrChange :: SockAddr }
+                  | SessionEstablished { contact :: PublicKey }
+                  | SessionTerminated  { contact :: PublicKey }
 
 data Contact = Contact
     { contactKeyPacket     :: TVar (Maybe (POSIXTime,DHT.DHTPublicKey))
@@ -105,6 +107,13 @@ setContactAddr now remoteUserKey addr acc = do
     updateAccount' remoteUserKey acc $ addrUpdate now addr
     writeTChan (eventChan acc) $ AddrChange remoteUserKey addr
 
+setEstablished :: POSIXTime -> PublicKey -> Account -> STM ()
+setEstablished now remoteUserKey acc =
+    writeTChan (eventChan acc) $ SessionEstablished remoteUserKey
+
+setTerminated :: POSIXTime -> PublicKey -> Account -> STM ()
+setTerminated now remoteUserKey acc =
+    writeTChan (eventChan acc) $ SessionTerminated remoteUserKey
 
 
 addContactInfo :: ContactInfo -> SecretKey -> STM ()