Merge pull request #121 from begriffs/fix-order-by

Do not add WHERE clause if only param is order

.travis.yml

@@ -1,22 +0,0 @@
-language: haskell
-ghc: 7.8
-addons:
-  postgresql: "9.3"
-notifications:
-  slack: looprecur:z2nPS2Cfx2D33LE2obyU2FyQ
-before_install:
-  - createuser --superuser --no-password dbapi_test
-  - createdb -O dbapi_test -U postgres dbapi_test
-  - travis_retry sudo add-apt-repository -y ppa:hvr/ghc
-  - travis_retry sudo apt-get update
-  - travis_retry sudo apt-get install --force-yes happy-1.19.3 alex-3.1.3
-  - export PATH=/opt/alex/3.1.3/bin:/opt/happy/1.19.3/bin:$PATH
-install:
-  - curl http://bin.begriffs.com/dbapi/cabal-sandbox.tar.xz | tar xJ
-  - chmod a+x .cabal-sandbox/bin/*
-  - cabal sandbox init
-  - cabal install --enable-test --dependencies-only
-  - cabal install --enable-test
-script:
-  - cabal test --show-details=always --test-options="--color"
-  - .cabal-sandbox/bin/hlint src/*.hs test/**/*.hs

README.md

@@ -1,43 +1,152 @@
-## Serve a RESTful API from any Postgres database
+![Logo](static/logo.png "Logo")
 
