feat: use viper and cobra to parameterize creation values with viper yaml

2023-02-15 19:30:05 +01:00
parent 0019941054
commit c1680106ab
8 changed files with 722 additions and 2 deletions
--- a/cmd/main.go
+++ b/cmd/main.go
@@ -0,0 +1,144 @@
+package cmd
+
+import (
+	"crypto/x509"
+	"fmt"
+	"log"
+	"math/big"
+	"os"
+
+	"github.com/joho/godotenv"
+	"github.com/kelseyhightower/envconfig"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+	"gitlab.com/urkob/go-cert-gen/internal/cert"
+	"gitlab.com/urkob/go-cert-gen/internal/io"
+	"gitlab.com/urkob/go-cert-gen/pkg/ca"
+	"gitlab.com/urkob/go-cert-gen/pkg/client"
+	pkgio "gitlab.com/urkob/go-cert-gen/pkg/io"
+	"gitlab.com/urkob/go-cert-gen/pkg/util"
+)
+
+var envConfig struct {
+	ViperConfig     string `required:"true" split_words:"true"`
+	ViperConfigType string `required:"true" split_words:"true"`
+	Env             string `required:"true" split_words:"true"`
+}
+
+var writer pkgio.WriterIface
+
+func intEnvConfig(isProd bool) {
+	if !isProd {
+		err := godotenv.Load(util.RootDir() + "/.env")
+		if err != nil {
+			log.Fatalf("environment variable ENV is empty and an error occurred while loading the .env file: %s\n", err)
+		}
+	}
+
+	err := envconfig.Process("", &envConfig)
+	if err != nil {
+		log.Fatalf("envconfig.Process: %s\n", err)
+	}
+}
+
+var rootCmd = &cobra.Command{
+	Use:   "go-gen-cert",
+	Short: "go-gen-cert is a CLI tool to generate TLS certificates for your gRPC client/server communication",
+	Run: func(cmd *cobra.Command, args []string) {
+		caCfg := &ca.CaConfig{
+			SerialNumber: big.NewInt(viper.GetInt64("ca.serial_number")),
+			Subject: ca.CaSubject{
+				Organization: viper.GetString("ca.subject.organization"),
+				CommonName:   viper.GetString("ca.subject.common_name"),
+			},
+			KeyUsage:    x509.KeyUsage(viper.GetInt("ca.key_usage")),
+			ExtKeyUsage: getExtKeyUsage(viper.GetIntSlice("ca.ext_key_usage")),
+			Duration:    viper.GetDuration("ca.duration"),
+		}
+
+		rootCA, err := cert.NewRootCA(caCfg)
+		if err != nil {
+			log.Fatalf("cert.NewRootCA: %s", err)
+		}
+
+		subjectKeyId, err := util.GetBytes(viper.Get("client.subject_key_id"))
+		if err != nil {
+			log.Fatalf("cert.NewRootCA: %s", err)
+		}
+
+		clientCfg := &client.ClientCertConfig{
+			Serial: &big.Int{},
+			Subject: client.Subject{
+				Organization:  viper.GetString("client.subject.organization"),
+				Country:       viper.GetString("client.subject.country"),
+				Province:      viper.GetString("client.subject.province"),
+				Locality:      viper.GetString("client.subject.locality"),
+				StreetAddress: viper.GetString("client.subject.street_address"),
+				PostalCode:    viper.GetString("client.subject.postal_code"),
+			},
+			Duration:     viper.GetDuration("client.duration"),
+			SubjectKeyId: subjectKeyId,
+			ExtKeyUsage:  getExtKeyUsage(viper.GetIntSlice("client.ext_key_usage")),
+			KeyUsage:     x509.KeyUsage(viper.GetInt("client.key_usage")),
+		}
+
+		clientCert, err := rootCA.WithClientCert(clientCfg)
+		if err != nil {
+			log.Fatalf("rootCA.WithClientCert: %s", err)
+		}
+
+		exportPem("root-ca.pem", rootCA.PEM())
+		exportPem("root-key.pem", rootCA.Key())
+
+		exportPem("client-cert.pem", clientCert.PEM())
+		exportPem("client-key.pem", clientCert.Key())
+	},
+}
+
+func getExtKeyUsage(intKeyUsageSlice []int) []x509.ExtKeyUsage {
+	extKeyUsage := make([]x509.ExtKeyUsage, len(intKeyUsageSlice))
+	for _, v := range intKeyUsageSlice {
+		extKeyUsage = append(extKeyUsage, x509.ExtKeyUsage(v))
+	}
+	return extKeyUsage
+}
+
+func exportPem(filename string, data []byte) {
+	outputPath, err := writer.WriteFile(filename, data)
+	if err != nil {
+		log.Fatalf("rootCA.WithClientCert: %s", err)
+	}
+	log.Printf("file created successfuly: %s\n", outputPath)
+}
+
+func init() {
+	intEnvConfig(false)
+	cobra.OnInitialize(initConfig)
+}
+
+func initConfig() {
+	if envConfig.ViperConfig != "" {
+		viper.AddConfigPath(util.RootDir())
+		viper.SetConfigName(envConfig.ViperConfig)
+		viper.SetConfigType(envConfig.ViperConfigType)
+	} else {
+		home, err := os.UserHomeDir()
+		cobra.CheckErr(err)
+		viper.AddConfigPath(home)
+		viper.SetConfigType("yaml")
+		viper.SetConfigName(".cobra")
+	}
+
+	if err := viper.ReadInConfig(); err != nil {
+		log.Fatalf("viper.ReadInConfig: %s", err)
+	}
+	log.Println("viper.GetString('export_dir)", viper.GetString("export_dir"))
+	writer = io.NewWriter(viper.GetString("export_dir"))
+}
+
+func Execute() {
+	if err := rootCmd.Execute(); err != nil {
+		fmt.Fprintln(os.Stderr, err)
+		os.Exit(1)
+	}
+}