implemented dynamic rate limiting

1 files changed, 13 insertions, 1 deletions

diff --git a/Storage/Local.hs b/Storage/Local.hs
index b9f0f3e..71d5aa7 100644
--- a/Storage/Local.hs
+++ b/Storage/Local.hs
@@ -3,7 +3,13 @@
  - Licensed under the GNU AGPL version 3 or higher.
  -}
 
-module Storage.Local (localStorage, storageDir, testStorageDir, uploadQueue) where
+module Storage.Local
+	( localStorage
+	, storageDir
+	, testStorageDir
+	, uploadQueue
+	, localDiskUsage
+	) where
 
 import Types
 import Types.Storage
@@ -22,6 +28,7 @@ import Raaz.Core.Encode
 import Control.DeepSeq
 import Control.Exception
 import Control.Monad
+import System.DiskSpace
 
 type GetShareDir = Section -> IO FilePath
 
@@ -127,6 +134,11 @@ storageDir (Just (LocalStorageDirectory d)) (Section section) =
 testStorageDir :: FilePath -> GetShareDir
 testStorageDir tmpdir = storageDir (Just (LocalStorageDirectory tmpdir))
 
+localDiskUsage :: Maybe LocalStorageDirectory -> IO DiskUsage
+localDiskUsage lsd = do
+	dir <- storageDir lsd (Section ".")
+	getDiskUsage dir
+
 -- | The takeFileName ensures that, if the StorableObjectIdent somehow
 -- contains a path (eg starts with "../" or "/"), it is not allowed
 -- to point off outside the shareDir.