Settings added.

Author: raul-brainattica

core/authentication/jwt_backend.go

@@ -3,11 +3,11 @@ package authentication
 import (
 	"api.jwt.auth/core/redis"
 	"api.jwt.auth/services/models"
+	"api.jwt.auth/settings"
 	"code.google.com/p/go-uuid/uuid"
 	jwt "github.com/dgrijalva/jwt-go"
 	"golang.org/x/crypto/bcrypt"
 	"io/ioutil"
-	"path/filepath"
 	"time"
 )
 
@@ -83,8 +83,7 @@ func (backend *JWTAuthenticationBackend) IsInBlacklist(token string) bool {
 }
 
 func getPrivateKey() []byte {
-	privateKeyPath, _ := filepath.Abs("./core/authentication/keys/private_key")
-	privateKey, err := ioutil.ReadFile(privateKeyPath)
+	privateKey, err := ioutil.ReadFile(settings.Get().PrivateKeyPath)
 	if err != nil {
 		panic(err)
 	}
@@ -93,8 +92,7 @@ func getPrivateKey() []byte {
 }
 
 func getPublicKey() []byte {
-	publicKeyPath, _ := filepath.Abs("./core/authentication/keys/public_key.pub")
-	publicKey, err := ioutil.ReadFile(publicKeyPath)
+	publicKey, err := ioutil.ReadFile(settings.Get().PublicKeyPath)
 	if err != nil {
 		panic(err)
 	}

server.go

@@ -2,11 +2,13 @@ package main
 
 import (
 	"api.jwt.auth/routers"
+	"api.jwt.auth/settings"
 	"github.com/codegangsta/negroni"
 	"net/http"
 )
 
 func main() {
+	settings.Init()
 	router := routers.InitRoutes()
 	n := negroni.Classic()
 	n.UseHandler(router)

core/authentication/keys/private_key renamed to settings/keys/private_key

@@ -0,0 +1,4 @@
+{
+  "PrivateKeyPath": "/home/vagrant/go/src/api.jwt.auth/settings/keys/private_key",
+  "PublicKeyPath": "/home/vagrant/go/src/api.jwt.auth/settings/keys/public_key.pub"
+}

core/authentication/keys/public_key.pub renamed to settings/keys/public_key.pub

@@ -0,0 +1,4 @@
+{
+  "PrivateKeyPath": "/home/vagrant/go/src/api.jwt.auth/settings/keys/private_key",
+  "PublicKeyPath": "/home/vagrant/go/src/api.jwt.auth/settings/keys/public_key.pub"
+}

settings/pre.json

@@ -0,0 +1,58 @@
+package settings
+
+import (
+	"encoding/json"
+	"fmt"
+	"io/ioutil"
+	"os"
+)
+
+var environments = map[string]string{
+	"production":    "settings/prod.json",
+	"preproduction": "settings/pre.json",
+	"tests":         "../../settings/tests.json",
+}
+
+type Settings struct {
+	PrivateKeyPath string
+	PublicKeyPath  string
+}
+
+var settings Settings = Settings{}
+var env = "preproduction"
+
+func Init() {
+	env = os.Getenv("GO_ENV")
+	if env == "" {
+		fmt.Println("Warning: Setting preproduction environment due to lack of GO_ENV value")
+		env = "preproduction"
+	}
+	LoadSettingsByEnv(env)
+}
+
+func LoadSettingsByEnv(env string) {
+	content, err := ioutil.ReadFile(environments[env])
+	if err != nil {
+		fmt.Println("Error while reading config file", err)
+	}
+	settings = Settings{}
+	jsonErr := json.Unmarshal(content, &settings)
+	if jsonErr != nil {
+		fmt.Println("Error while parsing config file", jsonErr)
+	}
+}
+
+func GetEnvironment() string {
+	return env
+}
+
+func Get() Settings {
+	if &settings == nil {
+		Init()
+	}
+	return settings
+}
+
+func IsTestEnvironment() bool {
+	return env == "tests"
+}

settings/prod.json

@@ -0,0 +1,4 @@
+{
+  "PrivateKeyPath": "/home/vagrant/go/src/api.jwt.auth/settings/keys/private_key",
+  "PublicKeyPath": "/home/vagrant/go/src/api.jwt.auth/settings/keys/public_key.pub"
+}