From 6a5b756cac81dac807147dddb8970bbfebcbf222 Mon Sep 17 00:00:00 2001
From: Joe Nelson <cred+github@begriffs.com>
Date: Mon, 27 Oct 2014 15:54:15 -0700
Subject: [PATCH] Report server and version in headers

---
 dbapi.cabal |  2 +-
 src/Main.hs | 13 +++++++++++--
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/dbapi.cabal b/dbapi.cabal
index dfbe0033c..7de9ec196 100644
--- a/dbapi.cabal
+++ b/dbapi.cabal
@@ -1,5 +1,5 @@
 name:                  dbapi
-version:               0.2.4.0
+version:               0.2.4.1
 synopsis:              The database is your api
 license:               MIT
 license-file:          LICENSE
diff --git a/src/Main.hs b/src/Main.hs
index 5de6edd08..09adc7927 100644
--- a/src/Main.hs
+++ b/src/Main.hs
@@ -1,6 +1,9 @@
 {-# LANGUAGE OverloadedStrings #-}
 
 module Main where
+
+import Paths_dbapi (version)
+
 import Dbapi
 import Middleware (inTransaction, authenticated, withSavepoint, clientErrors,
   redirectInsecure, withDBConnection)
@@ -15,8 +18,10 @@ import Network.Wai.Middleware.Gzip (gzip, def)
 import Network.Wai.Middleware.Cors (cors)
 import Network.Wai.Middleware.Static (staticPolicy, only)
 import Database.HDBC (disconnect)
-import Database.HDBC.PostgreSQL(connectPostgreSQL')
+import Database.HDBC.PostgreSQL (connectPostgreSQL')
 import Data.Pool(createPool, destroyAllResources)
+import Data.List (intercalate)
+import Data.Version (versionBranch)
 
 argParser :: Parser AppConfig
 argParser = AppConfig
@@ -45,7 +50,10 @@ main = do
         putStrLn "WARNING, running in insecure mode, auth will be in plaintext"
 
       Prelude.putStrLn $ "Listening on port " ++ (show $ configPort conf :: String)
-      run port $ (if configSecure conf then redirectInsecure else id)
+      let settings = setPort port
+                   . setServerName (cs $ "dbapi/" <> prettyVersion)
+                   $ defaultSettings
+      runSettings settings $ (if configSecure conf then redirectInsecure else id)
         . gzip def . cors corsPolicy . clientErrors
         . staticPolicy (only [("favicon.ico", "static/favicon.ico")])
         . withDBConnection pool . inTransaction
@@ -53,3 +61,4 @@ main = do
       )
   where
     describe = progDesc "create a REST API to an existing Postgres database"
+    prettyVersion = intercalate "." $ map show $ versionBranch version