From 0424ea77d6c09624f35c291a68dde60ea9ca8453 Mon Sep 17 00:00:00 2001
From: Niall Sheridan <nsheridan@gmail.com>
Date: Sat, 27 Aug 2016 21:34:53 +0100
Subject: Allow setting some config from environment

---
 server/config/config.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

(limited to 'server')

diff --git a/server/config/config.go b/server/config/config.go
index f1341c1..3587e9f 100644
--- a/server/config/config.go
+++ b/server/config/config.go
@@ -3,6 +3,8 @@ package config
 import (
 	"errors"
 	"io"
+	"os"
+	"strconv"
 
 	"github.com/hashicorp/go-multierror"
 	"github.com/spf13/viper"
@@ -82,6 +84,28 @@ func verifyConfig(u *unmarshalled) error {
 	return err
 }
 
+func setFromEnv(u *unmarshalled) {
+	port, err := strconv.Atoi(os.Getenv("PORT"))
+	if err == nil {
+		u.Server[0].Port = port
+	}
+	if os.Getenv("DATASTORE") != "" {
+		u.Server[0].Datastore = os.Getenv("DATASTORE")
+	}
+	if os.Getenv("OAUTH_CLIENT_ID") != "" {
+		u.Auth[0].OauthClientID = os.Getenv("OAUTH_CLIENT_ID")
+	}
+	if os.Getenv("OAUTH_CLIENT_SECRET") != "" {
+		u.Auth[0].OauthClientSecret = os.Getenv("OAUTH_CLIENT_SECRET")
+	}
+	if os.Getenv("CSRF_SECRET") != "" {
+		u.Server[0].CSRFSecret = os.Getenv("CSRF_SECRET")
+	}
+	if os.Getenv("COOKIE_SECRET") != "" {
+		u.Server[0].CookieSecret = os.Getenv("COOKIE_SECRET")
+	}
+}
+
 // ReadConfig parses a JSON configuration file into a Config struct.
 func ReadConfig(r io.Reader) (*Config, error) {
 	u := &unmarshalled{}
@@ -93,6 +117,7 @@ func ReadConfig(r io.Reader) (*Config, error) {
 	if err := v.Unmarshal(u); err != nil {
 		return nil, err
 	}
+	setFromEnv(u)
 	if err := verifyConfig(u); err != nil {
 		return nil, err
 	}
-- 
cgit v1.2.3