[MERGE]
[gargantext.git] / src / Gargantext / Database / Admin / Trigger / NodeNodeNgrams.hs
index d92095aec2bd246c208a9af9e83fc7e3eaa25c2e..56bdf759678b70cd8e896f8802abbffb80d0e417 100644 (file)
@@ -11,25 +11,21 @@ Triggers on NodeNodeNgrams table.
 
 -}
 
-{-# LANGUAGE FlexibleContexts  #-}
 {-# LANGUAGE QuasiQuotes       #-}
-{-# LANGUAGE NoImplicitPrelude #-}
-{-# LANGUAGE OverloadedStrings #-}
-{-# LANGUAGE RankNTypes        #-}
 
 module Gargantext.Database.Admin.Trigger.NodeNodeNgrams
   where
 
 import Database.PostgreSQL.Simple.SqlQQ (sql)
-import Gargantext.Core.Types.Main (listTypeId, ListType(CandidateTerm))
-import Gargantext.Database.Admin.Config (nodeTypeId)
+import Gargantext.Core
+import Gargantext.Core.Types.Main (ListType(CandidateTerm))
 import Gargantext.Database.Admin.Types.Node -- (ListId, CorpusId, NodeId)
-import Gargantext.Database.Admin.Utils (Cmd, execPGSQuery)
+import Gargantext.Database.Prelude (Cmd, execPGSQuery)
 import Gargantext.Prelude
 import qualified Database.PostgreSQL.Simple as DPS
 
-triggerCountInsert :: Cmd err Int64
-triggerCountInsert = execPGSQuery query (nodeTypeId NodeDocument, nodeTypeId NodeList)
+triggerCountInsert :: HasDBid NodeType => Cmd err Int64
+triggerCountInsert = execPGSQuery query (toDBid NodeDocument, toDBid NodeList)
   where
     query :: DPS.Query
     query = [sql|
@@ -64,10 +60,10 @@ triggerCountInsert = execPGSQuery query (nodeTypeId NodeDocument, nodeTypeId Nod
           EXECUTE PROCEDURE set_ngrams_global_count();
    |]
 
-triggerCountInsert2 :: Cmd err Int64
-triggerCountInsert2 = execPGSQuery query ( nodeTypeId NodeCorpus
-                                         , nodeTypeId NodeDocument
-                                         , nodeTypeId NodeList
+triggerCountInsert2 :: HasDBid NodeType => Cmd err Int64
+triggerCountInsert2 = execPGSQuery query ( toDBid NodeCorpus
+                                         , toDBid NodeDocument
+                                         , toDBid NodeList
                                          )
   where
     query :: DPS.Query
@@ -108,12 +104,12 @@ triggerCountInsert2 = execPGSQuery query ( nodeTypeId NodeCorpus
    |]
 
 -- TODO add the groups
-triggerCoocInsert :: Cmd err Int64
-triggerCoocInsert = execPGSQuery query ( nodeTypeId NodeCorpus
-                                       , nodeTypeId NodeDocument
-                                       , nodeTypeId NodeList
-                                       , listTypeId CandidateTerm
-                                       , listTypeId CandidateTerm
+triggerCoocInsert :: HasDBid NodeType => Cmd err Int64
+triggerCoocInsert = execPGSQuery query ( toDBid NodeCorpus
+                                       , toDBid NodeDocument
+                                       , toDBid NodeList
+                                       , toDBid CandidateTerm
+                                       , toDBid CandidateTerm
                                        )
   where
     query :: DPS.Query