log: bannerResource can supply replacement map

rundeck · Nov 17, 2017 · 277d1b9 · 277d1b9
1 parent f808f36
commit 277d1b9
Showing 1 changed file with 23 additions and 1 deletion.
diff --git a/toolbelt/src/main/java/com/simplifyops/toolbelt/ToolBelt.java b/toolbelt/src/main/java/com/simplifyops/toolbelt/ToolBelt.java
@@ -8,6 +8,8 @@
 import java.nio.file.Files;
 import java.util.*;
 import java.util.function.Supplier;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 
 /**
@@ -221,13 +223,33 @@ public ToolBelt banner(String text) {
      * @return this
      */
     public ToolBelt bannerResource(String resource) {
+        return bannerResource(resource, null);
+    }
+
+    /**
+     * Display banner for top level help
+     *
+     * @param resource     resource path
+     * @param replacements a map of Regex->replacement, to replace values in the loaded resource
+     *
+     * @return this
+     */
+    public ToolBelt bannerResource(String resource, Map<String, String> replacements) {
         this.commands.banner = () -> {
             InputStream resourceAsStream = this.getClass().getClassLoader().getResourceAsStream(resource);
             if (null != resourceAsStream) {
                 try {
+                    String result;
                     try (BufferedReader is = new BufferedReader(new InputStreamReader(resourceAsStream))) {
-                        return is.lines().collect(Collectors.joining("\n"));
+                        result = is.lines().collect(Collectors.joining("\n"));
+                    }
+                    if (replacements != null && !replacements.isEmpty()) {
+                        for (String s : replacements.keySet()) {
+                            String val = replacements.get(s);
+                            result = result.replaceAll(s, Matcher.quoteReplacement(val));
+                        }
                     }
+                    return result;
                 } catch (IOException e) {
 
                 }