config.go

package main

import (
	"encoding/json"
	"fmt"
	"os"
)

// Google Cloud Storage Buckets (uncomment below to enable)
// StorageBucketName = "<your-storage-bucket>"
// StorageBucket, err = configureStorage(StorageBucketName)

type PostgresConfig struct {
	Host     string `json:"host"`
	Port     int    `json:"port"`
	User     string `json:"user"`
	Password string `json:"password"`
	Name     string `json:"name"`
}

func (c PostgresConfig) Dialect() string {
	return "postgres"
}

func (c PostgresConfig) ConnectionInfo() string {
	// We are going to provide two potential  connection
	// info strings based on whether a password is present.
	if c.Password == "" {
		return fmt.Sprintf("host=%s port=%d user=%s dbname=%s "+
			"sslmode=disable", c.Host, c.Port, c.User, c.Name)
	}
	return fmt.Sprintf("host=%s port=%d user=%s password=%s "+
		"dbname=%s sslmode=disable", c.Host, c.Port, c.User,
		c.Password, c.Name)
}

func DefaultPostgresConfig() PostgresConfig {
	return PostgresConfig{
		Host:     "localhost",
		Port:     5432,
		User:     "postgres",
		Password: "Mutoworld2013!",
		Name:     "muto_dev",
	}
}

type Config struct {
	Port     int            `json:"port"`
	Env      string         `json:"env"`
	Pepper   string         `json:"pepper"`
	HMACKey  string         `json:"hmac_key"`
	Database PostgresConfig `json:"database"`
}

func (c Config) IsProd() bool {
	return c.Env == "prod"
}

func DefaultConfig() Config {
	return Config{
		Port:     8080,
		Env:      "dev",
		Pepper:   "secret-random-string",
		HMACKey:  "secret-hmac-key",
		Database: DefaultPostgresConfig(),
	}
}

func LoadConfig(configReq bool) Config {
	// Open the config file
	f, err := os.Open(".config")
	if err != nil {
		if configReq {
			panic(err)
		}
		// If there was an error opening the file,
		// print out a message saying we are
		// using the default config and return it.
		fmt.Println("Using the default config...")
		return DefaultConfig()
	}
	// If we opened the config file successfully we
	// are going to create a Config variable to load it into.
	var c Config
	// We also need a JSON decoder, which will
	// read from the file we opened when decoding.
	dec := json.NewDecoder(f)
	// We then decode the file and place the results in c,
	// the Config variable we created for the results. The decoder
	// knows how to decode the data because of the struct tags
	// (eg `json:"port"`) we added to our Config and PostgresConfig
	// fields, much like GORM uses struct tags to know
	// which database column each field maps to.
	err = dec.Decode(&c)
	if err != nil {
		panic(err)
	}
	// If all goes well, return the loaded config.
	fmt.Println("Successfully loaded .config")
	return c
}