-[![Build Status](https://travis-ci.org/begriffs/dbapi.svg?branch=master)](https://travis-ci.org/begriffs/dbapi)
+[![Build Status](https://circleci.com/gh/begriffs/postgrest.png?circle-token=f723c01686abf0364de1e2eaae5aff1f68bd3ff2)](https://circleci.com/gh/begriffs/postgrest/tree/master)
 
-### Installation
+PostgREST serves a fully RESTful API from any existing PostgreSQL
+database. It provides a cleaner, more standards-compliant, faster
+API than you are likely to write from scratch.
 
-```sh
-brew install postgres
-cabal install -j --enable-tests
+### Demo [postgrest.herokuapp.com](https://postgrest.herokuapp.com) | Watch [Video](https://begriffs.com/posts/2014-12-30-intro-to-postgrest.html)
+
+Try making requests to the live demo server with an HTTP client
+such as [postman](http://www.getpostman.com/). The structure of the
+demo database is defined by
+[begriffs/postgrest-example](https://github.com/begriffs/postgrest-example).
+You can use it as inspiration for test-driven server migrations in
+your own projects.
+
+### Usage
+
+Download the binary ([OS X](http://bin.begriffs.com/dbapi/osx/postgrest-0.2.4.10.tar.xz) /  [Ubuntu](http://bin.begriffs.com/dbapi/heroku/postgrest-0.2.4.10.tar.xz)) and invoke like so:
+
+```bash
+postgrest  --db-host localhost  --db-port 5432     \
+           --db-name my_db      --db-user postgres \
+           --db-pass foobar     --db-pool 200      \
+           --anonymous postgres --secure           \
+           --port 3000
 ```
 
-Example usage:
+### Performance
 
-```sh
-cabal run -p 3000 -d postgres://[auth-role]:@localhost:5432/[database] -a [anonymous-role]
-```
+TLDR; subsecond response times for up to 2000 requests/sec on Heroku free tier. ([see the load test](https://github.com/begriffs/postgrest/wiki/Performance-and-Scaling))
 
-You will need to provide two database roles (which are allowed to
-be the same). One is called the authenticator role (`auth-role`
-above) which should have enough privileges to read the `auth` table
-in the `dbapi` schema if you intend to support multi-user applications.
+If you're used to servers written in interpreted languages (or named
+after precious gems), prepare to be pleasantly surprised by PostgREST
+performance.
 
-The other role is for anonymous access (`anonymous-role` above).
-Immediately upon acceping any unauthenticated HTTP connection dbapi
-assumes this role in its queries to postgres. Give this role as
-much or little permissions as you would like.
+Three factors contribute to the speed. First the server is written
+in [Haskell](https://new-www.haskell.org/) using the
+[Warp](http://www.yesodweb.com/blog/2011/03/preliminary-warp-cross-language-benchmarks)
+HTTP server (aka a compiled language with lightweight threads).
+Next it delegates as much calculation as possible to the database
+including
 
-### Running tests
+* Serializing JSON responses directly in SQL
+* Data validation
+* Authorization
+* Combined row counting and retrieval
+* Data post in single command (`returning *`)
 
-```sh
-createuser --superuser --no-password dbapi_test
-createdb -O dbapi_test -U postgres dbapi_test
+Finally it uses the database efficiently with the
+[Hasql](https://nikita-volkov.github.io/hasql-benchmarks/) library
+by
 
-cabal test --show-details=always --test-options="--color"
-```
+* Reusing prepared statements
+* Keeping a pool of db connections
+* Using the Postgres binary protocol
+* Being stateless to allow horizontal scaling
 
-### Accessing server on localhost
+Ultimately the server (when load balanced) is constrained by database
+performance. This may make it inappropriate for very large traffic
+load. To learn more about scaling with Heroku and Amazon RDS see
+the [performance guide](https://github.com/begriffs/postgrest/wiki/Performance-and-Scaling).
 
-Dbapi permits only secure https access. By default it uses a
-self-signed key which will cause Chrome and other rest clients to
-complain. You will need to configure your system to trust this key.
-On mac follow [these
-instructions](http://www.robpeck.com/2010/10/google-chrome-mac-os-x-and-self-signed-ssl-certificates/).
+Other optimizations are possible, and some are outlined in the
+[Future Features](#future-features).
+
+### Security
+
+PostgREST handles authentication (HTTP Basic over SSL) and delegates
+authorization to the role information defined in the database. This
+ensures there is a single declarative source of truth for security.
+When dealing with the database the server assumes the identity of
+the currently authenticated user, and for the duration of the
+connection cannot do anything the user themselves couldn't.
+
+Postgres 9.5 will soon support true [row-level
+security](http://michael.otacoo.com/postgresql-2/postgres-9-5-feature-highlight-row-level-security/).
+In the meantime what isn't yet implemented can be simulated with
+triggers and security-barrier views. Because the possible queries
+to the database are limited to certain templates using
+[leakproof](http://blog.2ndquadrant.com/how-do-postgresql-security_barrier-views-work/)
+functions, the trigger workaround does not compromise row-level
+security.
+
+For example security patterns see the [security
+guide](https://github.com/begriffs/postgrest/wiki/Security-and-Permissions).
+
+### Versioning
+
+A robust long-lived API needs the freedom to exist in multiple
+versions. PostgREST supports versioning through HTTP content
+negotiation. Requests for a certain version translate into switching
+which database schema to search for tables. PostgreSQL schema search
+paths allow tables from earlier versions to be reused verbatim in
+later versions.
+
+To learn more, see the [guide to versioning](https://github.com/begriffs/postgrest/wiki/API-Versioning).
+
+### Self-documention
+
+Rather than writing and maintaining separate docs yourself let the
+API explain its own affordances using HTTP. All PostgREST endpoints
+respond to the OPTIONS verb and explain what they support as well
+as the data format of their JSON payload.
+
+The number of rows returned by an endpoint is reported by - and
+limited with - range headers. More about
+[that](http://begriffs.com/posts/2014-03-06-beyond-http-header-links.html).
+
+There are more opportunities for self-documentation listed in [Future
+Features](#future-features).
+
+### Data Integrity
+
+Rather than relying on an Object Relational Mapper and custom
+imperative coding, this system requires you put declarative constraints
+directly into your database. Hence no application can corrupt your
+data (including your API server).
+
+The PostgREST exposes HTTP interface with safeguards to prevent
+surprises, such as enforcing idempotent PUT requests, and
+
+See examples of [Postgres
+constraints](http://www.tutorialspoint.com/postgresql/postgresql_constraints.htm)
+and the [guide to routing](https://github.com/begriffs/postgrest/wiki/Routing).
+
+### Future Features
+
+* Watching endpoint changes with sockets and Postgres pubsub
+* Specifying per-view HTTP caching
+* Inferring good default caching policies from the Postgres stats collector
+* Generating mock data for test clients
+* Maintaining separate connection pools per role to avoid "set/reset
+  role" performance penalty
+* Describe more relationships with Link headers
+* Depending on accept headers, render OPTIONS as [RAML](http://raml.org/) or a
+  relational diagram
+* Add two-legged auth with OAuth 1.0a(?)
+* ... the other [issues](https://github.com/begriffs/postgrest/issues)
+
+### Guides
+
+* [Routing](https://github.com/begriffs/postgrest/wiki/Routing)
+* [Versioning](https://github.com/begriffs/postgrest/wiki/API-Versioning)
+* [Performance](https://github.com/begriffs/postgrest/wiki/Performance-and-Scaling)
+* [Security](https://github.com/begriffs/postgrest/wiki/Security-and-Permissions)
+
+### Thanks
+
+Thanks to [Adam Baker](https://github.com/adambaker) for code
+contributions and many fundamental design discussions.

circle.yml

@@ -0,0 +1,6 @@
+machine:
+  pre:
+    - createuser --superuser --no-password postgrest_test
+    - createdb -O postgrest_test -U ubuntu postgrest_test
+  ghc:
+    version: 7.8.3

dbapi.cabal

@@ -1,69 +0,0 @@
-name:                  dbapi
-version:               0.1.0.0
-synopsis:              The database is your api
-license:               MIT
-license-file:          LICENSE
-author:                Joe Nelson, Adam Baker
-maintainer:            cred+github@begriffs.com
-category:              Web
-build-type:            Simple
-cabal-version:         >=1.10
-
-executable dbapi
-  main-is:             Main.hs
-  ghc-options:         -Wall -W -Werror
-  default-language:    Haskell2010
-  build-depends:       base >=4.6 && <5
-                     , HDBC, HDBC-postgresql
-                     , warp, wai >= 3.0.1 && < 3.0.2
-                     , wai-extra, wai-cors
-                     , HTTP, convertible
-                     , case-insensitive
-                     , http-types, scientific, time
-                     , bytestring, aeson, network
-                     , text , containers
-                     , optparse-applicative >= 0.9.1 && < 0.10
-                     , unordered-containers
-                     , regex-base
-                     , string-conversions
-                     , http-media, regex-tdfa
-                     , Ranged-sets
-                     , transformers
-                     , warp-tls
-                     , bcrypt
-                     , base64-string
-                     , split
-  Other-Modules:       Dbapi
-                     , PgStructure
-                     , PgQuery
-                     , RangeQuery
-                     , Middleware
-  hs-source-dirs:      src
-
-Test-Suite spec
-  Type:                exitcode-stdio-1.0
-  Default-Language:    Haskell2010
-  Hs-Source-Dirs:      test, src
-  ghc-options:         -Wall -W -Werror
-  Main-Is:             Main.hs
-  Other-Modules:       Dbapi, Spec, SpecHelper
-  Build-Depends:       base, hspec2
-                     , hspec-wai >= 0.5.0, hspec-wai-json
-                     , HDBC, HDBC-postgresql
-                     , warp, wai >= 3.0.1 && < 3.0.2
-                     , HTTP, convertible
-                     , case-insensitive
-                     , wai-extra, wai-cors, containers
-                     , http-types, scientific, time
-                     , bytestring, aeson, network
-                     , text, optparse-applicative
-                     , unordered-containers
-                     , regex-base
-                     , string-conversions
-                     , http-media, regex-tdfa
-                     , Ranged-sets
-                     , transformers
-                     , warp-tls
-                     , bcrypt
-                     , base64-string
-                     , split

postgrest.cabal

@@ -0,0 +1,89 @@
+name:                  postgrest
+version:               0.2.4.10
+synopsis:              The database is your api
+license:               MIT
+license-file:          LICENSE
+author:                Joe Nelson, Adam Baker
+maintainer:            cred+github@begriffs.com
+category:              Web
+build-type:            Simple
+cabal-version:         >=1.10
+
+executable postgrest
+  main-is:             Main.hs
+  ghc-options:         -Wall -W -Werror -O2
+  default-language:    Haskell2010
+  default-extensions:  OverloadedStrings
+  other-extensions:    QuasiQuotes
+  build-depends:       base >=4.6 && <5
+                     , hasql == 0.4.*, hasql-backend
+                     , hasql-postgres == 0.8.*
+                     , warp >= 3.0.2, wai >= 3.0.1
+                     , wai-extra, wai-cors
+                     , wai-middleware-static >= 0.6.0
+                     , HTTP, convertible, http-types
+                     , case-insensitive
+                     , scientific, time
+                     , aeson, network >= 2.6
+                     , bytestring, text, split, string-conversions
+                     , stringsearch, parsec
+                     , containers, unordered-containers
+                     , optparse-applicative >= 0.9.1 && < 0.10
+                     , regex-base, regex-tdfa
+                     , regex-tdfa-text
+                     , Ranged-sets
+                     , transformers
+                     , bcrypt, base64-string
+                     , network-uri >= 2.6
+                     , resource-pool
+                     , blaze-builder
+                     , vector
+                     , mtl
+  Other-Modules:       App
+                     , Auth
+                     , Config
+                     , PgStructure
+                     , PgQuery
+                     , PgError
+                     , RangeQuery
+                     , Middleware
+  hs-source-dirs:      src
+
+Test-Suite spec
+  Type:                exitcode-stdio-1.0
+  Default-Language:    Haskell2010
+  default-extensions:  OverloadedStrings
+  other-extensions:    QuasiQuotes
+  Hs-Source-Dirs:      test, src
+  ghc-options:         -Wall -W -Werror
+  Main-Is:             Main.hs
+  Other-Modules:       App, Auth, Config, Spec, SpecHelper
+  Build-Depends:       base, hspec >= 2.1.2, QuickCheck
+                     , hspec-wai >= 0.5.0, hspec-wai-json
+                     , hasql == 0.4.*, hasql-backend
+                     , hasql-postgres == 0.8.*
+                     , warp >= 3.0.2, wai >= 3.0.1
+                     , HTTP, convertible
+                     , case-insensitive
+                     , wai-extra, wai-cors, containers
+                     , wai-middleware-static >= 0.6.0
+                     , http-types, scientific, time
+                     , bytestring, aeson, network >= 2.6
+                     , text, optparse-applicative
+                     , stringsearch, parsec
+                     , unordered-containers
+                     , regex-base
+                     , string-conversions
+                     , http-media, regex-tdfa
+                     , regex-tdfa-text
+                     , Ranged-sets
+                     , transformers
+                     , bcrypt
+                     , base64-string
+                     , split
+                     , network-uri >= 2.6
+                     , resource-pool
+                     , blaze-builder
+                     , vector
+                     , mtl
+                     , process

scripts/release_s3.sh

@@ -0,0 +1,10 @@
+export POSTGREST_VER=`grep ^version /app/postgrest.cabal | sed -En 's/.*\s+([0-9\.]+)/\1/p'`
+
+curl -L http://softlayer-ams.dl.sourceforge.net/project/s3tools/s3cmd/1.5.0-alpha1/s3cmd-1.5.0-alpha1.tar.gz | tar zx
+
+cp /app/dist/build/postgrest/postgrest postgrest-${POSTGREST_VER}
+tar cJf postgrest-${POSTGREST_VER}.tar.xz postgrest-${POSTGREST_VER}
+
+touch ~/.s3cfg
+
+s3cmd-1.5.0-alpha1/s3cmd put --access_key=${S3_ACCESS_KEY} --secret_key=${S3_SECRET_KEY} -P -f postgrest-${POSTGREST_VER}.tar.xz $S3_BUCKET/postgrest-${POSTGREST_VER}.tar.xz

src/App.hs

@@ -0,0 +1,254 @@
+{-# LANGUAGE FlexibleContexts #-}
+module App (app, sqlError, isSqlError) where
+
+import Control.Monad (join)
+import Control.Arrow ((***))
+import Control.Applicative
+
+import Data.Text hiding (map)
+import Data.Maybe (fromMaybe)
+import Text.Regex.TDFA ((=~))
+import Data.Ord (comparing)
+import Data.Ranged.Ranges (emptyRange)
+import Data.HashMap.Strict (keys, elems, filterWithKey, toList)
+import Data.String.Conversions (cs)
+import Data.List (sortBy)
+import Data.Functor.Identity
+import qualified Data.Set as S
+import qualified Data.ByteString.Lazy as BL
+
+import Network.HTTP.Types.Status
+import Network.HTTP.Types.Header
+import Network.HTTP.Types.URI (parseSimpleQuery)
+import Network.HTTP.Base (urlEncodeVars)
+import Network.Wai
+
+import Data.Aeson
+import Data.Monoid
+import qualified Hasql as H
+import qualified Hasql.Postgres as H
+
+import Auth
+import PgQuery
+import RangeQuery
+import PgStructure
+import PgError
+import Text.Parsec hiding (Column)
+
+app :: BL.ByteString -> Request -> H.Tx H.Postgres s Response
+app reqBody req =
+  case (path, verb) of
+    ([], _) -> do
+      body <- encode <$> tables (cs schema)
+      return $ responseLBS status200 [jsonH] $ cs body
+
+    ([table], "OPTIONS") -> do
+      let t = QualifiedTable schema (cs table)
+      cols <- columns t
+      pkey <- map cs <$> primaryKeyColumns t
+      return $ responseLBS status200 [jsonH, allOrigins]
+        $ encode (TableOptions cols pkey)
+
+    ([table], "GET") ->
+      if range == Just emptyRange
+      then return $ responseLBS status416 [] "HTTP Range error"
+      else do
+        let qt = QualifiedTable schema (cs table)
+        let select = coerce $
+              ("select ",[],mempty) <>
+                  parentheticT (
+                    whereT qq $ countRows qt
+                  ) <> commaq <> (
+                  asJsonWithCount
+                  . limitT range
+                  . orderT (orderParse qq)
+                  . whereT qq
+                  $ selectStar qt
+                )
+        row <- H.single select
+        let (tableTotal, queryTotal, body) =
+              fromMaybe (0, 0, Just "" :: Maybe Text) row
+            from = fromMaybe 0 $ rangeOffset <$> range
+            to = from+queryTotal-1
+            contentRange = contentRangeH from to tableTotal
+            status = rangeStatus from to tableTotal
+            canonical = urlEncodeVars
+                          . sortBy (comparing fst)
+                          . map (join (***) cs)
+                          . parseSimpleQuery
+                          $ rawQueryString req
+        return $ responseLBS status
+          [jsonH, contentRange,
+            ("Content-Location",
+             "/" <> cs table <>
+                if Prelude.null canonical then "" else "?" <> cs canonical
+            )
+          ] (cs $ fromMaybe "[]" body)
+
+    (["postgrest", "users"], "POST") -> do
+      let user = decode reqBody :: Maybe AuthUser
+
+      case user of
+        Nothing -> return $ responseLBS status400 [jsonH] $
+          encode . object $ [("message", String "Failed to parse user.")]
+        Just u -> do
+          _ <- addUser (cs $ userId u)
+            (cs $ userPass u) (cs $ userRole u)
+          return $ responseLBS status201
+            [ jsonH
+            , (hLocation, "/postgrest/users?id=eq." <> cs (userId u))
+            ] ""
+
+    ([table], "POST") ->
+      handleJsonObj reqBody $ \obj -> do
+        let qt = QualifiedTable schema (cs table)
+            query = coerce $
+              insertInto qt (map cs $ keys obj) (elems obj)
+        row <- H.single query
+        let (Identity insertedJson) = fromMaybe (Identity "{}" :: Identity Text) row
+            Just inserted = decode (cs insertedJson) :: Maybe Object
+
+        primaryKeys <- map cs <$> primaryKeyColumns qt
+        let primaries = if Prelude.null primaryKeys
+            then inserted
+            else filterWithKey (const . (`elem` primaryKeys)) inserted
+        let params = urlEncodeVars
+              $ map (\t -> (cs $ fst t, "eq." <> cs (unquoted $ snd t)))
+              $ sortBy (comparing fst) $ toList primaries
+        return $ responseLBS status201
+          [ jsonH
+          , (hLocation, "/" <> cs table <> "?" <> cs params)
+          ] ""
+
+    ([table], "PUT") ->
+      handleJsonObj reqBody $ \obj -> do
+        let qt = QualifiedTable schema (cs table)
+        primaryKeys <- primaryKeyColumns qt
+        let specifiedKeys = map (cs . fst) qq
+        if S.fromList primaryKeys /= S.fromList specifiedKeys
+          then return $ responseLBS status405 []
+               "You must speficy all and only primary keys as params"
+          else do
+            tableCols <- map (cs . colName) <$> columns qt
+            let cols = map cs $ keys obj
+            if S.fromList tableCols == S.fromList cols
+              then do
+                let vals = elems obj
+                H.unit . coerce $ iffNotT
+                        (whereT qq $ update qt cols vals)
+                        (insertSelect qt cols vals)
+                return $ responseLBS status204 [ jsonH ] ""
+
+              else return $ if Prelude.null tableCols
+                then responseLBS status404 [] ""
+                else responseLBS status400 []
+                   "You must specify all columns in PUT request"
+
+    ([table], "PATCH") ->
+      handleJsonObj reqBody $ \obj -> do
+        let qt = QualifiedTable schema (cs table)
+        H.unit
+          $ coerce
+          $ whereT qq
+          $ update qt (map cs $ keys obj) (elems obj)
+        return $ responseLBS status204 [ jsonH ] ""
+
+    (_, _) ->
+      return $ responseLBS status404 [] ""
+
+  where
+    path   = pathInfo req
+    verb   = requestMethod req
+    qq     = queryString req
+    hdrs   = requestHeaders req
+    schema = requestedSchema hdrs
+    range  = rangeRequested hdrs
+    allOrigins = ("Access-Control-Allow-Origin", "*") :: Header
+    coerce (q, args, All b) = (q, args, b)
+
+
+isSqlError :: H.Error -> Maybe H.Error
+isSqlError = Just
+
+sqlError :: H.Error -> Response
+sqlError err =
+  let inside = case err of
+        H.CantConnect _ ->
+          "Message: \"Cannot connect to postgres server\""
+        H.ConnectionLost t -> t
+        H.ErroneousResult t -> t
+        H.UnexpectedResult t -> t
+        H.UnparsableTemplate t -> t
+        H.UnparsableRow t -> t
+        H.NotInTransaction -> "An operation which requires a"
+          <> "database transaction was executed without one" in
+  either
+    (\hint ->
+      responseLBS status500
+      [(hContentType, "application/json")]
+      (cs . encode . object $ [
+          ("message", String $
+            "Failed to parse exception:" <> inside)
+        , ("hint", String . cs . show $ hint)]))
+    (\msg ->
+      responseLBS (httpStatus msg)
+      [(hContentType, "application/json")]
+      (encode msg))
+    (parse message "" inside)
+
+
+rangeStatus :: Int -> Int -> Int -> Status
+rangeStatus from to total
+  | from > total            = status416
+  | (1 + to - from) < total = status206
+  | otherwise               = status200
+
+contentRangeH :: Int -> Int -> Int -> Header
+contentRangeH from to total =
+  ("Content-Range",
+    if total == 0 || from > total
+    then "*/" <> cs (show total)
+    else cs (show from)  <> "-"
+       <> cs (show to)    <> "/"
+       <> cs (show total)
+  )
+
+requestedSchema :: RequestHeaders -> Text
+requestedSchema hdrs =
+  case verStr of
+       Just [[_, ver]] -> ver
+       _ -> "1"
+
+  where verRegex = "version[ ]*=[ ]*([0-9]+)" :: String
+        accept = cs <$> lookup hAccept hdrs :: Maybe Text
+        verStr = (=~ verRegex) <$> accept :: Maybe [[Text]]
+
+jsonH :: Header
+jsonH = (hContentType, "application/json")
+
+handleJsonObj :: BL.ByteString -> (Object -> H.Tx H.Postgres s Response)
+              -> H.Tx H.Postgres s Response
+handleJsonObj reqBody handler = do
+  let p = eitherDecode reqBody
+  case p of
+    Left err ->
+      return $ responseLBS status400 [jsonH] jErr
+      where
+        jErr = encode . object $
+          [("message", String $ "Failed to parse JSON payload. " <> cs err)]
+    Right (Object o) -> handler o
+    Right _ ->
+      return $ responseLBS status400 [jsonH] jErr
+      where
+        jErr = encode . object $
+          [("message", String "Expecting a JSON object")]
+
+data TableOptions = TableOptions {
+  tblOptcolumns :: [Column]
+, tblOptpkey :: [Text]
+}
+
+instance ToJSON TableOptions where
+  toJSON t = object [
+      "columns" .= tblOptcolumns t
+    , "pkey"   .= tblOptpkey t ]

src/Auth.hs

@@ -0,0 +1,69 @@
+{-# LANGUAGE QuasiQuotes, ScopedTypeVariables, OverloadedStrings #-}
+module Auth where
+
+import Data.Aeson
+import Control.Monad (mzero)
+import Control.Applicative ( (<*>), (<$>) )
+import Crypto.BCrypt
+import Data.Text
+import Data.Monoid
+import qualified Hasql as H
+import qualified Hasql.Postgres as H
+import Data.String.Conversions (cs)
+import PgQuery (pgFmtLit)
+
+import System.IO.Unsafe
+
+data AuthUser = AuthUser {
+    userId :: String
+  , userPass :: String
+  , userRole :: String
+  } deriving (Show)
+
+instance FromJSON AuthUser where
+  parseJSON (Object v) = AuthUser <$>
+                         v .: "id" <*>
+                         v .: "pass" <*>
+                         v .: "role"
+  parseJSON _ = mzero
+
+instance ToJSON AuthUser where
+  toJSON u = object [
+      "id" .= userId u
+    , "pass" .= userPass u
+    , "role" .= userRole u ]
+
+type DbRole = Text
+
+data LoginAttempt =
+    NoCredentials
+  | MalformedAuth
+  | LoginFailed
+  | LoginSuccess DbRole
+  deriving (Eq, Show)
+
+checkPass :: Text -> Text -> Bool
+checkPass = (. cs) . validatePassword . cs
+
+setRole :: Text -> H.Tx H.Postgres s ()
+setRole role = H.unit ("set role " <> cs (pgFmtLit role), [], True)
+
+resetRole :: H.Tx H.Postgres s ()
+resetRole = H.unit [H.q|reset role|]
+
+addUser :: Text -> Text -> Text -> H.Tx H.Postgres s ()
+addUser identity pass role = do
+  let Just hashed = unsafePerformIO $ hashPasswordUsingPolicy fastBcryptHashingPolicy (cs pass)
+  H.unit $
+    [H.q|insert into postgrest.auth (id, pass, rolname) values (?, ?, ?)|]
+      identity (cs hashed :: Text) role
+
+signInRole :: Text -> Text -> H.Tx H.Postgres s LoginAttempt
+signInRole user pass = do
+  u <- H.single $ [H.q|select pass, rolname from postgrest.auth where id = ?|] user
+  return $ maybe LoginFailed (\r ->
+      let (hashed, role) = r in
+      if checkPass hashed pass
+         then LoginSuccess role
+         else LoginFailed
+    ) u

src/Config.hs

@@ -0,0 +1,54 @@
+module Config where
+
+import Network.Wai
+import Control.Applicative
+import Data.Text (strip)
+import qualified Data.CaseInsensitive as CI
+import qualified Data.ByteString.Char8 as BS
+import Data.String.Conversions (cs)
+import Options.Applicative hiding (columns)
+import Network.Wai.Middleware.Cors (CorsResourcePolicy(..))
+
+data AppConfig = AppConfig {
+    configDbName :: String
+  , configDbPort :: Int
+  , configDbUser :: String
+  , configDbPass :: String
+  , configDbHost :: String
+
+  , configPort  :: Int
+  , configAnonRole :: String
+  , configSecure :: Bool
+  , configPool :: Int
+  }
+
+argParser :: Parser AppConfig
+argParser = AppConfig
+  <$> strOption (long "db-name" <> short 'd'                     <> help "name of database")
+  <*> option    (long "db-port" <> short 'P' <> value 5432        <> help "postgres server port")
+  <*> strOption (long "db-user" <> short 'U'                     <> help "postgres authenticator role")
+  <*> strOption (long "db-pass"             <> value ""          <> help "password for authenticator role")
+  <*> strOption (long "db-host" <> short 'h' <> value "localhost" <> help "postgres server hostname")
+
+  <*> option    (long "port" <> short 'p' <> value 3000 <> help "port number on which to run HTTP server")
+  <*> strOption (long "anonymous" <> short 'a' <> help "postgres role to use for non-authenticated requests")
+  <*> switch    (long "secure" <> short 's' <> help "Redirect all requests to HTTPS")
+  <*> option    (long "db-pool" <> value 10 <> help "Max connections in database pool")
+
+defaultCorsPolicy :: CorsResourcePolicy
+defaultCorsPolicy =  CorsResourcePolicy Nothing
+  ["GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"] ["Authorization"] Nothing
+  (Just $ 60*60*24) False False True
+
+corsPolicy :: Request -> Maybe CorsResourcePolicy
+corsPolicy req = case lookup "origin" headers of
+  Just origin -> Just defaultCorsPolicy {
+      corsOrigins = Just ([origin], True)
+    , corsRequestHeaders = "Authentication":accHeaders
+    }
+  Nothing -> Nothing
+  where
+    headers = requestHeaders req
+    accHeaders = case lookup "access-control-request-headers" headers of
+      Just hdrs -> map (CI.mk . cs . strip . cs) $ BS.split ',' hdrs
+      Nothing -> []

src/Dbapi.hs

@@ -1,234 +0,0 @@
-{-# LANGUAGE OverloadedStrings #-}
-
--- {{{ Imports
-module Dbapi where
-
-import Types (SqlRow, getRow)
-
-import Control.Monad (join)
-import Control.Exception.Base (bracket_)
-import Control.Arrow ((***))
-import Control.Applicative
-import Options.Applicative hiding (columns)
-
-import Data.Maybe (fromMaybe, isJust)
-import Text.Regex.TDFA ((=~))
-import Data.Map (intersection, fromList, toList, Map)
-import Data.List (sort)
-import qualified Data.Set as S
-import Data.Convertible.Base (convert)
-import Data.Text (strip)
-
-import Network.HTTP.Types.Status
-import Network.HTTP.Types.Header
-import Network.HTTP.Types.URI
-
-import Network.HTTP.Base (urlEncodeVars)
-
-import Network.Wai
-import Network.Wai.Internal
-import Network.Wai.Middleware.Cors (CorsResourcePolicy(..))
-
-import qualified Data.ByteString.Char8 as BS
-import Data.String.Conversions (cs)
-import qualified Data.CaseInsensitive as CI
-
-import Database.HDBC.PostgreSQL (Connection)
-import PgStructure (printTables, printColumns, primaryKeyColumns,
-                    columns, Column(colName))
-
-import qualified Data.Aeson as JSON
-
-import PgQuery
-import RangeQuery
-import Data.Ranged.Ranges (emptyRange)
-import Codec.Binary.Base64.String (decode)
-
--- }}}
-
-data AppConfig = AppConfig {
-    configDbUri :: String
-  , configPort  :: Int
-  , configSslCert :: FilePath
-  , configSslKey  :: FilePath
-  , configAnonRole :: String
-  }
-
-jsonContentType :: (HeaderName, BS.ByteString)
-jsonContentType = (hContentType, "application/json")
-
-jsonBodyAction :: Request -> (SqlRow -> IO Response) -> IO Response
-jsonBodyAction req handler = do
-  parse <- jsonBody req
-  case parse of
-    Left err -> return $ responseLBS status400 [jsonContentType] json
-      where json = JSON.encode . JSON.object $ [("error", JSON.String $ cs err)]
-    Right body -> handler body
-
-jsonBody :: Request -> IO (Either String SqlRow)
-jsonBody = fmap JSON.eitherDecode . strictRequestBody
-
-filterByKeys :: Ord a => Map a b -> [a] -> Map a b
-filterByKeys m keys =
-  if null keys then m else
-    m `intersection` fromList (zip keys $ repeat undefined)
-
-httpRequesterRole :: RequestHeaders -> Connection -> IO LoginAttempt
-httpRequesterRole hdrs conn = do
-  let auth = fromMaybe "" $ lookup hAuthorization hdrs
-  case BS.split ' ' (cs auth) of
-    ("Basic" : b64 : _) ->
-      case BS.split ':' $ cs (decode $ cs b64) of
-        (u:p:_) -> signInRole u p conn
-        _ -> return MalformedAuth
-    _ -> return NoCredentials
-
-
-app :: Connection -> DbRole -> Application
-app conn anonymous req respond = do
-  attempt <- httpRequesterRole (requestHeaders req) conn
-
-  case attempt of
-    MalformedAuth ->
-      respond $ responseLBS status400 [] "Malformed basic auth header"
-    LoginFailed ->
-      respond $ responseLBS status401 [] "Invalid username or password"
-    LoginSuccess role ->
-      bracket_ (pgSetRole conn role) (pgResetRole conn) $ appWithRole conn req respond
-    NoCredentials ->
-      bracket_ (pgSetRole conn anonymous) (pgResetRole conn) $ appWithRole conn req respond
-
-
-appWithRole :: Connection -> Application
-appWithRole conn req respond =
-  respond =<< case (path, verb) of
-    ([], _) ->
-      responseLBS status200 [jsonContentType] <$> printTables ver conn
-
-    ([table], "OPTIONS") ->
-      responseLBS status200 [jsonContentType, allOrigins] <$>
-        printColumns ver (cs table) conn
-
-    ([table], "GET") ->
-      if range == Just emptyRange
-      then return $ responseLBS status416 [] "HTTP Range error"
-      else do
-        r <- respondWithRangedResult <$> getRows ver (cs table) qq range conn
-        let canonical = urlEncodeVars $ sort $
-                        map (join (***) cs) $
-                        parseSimpleQuery $
-                        rawQueryString req
-        return $ addHeaders [
-          ("Content-Location",
-           "/" <> cs table <> "?" <> cs canonical
-          )] r
-
-    ([table], "POST") ->
-      jsonBodyAction req (\row -> do
-        allvals <- insert ver table row conn
-        keys <- primaryKeyColumns ver (cs table) conn
-        let params = urlEncodeVars $ map (\t -> (fst t, "eq." <> convert (snd t) :: String)) $ toList $ filterByKeys allvals keys
-        return $ responseLBS status201
-          [ jsonContentType
-          , (hLocation, "/" <> cs table <> "?" <> cs params)
-          ] ""
-      )
-
-    ([table], "PUT") ->
-      jsonBodyAction req (\row -> do
-        keys <- primaryKeyColumns ver (cs table) conn
-        let specifiedKeys = map (cs . fst) qq
-        if S.fromList keys /= S.fromList specifiedKeys
-          then return $ responseLBS status405 []
-               "You must speficy all and only primary keys as params"
-          else
-            if isJust cRange
-               then return $ responseLBS status400 []
-                    "Content-Range is not allowed in PUT request"
-            else do
-              cols <- columns ver (cs table) conn
-              let colNames = S.fromList $ map (cs . colName) cols
-              let specifiedCols = S.fromList $ map fst $ getRow row
-              return $ if colNames == specifiedCols then
-                responseLBS status200 [ jsonContentType ] ""
-
-                else if S.null colNames then responseLBS status404 [] ""
-                  else responseLBS status400 []
-                     "You must specify all columns in PUT request"
-      )
-
-    (_, _) ->
-      return $ responseLBS status404 [] ""
-
-  where
-    path   = pathInfo req
-    verb   = requestMethod req
-    qq     = queryString req
-    hdrs   = requestHeaders req
-    ver    = fromMaybe "1" $ requestedVersion hdrs
-    range  = requestedRange hdrs
-    cRange = requestedContentRange hdrs
-    allOrigins = ("Access-Control-Allow-Origin", "*") :: Header
-
-defaultCorsPolicy :: CorsResourcePolicy
-defaultCorsPolicy =  CorsResourcePolicy Nothing
-  ["GET", "POST", "PUT", "PATCH", "DELETE", "OPTIONS"] ["Authorization"] Nothing
-  (Just $ 60*60*24) False False True
-
-corsPolicy :: Request -> Maybe CorsResourcePolicy
-corsPolicy req = case lookup "origin" headers of
-  Just origin -> Just defaultCorsPolicy {
-      corsOrigins = Just ([origin], True)
-    , corsRequestHeaders = "Authentication":accHeaders
-    }
-  Nothing -> Nothing
-  where
-    headers = requestHeaders req
-    accHeaders = case lookup "access-control-request-headers" headers of
-      Just hdrs -> map (CI.mk . cs . strip . cs) $ BS.split ',' hdrs
-      Nothing -> []
-
-
-respondWithRangedResult :: RangedResult -> Response
-respondWithRangedResult rr =
-  responseLBS status [
-    jsonContentType,
-    ("Content-Range",
-      if total == 0 || from > total
-      then "*/" <> cs (show total)
-      else cs (show from)  <> "-"
-         <> cs (show to)    <> "/"
-         <> cs (show total)
-    )
-  ] (rrBody rr)
-
-  where
-    from   = rrFrom rr
-    to     = rrTo   rr
-    total  = rrTotal rr
-    status
-      | from > total            = status416
-      | total == 0               = status204
-      | (1 + to - from) < total = status206
-      | otherwise               = status200
-
-requestedVersion :: RequestHeaders -> Maybe String
-requestedVersion hdrs =
-  case verStr of
-       Just [[_, ver]] -> Just ver
-       _ -> Nothing
-
-  where verRegex = "version[ ]*=[ ]*([0-9]+)" :: String
-        accept = cs <$> lookup hAccept hdrs :: Maybe String
-        verStr = (=~ verRegex) <$> accept :: Maybe [[String]]
-
-
-addHeaders :: ResponseHeaders -> Response -> Response
-addHeaders hdrs (ResponseFile    s headers fp m) =
-                 ResponseFile    s (headers ++ hdrs) fp m
-addHeaders hdrs (ResponseBuilder s headers b)    =
-                 ResponseBuilder s (headers ++ hdrs) b
-addHeaders hdrs (ResponseStream  s headers b)    =
-                 ResponseStream  s (headers ++ hdrs) b
-addHeaders hdrs (ResponseRaw     s resp)         =
-                 ResponseRaw     s (addHeaders hdrs resp)

src/Main.hs

@@ -1,47 +1,64 @@
-{-# LANGUAGE OverloadedStrings #-}
-
--- {{{ Imports
-
 module Main where
-import Dbapi
-import Middleware (reportPgErrors)
-import Network.Wai.Handler.Warp hiding (Connection)
-import Database.HDBC.PostgreSQL (connectPostgreSQL')
+
+import Paths_postgrest (version)
+
+import App
+import Middleware
+
+import Control.Monad (unless)
+import Control.Monad.IO.Class (liftIO)
+import Control.Exception
 import Data.String.Conversions (cs)
-
-import Control.Applicative
-import Options.Applicative hiding (columns)
-import Network.Wai.Handler.WarpTLS (tlsSettings, runTLS)
-import Network.Wai.Middleware.Gzip (gzip, def)
+import Network.Wai (strictRequestBody)
 import Network.Wai.Middleware.Cors (cors)
+import Network.Wai.Handler.Warp hiding (Connection)
+import Network.Wai.Middleware.Gzip (gzip, def)
+import Network.Wai.Middleware.Static (staticPolicy, only)
+import Data.List (intercalate)
+import Data.Version (versionBranch)
+import qualified Hasql as H
+import qualified Hasql.Postgres as H
+import Options.Applicative hiding (columns)
 
--- }}}
-
-argParser :: Parser AppConfig
-argParser = AppConfig
-  <$> strOption (long "db" <> short 'd' <> metavar "URI"
-    <> help "database uri to expose, e.g. postgres://user:pass@host:port/database")
-  <*> option (long "port" <> short 'p' <> metavar "NUMBER" <> value 3000
-    <> help "port number on which to run HTTP server")
-  <*> strOption (long "sslcert" <> short 'c' <> metavar "PATH" <> value "test/test.crt"
-    <> help "path to SSL cert file")
-  <*> strOption (long "sslkey" <> short 'k' <> metavar "PATH" <> value "test/test.key"
-    <> help "path to SSL key file")
-  <*> strOption (long "anonymous" <> short 'a' <> metavar "ROLE"
-    <> help "postgres role to use for non-authenticated requests")
+import Config (AppConfig(..), argParser, corsPolicy)
 
 main :: IO ()
 main = do
   conf <- execParser (info (helper <*> argParser) describe)
   let port = configPort conf
-  let dburi = configDbUri conf
 
-  let tls = tlsSettings (configSslCert conf) (configSslKey conf)
-  let settings = setPort port defaultSettings
+  unless (configSecure conf) $
+    putStrLn "WARNING, running in insecure mode, auth will be in plaintext"
+  Prelude.putStrLn $ "Listening on port " ++
+    (show $ configPort conf :: String)
 
-  Prelude.putStrLn $ "Listening on port " ++ (show $ configPort conf :: String)
-  conn <- connectPostgreSQL' dburi
-  runTLS tls settings $ gzip def $ cors corsPolicy $ reportPgErrors $ app conn (cs $ configAnonRole conf)
+  let pgSettings = H.ParamSettings (cs $ configDbHost conf)
+                     (fromIntegral $ configDbPort conf)
+                     (cs $ configDbUser conf)
+                     (cs $ configDbPass conf)
+                     (cs $ configDbName conf)
+
+  sessSettings <- maybe (fail "Improper session settings") return $
+                    H.sessionSettings (fromIntegral $ configPool conf) 30
+
+  let appSettings = setPort port
+                  . setServerName (cs $ "postgrest/" <> prettyVersion)
+                  $ defaultSettings
+      middle =
+        (if configSecure conf then redirectInsecure else id)
+        . gzip def . cors corsPolicy
+        . staticPolicy (only [("favicon.ico", "static/favicon.ico")])
+      anonRole = cs $ configAnonRole conf
+      currRole = cs $ configDbUser conf
+
+  H.session pgSettings sessSettings $ H.sessionUnlifter >>= \unlift ->
+    liftIO $ runSettings appSettings $ middle $ \req respond -> do
+      body <- strictRequestBody req
+      respond =<< catchJust isSqlError
+        (unlift $ H.tx Nothing
+                $ authenticated currRole anonRole (app body) req)
+        (return . sqlError)
 
   where
     describe = progDesc "create a REST API to an existing Postgres database"
+    prettyVersion = intercalate "." $ map show $ versionBranch version

src/Middleware.hs

@@ -1,33 +1,76 @@
-{-# LANGUAGE OverloadedStrings #-}
 {-# OPTIONS_GHC -fno-warn-orphans #-}
+{-# LANGUAGE ScopedTypeVariables #-}
 
 module Middleware where
 
-import Data.Aeson
+import Data.Maybe (fromMaybe)
+import Data.Monoid (mconcat)
+import Data.Text
+-- import Data.Pool(withResource, Pool)
 
-import Network.HTTP.Types.Header (hContentType)
-import Network.HTTP.Types.Status (status400)
-import Database.HDBC.Types (SqlError(..))
-import Control.Exception (catchJust)
-import Network.Wai
+import qualified Hasql as H
+import qualified Hasql.Postgres as H
+import Data.String.Conversions(cs)
+
+import Network.HTTP.Types.Header (hLocation, hAuthorization)
+import Network.HTTP.Types (RequestHeaders)
+import Network.HTTP.Types.Status (status400, status401, status301)
+import Network.Wai (Application, requestHeaders, responseLBS, rawPathInfo,
+                   rawQueryString, isSecure, Request(..), Response)
+import Network.URI (URI(..), parseURI)
+
+import Auth (LoginAttempt(..), signInRole, setRole, resetRole)
+import Codec.Binary.Base64.String (decode)
+
+authenticated :: forall s. Text -> Text ->
+                 (Request -> H.Tx H.Postgres s Response) ->
+                 Request -> H.Tx H.Postgres s Response
+authenticated currentRole anon app req = do
+  attempt <- httpRequesterRole (requestHeaders req)
+  case attempt of
+    MalformedAuth ->
+      return $ responseLBS status400 [] "Malformed basic auth header"
+    LoginFailed ->
+      return $ responseLBS status401 [] "Invalid username or password"
+    LoginSuccess role -> if role /= currentRole then runInRole role else app req
+    NoCredentials ->     if anon /= currentRole then runInRole anon else app req
+
+ where
+   httpRequesterRole :: RequestHeaders -> H.Tx H.Postgres s LoginAttempt
+   httpRequesterRole hdrs = do
+    let auth = fromMaybe "" $ lookup hAuthorization hdrs
+    case split (==' ') (cs auth) of
+      ("Basic" : b64 : _) ->
+        case split (==':') (cs . decode . cs $ b64) of
+          (u:p:_) -> signInRole u p
+          _ -> return MalformedAuth
+      _ -> return NoCredentials
+
+   runInRole :: Text -> H.Tx H.Postgres s Response
+   runInRole r = do
+     setRole r
+     res <- app req
+     resetRole
+     return res
 
 
-instance ToJSON SqlError where
-  toJSON t = object [
-      "error" .= object [
-          "code"    .= seNativeError t
-        , "message" .= seErrorMsg t
-        , "state"   .= seState t
-      ]
-    ]
+redirectInsecure :: Application -> Application
+redirectInsecure app req respond = do
+  let hdrs = requestHeaders req
+      host = lookup "host" hdrs
+      uriM = parseURI . cs =<< mconcat [
+        Just "https://",
+        host,
+        Just $ rawPathInfo req,
+        Just $ rawQueryString req]
+      isHerokuSecure = lookup "x-forwarded-proto" hdrs == Just "https"
 
-reportPgErrors :: Middleware
-reportPgErrors app req respond =
-  catchJust isPgException (app req respond) (
-      respond . responseLBS status400 [(hContentType, "application/json")]
-              . encode
-    )
-
-  where
-    isPgException :: SqlError -> Maybe SqlError
-    isPgException = Just
+  if not (isSecure req || isHerokuSecure)
+    then case uriM of
+              Just uri ->
+                respond $ responseLBS status301 [
+                    (hLocation, cs . show $ uri { uriScheme = "https:" })
+                  ] ""
+              Nothing ->
+                respond $ responseLBS status400 [] "SSL is required"
+    else app req respond

src/PgError.hs

@@ -0,0 +1,86 @@
+{-# LANGUAGE OverloadedStrings #-}
+
+module PgError (Message(..), message, httpStatus) where
+
+import Text.Parsec
+import Text.Parsec.Text
+import qualified Data.Map as M
+import Text.Regex.TDFA.Text ()
+import Data.Text hiding (drop, concat, head)
+import Data.Aeson
+import Data.Maybe
+import Control.Monad (void)
+
+import Data.String.Conversions (cs)
+import Data.CaseInsensitive (CI, mk)
+
+import Network.HTTP.Types.Status
+
+data Message = Message {
+    msgStatus :: Maybe Text
+  , msgCode   :: Text
+  , msgText   :: Maybe Text
+  , msgHint   :: Maybe Text
+} deriving (Show, Eq)
+
+message :: Parser Message
+message = do
+  ps <- sepBy valPair (char ';')
+  let m = M.fromList ps
+  return $ Message
+    (M.lookup "status" m)
+    (fromMaybe "" $ M.lookup "code" m)
+    (M.lookup "message" m)
+    (M.lookup "hint" m)
+
+valPair :: Parser (CI Text, Text)
+valPair = do
+  _ <- spaces
+  name <- many1 letter
+  _ <- char ':'
+  spaces
+  _ <- many $ char '"'
+  val <- manyTill anyChar $
+    try
+      (void $ many (char '"') >> (
+        (void . lookAhead $ (char ';'))
+        <|> ((optional $ char '.') >> eof)
+      ))
+  return (mk (cs name), cs val)
+
+
+instance ToJSON Message where
+  toJSON t = object [
+      "message" .= msgText t
+    , "code"    .= msgCode t
+    , "status"  .= msgStatus t
+    , "hint"    .= msgHint t
+    ]
+
+httpStatus :: Message -> Status
+httpStatus m =
+  let code = cs $ msgCode m :: String in
+  case code of
+    '0' : '8' : _ -> status503 -- pg connection err
+    '0' : '9' : _ -> status500 -- triggered action exception
+    '0' : 'L' : _ -> status403 -- invalid grantor
+    '0' : 'P' : _ -> status403 -- invalid role specification
+    '2' : '5' : _ -> status500 -- invalid tx state
+    '2' : '8' : _ -> status403 -- invalid auth specification
+    '2' : 'D' : _ -> status500 -- invalid tx termination
+    '3' : '8' : _ -> status500 -- external routine exception
+    '3' : '9' : _ -> status500 -- external routine invocation
+    '3' : 'B' : _ -> status500 -- savepoint exception
+    '4' : '0' : _ -> status500 -- tx rollback
+    '5' : '3' : _ -> status503 -- insufficient resources
+    '5' : '4' : _ -> status413 -- too complex
+    '5' : '5' : _ -> status500 -- obj not on prereq state
+    '5' : '7' : _ -> status500 -- operator intervention
+    '5' : '8' : _ -> status500 -- system error
+    'F' : '0' : _ -> status500 -- conf file error
+    'H' : 'V' : _ -> status500 -- foreign data wrapper error
+    'P' : '0' : _ -> status500 -- PL/pgSQL Error
+    'X' : 'X' : _ -> status500 -- internal Error
+    "42P01" -> status404 -- undefined table
+    "42501" -> status404 -- insufficient privilege
+    _ -> status400

src/PgQuery.hs

@@ -1,210 +1,220 @@
-{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE TypeSynonymInstances, FlexibleInstances #-}
+module PgQuery where
 
--- {{{ Imports
+import RangeQuery
 
-module PgQuery (
-  getRows
-, insert
-, upsert
-, addUser
-, signInRole
-, pgSetRole
-, pgResetRole
-, checkPass
-, RangedResult(..)
-, LoginAttempt(..)
-, DbRole
-) where
-
-import Data.Text (Text)
-import Data.String.Conversions (cs)
-import Data.Functor ( (<$>) )
-import Data.Maybe (fromMaybe)
-import Data.List (intersperse, intercalate)
-import Data.Monoid ((<>), mconcat)
-import qualified Data.Map as M
-
-import Control.Monad (join, void)
-
-import qualified RangeQuery as R
-import qualified Data.ByteString.Char8 as BS
-import qualified Data.ByteString.Lazy as BL
-
-import Database.HDBC hiding (colType, colNullable)
-import Database.HDBC.PostgreSQL
+import qualified Hasql.Postgres as H
+import qualified Hasql.Backend as H
 
+import Data.Text hiding (map)
+import Text.Regex.TDFA ( (=~) )
+import Text.Regex.TDFA.Text ()
 import qualified Network.HTTP.Types.URI as Net
+import qualified Data.ByteString.Char8 as BS
+import Data.Monoid
+import Data.Maybe (fromMaybe, mapMaybe)
+import Data.Functor ( (<$>) )
+import Control.Monad (join)
+import Data.String.Conversions (cs)
+import qualified Data.Aeson as JSON
+import qualified Data.List as L
+import Data.Scientific (isInteger, formatScientific, FPFormat(..))
 
-import Types (SqlRow(..), getRow, sqlRowColumns, sqlRowValues)
-import Crypto.BCrypt (hashPasswordUsingPolicy, fastBcryptHashingPolicy, validatePassword)
+type DynamicSQL = (BS.ByteString, [H.StatementArgument H.Postgres], All)
 
--- }}}
+type StatementT = DynamicSQL -> DynamicSQL
 
-data RangedResult = RangedResult {
-  rrFrom  :: Int
-, rrTo    :: Int
-, rrTotal :: Int
-, rrBody  :: BL.ByteString
+data QualifiedTable = QualifiedTable {
+  qtSchema :: Text
+, qtName   :: Text
 } deriving (Show)
 
-type QuotedSql = (String, [SqlValue])
-type Schema = String
-type DbRole = BS.ByteString
+data OrderTerm = OrderTerm {
+  otTerm :: Text
+, otDirection :: BS.ByteString
+}
 
-data LoginAttempt =
-    NoCredentials
-  | MalformedAuth
-  | LoginFailed
-  | LoginSuccess DbRole
-  deriving (Eq, Show)
+limitT :: Maybe NonnegRange -> StatementT
+limitT r q =
+  q <> (" LIMIT " <> limit <> " OFFSET " <> offset <> " ", [], mempty)
+  where
+    limit  = maybe "ALL" (cs . show) $ join $ rangeLimit <$> r
+    offset = cs . show $ fromMaybe 0 $ rangeOffset <$> r
 
-getRows :: Schema -> String -> Net.Query -> Maybe R.NonnegRange -> Connection -> IO RangedResult
-getRows schema table qq range conn = do
-  query <- populateSql conn
-    $ globalAndLimitedCounts schema table qq <>
-      jsonArrayRows
-      (selectStarClause schema table
-        <> whereClause qq
-        <> limitClause range)
-  r <- quickQuery conn query []
+whereT :: Net.Query -> StatementT
+whereT params q =
+ if L.null cols
+   then q
+   else q <> (" where ",[],mempty) <> conjunction
+ where
+   cols = [ col | col <- params, fst col `notElem` ["order"] ]
+   conjunction = mconcat $ L.intersperse andq (map wherePred cols)
 
-  return $ case r of
-           [[total, _, SqlNull]] -> RangedResult offset 0 (fromSql total) ""
-           [[total, limited_total, json]] ->
-            RangedResult offset (offset + fromSql limited_total - 1)
-                         (fromSql total) (fromSql json)
-           _ -> RangedResult 0 0 0 ""
+orderT :: [OrderTerm] -> StatementT
+orderT ts q =
+  if L.null ts
+    then q
+    else q <> (" order by ",[],mempty) <> clause
+  where
+   clause = mconcat $ L.intersperse commaq (map queryTerm ts)
+   queryTerm :: OrderTerm -> DynamicSQL
+   queryTerm t = (" " <> cs (pgFmtIdent $ otTerm t) <> " "
+                      <> otDirection t              <> " "
+                  , [], mempty)
+
+parentheticT :: StatementT
+parentheticT (sql, params, pre) =
+  (" (" <> sql <> ") ", params, pre)
+
+iffNotT :: DynamicSQL -> StatementT
+iffNotT (aq, ap, apre) (bq, bp, bpre) =
+  ("WITH aaa AS (" <> aq <> " returning *) " <>
+    bq <> " WHERE NOT EXISTS (SELECT * FROM aaa)"
+  , ap ++ bp
+  , All $ getAll apre && getAll bpre
+  )
+
+countRows :: QualifiedTable -> DynamicSQL
+countRows t =
+  ("select count(1) from " <> fromQt t, [], mempty)
+
+asJsonWithCount :: StatementT
+asJsonWithCount (sql, params, pre) = (
+    "count(t), array_to_json(array_agg(row_to_json(t)))::character varying from (" <> sql <> ") t"
+  , params, pre
+  )
+
+asJsonRow :: StatementT
+asJsonRow (sql, params, pre) = (
+    "row_to_json(t) from (" <> sql <> ") t", params, pre
+  )
+
+selectStar :: QualifiedTable -> DynamicSQL
+selectStar t =
+  ("select * from " <> fromQt t, [], mempty)
+
+insertInto :: QualifiedTable -> [Text] -> [JSON.Value] -> DynamicSQL
+insertInto t [] _ =
+  ("insert into " <> fromQt t <> " default values returning *", [], mempty)
+insertInto t cols vals =
+  ("insert into " <> fromQt t <> " (" <>
+    cs (intercalate ", " (map pgFmtIdent cols)) <>
+    ") values (" <>
+    cs (
+      intercalate ", " (map
+        ((<> "::unknown") . pgFmtLit . unquoted)
+        vals)
+    ) <> ") returning row_to_json(" <> fromQt t <> ".*)"
+  , []
+  , mempty
+  )
+
+insertSelect :: QualifiedTable -> [Text] -> [JSON.Value] -> DynamicSQL
+insertSelect t [] _ =
+  ("insert into " <> fromQt t <> " default values returning *", [], mempty)
+insertSelect t cols vals =
+  ("insert into " <> fromQt t <> " (" <>
+    cs (intercalate ", " (map pgFmtIdent cols)) <>
+    ") select " <>
+    cs (
+      intercalate ", " (map
+        ((<> "::unknown") . pgFmtLit . unquoted)
+        vals)
+    )
+  , []
+  , mempty
+  )
+
+update :: QualifiedTable -> [Text] -> [JSON.Value] -> DynamicSQL
+update t cols vals =
+  ("update " <> fromQt t <> " set (" <>
+    cs (intercalate ", " (map pgFmtIdent cols)) <>
+    ") = (" <>
+    cs (
+      intercalate ", " (map
+        ((<> "::unknown") . pgFmtLit . unquoted)
+        vals)
+    ) <> ")"
+  , []
+  , mempty
+  )
+
+wherePred :: Net.QueryItem -> DynamicSQL
+wherePred (col, predicate) =
+  (" " <> cs (pgFmtIdent $ cs col) <> " " <> op <> " " <> cs (pgFmtLit value) <> "::unknown ", [], mempty)
 
   where
-    offset = fromMaybe 0 $ R.offset <$> range
-
-whereClause :: Net.Query -> QuotedSql
-whereClause qs =
-  if null qs then ("", []) else (" where ", []) <> conjunction
-
-  where
-    conjunction = mconcat $ intersperse (" and ", []) (map wherePred qs)
-
-wherePred :: Net.QueryItem -> QuotedSql
-wherePred (column, predicate) =
-  ("%I " <> op <> "%L", map toSql [column, value])
-
-  where
-    opCode:rest = BS.split '.' $ fromMaybe "." predicate
-    value = BS.intercalate "." rest
+    opCode:rest = split (=='.') $ cs $ fromMaybe "." predicate
+    value = intercalate "." rest
     op = case opCode of
-              "eq"  -> "="
-              "gt"  -> ">"
-              "lt"  -> "<"
-              "gte" -> ">="
-              "lte" -> "<="
-              "neq" -> "<>"
-              _     -> "="
-
-limitClause :: Maybe R.NonnegRange -> QuotedSql
-limitClause range =
-  (" LIMIT %s OFFSET %s ", [toSql limit, toSql offset])
+         "eq"  -> "="
+         "gt"  -> ">"
+         "lt"  -> "<"
+         "gte" -> ">="
+         "lte" -> "<="
+         "neq" -> "<>"
+         _     -> "="
 
+orderParse :: Net.Query -> [OrderTerm]
+orderParse q =
+  mapMaybe orderParseTerm . split (==',') $ cs order
   where
-    limit  = fromMaybe "ALL" $ show <$> (R.limit =<< range)
-    offset = fromMaybe 0     $ R.offset <$> range
+    order = fromMaybe "" $ join (lookup "order" q)
 
-globalAndLimitedCounts :: Schema -> String -> Net.Query -> QuotedSql
-globalAndLimitedCounts schema table qq =
-  (" select ", [])
-  <> ("(select count(1) from %I.%I ", map toSql [schema, table])
-  <> whereClause qq
-  <> ("), count(t), ", [])
+orderParseTerm :: Text -> Maybe OrderTerm
+orderParseTerm s =
+  case split (=='.') s of
+       [d,c] ->
+         if d `elem` ["asc", "desc"]
+            then Just $ OrderTerm c $
+              if d == "asc" then "asc" else "desc"
+            else Nothing
+       _ -> Nothing
 
-selectStarClause :: Schema -> String -> QuotedSql
-selectStarClause schema table =
-  (" select * from %I.%I ", map toSql [schema, table])
+commaq :: DynamicSQL
+commaq  = (", ", [], mempty)
 
-jsonArrayRows :: QuotedSql -> QuotedSql
-jsonArrayRows q =
-  ("array_to_json(array_agg(row_to_json(t))) from (", []) <> q <> (") t", [])
+andq :: DynamicSQL
+andq = (" and ", [], mempty)
 
-insert :: Schema -> Text -> SqlRow -> Connection -> IO (M.Map String SqlValue)
-insert schema table row conn = do
-  sql    <- populateSql conn $ insertClause schema table row
-  stmt   <- prepare conn sql
-  _      <- execute stmt $ sqlRowValues row
-  Just m <- fetchRowMap stmt
-  return m
+pgFmtIdent :: Text -> Text
+pgFmtIdent x =
+  let escaped = replace "\"" "\"\"" (trimNullChars $ cs x) in
+  if escaped =~ danger
+    then "\"" <> escaped <> "\""
+    else escaped
 
-addUser :: BS.ByteString -> BS.ByteString -> BS.ByteString -> Connection -> IO ()
-addUser identity pass role conn = do
-  hashed <- hashPasswordUsingPolicy fastBcryptHashingPolicy $ cs pass
-  _ <- insert "dbapi" "auth" (SqlRow [
-      ("id", toSql identity), ("pass", toSql hashed), ("rolname", toSql role)
-    ]) conn
-  return ()
+  where danger = "^$|^[^a-z_]|[^a-z_0-9]" :: Text
 
-signInRole :: BS.ByteString -> BS.ByteString -> Connection -> IO LoginAttempt
-signInRole user pass conn = do
-  u <- quickQuery conn "select pass, rolname from dbapi.auth where id = ?" [toSql user]
-  return $ case u of
-    [[hashed, role]] ->
-      if checkPass (fromSql hashed) (cs pass)
-         then LoginSuccess $ fromSql role
-         else LoginFailed
-    _ -> LoginFailed
+pgFmtLit :: Text -> Text
+pgFmtLit x =
+  let trimmed = trimNullChars x
+      escaped = "'" <> replace "'" "''" trimmed <> "'"
+      slashed = replace "\\" "\\\\" escaped in
+  cs $ if escaped =~ ("\\\\" :: Text)
+    then "E" <> slashed
+    else slashed
 
-checkPass :: BS.ByteString -> BS.ByteString -> Bool
-checkPass = validatePassword
+trimNullChars :: Text -> Text
+trimNullChars = Data.Text.takeWhile (/= '\x0')
 
-upsert :: Schema -> Text -> SqlRow -> Net.Query -> Connection -> IO (M.Map String SqlValue)
-upsert schema table row qq conn = do
-  sql    <- populateSql conn $ upsertClause schema table row qq
-  stmt   <- prepare conn sql
-  _      <- execute stmt $ join $ replicate 2 $ sqlRowValues row
-  Just m <- fetchRowMap stmt
-  return m
+fromQt :: QualifiedTable -> BS.ByteString
+fromQt t = cs $ pgFmtIdent (qtSchema t) <> "." <> pgFmtIdent (qtName t)
 
-placeholders :: String -> SqlRow -> String
-placeholders symbol = intercalate ", " . map (const symbol) . getRow
+unquoted :: JSON.Value -> Text
+unquoted (JSON.String t) = t
+unquoted (JSON.Number n) =
+  cs $ formatScientific Fixed (if isInteger n then Just 0 else Nothing) n
+unquoted (JSON.Bool b) = cs . show $ b
+unquoted _ = ""
 
-insertClause :: Schema -> Text -> SqlRow -> QuotedSql
-insertClause schema table row =
-    ("insert into %I.%I (" ++ placeholders "%I" row ++ ")",
-     map toSql $ cs schema : table : sqlRowColumns row)
-  <> (" values (" ++ placeholders "?" row ++ ") returning *", sqlRowValues row)
-
-
-insertClauseViaSelect :: Schema -> Text -> SqlRow -> QuotedSql
-insertClauseViaSelect schema table row =
-    ("insert into %I.%I (" ++ placeholders "%I" row ++ ")",
-     map toSql $ cs schema : table : sqlRowColumns row)
-  <> (" select " ++ placeholders "?" row, sqlRowValues row)
-
-updateClause :: Schema -> Text -> SqlRow -> QuotedSql
-updateClause schema table row =
-    ("update %I.%I set (" ++ placeholders "%I" row ++ ")",
-     map toSql $ cs schema : table : sqlRowColumns row)
-  <> (" = (" ++ placeholders "?" row ++ ")", [])
-
-upsertClause :: Schema -> Text -> SqlRow -> Net.Query -> QuotedSql
-upsertClause schema table row qq =
-  ("with upsert as (", []) <> updateClause schema table row
-  <> whereClause qq
-  <> (" returning *) ", []) <> insertClauseViaSelect schema table row
-  <> (" where not exists (select * from upsert) returning *", [])
-
-populateSql :: Connection -> QuotedSql -> IO String
-populateSql conn sql = do
-  [[escaped]] <- quickQuery conn q (snd sql)
-  return $ fromSql escaped
-
-  where
-    q = concat [ "select format('", fst sql, "', ", ph (snd sql), ")" ]
-
-    ph :: [a] -> String
-    ph = intercalate ", " . map (const "?::varchar")
-
-pgSetRole :: Connection -> DbRole -> IO ()
-pgSetRole conn role = do
-  query <- populateSql conn ("set role %I", [toSql role])
-  void $ run conn query []
-
-pgResetRole :: Connection -> IO ()
-pgResetRole conn = void $ run conn "reset role" []
+pgParam :: JSON.Value -> H.StatementArgument H.Postgres
+pgParam (JSON.Number n) = H.renderValue
+  (cs $ formatScientific Fixed
+       (if isInteger n then Just 0 else Nothing) n :: Text)
+pgParam (JSON.String s) = H.renderValue s
+pgParam (JSON.Bool      b) = H.renderValue $
+  if b then "t" else "f" :: Text
+pgParam JSON.Null       = H.renderValue (Nothing :: Maybe Text)
+pgParam (JSON.Object o) = H.renderValue $ JSON.encode o
+pgParam (JSON.Array  a) = H.renderValue $ JSON.encode a

src/PgStructure.hs

@@ -1,80 +1,171 @@
-{-# LANGUAGE OverloadedStrings #-}
-
+{-# LANGUAGE QuasiQuotes, OverloadedStrings, TypeSynonymInstances,
+             MultiParamTypeClasses, ScopedTypeVariables #-}
 module PgStructure where
 
+import PgQuery (QualifiedTable(..))
 import Data.Functor ( (<$>) )
-import Data.Maybe (mapMaybe)
+import Data.Text hiding (foldl, map, zipWith, concat)
+import Data.Aeson
+import Data.Functor.Identity
+import qualified Data.Vector as V
+import Data.String.Conversions (cs)
 
 import Control.Applicative ( (<*>) )
 
-import qualified Data.ByteString.Lazy as BL
-import Data.List.Split (splitOn)
-
-import qualified Data.Aeson as JSON
+import qualified Data.List as L
 import qualified Data.Map as Map
 
-import Database.HDBC hiding (colType, colNullable)
-import Database.HDBC.PostgreSQL
+import qualified Hasql as H
+import qualified Hasql.Backend as H
+import qualified Hasql.Postgres as H
 
-import Data.Aeson ((.=))
+foreignKeys :: QualifiedTable -> H.Tx H.Postgres s (Map.Map Text ForeignKey)
+foreignKeys table = do
+  r :: [(Text, Text, Text)] <- H.list $ [H.q|
+      select kcu.column_name, ccu.table_name AS foreign_table_name,
+        ccu.column_name AS foreign_column_name
+      from information_schema.table_constraints AS tc
+        join information_schema.key_column_usage AS kcu
+          on tc.constraint_name = kcu.constraint_name
+        join information_schema.constraint_column_usage AS ccu
+          on ccu.constraint_name = tc.constraint_name
+      where constraint_type = 'FOREIGN KEY'
+        and tc.table_name=? and tc.table_schema = ?
+        order by kcu.column_name
+    |] (qtName table) (qtSchema table)
+
+  return $ foldl addKey Map.empty r
+  where
+    addKey m (col, ftab, fcol) = Map.insert col (ForeignKey (cs ftab) (cs fcol)) m
+
+
+tables :: Text -> H.Tx H.Postgres s [Table]
+tables schema =
+  H.list $ [H.q|
+      select table_schema, table_name,
+             is_insertable_into
+        from information_schema.tables
+       where table_schema = ?
+       order by table_name
+    |] schema
+
+
+columns :: QualifiedTable -> H.Tx H.Postgres s [Column]
+columns table = do
+  cols <- H.list $ [H.q|
+    select info.table_schema as schema, info.table_name as table_name,
+           info.column_name as name, info.ordinal_position as position,
+           info.is_nullable as nullable, info.data_type as col_type,
+           info.is_updatable as updatable,
+           info.character_maximum_length as max_len,
+           info.numeric_precision as precision,
+           info.column_default as default_value,
+           array_to_string(enum_info.vals, ',') as enum
+       from (
+         select table_schema, table_name, column_name, ordinal_position,
+                is_nullable, data_type, is_updatable,
+                character_maximum_length, numeric_precision,
+                column_default, udt_name
+           from information_schema.columns
+          where table_schema = ? and table_name = ?
+       ) as info
+       left outer join (
+         select n.nspname as s,
+                t.typname as n,
+                array_agg(e.enumlabel ORDER BY e.enumsortorder) as vals
+         from pg_type t
+            join pg_enum e on t.oid = e.enumtypid
+            join pg_catalog.pg_namespace n ON n.oid = t.typnamespace
+         group by s, n
+       ) as enum_info
+       on (info.udt_name = enum_info.n)
+    order by position |] (qtSchema table) (qtName table)
+
+  fks <- foreignKeys table
+  return $ map (\col -> col { colFK = Map.lookup (cs . colName $ col) fks }) cols
+
+
+primaryKeyColumns :: QualifiedTable -> H.Tx H.Postgres s [Text]
+primaryKeyColumns table = do
+  r :: [Identity Text] <- H.list $ [H.q|
+    select kc.column_name
+      from
+        information_schema.table_constraints tc,
+        information_schema.key_column_usage kc
+    where
+      tc.constraint_type = 'PRIMARY KEY'
+      and kc.table_name = tc.table_name and kc.table_schema = tc.table_schema
+      and kc.constraint_name = tc.constraint_name
+      and kc.table_schema = ?
+      and kc.table_name  = ? |] (qtSchema table) (qtName table)
+  return $ map runIdentity r
+
+
+vanishNull :: [a] -> Maybe [a]
+vanishNull xs = if L.null xs then Nothing else Just xs
+
+toBool :: Text -> Bool
+toBool = (== "YES")
 
 data Table = Table {
-  tableSchema :: String
-, tableName :: String
+  tableSchema :: Text
+, tableName :: Text
 , tableInsertable :: Bool
 } deriving (Show)
 
-instance JSON.ToJSON Table where
-  toJSON v = JSON.object [
-      "schema"     .= tableSchema v
-    , "name"       .= tableName v
-    , "insertable" .= tableInsertable v ]
-
-toBool :: String -> Bool
-toBool = (== "YES")
-
 data ForeignKey = ForeignKey {
-  fkTable::String, fkCol::String
+  fkTable::Text, fkCol::Text
 } deriving (Eq, Show)
 
-instance JSON.ToJSON ForeignKey where
-  toJSON fk = JSON.object ["table".=fkTable fk, "column".=fkCol fk]
-
-foreignKeys :: String -> String -> Connection -> IO (Map.Map String ForeignKey)
-foreignKeys schema table conn = do
-  r <- quickQuery conn
-    "select kcu.column_name, ccu.table_name AS foreign_table_name,\
-    \  ccu.column_name AS foreign_column_name \
-    \from information_schema.table_constraints AS tc \
-    \  join information_schema.key_column_usage AS kcu \
-    \    on tc.constraint_name = kcu.constraint_name \
-    \  join information_schema.constraint_column_usage AS ccu \
-    \    on ccu.constraint_name = tc.constraint_name \
-    \where constraint_type = 'FOREIGN KEY' \
-    \  and tc.table_name=? and tc.table_schema = ? \
-    \order by kcu.column_name" (map toSql [table, schema])
-  return $ foldl addKey Map.empty $ map (map fromSql) r
-  where
-    addKey m [col, ftab, fcol] = Map.insert col (ForeignKey ftab fcol) m
-    addKey m _ = m --should never happen
-
 data Column = Column {
-  colSchema :: String
-, colTable :: String
-, colName :: String
+  colSchema :: Text
+, colTable :: Text
+, colName :: Text
 , colPosition :: Int
 , colNullable :: Bool
-, colType :: String
+, colType :: Text
 , colUpdatable :: Bool
 , colMaxLen :: Maybe Int
 , colPrecision :: Maybe Int
-, colDefault :: Maybe String
-, colEnum :: Maybe [String]
+, colDefault :: Maybe Text
+, colEnum :: [Text]
 , colFK :: Maybe ForeignKey
 } deriving (Show)
 
-instance JSON.ToJSON Column where
-  toJSON c = JSON.object [
+instance H.RowParser H.Postgres Column where
+  parseRow r =
+    let schema     = H.parseResult $ r V.! 0
+        table      = H.parseResult $ r V.! 1
+        name       = H.parseResult $ r V.! 2
+        position   = H.parseResult $ r V.! 3
+        nullable   = toBool <$> (H.parseResult $ r V.! 4 :: Either Text Text)
+        typ        = H.parseResult $ r V.! 5
+        updatable  = toBool <$> (H.parseResult $ r V.! 6 :: Either Text Text)
+        maxLen     = H.parseResult $ r V.! 7
+        precision  = H.parseResult $ r V.! 8
+        defValue   = H.parseResult $ r V.! 9
+        enum       = either (const $ Right []) (Right . split (==','))
+                       (H.parseResult $ r V.! 10 :: Either Text Text)
+      in
+    if V.length r /= 11
+       then Left "Wrong number of fields in Column"
+       else Column <$> schema <*> table <*> name <*> position <*> nullable
+                   <*> typ <*> updatable <*> maxLen <*> precision
+                   <*> defValue <*> enum
+                   <*> return Nothing
+
+
+instance H.RowParser H.Postgres Table where
+  parseRow r =
+    let schema     = H.parseResult $ r V.! 0
+        name       = H.parseResult $ r V.! 1
+        insertable = toBool <$> (H.parseResult $ r V.! 2 :: Either Text Text) in
+    if V.length r /= 3
+       then Left "Wrong number of fields in Table"
+       else Table <$> schema <*> name <*> insertable
+
+instance ToJSON Column where
+  toJSON c = object [
       "schema"    .= colSchema c
     , "name"      .= colName c
     , "position"  .= colPosition c
@@ -87,107 +178,11 @@ instance JSON.ToJSON Column where
     , "default"   .= colDefault c
     , "enum"      .= colEnum c ]
 
-data TableOptions = TableOptions {
-  tblOptcolumns :: [Column]
-, tblOptpkey :: [String]
-}
+instance ToJSON ForeignKey where
+  toJSON fk = object ["table".=fkTable fk, "column".=fkCol fk]
 
-instance JSON.ToJSON TableOptions where
-  toJSON t = JSON.object [
-      "columns" .= tblOptcolumns t
-    , "pkey"    .= tblOptpkey t ]
-
-tables :: String -> Connection -> IO [Table]
-tables s conn = do
-  r <- quickQuery conn
-        "select table_schema, table_name,\
-        \       is_insertable_into\
-        \  from information_schema.tables\
-        \ where table_schema = ?\
-        \ order by table_name" [toSql s]
-  return $ mapMaybe mkTable r
-
-  where
-    mkTable [schema, name, insertable] =
-      Just $ Table (fromSql schema)
-        (fromSql name)
-        (toBool (fromSql insertable))
-    mkTable _ = Nothing
-
-columns :: String -> String -> Connection -> IO [Column]
-columns s t conn = do
-  r <- quickQuery conn
-    "select info.table_schema as schema, info.table_name as table_name, \
-    \       info.column_name as name, info.ordinal_position as position, \
-    \       info.is_nullable as nullable, info.data_type as col_type, \
-    \       info.is_updatable as updatable, \
-    \       info.character_maximum_length as max_len, \
-    \       info.numeric_precision as precision, \
-    \       info.column_default as default_value, \
-    \       array_to_string(enum_info.vals, ',') as enum \
-    \   from ( \
-    \     select table_schema, table_name, column_name, ordinal_position, \
-    \            is_nullable, data_type, is_updatable, \
-    \            character_maximum_length, numeric_precision, \
-    \            column_default, udt_name \
-    \       from information_schema.columns \
-    \      where table_schema = ? and table_name = ? \
-    \   ) as info \
-    \   left outer join ( \
-    \     select n.nspname as s, \
-    \            t.typname as n, \
-    \            array_agg(e.enumlabel) as vals \
-    \     from pg_type t \
-    \        join pg_enum e on t.oid = e.enumtypid \
-    \        join pg_catalog.pg_namespace n ON n.oid = t.typnamespace \
-    \     group by s, n \
-    \   ) as enum_info \
-    \   on (info.udt_name = enum_info.n) \
-    \order by position" [toSql s, toSql t]
-  fks <- foreignKeys s t conn
-  let lookupFK (_:_:name:_) = Map.lookup (fromSql name) fks
-      lookupFK _ = Nothing
-  let cols = zipWith ($) (map mkColumn r) (map lookupFK r)
-  return cols
-
-  where
-    mkColumn [schema, table, name, pos, nullable, colT, updatable, maxlen, precision, defVal, enum] = Column (fromSql schema)
-        (fromSql table)
-        (fromSql name)
-        (fromSql pos)
-        (toBool (fromSql nullable))
-        (fromSql colT)
-        (toBool (fromSql updatable))
-        (fromSql maxlen)
-        (fromSql precision)
-        (fromSql defVal)
-        (splitOn "," <$> fromSql enum)
-    mkColumn _ =  error $ "Incomplete column data received for table " ++
-      t ++ " in schema " ++ s ++ "."
-
-printTables :: String -> Connection -> IO BL.ByteString
-printTables schema conn = JSON.encode <$> tables schema conn
-
-printColumns :: String -> String -> Connection -> IO BL.ByteString
-printColumns schema table conn =
-  JSON.encode <$> (TableOptions <$> cols <*> pkey)
-  where
-    cols :: IO [Column]
-    cols = columns schema table conn
-    pkey :: IO [String]
-    pkey = primaryKeyColumns schema table conn
-
-primaryKeyColumns :: String -> String -> Connection -> IO [String]
-primaryKeyColumns s t conn = do
-  r <- quickQuery conn
-        "select kc.column_name \
-        \  from \
-        \    information_schema.table_constraints tc, \
-        \    information_schema.key_column_usage kc \
-        \where \
-        \  tc.constraint_type = 'PRIMARY KEY' \
-        \  and kc.table_name = tc.table_name and kc.table_schema = tc.table_schema \
-        \  and kc.constraint_name = tc.constraint_name \
-        \  and kc.table_schema = ? \
-        \  and kc.table_name  = ?" [toSql s, toSql t]
-  return $ map fromSql (concat r)
+instance ToJSON Table where
+  toJSON v = object [
+      "schema"     .= tableSchema v
+    , "name"       .= tableName v
+    , "insertable" .= tableInsertable v ]

src/RangeQuery.hs

@@ -1,10 +1,16 @@
-{-# LANGUAGE OverloadedStrings #-}
-
-module RangeQuery where
+module RangeQuery (
+  rangeParse
+, rangeRequested
+, rangeLimit
+, rangeOffset
+, NonnegRange
+) where
 
 import Control.Applicative
 import Network.HTTP.Types.Header
 
+import qualified Data.ByteString.Char8 as BS
+
 import Data.Ranged.Boundaries
 import Data.Ranged.Ranges
 
@@ -16,6 +22,33 @@ import Data.Maybe (fromMaybe, listToMaybe)
 
 type NonnegRange = Range Int
 
+rangeParse :: BS.ByteString -> Maybe NonnegRange
+rangeParse range = do
+  let rangeRegex = "^([0-9]+)-([0-9]*)$" :: BS.ByteString
+
+  parsedRange <- listToMaybe (range =~ rangeRegex :: [[BS.ByteString]])
+
+  let [_, from, to] = readMaybe . cs <$> parsedRange
+  let lower = fromMaybe emptyRange   (rangeGeq <$> from)
+  let upper = fromMaybe (rangeGeq 0) (rangeLeq <$> to)
+
+  return $ rangeIntersection lower upper
+
+rangeRequested :: RequestHeaders -> Maybe NonnegRange
+rangeRequested = (rangeParse =<<) . lookup hRange
+
+rangeLimit :: NonnegRange -> Maybe Int
+rangeLimit range =
+  case [rangeLower range, rangeUpper range]
+    of [BoundaryBelow from, BoundaryAbove to] -> Just (1 + to - from)
+       _ -> Nothing
+
+rangeOffset :: NonnegRange -> Int
+rangeOffset range =
+  case rangeLower range
+    of BoundaryBelow from -> from
+       _ -> error "range without lower bound" -- should never happen
+
 rangeGeq :: Int -> NonnegRange
 rangeGeq n =
   Range (BoundaryBelow n) BoundaryAboveAll
@@ -23,33 +56,3 @@ rangeGeq n =
 rangeLeq :: Int -> NonnegRange
 rangeLeq n =
   Range BoundaryBelowAll (BoundaryAbove n)
-
-parseRange :: String -> Maybe NonnegRange
-parseRange range = do
-  let rangeRegex = "^([0-9]+)-([0-9]*)$" :: String
-
-  parsedRange <- listToMaybe (range =~ rangeRegex :: [[String]])
-
-  let [_, from, to] = readMaybe <$> parsedRange
-  let lower = fromMaybe emptyRange   (rangeGeq <$> from)
-  let upper = fromMaybe (rangeGeq 0) (rangeLeq <$> to)
-
-  return $ rangeIntersection lower upper
-
-requestedRange :: RequestHeaders -> Maybe NonnegRange
-requestedRange hdrs = parseRange =<< cs <$> lookup hRange hdrs
-
-requestedContentRange :: RequestHeaders -> Maybe NonnegRange
-requestedContentRange hdrs = parseRange =<< cs <$> lookup "Content-Range" hdrs
-
-limit :: NonnegRange -> Maybe Int
-limit range =
-  case [rangeLower range, rangeUpper range]
-    of [BoundaryBelow from, BoundaryAbove to] -> Just (1 + to - from)
-       _ -> Nothing
-
-offset :: NonnegRange -> Int
-offset range =
-  case rangeLower range
-    of BoundaryBelow from -> from
-       _ -> error "range without lower bound" -- should never happen

src/Types.hs

@@ -1,9 +1,6 @@
 {-# OPTIONS_GHC -fno-warn-orphans #-}
-
 module Types where
 
-import Database.HDBC (toSql, iToSql, SqlValue(..))
-
 import qualified Data.Aeson as JSON
 import Data.Aeson.Types (Parser)
 
@@ -12,7 +9,6 @@ import Data.HashMap.Strict (foldlWithKey')
 import Data.Text (Text)
 import Data.Text.Encoding (decodeUtf8)
 import Data.Time.Calendar (showGregorian)
-
 import Control.Monad (mzero)
 
 instance JSON.FromJSON SqlValue where

test/Feature/AuthSpec.hs

@@ -1,27 +1,25 @@
-{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE QuasiQuotes #-}
 module Feature.AuthSpec where
 
 -- {{{ Imports
 import Test.Hspec
 import Test.Hspec.Wai
+import Test.Hspec.Wai.JSON
 import Network.HTTP.Types
 
 import SpecHelper
 -- }}}
 
 spec :: Spec
-spec = around appWithFixture $
-  describe "authorization" $ do
-    it "hides tables that anonymous does not own" $ do
-      pendingWith_ "Fix pg exception"
-      get "/authors_only" `shouldRespondWith` 400 -- TODO: should be 404
+spec = around withApp $ describe "authorization" $ do
+    it "hides tables that anonymous does not own" $
+      get "/authors_only" `shouldRespondWith` 404
     it "indicates login failure" $ do
-      pendingWith_ "Fix pg exception"
-      let auth = authHeader "dbapi_test_author_a" "fakefake"
+      let auth = authHeader "postgrest_test_author" "fakefake"
       request methodGet "/authors_only" [auth] ""
         `shouldRespondWith` 401
     it "allows users with permissions to see their tables" $ do
-      pendingWith_ "Fix pg exception"
-      let auth = authHeader "dbapi_test_author_a" ""
+      _ <- post "/postgrest/users" [json| { "id":"jdoe", "pass": "1234", "role": "postgrest_test_author" } |]
+      let auth = authHeader "jdoe" "1234"
       request methodGet "/authors_only" [auth] ""
-        `shouldRespondWith` 400
+        `shouldRespondWith` 200

test/Feature/CorsSpec.hs

@@ -1,5 +1,3 @@
-{-# LANGUAGE OverloadedStrings #-}
-
 module Feature.CorsSpec where
 
 -- {{{ Imports
@@ -14,8 +12,7 @@ import Network.HTTP.Types
 -- }}}
 
 spec :: Spec
-spec = around appWithFixture $
-  describe "CORS" $ do
+spec = around withApp $ describe "CORS" $ do
     let preflightHeaders = [
           ("Accept", "*/*"),
           ("Origin", "http://example.com"),

test/Feature/InsertSpec.hs

@@ -1,8 +1,6 @@
-
-{-# LANGUAGE OverloadedStrings, QuasiQuotes #-}
+{-# LANGUAGE QuasiQuotes #-}
 module Feature.InsertSpec where
 
--- {{{ Imports
 import Test.Hspec
 import Test.Hspec.Wai
 import Test.Hspec.Wai.JSON
@@ -14,32 +12,35 @@ import qualified Data.Aeson as JSON
 import Data.Maybe (fromJust)
 import Network.HTTP.Types.Header
 import Network.HTTP.Types
+import Control.Monad (replicateM_)
 
-import TestTypes(IncPK, incStr, incNullableStr)
+import TestTypes(IncPK(..), CompoundPK(..))
 
--- }}}
+--import Debug.Trace
 
 spec :: Spec
-spec = around appWithFixture $ do
+spec = around withApp $ do
   describe "Posting new record" $ do
-    it "accepts disparate json types" $
-      post "/menagerie"
+    after_ (clearTable "menagerie") . it "accepts disparate json types" $ do
+      p <- post "/menagerie"
         [json| {
           "integer": 13, "double": 3.14159, "varchar": "testing!"
-        , "boolean": false, "date": "01/01/1900", "money": "$3.99"
-        , "enum": ["foo"]
+        , "boolean": false, "date": "1900-01-01", "money": "$3.99"
+        , "enum": "foo"
         } |]
-        `shouldRespondWith` 201
+      liftIO $ do
+        simpleBody p `shouldBe` ""
+        simpleStatus p `shouldBe` created201
 
     context "with no pk supplied" $ do
-      context "into a table with auto-incrementing pk" $
+      context "into a table with auto-incrementing pk" . after_ (clearTable "auto_incrementing_pk") $
         it "succeeds with 201 and link" $ do
           p <- post "/auto_incrementing_pk" [json| { "non_nullable_string":"not null"} |]
           liftIO $ do
             simpleBody p `shouldBe` ""
             simpleHeaders p `shouldSatisfy` matchHeader hLocation "/auto_incrementing_pk\\?id=eq\\.[0-9]+"
             simpleStatus p `shouldBe` created201
-          let Just location = getHeader hLocation $ simpleHeaders p
+          let Just location = lookup hLocation $ simpleHeaders p
           r <- get location
           let [record] = fromJust (JSON.decode $ simpleBody r :: Maybe [IncPK])
           liftIO $ do
@@ -47,12 +48,11 @@ spec = around appWithFixture $ do
             incNullableStr record `shouldBe` Nothing
 
       context "into a table with simple pk" $
-        it "fails with 400 and error" $ do
-          pendingWith_ "Fix pg exception"
+        it "fails with 400 and error" $
           post "/simple_pk" [json| { "extra":"foo"} |]
             `shouldRespondWith` 400
 
-      context "into a table with no pk" $
+      context "into a table with no pk" . after_ (clearTable "no_pk") $
         it "succeeds with 201 and a link including all fields" $ do
           p <- post "/no_pk" [json| { "a":"foo", "b":"bar" } |]
           liftIO $ do
@@ -60,7 +60,7 @@ spec = around appWithFixture $ do
             simpleHeaders p `shouldSatisfy` matchHeader hLocation "/no_pk\\?a=eq.foo&b=eq.bar"
             simpleStatus p `shouldBe` created201
 
-    context "with compound pk supplied" $
+    context "with compound pk supplied" . after_ (clearTable "compound_pk") $
       it "builds response location header appropriately" $
         post "/compound_pk" [json| { "k1":12, "k2":42 } |]
           `shouldRespondWith` ResponseMatcher {
@@ -69,6 +69,15 @@ spec = around appWithFixture $ do
             matchHeaders = ["Location" <:> "/compound_pk?k1=eq.12&k2=eq.42"]
           }
 
+    context "with invalid json payload" $
+      it "fails with 400 and error" $
+        post "/simple_pk" "}{ x = 2"
+          `shouldRespondWith` ResponseMatcher {
+            matchBody    = Just [json| {"message":"Failed to parse JSON payload. Failed reading: satisfy"} |]
+          , matchStatus  = 400
+          , matchHeaders = []
+          }
+
   describe "Putting record" $ do
 
     context "to unkonwn uri" $
@@ -86,39 +95,93 @@ spec = around appWithFixture $ do
 
       context "with a fully-specified primary key" $ do
 
-        context "with Content-Range header" $
-          it "fails as per RFC7231" $
-            request methodPut "/compound_pk?k1=eq.1&k2=eq.2"
-              [("Content-Range", "0-0")]
-              [json| { "k1":1, "k2":2, "extra":3 } |]
-                `shouldRespondWith` 400
-
         context "not specifying every column in the table" $
           it "is rejected for lack of idempotence" $
             request methodPut "/compound_pk?k1=eq.12&k2=eq.42" []
               [json| { "k1":12, "k2":42 } |]
                 `shouldRespondWith` 400
 
-        context "specifying every column in the table" $
-          it "succeeds with 201 and link" $ do
+        context "specifying every column in the table" . after_ (clearTable "compound_pk") $ do
+          it "can create a new record" $ do
             p <- request methodPut "/compound_pk?k1=eq.12&k2=eq.42" []
                  [json| { "k1":12, "k2":42, "extra":3 } |]
             liftIO $ do
               simpleBody p `shouldBe` ""
-              simpleStatus p `shouldBe` status200
+              simpleStatus p `shouldBe` status204
 
-      context "with an auto-incrementing primary key" $
+            r <- get "/compound_pk?k1=eq.12&k2=eq.42"
+            let rows = fromJust (JSON.decode $ simpleBody r :: Maybe [CompoundPK])
+            liftIO $ do
+              length rows `shouldBe` 1
+              let record = head rows
+              compoundK1 record `shouldBe` 12
+              compoundK2 record `shouldBe` 42
+              compoundExtra record `shouldBe` Just 3
 
-        it "succeeds with 201 and link" $
+          it "can update an existing record" $ do
+            _ <- request methodPut "/compound_pk?k1=eq.12&k2=eq.42" []
+                 [json| { "k1":12, "k2":42, "extra":4 } |]
+            _ <- request methodPut "/compound_pk?k1=eq.12&k2=eq.42" []
+                 [json| { "k1":12, "k2":42, "extra":5 } |]
+
+            r <- get "/compound_pk?k1=eq.12&k2=eq.42"
+            let rows = fromJust (JSON.decode $ simpleBody r :: Maybe [CompoundPK])
+            liftIO $ do
+              length rows `shouldBe` 1
+              let record = head rows
+              compoundExtra record `shouldBe` Just 5
+
+      context "with an auto-incrementing primary key" . after_ (clearTable "auto_incrementing_pk") $
+
+        it "succeeds with 204" $
           request methodPut "/auto_incrementing_pk?id=eq.1" []
                [json| {
                  "id":1,
                  "nullable_string":"hi",
                  "non_nullable_string":"bye",
-                 "inserted_at": "now()"
+                 "inserted_at": "2020-11-11"
                } |]
             `shouldRespondWith` ResponseMatcher {
               matchBody    = Nothing,
-              matchStatus  = 200,
+              matchStatus  = 204,
               matchHeaders = []
             }
+
+  describe "Patching record" $ do
+
+    context "to unkonwn uri" $
+      it "gives a 404" $
+        request methodPatch "/fake" []
+          [json| { "real": false } |]
+            `shouldRespondWith` 404
+
+    context "on an empty table" $
+      it "succeeds with no effect" $
+        request methodPatch "/simple_pk" []
+          [json| { "extra":20 } |]
+            `shouldRespondWith` 204
+
+    context "in a nonempty table" . before_ (clearTable "items" >> createItems 15) .
+      after_ (clearTable "items") $ do
+      it "can update a single item" $ do
+        g <- get "/items?id=eq.42"
+        liftIO $ simpleHeaders g
+          `shouldSatisfy` matchHeader "Content-Range" "\\*/0"
+        request methodPatch "/items?id=eq.1" []
+          [json| { "id":42 } |]
+            `shouldRespondWith` 204
+        g' <- get "/items?id=eq.42"
+        liftIO $ simpleHeaders g'
+          `shouldSatisfy` matchHeader "Content-Range" "0-0/1"
+
+      it "can update multiple items" $ do
+        replicateM_ 10 $ post "/auto_incrementing_pk"
+          [json| { non_nullable_string: "a" } |]
+        replicateM_ 10 $ post "/auto_incrementing_pk"
+          [json| { non_nullable_string: "b" } |]
+        _ <- request methodPatch
+          "/auto_incrementing_pk?non_nullable_string=eq.a" []
+          [json| { non_nullable_string: "c" } |]
+        g <- get "/auto_incrementing_pk?non_nullable_string=eq.c"
+        liftIO $ simpleHeaders g
+          `shouldSatisfy` matchHeader "Content-Range" "0-9/10"

test/Feature/QuerySpec.hs

@@ -1,5 +1,3 @@
-
-{-# LANGUAGE OverloadedStrings #-}
 module Feature.QuerySpec where
 
 import Test.Hspec
@@ -8,7 +6,12 @@ import Test.Hspec.Wai
 import SpecHelper
 
 spec :: Spec
-spec = around appWithFixture $ do
+spec = beforeAll (clearTable "items" >> createItems 15)
+  . afterAll_ (clearTable "items") . around withApp $ do
+  describe "Querying a nonexistent table" $
+    it "causes a 404" $
+      get "/faketable" `shouldRespondWith` 404
+
   describe "Filtering response" $
     context "column equality" $
 
@@ -20,11 +23,38 @@ spec = around appWithFixture $ do
           , matchHeaders = ["Content-Range" <:> "0-0/1"]
           }
 
-  describe "Canonical location" $
+  describe "ordering response" $ do
+    it "by a column asc" $
+      get "/items?id=lte.2&order=asc.id"
+        `shouldRespondWith` ResponseMatcher {
+          matchBody    = Just "[{\"id\":1},{\"id\":2}]"
+        , matchStatus  = 200
+        , matchHeaders = ["Content-Range" <:> "0-1/2"]
+        }
+    it "by a column desc" $
+      get "/items?id=lte.2&order=desc.id"
+        `shouldRespondWith` ResponseMatcher {
+          matchBody    = Just "[{\"id\":2},{\"id\":1}]"
+        , matchStatus  = 200
+        , matchHeaders = ["Content-Range" <:> "0-1/2"]
+        }
+
+    it "without other constraints" $
+      get "/items?order=asc.id" `shouldRespondWith` 200
+
+  describe "Canonical location" $ do
     it "Sets Content-Location with alphabetized params" $
       get "/no_pk?b=eq.1&a=eq.1"
         `shouldRespondWith` ResponseMatcher {
-          matchBody    = Nothing
-        , matchStatus  = 204
+          matchBody    = Just "[]"
+        , matchStatus  = 200
         , matchHeaders = ["Content-Location" <:> "/no_pk?a=eq.1&b=eq.1"]
         }
+
+    it "Omits question mark when there are no params" $
+      get "/no_pk"
+        `shouldRespondWith` ResponseMatcher {
+          matchBody    = Just "[]"
+        , matchStatus  = 200
+        , matchHeaders = ["Content-Location" <:> "/no_pk"]
+        }

test/Feature/RangeSpec.hs

@@ -1,4 +1,3 @@
-{-# LANGUAGE OverloadedStrings #-}
 module Feature.RangeSpec where
 
 import Test.Hspec
@@ -9,7 +8,8 @@ import Network.Wai.Test (SResponse(simpleHeaders,simpleStatus))
 import SpecHelper
 
 spec :: Spec
-spec = around appWithFixture $
+spec = beforeAll (clearTable "items" >> createItems 15) . afterAll_ (clearTable "items")
+  . around withApp $
   describe "GET /items" $ do
 
     context "without range headers" $
@@ -37,8 +37,8 @@ spec = around appWithFixture $
           request methodGet "/menagerie"
                   (rangeHdrs $ ByteRangeFromTo 0 1) ""
             `shouldRespondWith` ResponseMatcher {
-              matchBody    = Nothing
-            , matchStatus  = 204
+              matchBody    = Just "[]"
+            , matchStatus  = 200
             , matchHeaders = ["Content-Range" <:> "*/0"]
             }
 

test/Feature/StructureSpec.hs

@@ -1,29 +1,21 @@
 {-# LANGUAGE OverloadedStrings, QuasiQuotes #-}
 module Feature.StructureSpec where
 
-import Test.Hspec
+import Test.Hspec hiding (pendingWith)
 import Test.Hspec.Wai
 import Test.Hspec.Wai.JSON
 
 import SpecHelper
 
 import Network.HTTP.Types
-import Codec.Binary.Base64.String (encode)
-import Data.Monoid ((<>))
-import Data.String.Conversions (cs)
 
 spec :: Spec
-spec = let {uName = "a user"; uPass = "nobody can ever know";
-uRole = "dbapi_test"} in
-  around withDatabaseConnection $
-  aroundWith (withUser uName uPass uRole) $ aroundWith withApp $ do
-  describe "GET /" $
+spec = around withApp $ do
+  describe "GET /" $ do
     it "lists views in schema" $
-      request methodGet "/"
-        [("Authorization", "Basic "<>(cs.encode $ cs uName<>":"<>cs uPass))] ""
+      request methodGet "/" [] ""
         `shouldRespondWith` [json| [
-        {"schema":"1","name":"authors_only","insertable":true}
-        , {"schema":"1","name":"auto_incrementing_pk","insertable":true}
+          {"schema":"1","name":"auto_incrementing_pk","insertable":true}
         , {"schema":"1","name":"compound_pk","insertable":true}
         , {"schema":"1","name":"has_fk","insertable":true}
         , {"schema":"1","name":"items","insertable":true}
@@ -33,6 +25,17 @@ uRole = "dbapi_test"} in
         ] |]
         {matchStatus = 200}
 
+    it "lists only views user has permission to see" $ do
+      _ <- post "/postgrest/users" [json| { "id":"jdoe", "pass": "1234", "role": "postgrest_test_author" } |]
+      let auth = authHeader "jdoe" "1234"
+
+      request methodGet "/" [auth] ""
+        `shouldRespondWith` [json| [
+            {"schema":"1","name":"authors_only","insertable":true}
+        ] |]
+        {matchStatus = 200}
+
+
   describe "Table info" $ do
     it "is available with OPTIONS verb" $
       request methodOptions "/menagerie" [] "" `shouldRespondWith`
@@ -48,7 +51,7 @@ uRole = "dbapi_test"} in
             "name": "integer",
             "type": "integer",
             "maxLen": null,
-            "enum": null,
+            "enum": [],
             "nullable": false,
             "position": 1,
             "references": null,
@@ -61,7 +64,7 @@ uRole = "dbapi_test"} in
             "name": "double",
             "type": "double precision",
             "maxLen": null,
-            "enum": null,
+            "enum": [],
             "nullable": false,
             "references": null,
             "position": 2
@@ -73,7 +76,7 @@ uRole = "dbapi_test"} in
             "name": "varchar",
             "type": "character varying",
             "maxLen": null,
-            "enum": null,
+            "enum": [],
             "nullable": false,
             "position": 3,
             "references": null,
@@ -86,7 +89,7 @@ uRole = "dbapi_test"} in
             "name": "boolean",
             "type": "boolean",
             "maxLen": null,
-            "enum": null,
+            "enum": [],
             "nullable": false,
             "references": null,
             "position": 4
@@ -98,7 +101,7 @@ uRole = "dbapi_test"} in
             "name": "date",
             "type": "date",
             "maxLen": null,
-            "enum": null,
+            "enum": [],
             "nullable": false,
             "references": null,
             "position": 5
@@ -110,7 +113,7 @@ uRole = "dbapi_test"} in
             "name": "money",
             "type": "money",
             "maxLen": null,
-            "enum": null,
+            "enum": [],
             "nullable": false,
             "position": 6,
             "references": null,
@@ -136,9 +139,10 @@ uRole = "dbapi_test"} in
       }
       |]
 
-    it "includes foreign key data" $
-      request methodOptions "/has_fk"
-        [("Authorization", "Basic "<>(cs.encode $ cs uName<>":"<>cs uPass))] ""
+    it "includes foreign key data" $ do
+      pendingWith "have to resolve issue #107"
+
+      request methodOptions "/has_fk" [] ""
         `shouldRespondWith` [json|
       {
         "pkey": ["id"],
@@ -153,7 +157,7 @@ uRole = "dbapi_test"} in
             "maxLen": null,
             "nullable": false,
             "position": 1,
-            "enum": null,
+            "enum": [],
             "references": null
           }, {
             "default": null,
@@ -165,7 +169,7 @@ uRole = "dbapi_test"} in
             "maxLen": null,
             "nullable": true,
             "position": 2,
-            "enum": null,
+            "enum": [],
             "references": {"table": "auto_incrementing_pk", "column": "id"}
           }, {
             "default": null,
@@ -177,7 +181,7 @@ uRole = "dbapi_test"} in
             "maxLen": 255,
             "nullable": true,
             "position": 3,
-            "enum": null,
+            "enum": [],
             "references": {"table": "simple_pk", "column": "k"}
           }
         ]

test/Main.hs

@@ -1,17 +1,9 @@
+{-# LANGUAGE QuasiQuotes #-}
 module Main where
 
-import Database.HDBC (runRaw, disconnect)
 import Test.Hspec
+import SpecHelper
 import Spec
-import SpecHelper (openConnection, loadFixture)
 
 main :: IO ()
-main = do
-  c <-openConnection
-  runRaw c "drop schema if exists \"1\" cascade"
-  runRaw c "drop schema if exists private cascade"
-  runRaw c "drop schema if exists dbapi cascade"
-  loadFixture "roles" c
-  loadFixture "schema" c
-  disconnect c
-  hspec spec
+main = resetDb >> hspec spec

test/SpecHelper.hs

@@ -1,81 +1,80 @@
-{-# LANGUAGE OverloadedStrings #-}
+{-# LANGUAGE QuasiQuotes, OverloadedStrings #-}
+
 module SpecHelper where
 
 import Network.Wai
 import Test.Hspec
 import Test.Hspec.Wai
 
-import Database.HDBC
-import Database.HDBC.PostgreSQL
+import Hasql as H
+import Hasql.Postgres as H
 
 import Data.String.Conversions (cs)
-import Control.Exception.Base (bracket, finally, tryJust)
-import Control.Monad (when)
+import Data.Monoid
+-- import Control.Exception.Base (bracket, finally)
+import Control.Monad (void)
+import Control.Exception
 
 import Network.HTTP.Types.Header (Header, ByteRange, renderByteRange,
                                   hRange, hAuthorization)
 import Codec.Binary.Base64.String (encode)
 import Data.CaseInsensitive (CI(..))
+import Data.Maybe (fromMaybe)
 import Text.Regex.TDFA ((=~))
-import qualified Data.HashMap.Strict as Hash
 import qualified Data.ByteString.Char8 as BS
 import Network.Wai.Middleware.Cors (cors)
+import System.Process (readProcess)
 
-import Dbapi (app, corsPolicy, AppConfig(..))
-import PgQuery(addUser)
+import App (app, sqlError, isSqlError)
+import Config (AppConfig(..), corsPolicy)
+import Middleware
+-- import Auth (addUser)
 
 isLeft :: Either a b -> Bool
 isLeft (Left _ ) = True
 isLeft _ = False
 
 cfg :: AppConfig
-cfg = AppConfig "postgres://dbapi_test:@localhost:5432/dbapi_test" 9000 "test/test.crt" "test/test.key" "dbapi_anonymous"
+cfg = AppConfig "postgrest_test" 5432 "postgrest_test" "" "localhost" 3000 "postgrest_anonymous" False 10
 
-openConnection :: IO Connection
-openConnection = connectPostgreSQL' $ configDbUri cfg
+testSettings :: SessionSettings
+testSettings = fromMaybe (error "bad settings") $ H.sessionSettings 1 30
 
-withDatabaseConnection :: (Connection -> IO ()) -> IO ()
-withDatabaseConnection = bracket openConnection disconnect
+pgSettings :: Postgres
+pgSettings = H.ParamSettings "localhost" 5432 "postgrest_test" "" "postgrest_test"
 
-loadFixture :: String -> Connection -> IO ()
-loadFixture name conn = do
-  sql <- readFile $ "test/fixtures/" ++ name ++ ".sql"
-  runRaw conn sql
+withApp :: ActionWith Application -> IO ()
+withApp perform =
+  let anonRole = cs $ configAnonRole cfg
+      currRole = cs $ configDbUser cfg in
+  perform $ middle $ \req resp ->
+    H.session pgSettings testSettings $ H.sessionUnlifter >>= \unlift ->
+      liftIO $ do
+        body <- strictRequestBody req
+        resp =<< catchJust isSqlError
+          (unlift $ H.tx Nothing
+                  $ authenticated currRole anonRole (app body) req)
+          (return . sqlError)
 
-dbWithSchema :: ActionWith Connection -> IO ()
-dbWithSchema action = withDatabaseConnection $ \c -> do
-  runRaw c "begin;"
-  action c
-  rollback c
+  where middle = cors corsPolicy
 
-withUser :: BS.ByteString -> BS.ByteString -> BS.ByteString ->
-            ActionWith Connection -> ActionWith Connection
-withUser name pass role action conn = do
-  addUser name pass role conn
-  finally (action conn) $ do
-    _ <- run conn "delete from dbapi.auth where id=?" [toSql name]
-    runRaw conn "commit"
 
-withApp :: ActionWith Application -> ActionWith Connection
-withApp action conn = do
-  runRaw conn "begin;"
-  action $ cors corsPolicy $ app conn "dbapi_anonymous"
-  rollback conn
+resetDb :: IO ()
+resetDb = do
+  H.session pgSettings testSettings $
+    H.tx Nothing $ do
+      H.unit [H.q| drop schema if exists "1" cascade |]
+      H.unit [H.q| drop schema if exists private cascade |]
+      H.unit [H.q| drop schema if exists postgrest cascade |]
 
-appWithFixture :: ActionWith Application -> IO ()
-appWithFixture action = withDatabaseConnection $ \c -> do
-  result <- tryJust transactionAborted $ do
-    runRaw c "begin;"
-    action $ cors corsPolicy $ app c "dbapi_anonymous"
-    rollback c
+  loadFixture "roles"
+  loadFixture "schema"
 
-  when (isLeft result) $
-    putStrLn "note: commands ignored after aborted transaction"
 
-  where
-    transactionAborted :: SqlError -> Maybe ()
-    transactionAborted e =
-      if seState e == "25P02" then Just () else Nothing
+loadFixture :: FilePath -> IO()
+loadFixture name =
+  void $ readProcess "psql" ["-U", "postgrest_test", "-d", "postgrest_test", "-a", "-f", "test/fixtures/" ++ name ++ ".sql"] []
+
 
 rangeHdrs :: ByteRange -> [Header]
 rangeHdrs r = [rangeUnit, (hRange, renderByteRange r)]
@@ -83,22 +82,28 @@ rangeHdrs r = [rangeUnit, (hRange, renderByteRange r)]
 rangeUnit :: Header
 rangeUnit = ("Range-Unit" :: CI BS.ByteString, "items")
 
-getHeader :: CI BS.ByteString -> [Header] -> Maybe BS.ByteString
-getHeader name headers =
-  Hash.lookup name $ Hash.fromList headers
-
 matchHeader :: CI BS.ByteString -> String -> [Header] -> Bool
 matchHeader name valRegex headers =
-  maybe False (=~ valRegex) $ getHeader name headers
+  maybe False (=~ valRegex) $ lookup name headers
 
 authHeader :: String -> String -> Header
-authHeader user pass =
-  (hAuthorization, cs $ "Basic: " ++ encode (user ++ ":" ++ pass))
+authHeader u p =
+  (hAuthorization, cs $ "Basic " ++ encode (u ++ ":" ++ p))
+
+clearTable :: BS.ByteString -> IO ()
+clearTable table = H.session pgSettings testSettings $ H.tx Nothing $
+  H.unit ("delete from \"1\"."<>table, [], True)
+
+createItems :: Int -> IO ()
+createItems n = H.session pgSettings testSettings $ H.tx Nothing txn
+  where
+    txn = sequence_ $ map H.unit stmts
+    stmts = map [H.q|insert into "1".items (id) values (?)|] [1..n]
 
 -- for hspec-wai
 pending_ :: WaiSession ()
-pending_ = liftIO pending
+pending_ = liftIO Test.Hspec.pending
 
 -- for hspec-wai
 pendingWith_ :: String -> WaiSession ()
-pendingWith_ = liftIO . pendingWith
+pendingWith_ = liftIO . Test.Hspec.pendingWith

test/TestTypes.hs

@@ -1,17 +1,16 @@
-{-# LANGUAGE OverloadedStrings #-}
 module TestTypes (
-  IncPK(..),
-  fromList
+  IncPK(..)
+, CompoundPK(..)
+-- , incFromList
+-- , compoundFromList
 ) where
 
 import qualified Data.Aeson as JSON
 import Data.Aeson ((.:))
-import Data.Maybe (fromJust)
+-- import Data.Maybe (fromJust)
 import Control.Applicative ((<$>), (<*>))
 import Control.Monad (mzero)
 
-import Database.HDBC (SqlValue, fromSql)
-
 data IncPK = IncPK {
   incId :: Int
 , incNullableStr :: Maybe String
@@ -27,9 +26,28 @@ instance JSON.FromJSON IncPK where
     r .: "inserted_at"
   parseJSON _ = mzero
 
-fromList :: [(String, SqlValue)] -> IncPK
-fromList row = IncPK
-  (fromSql . fromJust $ lookup "id" row)
-  (fromSql . fromJust $ lookup "nullable_string" row)
-  (fromSql . fromJust $ lookup "non_nullable_string" row)
-  (fromSql . fromJust $ lookup "inserted_at" row)
+-- incFromList :: [(String, SqlValue)] -> IncPK
+-- incFromList row = IncPK
+--   (fromSql . fromJust $ lookup "id" row)
+--   (fromSql . fromJust $ lookup "nullable_string" row)
+--   (fromSql . fromJust $ lookup "non_nullable_string" row)
+--   (fromSql . fromJust $ lookup "inserted_at" row)
+
+data CompoundPK = CompoundPK {
+  compoundK1 :: Int
+, compoundK2 :: Int
+, compoundExtra :: Maybe Int
+}
+
+instance JSON.FromJSON CompoundPK where
+  parseJSON (JSON.Object r) = CompoundPK <$>
+    r .: "k1" <*>
+    r .: "k2" <*>
+    r .: "extra"
+  parseJSON _ = mzero
+
+-- compoundFromList :: [(String, SqlValue)] -> CompoundPK
+-- compoundFromList row = CompoundPK
+--   (fromSql . fromJust $ lookup "k1" row)
+--   (fromSql . fromJust $ lookup "k2" row)
+--   (fromSql . fromJust $ lookup "extra" row)

test/Unit/ErrorsSpec.hs

@@ -0,0 +1,41 @@
+module Unit.ErrorsSpec where
+
+import Test.Hspec
+
+import Text.Parsec
+import PgError
+import Data.Either (rights)
+
+spec :: Spec
+spec =
+  describe "Parsing Hasql errors" $ do
+    it "can handle status and code" $
+      let p = parse message "" "Status: \"foo\"; Code: \"abc\"." in
+      rights [p] `shouldBe` [
+          Message (Just "foo") "abc" Nothing Nothing
+        ]
+    it "can handle weird redundant quotes in status" $
+      let p = parse message "" "Status: \"\"foo\"\"; Code: \"abc\"." in
+      rights [p] `shouldBe` [
+          Message (Just "foo") "abc" Nothing Nothing
+        ]
+    it "can handle text and code" $
+      let p = parse message "" "Message: \"foo\"; Code: \"abc\"." in
+      rights [p] `shouldBe` [
+          Message Nothing "abc" (Just "foo") Nothing
+        ]
+    it "can handle status, text and code" $
+      let p = parse message "" "Status: \"hi\"; Message: \"foo\"; Code: \"abc\"." in
+      rights [p] `shouldBe` [
+          Message (Just "hi") "abc" (Just "foo") Nothing
+        ]
+    it "can handle unescaped quotes in message" $
+      let p = parse message "" "Status: \"hi\"; Message: \"unknown \"foo\"!\"; Code: \"abc\"." in
+      rights [p] `shouldBe` [
+          Message (Just "hi") "abc" (Just "unknown \"foo\"!") Nothing
+        ]
+    it "can handle periods in message" $
+      let p = parse message "" "Message: \"unknown \"foo\".bar\"; Code: \"42P01\"." in
+      rights [p] `shouldBe` [
+          Message Nothing "42P01" (Just "unknown \"foo\".bar") Nothing
+        ]

test/Unit/PgQuerySpec.hx

@@ -1,15 +1,16 @@
-{-# LANGUAGE OverloadedStrings #-}
-
 module Unit.PgQuerySpec where
 
 import Test.Hspec
+import Test.QuickCheck
+import Test.QuickCheck.Monadic
 
 import Database.HDBC (IConnection, SqlValue, toSql, prepare,
   quickQuery, fromSql, execute, seState, fetchAllRowsAL)
 
-import PgQuery (LoginAttempt(..), insert, addUser, signInRole, checkPass)
+import PgQuery (LoginAttempt(..), insert, addUser, signInRole, checkPass
+              , pgFmtIdent, pgFmtLit)
 import Types (SqlRow(SqlRow))
-import TestTypes (fromList, incStr, incNullableStr, incInsert, incId)
+import TestTypes (incFromList, incStr, incNullableStr, incInsert, incId)
 import Data.Map (toList)
 import Data.String.Conversions (cs)
 import Data.Monoid ((<>))
@@ -33,13 +34,13 @@ spec = around dbWithSchema $ do
       it "inserts and responds with a full object description" $ \conn -> do
         r <- insert "1" "auto_incrementing_pk" (SqlRow [
             ("non_nullable_string", toSql ("a string"::String))]) conn
-        let returnRow = fromList . toList $ r
+        let returnRow = incFromList . toList $ r
         incStr returnRow `shouldBe` "a string"
         incNullableStr returnRow `shouldBe` Nothing
         incInsert returnRow `shouldSatisfy` not . null
         incId returnRow `shouldSatisfy` (>= 0)
         tRows <- quickALQuery conn "select * from \"1\".auto_incrementing_pk" []
-        [returnRow] `shouldBe` map fromList tRows
+        [returnRow] `shouldBe` map incFromList tRows
 
       it "throws an exception if the PK is not unique" $ \conn -> do
         r <- insert "1" "auto_incrementing_pk" (SqlRow [
@@ -53,11 +54,17 @@ spec = around dbWithSchema $ do
           ("nullable_string", toSql ("a string"::String))]) conn
           `shouldThrow` \e -> seState e == "23502"
 
+      it "generates a default values query if no data is provided" $ \c -> do
+        r <- insert "1" "items" (SqlRow []) c
+        let [row] = toList r
+        quickALQuery c "select * from \"1\".items where id = ?" [snd row]
+          `shouldReturn` [[row]]
+
   let {user = "jdoe"; pass = "secret"; role = "test_default_role"}
   describe "addUser" $ do
     it "adds a correct user to the right table" $ \conn -> do
       addUser user pass role conn
-      [r] <- quickQuery conn "select * from dbapi.auth" []
+      [r] <- quickQuery conn "select * from postgrest.auth" []
       let [newUser, newRole, encryptedPass] = map fromSql r :: [String]
       cs newUser `shouldBe` user
       cs newRole `shouldBe` role
@@ -77,3 +84,16 @@ spec = around dbWithSchema $ do
     it "returns nothing with bad creds" $ \conn -> do
       signInRole "not-a-user" pass conn `shouldReturn` LoginFailed
       signInRole user (pass <> "crap") conn `shouldReturn` LoginFailed
+
+  describe "pgFmtIdent" $
+    it "Does what format %I would do" $ \conn -> property $ \fuzz ->
+      monadicIO $ do
+        [[row]] <- run $ quickALQuery conn "select format('%I', ? :: varchar)" [toSql (fuzz :: String)]
+        assert $ fromSql (snd row) == pgFmtIdent (cs fuzz)
+
+  describe "pgFmtLit" $
+    it "Does what format %L would do" $ \conn ->
+      property $ monadicIO $ do
+        fuzz <- pick arbitrary
+        [[row]] <- run $ quickALQuery conn "select format('%L', ? :: varchar)" [toSql (fuzz :: String)]
+        assert $ fromSql (snd row) == pgFmtLit (cs fuzz)

test/Unit/PgStructureSpec.hx

@@ -34,4 +34,4 @@ spec = around dbWithSchema $ beforeWith setRole $ do
         ("auto_inc_fk", ForeignKey {fkTable="auto_incrementing_pk", fkCol="id"}),
         ("simple_fk", ForeignKey { fkTable="simple_pk", fkCol="k"})]
 
-  where setRole conn = quickQuery conn "set role dbapi_test" [] >> return conn
+  where setRole conn = quickQuery conn "set role postgrest_test" [] >> return conn

test/fixtures/dumpfixture.sh

@@ -1 +1 @@
-pg_dump --host localhost --port 5432 --username "postgres" --no-password  --format plain --column-inserts --verbose --file "./test/fixtures/schema.sql" "dbapi_test"
+pg_dump --host localhost --port 5432 --username "postgres" --no-password  --format plain --column-inserts --verbose --file "./test/fixtures/schema.sql" "postgrest_test"

test/fixtures/roles.sql

@@ -11,9 +11,6 @@ BEGIN
 END;
 $$;
 
-select pg_temp.create_role_if_not_exists('dbapi_anonymous', 'with nologin');
-select pg_temp.create_role_if_not_exists('test_default_role', 'with nologin');
-
-select pg_temp.create_role_if_not_exists('dbapi_test_author', 'with nologin');
-select pg_temp.create_role_if_not_exists('dbapi_test_author_a', 'with nologin in role dbapi_test_author');
-select pg_temp.create_role_if_not_exists('dbapi_test_author_b', 'with nologin in role dbapi_test_author');
+select pg_temp.create_role_if_not_exists('postgrest_anonymous', 'with nologin') as a
+     , pg_temp.create_role_if_not_exists('test_default_role', 'with nologin') as b
+     , pg_temp.create_role_if_not_exists('postgrest_test_author', 'with nologin') into temp shh;

test/fixtures/schema.sql

@@ -2,50 +2,48 @@
 -- PostgreSQL database dump
 --
 
--- Dumped from database version 9.3.4
--- Dumped by pg_dump version 9.3.4
--- Started on 2014-10-01 13:41:39 PDT
--- Started on 2014-10-07 16:46:34 PDT
+-- Dumped from database version 9.3.5
+-- Dumped by pg_dump version 9.3.5
+-- Started on 2014-10-21 15:12:44 PDT
 
 SET statement_timeout = 0;
-SET lock_timeout = 0;
 SET client_encoding = 'UTF8';
 SET standard_conforming_strings = on;
 SET check_function_bodies = false;
 SET client_min_messages = warning;
 
 --
--- TOC entry 11 (class 2615 OID 280932)
--- Name: 1; Type: SCHEMA; Schema: -; Owner: dbapi_test
+-- TOC entry 20 (class 2615 OID 337536)
+-- Name: 1; Type: SCHEMA; Schema: -; Owner: postgrest_test
 --
 
 CREATE SCHEMA "1";
 
 
-ALTER SCHEMA "1" OWNER TO dbapi_test;
+ALTER SCHEMA "1" OWNER TO postgrest_test;
 
 --
--- TOC entry 8 (class 2615 OID 50928)
--- Name: dbapi; Type: SCHEMA; Schema: -; Owner: dbapi_test
+-- TOC entry 19 (class 2615 OID 337537)
+-- Name: postgrest; Type: SCHEMA; Schema: -; Owner: postgrest_test
 --
 
-CREATE SCHEMA dbapi;
+CREATE SCHEMA postgrest;
 
 
-ALTER SCHEMA dbapi OWNER TO dbapi_test;
+ALTER SCHEMA postgrest OWNER TO postgrest_test;
 
 --
--- TOC entry 9 (class 2615 OID 50929)
--- Name: private; Type: SCHEMA; Schema: -; Owner: dbapi_test
+-- TOC entry 16 (class 2615 OID 337538)
+-- Name: private; Type: SCHEMA; Schema: -; Owner: postgrest_test
 --
 
 CREATE SCHEMA private;
 
 
-ALTER SCHEMA private OWNER TO dbapi_test;
+ALTER SCHEMA private OWNER TO postgrest_test;
 
 --
--- TOC entry 187 (class 3079 OID 12018)
+-- TOC entry 205 (class 3079 OID 12018)
 -- Name: plpgsql; Type: EXTENSION; Schema: -; Owner: 
 --
 
@@ -53,29 +51,37 @@ CREATE EXTENSION IF NOT EXISTS plpgsql WITH SCHEMA pg_catalog;
 
 
 --
+-- TOC entry 2321 (class 0 OID 0)
+-- Dependencies: 205
 -- Name: EXTENSION plpgsql; Type: COMMENT; Schema: -; Owner: 
 --
 
 COMMENT ON EXTENSION plpgsql IS 'PL/pgSQL procedural language';
 
+
 SET search_path = "1", pg_catalog;
 
 --
--- TOC entry 571 (class 1247 OID 309112)
--- Name: enum_menagerie_type; Type: TYPE; Schema: 1; Owner: postgres
+-- TOC entry 553 (class 1247 OID 337540)
+-- Name: enum_menagerie_type; Type: TYPE; Schema: 1; Owner: postgrest_test
 --
 
-CREATE TYPE "1".enum_menagerie_type AS ENUM (
-  'foo',
-  'bar'
+CREATE TYPE enum_menagerie_type AS ENUM (
+    'foo',
+    'bar'
 );
 
 
+ALTER TYPE "1".enum_menagerie_type OWNER TO postgrest_test;
+
+SET search_path = postgrest, pg_catalog;
+
 --
--- Name: check_role_exists(); Type: FUNCTION; Schema: dbapi; Owner: dbapi_test
+-- TOC entry 218 (class 1255 OID 337545)
+-- Name: check_role_exists(); Type: FUNCTION; Schema: postgrest; Owner: postgrest_test
 --
 
-CREATE FUNCTION dbapi.check_role_exists() RETURNS trigger
+CREATE FUNCTION check_role_exists() RETURNS trigger
     LANGUAGE plpgsql
     AS $$
 begin 
@@ -88,10 +94,14 @@ end
 $$;
 
 
-ALTER FUNCTION dbapi.check_role_exists() OWNER TO dbapi_test;
+ALTER FUNCTION postgrest.check_role_exists() OWNER TO postgrest_test;
 
+--
+-- TOC entry 219 (class 1255 OID 337546)
+-- Name: update_owner(); Type: FUNCTION; Schema: postgrest; Owner: postgrest_test
+--
 
-CREATE FUNCTION dbapi.update_owner() RETURNS trigger
+CREATE FUNCTION update_owner() RETURNS trigger
     LANGUAGE plpgsql
     AS $$
 BEGIN
@@ -101,7 +111,7 @@ END;
 $$;
 
 
-ALTER FUNCTION dbapi.update_owner() OWNER TO dbapi_test;
+ALTER FUNCTION postgrest.update_owner() OWNER TO postgrest_test;
 
 SET search_path = "1", pg_catalog;
 
@@ -109,6 +119,9 @@ SET default_tablespace = '';
 
 SET default_with_oids = false;
 
+--
+-- TOC entry 191 (class 1259 OID 337547)
+-- Name: authors_only; Type: TABLE; Schema: 1; Owner: postgrest_test_author; Tablespace: 
 --
 
 CREATE TABLE authors_only (
@@ -116,10 +129,11 @@ CREATE TABLE authors_only (
 );
 
 
-ALTER TABLE "1".authors_only OWNER TO dbapi_test_author;
+ALTER TABLE "1".authors_only OWNER TO postgrest_test_author;
 
 --
--- TOC entry 175 (class 1259 OID 280937)
+-- TOC entry 192 (class 1259 OID 337553)
+-- Name: auto_incrementing_pk; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE auto_incrementing_pk (
@@ -130,10 +144,11 @@ CREATE TABLE auto_incrementing_pk (
 );
 
 
-ALTER TABLE "1".auto_incrementing_pk OWNER TO dbapi_test;
+ALTER TABLE "1".auto_incrementing_pk OWNER TO postgrest_test;
 
 --
--- Name: auto_incrementing_pk_id_seq; Type: SEQUENCE; Schema: 1; Owner: dbapi_test
+-- TOC entry 193 (class 1259 OID 337560)
+-- Name: auto_incrementing_pk_id_seq; Type: SEQUENCE; Schema: 1; Owner: postgrest_test
 --
 
 CREATE SEQUENCE auto_incrementing_pk_id_seq
@@ -144,16 +159,20 @@ CREATE SEQUENCE auto_incrementing_pk_id_seq
     CACHE 1;
 
 
-ALTER TABLE "1".auto_incrementing_pk_id_seq OWNER TO dbapi_test;
+ALTER TABLE "1".auto_incrementing_pk_id_seq OWNER TO postgrest_test;
 
 --
+-- TOC entry 2324 (class 0 OID 0)
+-- Dependencies: 193
+-- Name: auto_incrementing_pk_id_seq; Type: SEQUENCE OWNED BY; Schema: 1; Owner: postgrest_test
 --
 
 ALTER SEQUENCE auto_incrementing_pk_id_seq OWNED BY auto_incrementing_pk.id;
 
 
 --
--- Name: compound_pk; Type: TABLE; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 194 (class 1259 OID 337562)
+-- Name: compound_pk; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE compound_pk (
@@ -163,10 +182,11 @@ CREATE TABLE compound_pk (
 );
 
 
-ALTER TABLE "1".compound_pk OWNER TO dbapi_test;
+ALTER TABLE "1".compound_pk OWNER TO postgrest_test;
 
 --
--- Name: has_fk; Type: TABLE; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 195 (class 1259 OID 337565)
+-- Name: has_fk; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE has_fk (
@@ -176,11 +196,11 @@ CREATE TABLE has_fk (
 );
 
 
-ALTER TABLE "1".has_fk OWNER TO dbapi_test;
+ALTER TABLE "1".has_fk OWNER TO postgrest_test;
 
 --
--- TOC entry 185 (class 1259 OID 50947)
--- Name: has_fk_id_seq; Type: SEQUENCE; Schema: 1; Owner: dbapi_test
+-- TOC entry 196 (class 1259 OID 337568)
+-- Name: has_fk_id_seq; Type: SEQUENCE; Schema: 1; Owner: postgrest_test
 --
 
 CREATE SEQUENCE has_fk_id_seq
@@ -191,20 +211,20 @@ CREATE SEQUENCE has_fk_id_seq
     CACHE 1;
 
 
-ALTER TABLE "1".has_fk_id_seq OWNER TO dbapi_test;
+ALTER TABLE "1".has_fk_id_seq OWNER TO postgrest_test;
 
 --
--- TOC entry 2042 (class 0 OID 0)
--- Dependencies: 185
--- Name: has_fk_id_seq; Type: SEQUENCE OWNED BY; Schema: 1; Owner: dbapi_test
+-- TOC entry 2327 (class 0 OID 0)
+-- Dependencies: 196
+-- Name: has_fk_id_seq; Type: SEQUENCE OWNED BY; Schema: 1; Owner: postgrest_test
 --
 
 ALTER SEQUENCE has_fk_id_seq OWNED BY has_fk.id;
 
 
 --
--- TOC entry 186 (class 1259 OID 50949)
--- Name: items; Type: TABLE; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 197 (class 1259 OID 337570)
+-- Name: items; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE items (
@@ -212,10 +232,11 @@ CREATE TABLE items (
 );
 
 
-ALTER TABLE "1".items OWNER TO dbapi_test;
+ALTER TABLE "1".items OWNER TO postgrest_test;
 
 --
--- Name: items_id_seq; Type: SEQUENCE; Schema: 1; Owner: dbapi_test
+-- TOC entry 198 (class 1259 OID 337573)
+-- Name: items_id_seq; Type: SEQUENCE; Schema: 1; Owner: postgrest_test
 --
 
 CREATE SEQUENCE items_id_seq
@@ -226,17 +247,20 @@ CREATE SEQUENCE items_id_seq
     CACHE 1;
 
 
-ALTER TABLE "1".items_id_seq OWNER TO dbapi_test;
+ALTER TABLE "1".items_id_seq OWNER TO postgrest_test;
 
 --
--- Name: items_id_seq; Type: SEQUENCE OWNED BY; Schema: 1; Owner: dbapi_test
+-- TOC entry 2329 (class 0 OID 0)
+-- Dependencies: 198
+-- Name: items_id_seq; Type: SEQUENCE OWNED BY; Schema: 1; Owner: postgrest_test
 --
 
 ALTER SEQUENCE items_id_seq OWNED BY items.id;
 
 
 --
--- Name: menagerie; Type: TABLE; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 199 (class 1259 OID 337575)
+-- Name: menagerie; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE menagerie (
@@ -246,14 +270,15 @@ CREATE TABLE menagerie (
     "boolean" boolean NOT NULL,
     date date NOT NULL,
     money money NOT NULL,
-    enum "1".enum_menagerie_type not null
+    enum enum_menagerie_type NOT NULL
 );
 
 
-ALTER TABLE "1".menagerie OWNER TO dbapi_test;
+ALTER TABLE "1".menagerie OWNER TO postgrest_test;
 
 --
--- Name: no_pk; Type: TABLE; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 200 (class 1259 OID 337581)
+-- Name: no_pk; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE no_pk (
@@ -262,10 +287,11 @@ CREATE TABLE no_pk (
 );
 
 
-ALTER TABLE "1".no_pk OWNER TO dbapi_test;
+ALTER TABLE "1".no_pk OWNER TO postgrest_test;
 
 --
--- Name: simple_pk; Type: TABLE; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 201 (class 1259 OID 337587)
+-- Name: simple_pk; Type: TABLE; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE simple_pk (
@@ -274,12 +300,13 @@ CREATE TABLE simple_pk (
 );
 
 
-ALTER TABLE "1".simple_pk OWNER TO dbapi_test;
+ALTER TABLE "1".simple_pk OWNER TO postgrest_test;
 
-SET search_path = dbapi, pg_catalog;
+SET search_path = postgrest, pg_catalog;
 
 --
--- Name: auth; Type: TABLE; Schema: dbapi; Owner: dbapi_test; Tablespace: 
+-- TOC entry 202 (class 1259 OID 337593)
+-- Name: auth; Type: TABLE; Schema: postgrest; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE auth (
@@ -289,12 +316,13 @@ CREATE TABLE auth (
 );
 
 
-ALTER TABLE dbapi.auth OWNER TO dbapi_test;
+ALTER TABLE postgrest.auth OWNER TO postgrest_test;
 
 SET search_path = private, pg_catalog;
 
 --
--- Name: articles; Type: TABLE; Schema: private; Owner: dbapi_test; Tablespace: 
+-- TOC entry 203 (class 1259 OID 337599)
+-- Name: articles; Type: TABLE; Schema: private; Owner: postgrest_test; Tablespace: 
 --
 
 CREATE TABLE articles (
@@ -304,10 +332,11 @@ CREATE TABLE articles (
 );
 
 
-ALTER TABLE private.articles OWNER TO dbapi_test;
+ALTER TABLE private.articles OWNER TO postgrest_test;
 
 --
--- Name: articles_id_seq; Type: SEQUENCE; Schema: private; Owner: dbapi_test
+-- TOC entry 204 (class 1259 OID 337605)
+-- Name: articles_id_seq; Type: SEQUENCE; Schema: private; Owner: postgrest_test
 --
 
 CREATE SEQUENCE articles_id_seq
@@ -318,10 +347,12 @@ CREATE SEQUENCE articles_id_seq
     CACHE 1;
 
 
-ALTER TABLE private.articles_id_seq OWNER TO dbapi_test;
+ALTER TABLE private.articles_id_seq OWNER TO postgrest_test;
 
 --
--- Name: articles_id_seq; Type: SEQUENCE OWNED BY; Schema: private; Owner: dbapi_test
+-- TOC entry 2336 (class 0 OID 0)
+-- Dependencies: 204
+-- Name: articles_id_seq; Type: SEQUENCE OWNED BY; Schema: private; Owner: postgrest_test
 --
 
 ALTER SEQUENCE articles_id_seq OWNED BY articles.id;
@@ -330,23 +361,24 @@ ALTER SEQUENCE articles_id_seq OWNED BY articles.id;
 SET search_path = "1", pg_catalog;
 
 --
--- Name: id; Type: DEFAULT; Schema: 1; Owner: dbapi_test
+-- TOC entry 2165 (class 2604 OID 337607)
+-- Name: id; Type: DEFAULT; Schema: 1; Owner: postgrest_test
 --
 
 ALTER TABLE ONLY auto_incrementing_pk ALTER COLUMN id SET DEFAULT nextval('auto_incrementing_pk_id_seq'::regclass);
 
 
 --
--- TOC entry 1886 (class 2604 OID 50987)
--- Name: id; Type: DEFAULT; Schema: 1; Owner: dbapi_test
+-- TOC entry 2166 (class 2604 OID 337608)
+-- Name: id; Type: DEFAULT; Schema: 1; Owner: postgrest_test
 --
 
 ALTER TABLE ONLY has_fk ALTER COLUMN id SET DEFAULT nextval('has_fk_id_seq'::regclass);
 
 
 --
--- TOC entry 1887 (class 2604 OID 50988)
--- Name: id; Type: DEFAULT; Schema: 1; Owner: dbapi_test
+-- TOC entry 2167 (class 2604 OID 337609)
+-- Name: id; Type: DEFAULT; Schema: 1; Owner: postgrest_test
 --
 
 ALTER TABLE ONLY items ALTER COLUMN id SET DEFAULT nextval('items_id_seq'::regclass);
@@ -355,7 +387,8 @@ ALTER TABLE ONLY items ALTER COLUMN id SET DEFAULT nextval('items_id_seq'::regcl
 SET search_path = private, pg_catalog;
 
 --
--- Name: id; Type: DEFAULT; Schema: private; Owner: dbapi_test
+-- TOC entry 2168 (class 2604 OID 337610)
+-- Name: id; Type: DEFAULT; Schema: private; Owner: postgrest_test
 --
 
 ALTER TABLE ONLY articles ALTER COLUMN id SET DEFAULT nextval('articles_id_seq'::regclass);
@@ -364,105 +397,101 @@ ALTER TABLE ONLY articles ALTER COLUMN id SET DEFAULT nextval('articles_id_seq':
 SET search_path = "1", pg_catalog;
 
 --
--- TOC entry 2279 (class 0 OID 281006)
--- Dependencies: 186
--- Data for Name: authors_only; Type: TABLE DATA; Schema: 1; Owner: dbapi_test_author
+-- TOC entry 2298 (class 0 OID 337547)
+-- Dependencies: 191
+-- Data for Name: authors_only; Type: TABLE DATA; Schema: 1; Owner: postgrest_test_author
 --
 
 
 
 --
--- TOC entry 2268 (class 0 OID 280937)
--- Dependencies: 175
--- TOC entry 2016 (class 0 OID 50932)
--- Dependencies: 181
--- Data for Name: auto_incrementing_pk; Type: TABLE DATA; Schema: 1; Owner: dbapi_test
---
-
-
--- TOC entry 2270 (class 0 OID 280946)
--- Dependencies: 177
--- TOC entry 2051 (class 0 OID 0)
--- Dependencies: 182
--- Name: auto_incrementing_pk_id_seq; Type: SEQUENCE SET; Schema: 1; Owner: dbapi_test
---
-
-SELECT pg_catalog.setval('auto_incrementing_pk_id_seq', 50, true);
-
-
---
--- TOC entry 2018 (class 0 OID 50941)
--- Dependencies: 183
--- Data for Name: compound_pk; Type: TABLE DATA; Schema: 1; Owner: dbapi_test
+-- TOC entry 2299 (class 0 OID 337553)
+-- Dependencies: 192
+-- Data for Name: auto_incrementing_pk; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
 --
 
 
 
 --
--- TOC entry 2271 (class 0 OID 280949)
--- Dependencies: 178
+-- TOC entry 2337 (class 0 OID 0)
+-- Dependencies: 193
+-- Name: auto_incrementing_pk_id_seq; Type: SEQUENCE SET; Schema: 1; Owner: postgrest_test
+--
+
+SELECT pg_catalog.setval('auto_incrementing_pk_id_seq', 54, true);
+
+
+--
+-- TOC entry 2301 (class 0 OID 337562)
+-- Dependencies: 194
+-- Data for Name: compound_pk; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
+--
 
 
 
 --
--- TOC entry 2052 (class 0 OID 0)
--- Dependencies: 185
--- Name: has_fk_id_seq; Type: SEQUENCE SET; Schema: 1; Owner: dbapi_test
+-- TOC entry 2302 (class 0 OID 337565)
+-- Dependencies: 195
+-- Data for Name: has_fk; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
+--
+
+
+
+--
+-- TOC entry 2338 (class 0 OID 0)
+-- Dependencies: 196
+-- Name: has_fk_id_seq; Type: SEQUENCE SET; Schema: 1; Owner: postgrest_test
 --
 
 SELECT pg_catalog.setval('has_fk_id_seq', 1, false);
 
 
 --
--- TOC entry 2021 (class 0 OID 50949)
--- Dependencies: 186
--- Data for Name: items; Type: TABLE DATA; Schema: 1; Owner: dbapi_test
+-- TOC entry 2304 (class 0 OID 337570)
+-- Dependencies: 197
+-- Data for Name: items; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
 --
 
 INSERT INTO items (id) VALUES (1);
-INSERT INTO items (id) VALUES (2);
-INSERT INTO items (id) VALUES (3);
-INSERT INTO items (id) VALUES (4);
-INSERT INTO items (id) VALUES (5);
-INSERT INTO items (id) VALUES (6);
-INSERT INTO items (id) VALUES (7);
-INSERT INTO items (id) VALUES (8);
-INSERT INTO items (id) VALUES (9);
-INSERT INTO items (id) VALUES (10);
-INSERT INTO items (id) VALUES (11);
-INSERT INTO items (id) VALUES (12);
-INSERT INTO items (id) VALUES (13);
-INSERT INTO items (id) VALUES (14);
-INSERT INTO items (id) VALUES (15);
+--
+-- TOC entry 2339 (class 0 OID 0)
+-- Dependencies: 198
+-- Name: items_id_seq; Type: SEQUENCE SET; Schema: 1; Owner: postgrest_test
+--
+
+SELECT pg_catalog.setval('items_id_seq', 19, true);
 
 
 --
--- Name: items_id_seq; Type: SEQUENCE SET; Schema: 1; Owner: dbapi_test
---
-
-SELECT pg_catalog.setval('items_id_seq', 15, true);
-
-
---
--- Data for Name: menagerie; Type: TABLE DATA; Schema: 1; Owner: dbapi_test
+-- TOC entry 2306 (class 0 OID 337575)
+-- Dependencies: 199
+-- Data for Name: menagerie; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
 --
 
 
 
 --
--- Data for Name: no_pk; Type: TABLE DATA; Schema: 1; Owner: dbapi_test
+-- TOC entry 2307 (class 0 OID 337581)
+-- Dependencies: 200
+-- Data for Name: no_pk; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
 --
 
 
 
+--
+-- TOC entry 2308 (class 0 OID 337587)
+-- Dependencies: 201
+-- Data for Name: simple_pk; Type: TABLE DATA; Schema: 1; Owner: postgrest_test
 --
 
 
 
-SET search_path = dbapi, pg_catalog;
+SET search_path = postgrest, pg_catalog;
 
 --
--- Data for Name: auth; Type: TABLE DATA; Schema: dbapi; Owner: dbapi_test
+-- TOC entry 2309 (class 0 OID 337593)
+-- Dependencies: 202
+-- Data for Name: auth; Type: TABLE DATA; Schema: postgrest; Owner: postgrest_test
 --
 
 
@@ -470,12 +499,17 @@ SET search_path = dbapi, pg_catalog;
 SET search_path = private, pg_catalog;
 
 --
+-- TOC entry 2310 (class 0 OID 337599)
+-- Dependencies: 203
+-- Data for Name: articles; Type: TABLE DATA; Schema: private; Owner: postgrest_test
 --
 
 
 
 --
--- Name: articles_id_seq; Type: SEQUENCE SET; Schema: private; Owner: dbapi_test
+-- TOC entry 2340 (class 0 OID 0)
+-- Dependencies: 204
+-- Name: articles_id_seq; Type: SEQUENCE SET; Schema: private; Owner: postgrest_test
 --
 
 SELECT pg_catalog.setval('articles_id_seq', 1, false);
@@ -484,7 +518,8 @@ SELECT pg_catalog.setval('articles_id_seq', 1, false);
 SET search_path = "1", pg_catalog;
 
 --
--- Name: authors_only_pkey; Type: CONSTRAINT; Schema: 1; Owner: dbapi_test_author; Tablespace: 
+-- TOC entry 2170 (class 2606 OID 337612)
+-- Name: authors_only_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test_author; Tablespace: 
 --
 
 ALTER TABLE ONLY authors_only
@@ -492,7 +527,8 @@ ALTER TABLE ONLY authors_only
 
 
 --
--- TOC entry 2144 (class 2606 OID 280990)
+-- TOC entry 2172 (class 2606 OID 337614)
+-- Name: auto_incrementing_pk_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY auto_incrementing_pk
@@ -500,7 +536,8 @@ ALTER TABLE ONLY auto_incrementing_pk
 
 
 --
--- Name: compound_pk_pkey; Type: CONSTRAINT; Schema: 1; Owner: dbapi_test; Tablespace: 
+-- TOC entry 2174 (class 2606 OID 337616)
+-- Name: compound_pk_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY compound_pk
@@ -508,7 +545,8 @@ ALTER TABLE ONLY compound_pk
 
 
 --
--- TOC entry 1900 (class 2606 OID 50995)
+-- TOC entry 2182 (class 2606 OID 337618)
+-- Name: contacts_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY simple_pk
@@ -516,6 +554,8 @@ ALTER TABLE ONLY simple_pk
 
 
 --
+-- TOC entry 2176 (class 2606 OID 337620)
+-- Name: has_fk_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY has_fk
@@ -523,7 +563,8 @@ ALTER TABLE ONLY has_fk
 
 
 --
--- TOC entry 1896 (class 2606 OID 50999)
+-- TOC entry 2178 (class 2606 OID 337622)
+-- Name: items_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY items
@@ -531,16 +572,19 @@ ALTER TABLE ONLY items
 
 
 --
+-- TOC entry 2180 (class 2606 OID 337624)
+-- Name: menagerie_pkey; Type: CONSTRAINT; Schema: 1; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY menagerie
     ADD CONSTRAINT menagerie_pkey PRIMARY KEY ("integer");
 
 
-SET search_path = dbapi, pg_catalog;
+SET search_path = postgrest, pg_catalog;
 
 --
--- Name: auth_pkey; Type: CONSTRAINT; Schema: dbapi; Owner: dbapi_test; Tablespace: 
+-- TOC entry 2184 (class 2606 OID 337626)
+-- Name: auth_pkey; Type: CONSTRAINT; Schema: postgrest; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY auth
@@ -550,16 +594,19 @@ ALTER TABLE ONLY auth
 SET search_path = private, pg_catalog;
 
 --
+-- TOC entry 2186 (class 2606 OID 337628)
+-- Name: articles_pkey; Type: CONSTRAINT; Schema: private; Owner: postgrest_test; Tablespace: 
 --
 
 ALTER TABLE ONLY articles
     ADD CONSTRAINT articles_pkey PRIMARY KEY (id);
 
 
-SET search_path = dbapi, pg_catalog;
+SET search_path = postgrest, pg_catalog;
 
 --
--- Name: ensure_auth_role_exists; Type: TRIGGER; Schema: dbapi; Owner: dbapi_test
+-- TOC entry 2189 (class 2620 OID 337630)
+-- Name: ensure_auth_role_exists; Type: TRIGGER; Schema: postgrest; Owner: postgrest_test
 --
 
 CREATE CONSTRAINT TRIGGER ensure_auth_role_exists AFTER INSERT OR UPDATE ON auth NOT DEFERRABLE INITIALLY IMMEDIATE FOR EACH ROW EXECUTE PROCEDURE check_role_exists();
@@ -568,17 +615,18 @@ CREATE CONSTRAINT TRIGGER ensure_auth_role_exists AFTER INSERT OR UPDATE ON auth
 SET search_path = private, pg_catalog;
 
 --
--- Name: articles_owner_track; Type: TRIGGER; Schema: private; Owner: dbapi_test
+-- TOC entry 2190 (class 2620 OID 337631)
+-- Name: articles_owner_track; Type: TRIGGER; Schema: private; Owner: postgrest_test
 --
 
-CREATE TRIGGER articles_owner_track BEFORE INSERT OR UPDATE ON articles FOR EACH ROW EXECUTE PROCEDURE dbapi.update_owner();
+CREATE TRIGGER articles_owner_track BEFORE INSERT OR UPDATE ON articles FOR EACH ROW EXECUTE PROCEDURE postgrest.update_owner();
 
 
 SET search_path = "1", pg_catalog;
 
 --
--- TOC entry 1905 (class 2606 OID 51009)
--- Name: has_fk_fk_fkey; Type: FK CONSTRAINT; Schema: 1; Owner: dbapi_test
+-- TOC entry 2187 (class 2606 OID 337632)
+-- Name: has_fk_fk_fkey; Type: FK CONSTRAINT; Schema: 1; Owner: postgrest_test
 --
 
 ALTER TABLE ONLY has_fk
@@ -586,95 +634,192 @@ ALTER TABLE ONLY has_fk
 
 
 --
--- TOC entry 1906 (class 2606 OID 51015)
--- Name: has_fk_simple_fk_fkey; Type: FK CONSTRAINT; Schema: 1; Owner: dbapi_test
+-- TOC entry 2188 (class 2606 OID 337637)
+-- Name: has_fk_simple_fk_fkey; Type: FK CONSTRAINT; Schema: 1; Owner: postgrest_test
 --
 
 ALTER TABLE ONLY has_fk
     ADD CONSTRAINT has_fk_simple_fk_fkey FOREIGN KEY (simple_fk) REFERENCES simple_pk(k);
 
 
+--
+-- TOC entry 2317 (class 0 OID 0)
+-- Dependencies: 20
+-- Name: 1; Type: ACL; Schema: -; Owner: postgrest_test
+--
 
-REVOKE ALL ON SCHEMA "1" FROM dbapi_test;
-GRANT ALL ON SCHEMA "1" TO dbapi_test;
-GRANT USAGE ON SCHEMA "1" TO dbapi_anonymous;
+REVOKE ALL ON SCHEMA "1" FROM PUBLIC;
+REVOKE ALL ON SCHEMA "1" FROM postgrest_test;
+GRANT ALL ON SCHEMA "1" TO postgrest_test;
+GRANT USAGE ON SCHEMA "1" TO postgrest_anonymous;
+GRANT USAGE ON SCHEMA "1" TO postgrest_test_author;
 
 
 --
--- TOC entry 2036 (class 0 OID 0)
--- Dependencies: 15
+-- TOC entry 2318 (class 0 OID 0)
+-- Dependencies: 19
+-- Name: postgrest; Type: ACL; Schema: -; Owner: postgrest_test
+--
+
+REVOKE ALL ON SCHEMA postgrest FROM PUBLIC;
+REVOKE ALL ON SCHEMA postgrest FROM postgrest_test;
+GRANT ALL ON SCHEMA postgrest TO postgrest_test;
+GRANT USAGE ON SCHEMA postgrest TO postgrest_anonymous;
+
+
+--
+-- TOC entry 2320 (class 0 OID 0)
+-- Dependencies: 21
 -- Name: public; Type: ACL; Schema: -; Owner: postgres
 --
 
 REVOKE ALL ON SCHEMA public FROM PUBLIC;
-REVOKE ALL ON SCHEMA public FROM postgres;
 GRANT ALL ON SCHEMA public TO postgres;
 GRANT ALL ON SCHEMA public TO PUBLIC;
 
 
+--
+-- TOC entry 2322 (class 0 OID 0)
+-- Dependencies: 191
+-- Name: authors_only; Type: ACL; Schema: 1; Owner: postgrest_test_author
 --
 
-REVOKE ALL ON TABLE authors_only FROM dbapi_test_author;
-GRANT ALL ON TABLE authors_only TO dbapi_test_author;
+REVOKE ALL ON TABLE authors_only FROM PUBLIC;
+REVOKE ALL ON TABLE authors_only FROM postgrest_test_author;
+GRANT ALL ON TABLE authors_only TO postgrest_test_author;
 
 
 --
--- TOC entry 2290 (class 0 OID 0)
--- Dependencies: 175
--- Name: auto_incrementing_pk; Type: ACL; Schema: 1; Owner: dbapi_test
+-- TOC entry 2323 (class 0 OID 0)
+-- Dependencies: 192
+-- Name: auto_incrementing_pk; Type: ACL; Schema: 1; Owner: postgrest_test
 --
 
-REVOKE ALL ON TABLE auto_incrementing_pk FROM dbapi_test;
-GRANT ALL ON TABLE auto_incrementing_pk TO dbapi_test;
-GRANT ALL ON TABLE auto_incrementing_pk TO dbapi_anonymous;
+REVOKE ALL ON TABLE auto_incrementing_pk FROM PUBLIC;
+REVOKE ALL ON TABLE auto_incrementing_pk FROM postgrest_test;
+GRANT ALL ON TABLE auto_incrementing_pk TO postgrest_test;
+GRANT ALL ON TABLE auto_incrementing_pk TO postgrest_anonymous;
 
 
-
-REVOKE ALL ON SEQUENCE auto_incrementing_pk_id_seq FROM dbapi_test;
-GRANT ALL ON SEQUENCE auto_incrementing_pk_id_seq TO dbapi_test;
-GRANT USAGE ON SEQUENCE auto_incrementing_pk_id_seq TO dbapi_anonymous;
-
-
-
-REVOKE ALL ON TABLE compound_pk FROM dbapi_test;
-GRANT ALL ON TABLE compound_pk TO dbapi_test;
-GRANT ALL ON TABLE compound_pk TO dbapi_anonymous;
-
 --
-
-REVOKE ALL ON TABLE items FROM dbapi_test;
-GRANT ALL ON TABLE items TO dbapi_test;
-GRANT ALL ON TABLE items TO dbapi_anonymous;
-
-
-
-REVOKE ALL ON SEQUENCE items_id_seq FROM dbapi_test;
-GRANT ALL ON SEQUENCE items_id_seq TO dbapi_test;
-GRANT USAGE ON SEQUENCE items_id_seq TO dbapi_anonymous;
-
+-- TOC entry 2325 (class 0 OID 0)
+-- Dependencies: 193
+-- Name: auto_incrementing_pk_id_seq; Type: ACL; Schema: 1; Owner: postgrest_test
 --
 
-REVOKE ALL ON TABLE menagerie FROM dbapi_test;
-GRANT ALL ON TABLE menagerie TO dbapi_test;
-GRANT ALL ON TABLE menagerie TO dbapi_anonymous;
+REVOKE ALL ON SEQUENCE auto_incrementing_pk_id_seq FROM PUBLIC;
+REVOKE ALL ON SEQUENCE auto_incrementing_pk_id_seq FROM postgrest_test;
+GRANT ALL ON SEQUENCE auto_incrementing_pk_id_seq TO postgrest_test;
+GRANT USAGE ON SEQUENCE auto_incrementing_pk_id_seq TO postgrest_anonymous;
+
 
+--
+-- TOC entry 2326 (class 0 OID 0)
+-- Dependencies: 194
+-- Name: compound_pk; Type: ACL; Schema: 1; Owner: postgrest_test
 --
 
-REVOKE ALL ON TABLE no_pk FROM dbapi_test;
-GRANT ALL ON TABLE no_pk TO dbapi_test;
-GRANT ALL ON TABLE no_pk TO dbapi_anonymous;
+REVOKE ALL ON TABLE compound_pk FROM PUBLIC;
+REVOKE ALL ON TABLE compound_pk FROM postgrest_test;
+GRANT ALL ON TABLE compound_pk TO postgrest_test;
+GRANT ALL ON TABLE compound_pk TO postgrest_anonymous;
 
 
+REVOKE ALL ON TABLE has_fk FROM PUBLIC;
+REVOKE ALL ON TABLE has_fk FROM postgrest_test;
+GRANT ALL ON TABLE has_fk TO postgrest_test;
+GRANT ALL ON TABLE has_fk TO postgrest_anonymous;
 
-REVOKE ALL ON TABLE simple_pk FROM dbapi_test;
-GRANT ALL ON TABLE simple_pk TO dbapi_test;
-GRANT ALL ON TABLE simple_pk TO dbapi_anonymous;
+--
+-- TOC entry 2328 (class 0 OID 0)
+-- Dependencies: 197
+-- Name: items; Type: ACL; Schema: 1; Owner: postgrest_test
+--
+
+REVOKE ALL ON TABLE items FROM PUBLIC;
+REVOKE ALL ON TABLE items FROM postgrest_test;
+GRANT ALL ON TABLE items TO postgrest_test;
+GRANT ALL ON TABLE items TO postgrest_anonymous;
+
+
+--
+-- TOC entry 2330 (class 0 OID 0)
+-- Dependencies: 198
+-- Name: items_id_seq; Type: ACL; Schema: 1; Owner: postgrest_test
+--
+
+REVOKE ALL ON SEQUENCE items_id_seq FROM PUBLIC;
+REVOKE ALL ON SEQUENCE items_id_seq FROM postgrest_test;
+GRANT ALL ON SEQUENCE items_id_seq TO postgrest_test;
+GRANT USAGE ON SEQUENCE items_id_seq TO postgrest_anonymous;
+
+
+--
+-- TOC entry 2331 (class 0 OID 0)
+-- Dependencies: 199
+-- Name: menagerie; Type: ACL; Schema: 1; Owner: postgrest_test
+--
+
+REVOKE ALL ON TABLE menagerie FROM PUBLIC;
+REVOKE ALL ON TABLE menagerie FROM postgrest_test;
+GRANT ALL ON TABLE menagerie TO postgrest_test;
+GRANT ALL ON TABLE menagerie TO postgrest_anonymous;
+
+
+--
+-- TOC entry 2332 (class 0 OID 0)
+-- Dependencies: 200
+-- Name: no_pk; Type: ACL; Schema: 1; Owner: postgrest_test
+--
+
+REVOKE ALL ON TABLE no_pk FROM PUBLIC;
+REVOKE ALL ON TABLE no_pk FROM postgrest_test;
+GRANT ALL ON TABLE no_pk TO postgrest_test;
+GRANT ALL ON TABLE no_pk TO postgrest_anonymous;
+
+
+--
+-- TOC entry 2333 (class 0 OID 0)
+-- Dependencies: 201
+-- Name: simple_pk; Type: ACL; Schema: 1; Owner: postgrest_test
+--
+
+REVOKE ALL ON TABLE simple_pk FROM PUBLIC;
+REVOKE ALL ON TABLE simple_pk FROM postgrest_test;
+GRANT ALL ON TABLE simple_pk TO postgrest_test;
+GRANT ALL ON TABLE simple_pk TO postgrest_anonymous;
+
+
+SET search_path = postgrest, pg_catalog;
+
+--
+-- TOC entry 2334 (class 0 OID 0)
+-- Dependencies: 202
+-- Name: auth; Type: ACL; Schema: postgrest; Owner: postgrest_test
+--
+
+REVOKE ALL ON TABLE auth FROM PUBLIC;
+REVOKE ALL ON TABLE auth FROM postgrest_test;
+GRANT ALL ON TABLE auth TO postgrest_test;
+GRANT INSERT ON TABLE auth TO postgrest_anonymous;
 
 
 SET search_path = private, pg_catalog;
 
+--
+-- TOC entry 2335 (class 0 OID 0)
+-- Dependencies: 203
+-- Name: articles; Type: ACL; Schema: private; Owner: postgrest_test
+--
 
-REVOKE ALL ON TABLE articles FROM dbapi_test;
-GRANT ALL ON TABLE articles TO dbapi_test;
+REVOKE ALL ON TABLE articles FROM PUBLIC;
+REVOKE ALL ON TABLE articles FROM postgrest_test;
+GRANT ALL ON TABLE articles TO postgrest_test;
+
+
+-- Completed on 2014-10-21 15:12:44 PDT
+
+--
+-- PostgreSQL database dump complete
 --
 

test/test.crt

@@ -1,14 +0,0 @@
------BEGIN CERTIFICATE-----
-MIICLTCCAZYCCQCj6GtISfdwNjANBgkqhkiG9w0BAQUFADBbMQswCQYDVQQGEwJV
-UzELMAkGA1UECBMCQ0ExFjAUBgNVBAcTDVNhbiBGcmFuY2lzY28xEzARBgNVBAoT
-Ckxvb3AgUmVjdXIxEjAQBgNVBAMTCWxvY2FsaG9zdDAeFw0xNDEwMDMyMDQyNDNa
-Fw0xNTEwMDMyMDQyNDNaMFsxCzAJBgNVBAYTAlVTMQswCQYDVQQIEwJDQTEWMBQG
-A1UEBxMNU2FuIEZyYW5jaXNjbzETMBEGA1UEChMKTG9vcCBSZWN1cjESMBAGA1UE
-AxMJbG9jYWxob3N0MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQC5gFnTRBge
-mXdkCMD+OycujZrCWOOLIDBqRr7kDbxVXqz/TKHRVx6bz88g9egzvR2HLyA418kd
-dAu+lMmjrRv/k0Lnk/UvC1aj0huoHpOVUgOwy3qS4cE663uU5qsrgf4RDP7bLDcQ
-FDW02SQ2n5ryv8nB9TSYpvQvYPiTMMKdrwIDAQABMA0GCSqGSIb3DQEBBQUAA4GB
-AAQwkC+GSaGArAKdMIqs8/55KAjyNd11MupiCWsu1cwBJ4QJc1PxrYOMLMYnU06J
-I0v1bJ6mG06/Js0r2FHM0NXSQO+7DLPWu4LchoBgFt4ZRm2+GbLzFrfu41yn5mJN
-VeUxCBQ9hOrE8Kwe+/9IUUVPxlISF+YHOyF3DxWUViak
------END CERTIFICATE-----

test/test.key

@@ -1,15 +0,0 @@
------BEGIN RSA PRIVATE KEY-----
-MIICXQIBAAKBgQC5gFnTRBgemXdkCMD+OycujZrCWOOLIDBqRr7kDbxVXqz/TKHR
-Vx6bz88g9egzvR2HLyA418kddAu+lMmjrRv/k0Lnk/UvC1aj0huoHpOVUgOwy3qS
-4cE663uU5qsrgf4RDP7bLDcQFDW02SQ2n5ryv8nB9TSYpvQvYPiTMMKdrwIDAQAB
-AoGAZZD0Haub9S0b5KayXMCwnFmmEaEvvR47xATGQgGPS8LRv9sKgp9LwA4RH7/k
-imeSglD4OIdVs421XH0ExlxjBiV5EzTCgLUyKbfA//xUy9ggXD1Ks4vIHL0c1DM4
-g6/zylN7CQtt4Bb6YdWDSAUXIl3U5Dj1kG7BWuXfJTxCx7ECQQDyLxoQRNkPwKP/
-xeOqfCgwyD52rnUJ7g9UViFcCU2ZLLEYxE4b3FECcwhOfUlxYvKdOx03Qts0taow
-dPU/BEXHAkEAxBVyKwTJ0T6rQmcQEx3WuaBhSjXgZigUN4OdXMg+LSO+1MON6Nhm
-J9ociy+xvfd0Cf3yABaAEU/hZ+Pmk2DI2QJBAO/XU8F+3VQrXH7l9HSXJppBBRM1
-7HScDRRhhAIIuI+UYgJ8DjqrMpLxZu2MSBqBenHZ5DIhBMOrkVMR0PrKeWsCQCHh
-gsSkIysgpP7oSALFmSCh8a2c+ZUtP7EH8NzjTLsH/iVNVOvy2FPygBQcvZ2RcF95
-naMeQCq9nrkQy/qTMqECQQDa1WjWS2ngwqGzHovqiKRg3lQ8MFNlvlStAQl8+9h0
-KJj8Rl6h+gO8eretBIe+y5j/hCC90xlJotzwO3jJF6Hc
------END RSA PRIVATE KEY-----