1 files changed, 7 insertions, 0 deletions

diff --git a/examples/dhtd.hs b/examples/dhtd.hs
index 57ee8deb..bb491bfc 100644
--- a/examples/dhtd.hs
+++ b/examples/dhtd.hs
@@ -395,6 +395,7 @@ data Session = Session
     , toxkeys           :: TVar Tox.AnnouncedKeys
     , userkeys          :: TVar [(SecretKey,PublicKey)]
     , roster            :: Tox.ContactInfo
+    , announceToLan     :: IO ()
     , sessions          :: TVar [PerSession]
     , connectionManager :: Maybe ConnectionManager
     , onionRouter       :: OnionRouter
@@ -469,6 +470,7 @@ clientSession s@Session{..} sock cnum h = do
             , ["threads"]
             , ["mem"]
             , ["nid"]
+            , ["lan"]
             , ["ls"]
             , ["k"]
             , ["roster"]
@@ -574,6 +576,10 @@ clientSession s@Session{..} sock cnum h = do
                                     Right nid -> show nid ++ " nospam:" ++ drop 64 s
                             Right nid -> show nid
 
+        ("lan", _)  -> cmd0 $ do
+                        announceToLan
+                        hPutClient h "ok"
+
         ("ls", _) | Just DHT{dhtBuckets} <- Map.lookup netname dhts
                     -> cmd0 $ do
                         bkts <- atomically $ readTVar dhtBuckets
@@ -1640,6 +1646,7 @@ main = runResourceT $ liftBaseWith $ \resT -> do
                 , toxkeys           = keysdb
                 , userkeys          = toxids
                 , roster            = rstr
+                , announceToLan     = fromMaybe (return ()) $ Tox.toxAnnounceToLan <$> mbtox
                 , sessions          = sessions
                 , connectionManager = ConnectionManager <$> mconns
                 , onionRouter       = orouter