{-# LANGUAGE DataKinds         #-}
{-# LANGUAGE DeriveGeneric     #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE PolyKinds         #-}
{-# LANGUAGE TypeFamilies      #-}
{-# LANGUAGE TypeOperators     #-}

import           Control.Applicative
import           Control.Monad.IO.Class
import           Data.Aeson
import           Data.Proxy
import           Data.Text                          (Text)
import           Database.PostgreSQL.Simple
import           Database.PostgreSQL.Simple.FromRow
import           Database.PostgreSQL.Simple.ToField
import           Database.PostgreSQL.Simple.ToRow
import           GHC.Generics
import           Network.Wai.Handler.Warp           hiding (Connection)

import           Servant

data Book = Book
  { title  :: Text
  , author :: Text
} deriving Generic

-- JSON instances
instance FromJSON Book
instance ToJSON Book

-- PostgreSQL instances
instance FromRow Book where
  fromRow = Book <$> field <*> field

instance ToRow Book where
  toRow book = [ toField (title book)
               , toField (author book)]

-- we explicitly say we expect a request body, of type Book
type BookApi = "books" :> ReqBody Book :> Post Book  -- POST /books
          :<|> "books" :> Get [Book]                 -- GET /books


server :: Connection -> Server BookApi
server conn = postBook
         :<|> getBooks
  where
    -- the aforementioned 'ReqBody' automatically makes this handler
    -- receive a Book argument
    postBook book = liftIO $ execute conn "insert into books values (?, ?)" book >> return book
    getBooks      = liftIO $ query_ conn "select * from books"

bookApi :: Proxy BookApi
bookApi = Proxy

main :: IO ()
main = do
  conn <- connectPostgreSQL "host=localhost user=qubit dbname=bookstore"
  run 8080 (serve bookApi $ server conn)