From cf5b024e189cb4ed85f80b521cf9e0e5dc5b5070 Mon Sep 17 00:00:00 2001
From: Ben Sima <ben@bsima.me>
Date: Sat, 16 Feb 2019 20:51:58 -0800
Subject: Make port configurable

---
 ibb/Main.hs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'ibb/Main.hs')

diff --git a/ibb/Main.hs b/ibb/Main.hs
index 544b0c8..05e7738 100644
--- a/ibb/Main.hs
+++ b/ibb/Main.hs
@@ -3,6 +3,7 @@
 
 module Main where
 
+import Data.Maybe (fromMaybe)
 import Data.ByteString.Lazy (ByteString)
 import Data.Text.Lazy (Text)
 import Data.Text.Lazy.Encoding (encodeUtf8)
@@ -13,6 +14,7 @@ import Text.Hamlet (shamlet)
 import Text.Lucius (lucius, renderCss)
 import Web.Scotty (ActionM, scotty, get, html, raw, setHeader)
 import Influencers
+import System.Environment (lookupEnv)
 
 render :: Html -> ActionM ()
 render = html . renderHtml
@@ -21,7 +23,11 @@ css :: ByteString -> ActionM ()
 css src = setHeader "content-type" "text/css" >> raw src
 
 main :: IO ()
-main = scotty 3000 $ do
+main = do
+    port <- read <$> fromMaybe "3000" <$> lookupEnv "PORT" :: IO Int
+    scotty port routes
+
+routes = do
     get "/" $ render homepage
     get "/custom.css" $ css stylesheet
 
-- 
cgit v1.2.3