Imported github.com/ardanlabs/service as base example project

2025-12-21 23:57:41 +02:00 · 2019-05-16 10:39:25 -04:00
parent a5af03321d
commit e6453bae45
304 changed files with 51148 additions and 0 deletions
--- a/example-project/cmd/sales-admin/main.go
+++ b/example-project/cmd/sales-admin/main.go
@@ -0,0 +1,137 @@
+// 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
+}