genenv.go

package cmd

import (
	"bufio"
	"fmt"
	"os"
	"strings"
	"text/template"

	"github.com/danjac/podbaby/cmd/Godeps/_workspace/src/golang.org/x/crypto/ssh/terminal"
	"github.com/danjac/podbaby/config"
)

func fileExists(name string) bool {
	_, err := os.Stat(name)
	return !os.IsNotExist(err)
}

func readLine(prompt, defaultValue string, required bool) string {
	for {
		fmt.Printf("\r\n%s ", prompt)
		reader := bufio.NewReader(os.Stdin)
		text, _ := reader.ReadString('\n')
		text = strings.TrimSpace(text)
		if text == "" {
			text = defaultValue
		}
		if text != "" || !required {
			return text
		}
	}
}

func readSecret(prompt, defaultValue string, required bool) string {
	for {
		fmt.Printf("\r\n%s ", prompt)
		bytePassword, _ := terminal.ReadPassword(int(os.Stdin.Fd()))
		text := string(bytePassword)
		text = strings.TrimSpace(text)
		if text == "" {
			text = defaultValue
		}
		if text != "" || !required {
			return text
		}
		return text
	}
}

// SecretKey prints a 32 byte random key you can use for secure keys
func SecretKey() {
	fmt.Println(config.RandomKey())
}

// Genenv interactively generates an environment settings file
// Usage: podbaby genenv [dest-file.env]
func Genenv(dst string) {
	if err := genenv(dst); err != nil {
		panic(err)
	}
}

func genenv(dst string) error {

	// does destination file exist?

	if fileExists(dst) {
		text := readLine("This file already exists. Overwrite (Y/N)?", "N", false)
		if strings.ToLower(text) != "y" {
			return nil
		}
	}

	pgHost := readLine("PostgreSQL Host (localhost:5432)?", "localhost:5432", true)

	pgName := readLine("PostgreSQL database name?", "podbaby", true)
	pgUser := readLine("PostgreSQL User?", "", false)
	pgPass := readSecret("PostgreSQL Password?", "", false)

	pgDisable := readLine("PostgreSQL SSL enabled (Y/N)", "N", false)

	smtpHost := readLine("SMTP Host? (localhost)", "localhost", true)
	smtpUser := readLine("SMTP User?", "", false)
	smtpPass := readSecret("SMPT Password?", "", false)

	secretKey := readSecret("Secret key?", config.RandomKey(), true)
	secureCookieKey := readSecret("Secure cookie key?", config.RandomKey(), true)

	gaKey := readLine("Google analytics key?", "", false)

	dbURL := fmt.Sprintf("postgres://%s:%s@%s/%s",
		pgUser, pgPass, pgHost, pgName)

	if strings.ToLower(pgDisable) == "n" {
		dbURL += "?sslmode=disable"
	}

	cfg := config.Default()
	cfg.DatabaseURL = dbURL
	cfg.SecretKey = secretKey
	cfg.SecureCookieKey = secureCookieKey
	cfg.GoogleAnalyticsID = gaKey
	cfg.Mail.Host = smtpHost
	cfg.Mail.User = smtpUser
	cfg.Mail.Password = smtpPass

	t, err := template.New("env").Parse(`
DB_URL="{{.DatabaseURL}}"
MAX_DB_CONNECTIONS="{{.MaxDBConnections}}"
GOOGLE_ANALYTICS_ID="{{.GoogleAnalyticsID}}"
SECRET_KEY="{{.SecretKey}}"
SECURE_COOKIE_KEY="{{.SecureCookieKey}}"
MAIL_ADDR="{{.Mail.Addr}}"
MAIL_HOST="{{.Mail.Host}}"
MAIL_USER="{{.Mail.User}}"
MAIL_PASSWORD="{{.Mail.Password}}"
`)
	if err != nil {
		return err
	}

	f, err := os.Create(dst)
	if err != nil {
		return err
	}
	defer f.Close()
	return t.Execute(f, cfg)
}