[FIX] clean groups + handling duplicates (TODO inside)
[gargantext.git] / src / Gargantext / Prelude / Utils.hs
index f7670d6532374b10e50c7db0a531319fdb2ab1ed..18dca2577df63bb6d5002f2aa20fc4e48c0c14b7 100644 (file)
@@ -9,26 +9,28 @@ Portability : POSIX
 
 -}
 
-{-# OPTIONS_GHC -fno-warn-orphans   #-}
-
 module Gargantext.Prelude.Utils
   where
 
+import Control.Exception
 import Control.Lens (view)
-import Control.Monad.Random.Class (MonadRandom)
 import Control.Monad.Reader (MonadReader)
-import Control.Monad.Reader (ask)
+import Control.Monad.Random.Class (MonadRandom)
 import Data.Text (Text)
+import qualified Data.Text             as Text
 import GHC.IO (FilePath)
-import Gargantext.API.Admin.Settings
-import Gargantext.Database.Admin.Types.Node (NodeId, NodeType)
-import Gargantext.Prelude
-import Gargantext.Core.Crypto.Hash
 import System.Directory (createDirectoryIfMissing)
+import qualified System.Directory as SD
+import System.IO.Error
 import System.Random (newStdGen)
-import qualified Data.Text             as Text
 import qualified System.Random.Shuffle as SRS
 
+import Gargantext.Prelude.Config
+import Gargantext.Prelude.Crypto.Hash
+import Gargantext.Database.Prelude (HasConfig(..))
+import Gargantext.Database.Admin.Types.Node (NodeId, NodeType)
+import Gargantext.Prelude
+
 --------------------------------------------------------------------------
 shuffle :: MonadRandom m => [a] -> m [a]
 shuffle ns = SRS.shuffleM ns 
@@ -49,7 +51,7 @@ type FileName   = FilePath
 -- ("gar/gan","texthello")
 
 
-toPath :: Int -> Text -> (FolderPath,FileName)
+toPath :: Int -> Text -> (FolderPath, FileName)
 toPath n x = (Text.unpack $ Text.intercalate "/" [x1,x2], Text.unpack xs)
   where
     (x1,x') = Text.splitAt n x
@@ -62,23 +64,42 @@ class ReadFile a where
   readFile' :: FilePath -> IO a
 
 
-writeFile :: (MonadReader env m, MonadBase IO m, HasSettings env, SaveFile a)
+folderFilePath :: (MonadReader env m, MonadBase IO m) => m (FolderPath, FileName)
+folderFilePath = do
+  (foldPath, fileName)  <- liftBase $ (toPath 3) . hash . show <$> newStdGen
+
+  pure (foldPath, fileName)
+
+
+writeFile :: (MonadReader env m, MonadBase IO m, HasConfig env, SaveFile a)
           => a -> m FilePath
 writeFile a = do
-  dataPath <- view (settings . fileFolder) <$> ask
-  (fp,fn)  <- liftBase $ (toPath 3) . hash . show <$> newStdGen
+  dataPath <- view $ hasConfig . gc_datafilepath
+
+  (foldPath, fileName) <- folderFilePath
 
-  let foldPath = dataPath <> "/" <> fp
-      filePath = foldPath <> "/" <> fn
+  let filePath = foldPath <> "/" <> fileName
+      dataFoldPath = dataPath <> "/" <> foldPath
+      dataFileName = dataPath <> "/" <> filePath
 
-  _ <- liftBase $ createDirectoryIfMissing True foldPath
-  _ <- liftBase $ saveFile' filePath a
+  _ <- liftBase $ createDirectoryIfMissing True dataFoldPath
+  _ <- liftBase $ saveFile' dataFileName a
 
   pure filePath
 
 
-readFile :: (MonadReader env m, MonadBase IO m, HasSettings env, ReadFile a)
+readFile :: (MonadReader env m, MonadBase IO m, HasConfig env, ReadFile a)
          => FilePath -> m a
 readFile fp = do
-  dataPath <- view (settings . fileFolder) <$> ask
+  dataPath <- view $ hasConfig . gc_datafilepath
   liftBase $ readFile' $ dataPath <> "/" <> fp
+
+removeFile :: (MonadReader env m, MonadBase IO m, HasConfig env)
+           => FilePath -> m ()
+removeFile fp = do
+  dataPath <- view $ hasConfig . gc_datafilepath
+  liftBase $ SD.removeFile (dataPath <> "/" <> fp) `catch` handleExists
+    where
+      handleExists e
+        | isDoesNotExistError e = return ()
+        | otherwise = throwIO e