1 files changed, 54 insertions, 0 deletions

diff --git a/cli/tolerant_pflag.go b/cli/tolerant_pflag.go
new file mode 100644
index 0000000..f2e9f22
--- /dev/null
+++ b/cli/tolerant_pflag.go
@@ -0,0 +1,54 @@
+package cli
+
+import (
+        "log"
+        "time"
+
+        "code.crute.us/mcrute/golib/vault"
+
+        "github.com/spf13/pflag"
+)
+
+type VaultCredential struct {
+        Username, Password string
+}
+
+type TolerantPflagSet struct {
+        *pflag.FlagSet
+}
+
+func (f *TolerantPflagSet) MayGetBool(n string) bool {
+        v, _ := f.GetBool(n)
+        return v
+}
+
+func (f *TolerantPflagSet) MayGetString(n string) string {
+        v, _ := f.GetString(n)
+        return v
+}
+
+func (f *TolerantPflagSet) MayGetDuration(n string) time.Duration {
+        v, _ := f.GetDuration(n)
+        return v
+}
+
+func (f *TolerantPflagSet) MayGetStringSlice(n string) []string {
+        v, _ := f.GetStringSlice(n)
+        return v
+}
+
+func (f *TolerantPflagSet) MayGetVaultCredential(n string) VaultCredential {
+        vk, err := vault.GetVaultKey(f.MayGetString(n))
+        if err != nil {
+                return VaultCredential{}
+        }
+        return VaultCredential{vk.Username, vk.Password}
+}
+
+func (f *TolerantPflagSet) MustGetVaultCredential(n string) VaultCredential {
+        vk, err := vault.GetVaultKey(f.MayGetString(n))
+        if err != nil {
+                log.Fatalf("Error getting %s from vault: %w", n, err)
+        }
+        return VaultCredential{vk.Username, vk.Password}
+}