Statements.hs

module Main.Statements where

import qualified Hasql.Decoders as HD
import qualified Hasql.Statement as HQ
import Main.Prelude

plain :: ByteString -> HQ.Statement () ()
plain sql =
  HQ.Statement sql mempty HD.noResult False

dropType :: ByteString -> HQ.Statement () ()
dropType name =
  plain
    $ "drop type if exists "
    <> name

createEnum :: ByteString -> [ByteString] -> HQ.Statement () ()
createEnum name values =
  plain
    $ "create type "
    <> name
    <> " as enum ("
    <> mconcat (intersperse ", " (map (\x -> "'" <> x <> "'") values))
    <> ")"

selectList :: HQ.Statement () ([] (Int64, Int64))
selectList =
  HQ.Statement sql mempty decoder True
  where
    sql =
      "values (1,2), (3,4), (5,6)"
    decoder =
      HD.rowList ((,) <$> (HD.column . HD.nonNullable) HD.int8 <*> (HD.column . HD.nonNullable) HD.int8)