[CLEAN] renaming unexplicit fun
[gargantext.git] / src / Gargantext / Database / Schema / Ngrams.hs
index a1dd4b6efba27397908282717df0621bd82b9fc0..5d1f4870319dc606685bb9254427b349b24b8ff8 100644 (file)
@@ -25,7 +25,7 @@ import Control.Monad (mzero)
 import Data.Aeson
 import Data.Aeson.Types (toJSONKeyText)
 import Data.Map (Map, fromList, lookup)
-import Data.Text (Text, splitOn, pack)
+import Data.Text (Text, splitOn, pack, strip)
 import Gargantext.Core.Types (TODO(..))
 import Gargantext.Prelude
 import Prelude (Functor)
@@ -140,16 +140,18 @@ fromNgramsTypeId id = lookup id
 
 ------------------------------------------------------------------------
 -- | TODO put it in Gargantext.Core.Text.Ngrams
-data Ngrams = Ngrams { _ngramsTerms :: Text
-                     , _ngramsSize  :: Int
-           } deriving (Generic, Show, Eq, Ord)
+data Ngrams = UnsafeNgrams { _ngramsTerms :: Text
+                           , _ngramsSize  :: Int
+                           } deriving (Generic, Show, Eq, Ord)
 
 makeLenses ''Ngrams
 instance PGS.ToRow Ngrams where
-  toRow (Ngrams t s) = [toField t, toField s]
+  toRow (UnsafeNgrams t s) = [toField t, toField s]
 
 text2ngrams :: Text -> Ngrams
-text2ngrams txt = Ngrams txt $ length $ splitOn " " txt
+text2ngrams txt = UnsafeNgrams txt' $ length $ splitOn " " txt'
+  where
+    txt' = strip txt
 
 -------------------------------------------------------------------------
 -- | TODO put it in Gargantext.Core.Text.Ngrams