src/Gargantext/Database/Query/Tree/Root.hs

   1 {-|
   2 Module      : Gargantext.Database.Root
   3 Description : Main requests to get root of users
   4 Copyright   : (c) CNRS, 2017-Present
   5 License     : AGPL + CECILL v3
   6 Maintainer  : team@gargantext.org
   7 Stability   : experimental
   8 Portability : POSIX
   9 -}
  10
  11 {-# OPTIONS_GHC -fno-warn-name-shadowing #-}
  12 {-# OPTIONS_GHC -fno-warn-orphans        #-}
  13
  14 {-# LANGUAGE Arrows                 #-}
  15 {-# LANGUAGE DeriveGeneric          #-}
  16 {-# LANGUAGE ConstraintKinds        #-}
  17 {-# LANGUAGE FlexibleContexts       #-}
  18 {-# LANGUAGE FlexibleInstances      #-}
  19 {-# LANGUAGE FunctionalDependencies #-}
  20 {-# LANGUAGE GeneralizedNewtypeDeriving #-}
  21 {-# LANGUAGE OverloadedStrings #-}
  22 {-# LANGUAGE MultiParamTypeClasses  #-}
  23 {-# LANGUAGE NoImplicitPrelude      #-}
  24 {-# LANGUAGE RankNTypes             #-}
  25 {-# LANGUAGE TemplateHaskell        #-}
  26
  27 module Gargantext.Database.Query.Tree.Root where
  28
  29 import Control.Arrow (returnA)
  30 import Gargantext.Core.Types.Individu (User(..))
  31 import Gargantext.Database.Config (nodeTypeId)
  32 import Gargantext.Database.Node.User (HyperdataUser)
  33 import Gargantext.Database.Schema.Node (NodeRead)
  34 import Gargantext.Database.Schema.Node (queryNodeTable)
  35 import Gargantext.Database.Schema.User (queryUserTable, UserPoly(..))
  36 import Gargantext.Database.Types.Node (Node, NodePoly(..), NodeType(NodeUser))
  37 import Gargantext.Database.Utils (Cmd, runOpaQuery)
  38 import Gargantext.Prelude
  39 import Opaleye (restrict, (.==), Query)
  40 import Opaleye.PGTypes (pgStrictText, pgInt4)
  41
  42 getRoot :: User -> Cmd err [Node HyperdataUser]
  43 getRoot = runOpaQuery . selectRoot
  44
  45 selectRoot :: User -> Query NodeRead
  46 selectRoot (UserName username) = proc () -> do
  47     row   <- queryNodeTable -< ()
  48     users <- queryUserTable -< ()
  49     restrict -< _node_typename row   .== (pgInt4 $ nodeTypeId NodeUser)
  50     restrict -< user_username  users .== (pgStrictText username)
  51     restrict -< _node_userId   row   .== (user_id users)
  52     returnA  -< row
  53
  54 selectRoot (UserDBId uid) = proc () -> do
  55     row   <- queryNodeTable -< ()
  56     restrict -< _node_typename row   .== (pgInt4 $ nodeTypeId NodeUser)
  57     restrict -< _node_userId   row   .== (pgInt4 uid)
  58     returnA  -< row
  59