adding try, flatten methods

bertramdev · Oct 3, 2024 · d76b5dc · d76b5dc
1 parent 448e2cd
commit d76b5dc
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 1 deletion.
diff --git a/build.gradle b/build.gradle
@@ -30,7 +30,7 @@ ext {
 
 
 group = 'com.bertramlabs.plugins'
-version = '0.9.2'
+version = '0.9.3'
 
 ext.isReleaseVersion = !version.endsWith("SNAPSHOT")
 sourceCompatibility = "1.8"

diff --git a/src/main/java/com/bertramlabs/plugins/hcl4j/HCLBaseFunctions.java b/src/main/java/com/bertramlabs/plugins/hcl4j/HCLBaseFunctions.java
@@ -204,6 +204,15 @@ static void registerBaseFunctions(HCLParser parser) {
             return java.util.UUID.randomUUID().toString();
         });
 
+        parser.registerFunction("try", (arguments) -> {
+            for(Object argument : arguments) {
+                if(argument != null) {
+                    return argument;
+                }
+            }
+            return null;
+        });
+
         registerNumericFunctions(parser);
         registerCollectionFunctions(parser);
         registerDateFunctions(parser);
@@ -417,6 +426,33 @@ static void registerCollectionFunctions(HCLParser parser) {
             return null;
         });
 
+        parser.registerFunction("flatten", (arguments) -> {
+            if(arguments.size() > 0) {
+                if(arguments.get(0) instanceof List) {
+                    List<Object> elements = ((List<Object>)(arguments.get(0)));
+                    ArrayList<Object> flattened = new ArrayList<>();
+                    flattenList(flattened,elements);
+
+                    return flattened;
+                } else {
+                    return null;
+                }
+            }
+            return null;
+        });
+
+
+    }
+
+    private static void flattenList(ArrayList<Object> flattened, List<Object> elements) {
+        for(Object element : elements) {
+            if(element instanceof List) {
+                List<Object> subElements = ((List<Object>)(element));
+                flattenList(flattened,subElements);
+            } else {
+                flattened.add(element);
+            }
+        }
 
     }