Merge pull request #131 from gdgib/G2-1607-ParseEnums

G2-1607 Parse enums
g2forge · Jul 4, 2024 · b04ae43 · b04ae43
2 parents 96becf0 + a1e6739
commit b04ae43
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/gb-argparse/src/main/java/com/g2forge/gearbox/argparse/BasicParameterParser.java b/gb-argparse/src/main/java/com/g2forge/gearbox/argparse/BasicParameterParser.java
@@ -3,6 +3,7 @@
 import java.nio.file.Paths;
 import java.util.ListIterator;
 
+import com.g2forge.alexandria.java.core.enums.HEnum;
 import com.g2forge.alexandria.java.fluent.optional.IOptional;
 import com.g2forge.alexandria.java.fluent.optional.NullableOptional;
 
@@ -43,5 +44,20 @@ public IOptional<Object> getDefault(IParameterInfo parameter) {
 		public Object parse(IParameterInfo parameter, ListIterator<String> argumentIterator) {
 			return argumentIterator.next();
 		}
+	},
+	ENUM {
+		@Override
+		public IOptional<Object> getDefault(IParameterInfo parameter) {
+			if (parameter.getSubject().bind(Parameter.class).isPresent()) return NullableOptional.of(null);
+			return NullableOptional.empty();
+		}
+
+		@SuppressWarnings("unchecked")
+		@Override
+		public Object parse(IParameterInfo parameter, ListIterator<String> argumentIterator) {
+			@SuppressWarnings("rawtypes")
+			final Class<? extends Enum> cast = (Class<? extends Enum>) parameter.getType();
+			return HEnum.valueOfInsensitive(cast, argumentIterator.next());
+		}
 	};
 }
diff --git a/gb-argparse/src/main/java/com/g2forge/gearbox/argparse/StandardParameterParserFactory.java b/gb-argparse/src/main/java/com/g2forge/gearbox/argparse/StandardParameterParserFactory.java
@@ -74,6 +74,7 @@ private IParameterParser computeParser(final Class<?> type) {
 		if (Path.class.equals(type)) return BasicParameterParser.PATH;
 		if (String.class.equals(type)) return BasicParameterParser.STRING;
 		if (Boolean.TYPE.equals(type) || Boolean.class.equals(type)) return BasicParameterParser.BOOLEAN;
+		if (Enum.class.isAssignableFrom(type)) return BasicParameterParser.ENUM;
 		if (type.isArray()) {
 			final Class<?> componentType = type.getComponentType();
 			final IParameterParser componentParser = computeParser(componentType);