feat: Support configuration file for hmsecrets.
This commit is contained in:
1 parent
3b29c2cb1d
commit
03247415b2
12 files changed
+106
-48
No files matched your search
@@ -1,6 +1,8 @@
|
||||
package main
|
||||
|
||||
import "git.handmadecities.com/HMC/shhh/internal/clis/hmsecrets"
|
||||
import (
|
||||
"git.handmadecities.com/HMC/shhh/internal/clis/hmsecrets"
|
||||
)
|
||||
|
||||
func main() {
|
||||
hmsecrets.Execute()
|
||||
|
||||
@@ -11,4 +11,5 @@ require (
|
||||
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
||||
github.com/spf13/pflag v1.0.10 // indirect
|
||||
golang.org/x/sys v0.45.0 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
)
|
||||
@@ -2,8 +2,8 @@ package hmsecrets
|
||||
|
||||
import "git.handmadecities.com/HMC/shhh/pkg/secretsclient"
|
||||
|
||||
func getClient(base, ca, cert, certKey string) secretsclient.Client {
|
||||
cli, err := secretsclient.New(base, ca, cert, certKey)
|
||||
func getClient() secretsclient.Client {
|
||||
cli, err := secretsclient.New(conf.APIBase, conf.CA, conf.ClientCert, conf.ClientKey)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
@@ -0,0 +1,83 @@
|
||||
package hmsecrets
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"path"
|
||||
"runtime"
|
||||
|
||||
"gopkg.in/yaml.v3"
|
||||
)
|
||||
|
||||
type config struct {
|
||||
APIBase string `yaml:"api-base"`
|
||||
CA string `yaml:"ca"`
|
||||
ClientCert string `yaml:"client-cert"`
|
||||
ClientKey string `yaml:"client-key"`
|
||||
}
|
||||
|
||||
func mustLoadConfig() config {
|
||||
var baseConfigDir string
|
||||
var err error
|
||||
switch runtime.GOOS {
|
||||
case "windows":
|
||||
fallthrough
|
||||
case "linux":
|
||||
baseConfigDir, err = os.UserConfigDir()
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("determining user config dir: %w", err))
|
||||
}
|
||||
case "darwin":
|
||||
// We want $HOME/.config in macOS.
|
||||
homeDir, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("determining user home dir: %w", err))
|
||||
}
|
||||
baseConfigDir = path.Join(homeDir, ".config")
|
||||
}
|
||||
|
||||
configDir := path.Join(baseConfigDir, "hmsecrets")
|
||||
|
||||
configBytes, err := os.ReadFile(path.Join(configDir, "config.yaml"))
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("reading config file: %w", err))
|
||||
}
|
||||
var conf config
|
||||
err = yaml.Unmarshal(configBytes, &conf)
|
||||
if err != nil {
|
||||
panic(fmt.Errorf("parsing config file: %w", err))
|
||||
}
|
||||
|
||||
if conf.APIBase == "" {
|
||||
panic(errors.New("api base must be present in config"))
|
||||
}
|
||||
|
||||
if conf.CA == "" {
|
||||
conf.CA = path.Join(configDir, "ca.crt")
|
||||
}
|
||||
if !fileAccesible(conf.CA) {
|
||||
panic(fmt.Errorf("ca file '%s' does not exist or is inaccesible", conf.CA))
|
||||
}
|
||||
|
||||
if conf.ClientCert == "" {
|
||||
conf.ClientCert = path.Join(configDir, "client.crt")
|
||||
}
|
||||
if !fileAccesible(conf.ClientCert) {
|
||||
panic(fmt.Errorf("client cert file '%s' does not exist or is inaccesible", conf.ClientCert))
|
||||
}
|
||||
|
||||
if conf.ClientKey == "" {
|
||||
conf.ClientKey = path.Join(configDir, "client.key")
|
||||
}
|
||||
if !fileAccesible(conf.ClientKey) {
|
||||
panic(fmt.Errorf("client key file '%s' does not exist or is inaccesible", conf.ClientKey))
|
||||
}
|
||||
|
||||
return conf
|
||||
}
|
||||
|
||||
func fileAccesible(f string) bool {
|
||||
_, err := os.Stat(f)
|
||||
return err == nil
|
||||
}
|
||||
@@ -13,7 +13,7 @@ var delCmd = &cobra.Command{
|
||||
Short: "del [key]",
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
err := cli.Delete(args[0])
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
@@ -15,7 +15,7 @@ var getArrCmd = &cobra.Command{
|
||||
Short: "get-array [key]",
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
secret, err := cli.GetStringArr(args[0])
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
@@ -15,7 +15,7 @@ var getMapCmd = &cobra.Command{
|
||||
Short: "get-map [key]",
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
secret, err := cli.GetStringMap(args[0])
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
@@ -15,7 +15,7 @@ var getCmd = &cobra.Command{
|
||||
Short: "get [key]",
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
secret, err := cli.GetString(args[0])
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
@@ -7,44 +7,7 @@ import (
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
var apiBase string
|
||||
var caFile string
|
||||
var clientCert string
|
||||
var clientKey string
|
||||
|
||||
func init() {
|
||||
rootCmd.PersistentFlags().StringVar(
|
||||
&apiBase,
|
||||
"api-base",
|
||||
"",
|
||||
"base url for the service",
|
||||
)
|
||||
rootCmd.MarkPersistentFlagRequired("api-base")
|
||||
|
||||
rootCmd.PersistentFlags().StringVar(
|
||||
&caFile,
|
||||
"ca-file",
|
||||
"",
|
||||
"CA cert file",
|
||||
)
|
||||
rootCmd.MarkPersistentFlagRequired("ca-file")
|
||||
|
||||
rootCmd.PersistentFlags().StringVar(
|
||||
&clientCert,
|
||||
"client-cert",
|
||||
"",
|
||||
"certificate to be used for client mTLS auth",
|
||||
)
|
||||
rootCmd.MarkPersistentFlagRequired("client-cert")
|
||||
|
||||
rootCmd.PersistentFlags().StringVar(
|
||||
&clientKey,
|
||||
"client-key",
|
||||
"",
|
||||
"private key for the certificate to be used for client mTLS auth",
|
||||
)
|
||||
rootCmd.MarkPersistentFlagRequired("client-key")
|
||||
}
|
||||
var conf config
|
||||
|
||||
var rootCmd = &cobra.Command{
|
||||
Use: "hmsecrets",
|
||||
@@ -52,6 +15,15 @@ var rootCmd = &cobra.Command{
|
||||
}
|
||||
|
||||
func Execute() {
|
||||
defer func() {
|
||||
if r := recover(); r != nil {
|
||||
fmt.Printf("%+v\n", r)
|
||||
os.Exit(1)
|
||||
}
|
||||
}()
|
||||
|
||||
conf = mustLoadConfig()
|
||||
|
||||
if err := rootCmd.Execute(); err != nil {
|
||||
fmt.Fprintln(os.Stderr, err)
|
||||
os.Exit(1)
|
||||
|
||||
@@ -24,7 +24,7 @@ var setArrCmd = &cobra.Command{
|
||||
Short: "set-array [key]",
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
err := cli.PutStringArr(args[0], arrValue)
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
@@ -24,7 +24,7 @@ var setMapCmd = &cobra.Command{
|
||||
Short: "set-map [key]",
|
||||
Args: cobra.ExactArgs(1),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
err := cli.PutStringMap(args[0], mapValue)
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
@@ -13,7 +13,7 @@ var setCmd = &cobra.Command{
|
||||
Short: "set [key] [value]",
|
||||
Args: cobra.ExactArgs(2),
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
cli := getClient(apiBase, caFile, clientCert, clientKey)
|
||||
cli := getClient()
|
||||
err := cli.PutString(args[0], args[1])
|
||||
if err != nil {
|
||||
return err
|
||||
|
||||
Reference in new issue
Block a user