-- | Support for [PostgreSQL window
-- functions](https://www.postgresql.org/docs/current/tutorial-window.html)

module Opaleye.Window
       (
         -- * Run window functions on a @Select@
         W.runWindows

         -- * Create @Windows@
       , W.Windows
       , W.over

         -- * Create a @Window@
       , W.Window
       , W.partitionBy

         -- * Create a @WindowFunction@
       , W.WindowFunction

         -- * Window functions

         -- | You might like to also refer to [the Postgres
         -- documentation page that describes its window
         -- functions](https://www.postgresql.org/docs/devel/functions-window.html).

       , W.noWindowFunction
       , W.aggregatorWindowFunction
       , rowNumber
       , rank
       , denseRank
       , percentRank
       , cumeDist
       , ntile
       , lag
       , lead
       , firstValue
       , lastValue
       , nthValue
       ) where

import qualified Opaleye.Internal.Column as IC
import qualified Opaleye.Internal.HaskellDB.PrimQuery as HPQ
import qualified Opaleye.Internal.Window as W

import qualified Opaleye.Field as F
import qualified Opaleye.SqlTypes as T

-- | [@row_number()@](https://www.postgresql.org/docs/current/functions-window.html)
rowNumber :: W.WindowFunction a (F.Field T.SqlInt8)
rowNumber :: forall a. WindowFunction a (Field SqlInt8)
rowNumber = forall a (n :: Nullability) b.
WndwOp -> WindowFunction a (Field_ n b)
W.makeWndwAny WndwOp
HPQ.WndwRowNumber


-- | [@rank()@](https://www.postgresql.org/docs/current/functions-window.html)
rank :: W.WindowFunction a (F.Field T.SqlInt8)
rank :: forall a. WindowFunction a (Field SqlInt8)
rank = forall a (n :: Nullability) b.
WndwOp -> WindowFunction a (Field_ n b)
W.makeWndwAny WndwOp
HPQ.WndwRank


-- | [@dense_rank()@](https://www.postgresql.org/docs/current/functions-window.html)
denseRank :: W.WindowFunction a (F.Field T.SqlInt8)
denseRank :: forall a. WindowFunction a (Field SqlInt8)
denseRank = forall a (n :: Nullability) b.
WndwOp -> WindowFunction a (Field_ n b)
W.makeWndwAny WndwOp
HPQ.WndwDenseRank


-- | [@percent_rank()@](https://www.postgresql.org/docs/current/functions-window.html)
percentRank :: W.WindowFunction a (F.Field T.SqlFloat8)
percentRank :: forall a. WindowFunction a (Field SqlFloat8)
percentRank = forall a (n :: Nullability) b.
WndwOp -> WindowFunction a (Field_ n b)
W.makeWndwAny WndwOp
HPQ.WndwPercentRank


-- | [@cume_dist()@](https://www.postgresql.org/docs/current/functions-window.html)
cumeDist :: W.WindowFunction a (F.Field T.SqlFloat8)
cumeDist :: forall a. WindowFunction a (Field SqlFloat8)
cumeDist = forall a (n :: Nullability) b.
WndwOp -> WindowFunction a (Field_ n b)
W.makeWndwAny WndwOp
HPQ.WndwCumeDist


-- | [@ntile(num_buckets)@](https://www.postgresql.org/docs/current/functions-window.html)
ntile :: F.Field T.SqlInt4
      -- ^ num_buckets
      -> W.WindowFunction a (F.Field T.SqlInt4)
ntile :: forall a. Field SqlInt4 -> WindowFunction a (Field SqlInt4)
ntile (IC.Column PrimExpr
buckets) = forall a (n :: Nullability) b.
WndwOp -> WindowFunction a (Field_ n b)
W.makeWndwAny forall a b. (a -> b) -> a -> b
$ PrimExpr -> WndwOp
HPQ.WndwNtile PrimExpr
buckets


-- | [@lag(value, offset, default)@](https://www.postgresql.org/docs/current/functions-window.html)
lag :: F.Field T.SqlInt4
    -- ^ offset
    -> F.Field_ n a
    -- ^ default
    -> W.WindowFunction (F.Field_ n a) (F.Field_ n a)
lag :: forall (n :: Nullability) a.
Field SqlInt4
-> Field_ n a -> WindowFunction (Field_ n a) (Field_ n a)
lag (IC.Column PrimExpr
offset) (IC.Column PrimExpr
def) =
  forall (n :: Nullability) a (n' :: Nullability) a'.
(PrimExpr -> WndwOp) -> WindowFunction (Field_ n a) (Field_ n' a')
W.makeWndwField forall a b. (a -> b) -> a -> b
$ \PrimExpr
a -> PrimExpr -> PrimExpr -> PrimExpr -> WndwOp
HPQ.WndwLag PrimExpr
a PrimExpr
offset PrimExpr
def


-- | [@lead(value, offset, default)@](https://www.postgresql.org/docs/current/functions-window.html)
lead :: F.Field T.SqlInt4
     -- ^ offset
     -> F.Field_ n a
     -- ^ default
     -> W.WindowFunction (F.Field_ n a) (F.Field_ n a)
lead :: forall (n :: Nullability) a.
Field SqlInt4
-> Field_ n a -> WindowFunction (Field_ n a) (Field_ n a)
lead (IC.Column PrimExpr
offset) (IC.Column PrimExpr
def) =
  forall (n :: Nullability) a (n' :: Nullability) a'.
(PrimExpr -> WndwOp) -> WindowFunction (Field_ n a) (Field_ n' a')
W.makeWndwField forall a b. (a -> b) -> a -> b
$ \PrimExpr
a -> PrimExpr -> PrimExpr -> PrimExpr -> WndwOp
HPQ.WndwLead PrimExpr
a PrimExpr
offset PrimExpr
def


-- | [@first_value(value)@](https://www.postgresql.org/docs/current/functions-window.html)
firstValue :: W.WindowFunction (F.Field_ n a) (F.Field_ n a)
firstValue :: forall (n :: Nullability) a.
WindowFunction (Field_ n a) (Field_ n a)
firstValue = forall (n :: Nullability) a (n' :: Nullability) a'.
(PrimExpr -> WndwOp) -> WindowFunction (Field_ n a) (Field_ n' a')
W.makeWndwField PrimExpr -> WndwOp
HPQ.WndwFirstValue


-- | [@last_value(value)@](https://www.postgresql.org/docs/current/functions-window.html)
lastValue :: W.WindowFunction (F.Field_ n a) (F.Field_ n a)
lastValue :: forall (n :: Nullability) a.
WindowFunction (Field_ n a) (Field_ n a)
lastValue = forall (n :: Nullability) a (n' :: Nullability) a'.
(PrimExpr -> WndwOp) -> WindowFunction (Field_ n a) (Field_ n' a')
W.makeWndwField PrimExpr -> WndwOp
HPQ.WndwLastValue


-- | [@nth_value(value, n)@](https://www.postgresql.org/docs/current/functions-window.html)
nthValue :: F.Field T.SqlInt4
         -- ^ n
         -> W.WindowFunction (F.Field_ n a) (F.FieldNullable a)
nthValue :: forall (n :: Nullability) a.
Field SqlInt4 -> WindowFunction (Field_ n a) (FieldNullable a)
nthValue (IC.Column PrimExpr
n) = forall (n :: Nullability) a (n' :: Nullability) a'.
(PrimExpr -> WndwOp) -> WindowFunction (Field_ n a) (Field_ n' a')
W.makeWndwField forall a b. (a -> b) -> a -> b
$ \PrimExpr
a -> PrimExpr -> PrimExpr -> WndwOp
HPQ.WndwNthValue PrimExpr
a PrimExpr
n