{-# LANGUAGE
    DataKinds
  , MagicHash
  , OverloadedLabels
  , OverloadedStrings
  , ScopedTypeVariables
  , TypeOperators
#-}

module Main (main) where

import Control.Category ((>>>))
import Control.Monad.Base
import Control.Monad.Except
import Control.Monad.Trans.Maybe
import Data.Function ((&))
import Data.Int
import Data.Monoid
import Generics.SOP
import Squeel.PostgreSQL

import qualified Data.ByteString.Char8 as Char8

main :: IO ()
main = do
  Char8.putStrLn "squeel"
  connectionString <- pure
    "host=localhost port=5432 dbname=exampledb"
  Char8.putStrLn $ "connecting to " <> connectionString
  connection0 <- connectdb connectionString
  Char8.putStrLn "creating tables"
  connection1 <- flip execPQ connection0 $ pqExec $
    createTable #students
      (  (text & notNull) `As` #name
      :* Nil )
    >>>
    createTable #table1
      (  (int4 & notNull) `As` #col1
      :* (int4 & notNull) `As` #col2
      :* Nil )
  connection2 <- flip execPQ connection1 $ do
    _insertTable1Result <- pqExec $
      insertInto #table1 ( 1 `As` #col1 :* 2 `As` #col2 :* Nil )
      >>>
      insertInto #table1 ( 3 `As` #col1 :* 4 `As` #col2 :* Nil )
    Just selectTable1Result <- flip pqExecParams Nil $
      (select $ starFrom #table1 :: Statement '[] Columns Tables Tables)
    Just (Right value00) <- runMaybeT . runExceptT . flip runValue selectTable1Result $
      getValue (RowNumber 0) colNum0
    Just (Right value01) <- runMaybeT . runExceptT . flip runValue selectTable1Result $
      getValue (RowNumber 0) colNum1
    Just (Right value10) <- runMaybeT . runExceptT . flip runValue selectTable1Result $
      getValue (RowNumber 1) colNum0
    Just (Right value11) <- runMaybeT . runExceptT . flip runValue selectTable1Result $
      getValue (RowNumber 1) colNum1
    Just (Right row0) <- runMaybeT . runExceptT . flip runValue selectTable1Result $
      getRow (RowNumber 0)
    Just (Right row1) <- runMaybeT . runExceptT . flip runValue selectTable1Result $
      getRow (RowNumber 1)
    liftBase $ do
      print (value00 :: Int32)
      print (value01 :: Int32)
      print (value10 :: Int32)
      print (value11 :: Int32)
      print (row0 :: NP I '[Int32,Int32])
      print (row1 :: NP I '[Int32,Int32])
  connection3 :: Connection '[] <- flip execPQ connection2 $ pqExec $
    (dropTable #table1 :: Statement '[] '[] Tables '["students" ::: StudentsColumns])
    >>> dropTable #students
  finish connection3

type Columns =
  '[ "col1" ::: 'Required ('NotNull 'PGInt4)
   , "col2" ::: 'Required ('NotNull 'PGInt4)
   ]
-- type SumAndCol1 = '[ "sum" ::: 'NotNull 'PGInt4, "col1" ::: 'NotNull 'PGInt4]
type StudentsColumns = '["name" ::: 'Required ('NotNull 'PGText)]
type Tables = '[ "table1" ::: Columns, "students" ::: StudentsColumns]
-- type OrderColumns =
--   [ "orderID"    ::: 'NotNull 'PGInt4
--   , "orderVal"   ::: 'NotNull 'PGText
--   , "customerID" ::: 'NotNull 'PGInt4
--   , "shipperID"  ::: 'NotNull 'PGInt4
--   ]
-- type CustomerColumns =
--   [ "customerID" ::: 'NotNull 'PGInt4, "customerVal" ::: 'NotNull 'PGFloat4 ]
-- type ShipperColumns =
--   [ "shipperID" ::: 'NotNull 'PGInt4, "shipperVal" ::: 'NotNull 'PGBool ]
-- type JoinTables =
--   [ "shippers"  ::: ShipperColumns
--   , "customers" ::: CustomerColumns
--   , "orders"    ::: OrderColumns
--   ]
-- type ValueColumns =
--   [ "orderVal"    ::: 'NotNull 'PGText
--   , "customerVal" ::: 'NotNull 'PGFloat4
--   , "shipperVal"  ::: 'NotNull 'PGBool
--   ]