1 files changed, 55 insertions, 0 deletions

diff --git a/Roster.hs b/Roster.hs
new file mode 100644
index 00000000..94ab462d
--- /dev/null
+++ b/Roster.hs
@@ -0,0 +1,55 @@
+{-# LANGUAGE NamedFieldPuns #-}
+module Roster where
+
+import Crypto.PubKey.Curve25519
+import Network.Tox.Onion.Transport as Onion
+import Network.Tox.DHT.Transport as DHT
+import Network.Tox.NodeId
+import Control.Concurrent.STM
+import qualified Data.HashMap.Strict as HashMap
+import Data.HashMap.Strict (HashMap)
+
+newtype Roster = Roster { accounts :: TVar (HashMap NodeId Account) }
+
+data Account = Account
+    { userSecret :: SecretKey                            -- local secret key
+    , contacts :: TVar (HashMap NodeId DHT.DHTPublicKey) -- received contact info
+    }
+
+newRoster :: IO Roster
+newRoster = atomically $ Roster <$> newTVar HashMap.empty
+
+newAccount :: SecretKey -> STM Account
+newAccount sk = Account sk <$> newTVar HashMap.empty
+
+addRoster :: Roster -> SecretKey -> STM ()
+addRoster (Roster as) sk = do
+    a <- newAccount sk
+    modifyTVar' as $ HashMap.insert (key2id $ toPublic sk) a
+
+delRoster :: Roster -> PublicKey -> STM ()
+delRoster (Roster as) pk = modifyTVar' as $ HashMap.delete (key2id pk)
+
+updateRoster :: Roster -> Onion.AnnouncedRendezvous -> (PublicKey,Onion.OnionData) -> IO ()
+updateRoster roster Onion.AnnouncedRendezvous{remoteUserKey} (localUserKey,omsg) = do
+    atomically $ do
+        as <- readTVar (accounts roster)
+        maybe (return ())
+              (updateAccount remoteUserKey omsg)
+              $ HashMap.lookup (key2id localUserKey) as
+
+
+updateConcts :: DHT.DHTPublicKey -> Maybe DHT.DHTPublicKey -> Maybe DHT.DHTPublicKey
+updateConcts new (Just old) | DHT.dhtpkNonce old < DHT.dhtpkNonce new = Just new
+updateConcts new Nothing                                              = Just new
+updateConcts _   old                                                  = old
+
+
+updateAccount :: PublicKey -> Onion.OnionData -> Account -> STM ()
+updateAccount remoteUserKey (Onion.OnionDHTPublicKey dhtpk) acc = do
+    modifyTVar' (contacts acc) $ HashMap.alter (updateConcts dhtpk)
+                                               (key2id remoteUserKey)
+
+updateAccount remoteUserKey (Onion.OnionFriendRequest fr) acc = do
+    -- TODO
+    return ()