golang-saas-starter-kit/example-project/cmd/sales-admin/main.go

// This program performs administrative tasks for the garage sale service.
//
// Run it with --cmd keygen or --cmd useradd

package main

import (
	"context"
	"crypto/rand"
	"crypto/rsa"
	"crypto/x509"
	"encoding/pem"
	"fmt"
	"log"
	"os"
	"time"

	"geeks-accelerator/oss/saas-starter-kit/example-project/internal/platform/auth"
	"geeks-accelerator/oss/saas-starter-kit/example-project/internal/platform/db"
	"geeks-accelerator/oss/saas-starter-kit/example-project/internal/platform/flag"
	"geeks-accelerator/oss/saas-starter-kit/example-project/internal/user"
	"github.com/kelseyhightower/envconfig"
	"github.com/pkg/errors"
)

func main() {

	// =========================================================================
	// Logging

	log := log.New(os.Stdout, "sales-admin : ", log.LstdFlags|log.Lmicroseconds|log.Lshortfile)

	// =========================================================================
	// Configuration

	var cfg struct {
		CMD string `envconfig:"CMD"`
		DB  struct {
			DialTimeout time.Duration `default:"5s" envconfig:"DIAL_TIMEOUT"`
			Host        string        `default:"localhost:27017/gotraining" envconfig:"HOST"`
		}
		Auth struct {
			PrivateKeyFile string `default:"private.pem" envconfig:"PRIVATE_KEY_FILE"`
		}
		User struct {
			Email    string
			Password string
		}
	}

	if err := envconfig.Process("SALES", &cfg); err != nil {
		log.Fatalf("main : Parsing Config : %v", err)
	}

	if err := flag.Process(&cfg); err != nil {
		if err != flag.ErrHelp {
			log.Fatalf("main : Parsing Command Line : %v", err)
		}
		return // We displayed help.
	}

	var err error
	switch cfg.CMD {
	case "keygen":
		err = keygen(cfg.Auth.PrivateKeyFile)
	case "useradd":
		err = useradd(cfg.DB.Host, cfg.DB.DialTimeout, cfg.User.Email, cfg.User.Password)
	default:
		err = errors.New("Must provide --cmd keygen or --cmd useradd")
	}

	if err != nil {
		log.Fatal(err)
	}
}

// keygen creates an x509 private key for signing auth tokens.
func keygen(path string) error {

	key, err := rsa.GenerateKey(rand.Reader, 2048)
	if err != nil {
		return errors.Wrap(err, "generating keys")
	}

	file, err := os.Create(path)
	if err != nil {
		return errors.Wrap(err, "creating private file")
	}

	block := pem.Block{
		Type:  "RSA PRIVATE KEY",
		Bytes: x509.MarshalPKCS1PrivateKey(key),
	}

	if err := pem.Encode(file, &block); err != nil {
		return errors.Wrap(err, "encoding to private file")
	}

	if err := file.Close(); err != nil {
		return errors.Wrap(err, "closing private file")
	}

	return nil
}

func useradd(dbHost string, dbTimeout time.Duration, email, pass string) error {

	dbConn, err := db.New(dbHost, dbTimeout)
	if err != nil {
		return err
	}
	defer dbConn.Close()

	if email == "" {
		return errors.New("Must provide --user_email")
	}
	if pass == "" {
		return errors.New("Must provide --user_password or set the env var SALES_USER_PASSWORD")
	}

	ctx := context.Background()

	newU := user.NewUser{
		Email:           email,
		Password:        pass,
		PasswordConfirm: pass,
		Roles:           []string{auth.RoleAdmin, auth.RoleUser},
	}

	usr, err := user.Create(ctx, dbConn, &newU, time.Now())
	if err != nil {
		return err
	}

	fmt.Printf("User created with id: %v\n", usr.ID.Hex())
	return nil
}