[gargantext.ini] add BACKEND_NAME
[gargantext.git] / src / Gargantext / Database / Action / Learn.hs
index 6f71341000769c3878784afe935e59f94e375100..46db08a27b0ba19d5289b052f30f9acfd6bb055b 100644 (file)
@@ -17,14 +17,14 @@ module Gargantext.Database.Action.Learn
 
 import Data.Maybe
 import Data.Text (Text)
-import Data.Tuple (snd)
+import Gargantext.Core
 import Gargantext.Core.Types (Offset, Limit)
 import Gargantext.Database.Query.Facet
 import Gargantext.Database.Admin.Types.Hyperdata
 import Gargantext.Database.Admin.Types.Node
 import Gargantext.Database.Prelude (Cmd)
 import Gargantext.Prelude
-import Gargantext.Text.Learn
+import Gargantext.Core.Text.Learn
 import qualified Data.List as List
 import qualified Data.Text as Text
 
@@ -32,21 +32,22 @@ data FavOrTrash = IsFav | IsTrash
   deriving (Eq)
 
 
-moreLike :: CorpusId   -> Maybe Offset -> Maybe Limit -> Maybe OrderBy
+moreLike :: HasDBid NodeType
+         => CorpusId   -> Maybe Offset -> Maybe Limit -> Maybe OrderBy
          -> FavOrTrash -> Cmd err [FacetDoc]
 moreLike cId o _l order ft = do
   priors <- getPriors ft cId
   moreLikeWith cId o (Just 3) order ft priors
 
 ---------------------------------------------------------------------------
-getPriors :: FavOrTrash -> CorpusId -> Cmd err (Events Bool)
+getPriors :: HasDBid NodeType => FavOrTrash -> CorpusId -> Cmd err (Events Bool)
 getPriors ft cId = do
 
-  docs_fav   <- filter (\(FacetDoc _ _ _ _ f _) -> f == Just 2)
-              <$> runViewDocuments cId False Nothing Nothing Nothing
+  docs_fav   <- filter (\(FacetDoc _ _ _ _ f _ _) -> f == Just 2)
+              <$> runViewDocuments cId False Nothing Nothing Nothing Nothing
 
   docs_trash <- List.take (List.length docs_fav)
-            <$> runViewDocuments cId True Nothing Nothing Nothing
+            <$> runViewDocuments cId True Nothing Nothing Nothing Nothing
 
 
   let priors = priorEventsWith text (fav2bool ft) (  List.zip (repeat False) docs_fav
@@ -55,12 +56,13 @@ getPriors ft cId = do
   pure priors
 
 
-moreLikeWith :: CorpusId   -> Maybe Offset -> Maybe Limit -> Maybe OrderBy
+moreLikeWith :: HasDBid NodeType
+             => CorpusId   -> Maybe Offset -> Maybe Limit -> Maybe OrderBy
              -> FavOrTrash -> Events Bool  -> Cmd err [FacetDoc]
 moreLikeWith cId o l order ft priors = do
 
-  docs_test  <- filter (\(FacetDoc _ _ _ _ f _) -> f == Just 1)
-            <$> runViewDocuments cId False o Nothing order
+  docs_test  <- filter (\(FacetDoc _ _ _ _ f _ _) -> f == Just 1)
+            <$> runViewDocuments cId False o Nothing order Nothing
 
   let results = map fst
        $ filter ((==) (Just $ not $ fav2bool ft) . snd)
@@ -74,7 +76,7 @@ fav2bool ft = if (==) ft IsFav then True else False
 
 
 text :: FacetDoc -> Text
-text (FacetDoc _ _ _ h _ _)  = title <> "" <> Text.take 100 abstr
+text (FacetDoc _ _ _ h _ _ _)  = title <> "" <> Text.take 100 abstr
   where
     title = maybe "" identity (_hd_title    h)
     abstr = maybe "" identity (_hd_abstract h)