From be48775177ac1d22f868b252b9517354f69a452d Mon Sep 17 00:00:00 2001
From: James Crayne <jim.crayne@gmail.com>
Date: Tue, 29 May 2018 03:37:28 +0000
Subject: oops, left out new module in last commit

---
 src/DPut.hs | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 src/DPut.hs

(limited to 'src/DPut.hs')

diff --git a/src/DPut.hs b/src/DPut.hs
new file mode 100644
index 00000000..abf926b5
--- /dev/null
+++ b/src/DPut.hs
@@ -0,0 +1,28 @@
+module DPut where
+
+import Control.Concurrent.STM
+import qualified Data.Map.Strict as Map
+import System.IO (stderr,hPutStrLn)
+import Data.Maybe
+import System.IO.Unsafe (unsafePerformIO)
+
+data DebugTag = XAnnounce | XDHT | XOnion | XNetCrypto | XMisc
+    deriving (Eq,Ord,Show,Read)
+
+{-# NOINLINE verbosityMap #-}
+verbosityMap :: TVar (Map.Map DebugTag Bool)
+verbosityMap = unsafePerformIO $ newTVarIO (Map.empty)
+
+dput :: DebugTag -> String -> IO ()
+dput tag msg
+    = do
+       mp <- atomically $ readTVar verbosityMap
+       if fromMaybe True (Map.lookup tag mp)
+        then System.IO.hPutStrLn stderr msg
+        else return ()
+
+setQuiet :: DebugTag -> IO ()
+setQuiet tag = atomically $ modifyTVar' (verbosityMap) (Map.insert tag False)
+
+setVerbose :: DebugTag -> IO ()
+setVerbose tag = atomically $ modifyTVar' (verbosityMap) (Map.insert tag True)
-- 
cgit v1.2.3