{-# LANGUAGE Arrows #-}

module Opaleye.Label (
  label',
  -- * Deprecated
  label
  ) where

import qualified Opaleye.Internal.PrimQuery as PQ
import qualified Opaleye.Internal.QueryArr as Q
import qualified Opaleye.Select            as S

import           Control.Arrow (returnA)

-- | Add a commented label to the generated SQL.
label' :: String -> S.Select ()
label' :: String -> Select ()
label' String
l = forall a b. State Tag (a -> (b, PrimQueryArr)) -> SelectArr a b
Q.selectArr StateT Tag Identity (() -> ((), PrimQueryArr))
f where
  f :: StateT Tag Identity (() -> ((), PrimQueryArr))
f = forall (f :: * -> *) a. Applicative f => a -> f a
pure (\() -> ((), String -> PrimQueryArr
PQ.aLabel String
l))

-- | Will be deprecated in version 0.10.  Use 'label\'' instead.
label :: String -> S.SelectArr a b -> S.SelectArr a b
label :: forall a b. String -> SelectArr a b -> SelectArr a b
label String
l SelectArr a b
s = proc a
a -> do
  b
b <- SelectArr a b
s -< a
a
  String -> Select ()
label' String
l -< ()
  forall (a :: * -> * -> *) b. Arrow a => a b b
returnA -< b
b