From b2a94c9e49434c108e33aa8aa695a5822d79d72f Mon Sep 17 00:00:00 2001
From: belevy <benjaminlevy007@gmail.com>
Date: Thu, 21 Jan 2021 21:03:03 -0600
Subject: [PATCH] Demonstrate a simple case of Aggregation

---
 esqueleto.cabal                               |   1 +
 .../Esqueleto/Experimental/Aggregates.hs      | 103 ++++++++++++++++++
 src/Database/Esqueleto/Internal/Internal.hs   |   3 +
 3 files changed, 107 insertions(+)
 create mode 100644 src/Database/Esqueleto/Experimental/Aggregates.hs

diff --git a/esqueleto.cabal b/esqueleto.cabal
index 73ff0ed..6d09a85 100644
--- a/esqueleto.cabal
+++ b/esqueleto.cabal
@@ -30,6 +30,7 @@ library
     exposed-modules:
         Database.Esqueleto
         Database.Esqueleto.Experimental
+        Database.Esqueleto.Experimental.Aggregates
         Database.Esqueleto.Internal.Language
         Database.Esqueleto.Internal.Sql
         Database.Esqueleto.Internal.Internal
diff --git a/src/Database/Esqueleto/Experimental/Aggregates.hs b/src/Database/Esqueleto/Experimental/Aggregates.hs
new file mode 100644
index 0000000..8edcdf2
--- /dev/null
+++ b/src/Database/Esqueleto/Experimental/Aggregates.hs
@@ -0,0 +1,103 @@
+{-# LANGUAGE FlexibleContexts #-}
+{-# LANGUAGE FlexibleInstances #-}
+{-# LANGUAGE FunctionalDependencies #-}
+{-# LANGUAGE MultiParamTypeClasses #-}
+{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE RankNTypes #-}
+{-# LANGUAGE UndecidableInstances #-}
+
+module Database.Esqueleto.Experimental.Aggregates
+    where
+
+import Control.Monad.IO.Class
+import qualified Control.Monad.Trans.Writer as W
+import Data.Coerce (Coercible, coerce)
+import Database.Esqueleto.Internal.Internal
+       ( GroupByClause(..)
+       , SideData(..)
+       , SqlExpr(..)
+       , SqlQuery(..)
+       , SqlSelect(..)
+       , ToSomeValues(..)
+       , Value(..)
+       , select
+       , unsafeSqlFunction
+       )
+import Database.Esqueleto.Internal.PersistentImport (Entity, SqlReadT)
+
+-- Phantom data type that doesn't admit a SqlSelect forcing the use of selectAggregate
+data Aggregate a
+
+test :: Integral n
+     => SqlExpr (Value a)
+     -> SqlExpr (Value b)
+     -> SqlExpr (Value c)
+     -> SqlQuery (SqlExpr (Value a), SqlExpr (Value b), SqlExpr (Value n))
+test x y other =
+    groupBy (x, y) $ \(x', y') ->
+        pure (x', y', sum_ other)
+
+
+-- Tuple magic, only SqlExprs are on the leaves.
+-- The Coercible instance from the SqlExpr a -> SqlExpr b allows 0 cost casting
+class Coercible a r => Aggregateable a r | a -> r, r -> a where
+    toAggregate :: a -> r
+    toAggregate = coerce
+    fromAggregate :: r -> a
+    fromAggregate = coerce
+instance Aggregateable (SqlExpr (Value a)) (SqlExpr (Aggregate (Value a))) where
+instance Aggregateable (SqlExpr (Entity a)) (SqlExpr (Aggregate (Entity a))) where
+instance (Aggregateable a ra, Aggregateable b rb) => Aggregateable (a,b) (ra, rb) where
+instance
+    ( Aggregateable a ra
+    , Aggregateable b rb
+    , Aggregateable c rc
+    ) => Aggregateable (a,b,c) (ra,rb,rc) where
+instance
+    ( Aggregateable a ra
+    , Aggregateable b rb
+    , Aggregateable c rc
+    , Aggregateable d rd
+    ) => Aggregateable (a,b,c,d) (ra,rb,rc,rd) where
+instance
+    ( Aggregateable a ra
+    , Aggregateable b rb
+    , Aggregateable c rc
+    , Aggregateable d rd
+    , Aggregateable e re
+    ) => Aggregateable (a,b,c,d,e) (ra,rb,rc,rd,re) where
+instance
+    ( Aggregateable a ra
+    , Aggregateable b rb
+    , Aggregateable c rc
+    , Aggregateable d rd
+    , Aggregateable e re
+    , Aggregateable f rf
+    ) => Aggregateable (a,b,c,d,e,f) (ra,rb,rc,rd,re,rf) where
+instance
+    ( Aggregateable a ra
+    , Aggregateable b rb
+    , Aggregateable c rc
+    , Aggregateable d rd
+    , Aggregateable e re
+    , Aggregateable f rf
+    , Aggregateable g rg
+    ) => Aggregateable (a,b,c,d,e,f,g) (ra,rb,rc,rd,re,rf,rg) where
+instance
+    ( Aggregateable a ra
+    , Aggregateable b rb
+    , Aggregateable c rc
+    , Aggregateable d rd
+    , Aggregateable e re
+    , Aggregateable f rf
+    , Aggregateable g rg
+    , Aggregateable h rh
+    ) => Aggregateable (a,b,c,d,e,f,g,h) (ra,rb,rc,rd,re,rf,rg,rh) where
+
+sum_ :: Integral n => SqlExpr (Value a) -> SqlExpr (Aggregate (Value n))
+sum_ = coerce . unsafeSqlFunction "SUM"
+
+groupBy :: (ToSomeValues a, Aggregateable a a', Aggregateable b b') => a -> (a' -> SqlQuery b') -> SqlQuery b
+groupBy a f = do
+    Q $ W.tell $ mempty{sdGroupByClause = GroupBy $ toSomeValues a }
+    fmap fromAggregate $ f $ toAggregate a
diff --git a/src/Database/Esqueleto/Internal/Internal.hs b/src/Database/Esqueleto/Internal/Internal.hs
index 96c6bce..149c4a9 100644
--- a/src/Database/Esqueleto/Internal/Internal.hs
+++ b/src/Database/Esqueleto/Internal/Internal.hs
@@ -1138,6 +1138,9 @@ data SomeValue where
 class ToSomeValues a where
     toSomeValues :: a -> [SomeValue]
 
+instance PersistEntity ent => ToSomeValues (SqlExpr (Entity ent)) where
+    toSomeValues ent = [SomeValue $ ent ^. persistIdField]
+
 instance
     ( ToSomeValues a
     , ToSomeValues b