From ef1ec36b42350d9abae5ea533ce38bf37d013ffc Mon Sep 17 00:00:00 2001
From: Grégoire Duchêne <gduchene@awhk.org>
Date: Sun, 26 Feb 2023 13:19:02 +0000
Subject: Add ParseStringerEnum

---
 flag.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'flag.go')

diff --git a/flag.go b/flag.go
index f9d048e..96c74d2 100644
--- a/flag.go
+++ b/flag.go
@@ -184,6 +184,20 @@ func ParseStringEnum(values ...string) ParseFunc[string] {
 	}
 }
 
+// ParseStringerEnum returns a ParseFunc that will return the first
+// value having a string value matching the string passed.
+func ParseStringerEnum[T fmt.Stringer](values ...T) ParseFunc[T] {
+	return func(s string) (T, error) {
+		for _, val := range values {
+			if s == val.String() {
+				return val, nil
+			}
+		}
+		var zero T
+		return zero, UnknownEnumValueError[T]{s, values}
+	}
+}
+
 // ParseTime parses a string according to the time.RFC3339 format.
 func ParseTime(s string) (time.Time, error) {
 	return time.Parse(time.RFC3339, s)
-- 
cgit v1.2.3-70-g09d2