More path resolver functions

Kale-Ko · Apr 10, 2023 · c054415 · c054415
1 parent 8f72000
commit c054415
Show file tree

Hide file tree

Showing 2 changed files with 113 additions and 1 deletion.
diff --git a/gradle.properties b/gradle.properties
@@ -1,6 +1,6 @@
 java_version=17
 
-project_version=2.0.0
+project_version=2.1.0
 project_description=A semi-advanced Java configuration library with support for local files, mysql, and more.
 
 bjsl_version=1.4.3

diff --git a/src/main/java/io/github/kale_ko/ejcl/PathResolver.java b/src/main/java/io/github/kale_ko/ejcl/PathResolver.java
@@ -78,6 +78,52 @@ public static Object resolve(ParsedElement element, String path) {
         return null;
     }
 
+    /**
+     * Resolve an element on an element
+     *
+     * @param element
+     *        The element to resolve on
+     * @param path
+     *        The path to resolve to
+     * @return The value resolved
+     * @since 1.0.0
+     */
+    public static ParsedElement resolveElement(ParsedElement element, String path) {
+        String[] keys = path.replaceAll("\\[([0-9])\\]", ".[$1]").split("\\.");
+
+        ParsedElement resolved = element;
+
+        for (String key : keys) {
+            if (resolved.isObject()) {
+                if (resolved.asObject().has(key)) {
+                    resolved = resolved.asObject().get(key);
+                } else {
+                    resolved = null;
+                    break;
+                }
+            } else if (resolved.isArray()) {
+                if (key.startsWith("[") && key.endsWith("]")) {
+                    int index = Integer.parseInt(key.replaceAll("\\[([0-9])\\]", "$1"));
+
+                    if (index >= 0 && index < resolved.asArray().getSize()) {
+                        resolved = resolved.asArray().get(index);
+                    } else {
+                        resolved = null;
+                        break;
+                    }
+                } else {
+                    resolved = null;
+                    break;
+                }
+            } else {
+                resolved = null;
+                break;
+            }
+        }
+
+        return resolved;
+    }
+
     /**
      * Update a value on an element
      *
@@ -164,6 +210,72 @@ public static ParsedElement update(ParsedElement element, String path, Object va
         return element;
     }
 
+    /**
+     * Update an element on an element
+     *
+     * @param element
+     *        The element to update on
+     * @param path
+     *        The path to update
+     * @param value
+     *        The value to update to
+     * @return element for chaining
+     * @since 1.0.0
+     */
+    public static ParsedElement updateElement(ParsedElement element, String path, ParsedElement value) {
+        String[] keys = path.replaceAll("\\[([0-9])\\]", ".[$1]").split("\\.");
+        String valueKey = keys[keys.length - 1];
+        keys = Arrays.copyOf(keys, keys.length - 1);
+
+        ParsedElement resolved = element;
+
+        for (String key : keys) {
+            if (resolved.isObject()) {
+                if (resolved.asObject().has(key)) {
+                    resolved = resolved.asObject().get(key);
+                } else {
+                    resolved = null;
+                    break;
+                }
+            } else if (resolved.isArray()) {
+                if (key.startsWith("[") && key.endsWith("]")) {
+                    int index = Integer.parseInt(key.replaceAll("\\[([0-9])\\]", "$1"));
+
+                    if (index >= 0 && index < resolved.asArray().getSize()) {
+                        resolved = resolved.asArray().get(index);
+                    } else {
+                        resolved = null;
+                        break;
+                    }
+                } else {
+                    resolved = null;
+                    break;
+                }
+            } else {
+                resolved = null;
+                break;
+            }
+        }
+
+        if (resolved != null) {
+            if (resolved.isObject()) {
+                resolved.asObject().set(valueKey, value);
+            } else if (resolved.isArray()) {
+                if (valueKey.startsWith("[") && valueKey.endsWith("]")) {
+                    Integer resolvedValueKey = Integer.parseInt(valueKey.replaceAll("\\[([0-9])\\]", "$1"));
+
+                    if (resolvedValueKey >= 0 && resolvedValueKey < resolved.asArray().getSize()) {
+                        resolved.asArray().set(resolvedValueKey, value);
+                    } else {
+                        resolved.asArray().add(value);
+                    }
+                }
+            }
+        }
+
+        return element;
+    }
+
     /**
      * Get all the keys in an element
      *