new_pki part 1

.gitignore

@@ -3,12 +3,14 @@
 !**/*.pem
 !**/*.dot
 !**/*.svg
+!**/*.sh
 !**/
 
 !/.gitignore
 !/go.mod
 !/go.sum
 !/LICENSE
 !/README.md
+!/Dockerfile
 
 vendor

Dockerfile

@@ -0,0 +1,22 @@
+FROM ubuntu:latest as setup-build-stage-git
+RUN apt-get update && \
+    apt-get upgrade -y && \
+    apt-get install -y git
+# temporary until merge to master
+RUN git clone -b new_pki https://github.com/ditrit/shoset.git
+
+FROM golang:1.18-bullseye as setup-build-stage-shoset
+WORKDIR /app
+COPY --from=setup-build-stage-git . /app
+WORKDIR /app/shoset
+RUN go get
+WORKDIR /app/shoset/test
+RUN go build -o shoset
+
+FROM ubuntu:21.04 as setup-stage-production
+COPY --from=setup-build-stage-shoset /app/shoset/test/shoset usr/bin/
+RUN apt-get update
+
+FROM setup-stage-production as stage-production
+WORKDIR /usr/bin
+CMD [ "./shoset", "4" ]

config.go

@@ -0,0 +1,87 @@
+package shoset
+
+import (
+	"os"
+	"sync"
+
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/viper"
+)
+
+// Config: collection of configuration information for a shoset.
+type Config struct {
+	baseDir  string
+	fileName string
+	viper    *viper.Viper
+	mu       sync.Mutex
+}
+
+// NewConfig returns a *Config object.
+// Initialize home directory and viper.
+func NewConfig() *Config {
+	homeDir := "."
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		log.Error().Msg("couldn't home dir folder: " + err.Error())
+	}
+	cfg := &Config{
+		baseDir: homeDir + "/.shoset/",
+		viper:   viper.New(),
+	}
+	if err := mkdir(cfg.baseDir); err != nil {
+		log.Error().Msg("couldn't create folder: " + err.Error())
+	}
+	return cfg
+}
+
+// GetBaseDir returns baseDir from config, baseDir corresponds to homeDir + shoset repertory.
+func (cfg *Config) GetBaseDir() string { return cfg.baseDir }
+// GetFileName returns fileName from config.
+func (cfg *Config) GetFileName() string { return cfg.fileName }
+// SetFileName sets fileName for a config.
+func (cfg *Config) SetFileName(fileName string) { cfg.fileName = fileName }
+
+// InitFolders creates following config folders if needed: <base>/<name>/{cert, config}.
+func (cfg *Config) InitFolders(name string) (string, error) {
+	if err := mkdir(cfg.baseDir + name + "/"); err != nil {
+		return VOID, err
+	}
+	if err := mkdir(cfg.baseDir + name + PATH_CONFIG_DIR); err != nil {
+		return VOID, err
+	}
+	if err := mkdir(cfg.baseDir + name + PATH_CERT_DIR); err != nil {
+		return VOID, err
+	}
+	return cfg.baseDir, nil
+}
+
+// ReadConfig will load the configuration file from disk for a given fileName.
+// Initialize viper parameters before reading.
+func (cfg *Config) ReadConfig(fileName string) error {
+	cfg.viper.AddConfigPath(cfg.baseDir + fileName + PATH_CONFIG_DIR)
+	cfg.viper.SetConfigName(fileName)
+	cfg.viper.SetConfigType(CONFIG_TYPE)
+
+	return cfg.viper.ReadInConfig()
+}
+
+// WriteConfig writes current configuration to a given shoset.
+func (cfg *Config) WriteConfig(fileName string) error {
+	cfg.mu.Lock()
+	defer cfg.mu.Unlock()
+
+	return cfg.viper.WriteConfigAs(cfg.baseDir + fileName + PATH_CONFIG_DIR + CONFIG_FILE)
+}
+
+// Set sets the value for a key for viper config.
+func (cfg *Config) Set(key string, value interface{}) {
+	cfg.mu.Lock()
+	defer cfg.mu.Unlock()
+
+	cfg.viper.Set(key, value)
+}
+
+// GetSlice returns the viper config for a specific protocol.
+func (cfg *Config) GetSlice(protocol string) []string {
+	return cfg.viper.GetStringSlice(protocol)
+}