feat: Add Vault provider

args.go

@@ -0,0 +1,40 @@
+// Copyright © 2023 Bank-Vaults Maintainers
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"fmt"
+	"os/exec"
+)
+
+// ExtractEntrypoint extracts entrypoint data in the form of binary path and its arguments from the
+// os.Args. Note that the path to the binary will be returned as the first element.
+func ExtractEntrypoint(args []string) (string, []string, error) {
+	if len(args) <= 1 {
+		return "", nil, fmt.Errorf("no args provided")
+	}
+
+	binaryPath, err := exec.LookPath(args[1])
+	if err != nil {
+		return "", nil, fmt.Errorf("binary %s not found", args[1])
+	}
+
+	var binaryArgs []string
+	if len(args) >= 2 {
+		binaryArgs = args[2:] // returns the arguments for the binary
+	}
+
+	return binaryPath, binaryArgs, nil
+}

args_test.go

@@ -0,0 +1,68 @@
+// Copyright © 2023 Bank-Vaults Maintainers
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package main
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestExtractEntrypoint(t *testing.T) {
+	tests := []struct {
+		name               string
+		args               []string
+		expectedBinaryPath string
+		expectedBinaryArgs []string
+		err                error
+	}{
+		{
+			name:               "Valid case with one argument",
+			args:               []string{"secret-init", "env"},
+			expectedBinaryPath: "/usr/bin/env",
+			expectedBinaryArgs: []string{},
+		},
+		{
+			name:               "Valid case with more than two arguments",
+			args:               []string{"secret-init", "env", "|", "grep", "secrets"},
+			expectedBinaryPath: "/usr/bin/env",
+			expectedBinaryArgs: []string{"|", "grep", "secrets"},
+		},
+		{
+			name: "Invalid case - no arguments",
+			args: []string{"secret-init"},
+			err:  fmt.Errorf("no args provided"),
+		},
+		{
+			name: "Invalid case - binary not found",
+			args: []string{"secret-init", "nonexistentBinary"},
+			err:  fmt.Errorf("binary nonexistentBinary not found"),
+		},
+	}
+
+	for _, tt := range tests {
+		ttp := tt
+		t.Run(ttp.name, func(t *testing.T) {
+			binaryPath, binaryArgs, err := ExtractEntrypoint(ttp.args)
+			if err != nil {
+				assert.EqualError(t, ttp.err, err.Error(), "Unexpected error message")
+			} else {
+				assert.Equal(t, ttp.expectedBinaryPath, binaryPath, "Unexpected binary path")
+				assert.Equal(t, ttp.expectedBinaryArgs, binaryArgs, "Unexpected binary args")
+			}
+		})
+	}
+}

common/config.go

@@ -0,0 +1,51 @@
+// Copyright © 2023 Bank-Vaults Maintainers
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package common
+
+import (
+	"os"
+	"time"
+
+	"github.com/spf13/cast"
+)
+
+const (
+	LogLevelEnv  = "SECRET_INIT_LOG_LEVEL"
+	JSONLogEnv   = "SECRET_INIT_JSON_LOG"
+	LogServerEnv = "SECRET_INIT_LOG_SERVER"
+	DaemonEnv    = "SECRET_INIT_DAEMON"
+	DelayEnv     = "SECRET_INIT_DELAY"
+	ProviderEnv  = "SECRET_INIT_PROVIDER"
+)
+
+type Config struct {
+	LogLevel  string        `json:"log_level"`
+	JSONLog   bool          `json:"json_log"`
+	LogServer string        `json:"log_server"`
+	Daemon    bool          `json:"daemon"`
+	Delay     time.Duration `json:"delay"`
+	Provider  string        `json:"provider"`
+}
+
+func LoadConfig() (*Config, error) {
+	return &Config{
+		LogLevel:  os.Getenv(LogLevelEnv),
+		JSONLog:   cast.ToBool(os.Getenv(JSONLogEnv)),
+		LogServer: os.Getenv(LogServerEnv),
+		Daemon:    cast.ToBool(os.Getenv(DaemonEnv)),
+		Delay:     cast.ToDuration(os.Getenv(DelayEnv)),
+		Provider:  os.Getenv(ProviderEnv),
+	}, nil
+}

common/config_test.go

@@ -0,0 +1,67 @@
+// Copyright © 2023 Bank-Vaults Maintainers
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package common
+
+import (
+	"os"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestConfig(t *testing.T) {
+	tests := []struct {
+		name       string
+		env        map[string]string
+		wantConfig *Config
+	}{
+		{
+			name: "Valid configuration",
+			env: map[string]string{
+				LogLevelEnv:  "debug",
+				JSONLogEnv:   "true",
+				LogServerEnv: "",
+				DaemonEnv:    "true",
+				ProviderEnv:  "vault",
+			},
+			wantConfig: &Config{
+				LogLevel:  "debug",
+				JSONLog:   true,
+				LogServer: "",
+				Daemon:    true,
+				Provider:  "vault",
+			},
+		},
+	}
+
+	for _, tt := range tests {
+		ttp := tt
+		t.Run(ttp.name, func(t *testing.T) {
+			for envKey, envVal := range ttp.env {
+				os.Setenv(envKey, envVal)
+			}
+
+			config, err := LoadConfig()
+			assert.Nil(t, err, "Unexpected error")
+
+			assert.Equal(t, ttp.wantConfig, config, "Unexpected config")
+
+			// unset envs for the next test
+			for envKey := range ttp.env {
+				os.Unsetenv(envKey)
+			}
+		})
+	}
+}