From d9184ad8716cd35e5dd67bef1ebf21a0db45b8e0 Mon Sep 17 00:00:00 2001
From: kohsuke <kohsuke@71c3de6d-444a-0410-be80-ed276b4c234a>
Date: Sat, 18 Nov 2006 17:52:04 +0000
Subject: [PATCH] added code to handle macro expansion.

git-svn-id: https://hudson.dev.java.net/svn/hudson/trunk/hudson/main@1166 71c3de6d-444a-0410-be80-ed276b4c234a
---
 core/src/main/java/hudson/Util.java     | 31 +++++++++++++++++++++++++
 core/src/test/java/hudson/UtilTest.java | 24 +++++++++++++++++++
 2 files changed, 55 insertions(+)
 create mode 100644 core/src/test/java/hudson/UtilTest.java

diff --git a/core/src/main/java/hudson/Util.java b/core/src/main/java/hudson/Util.java
index 20d271891b..e57d2c2032 100644
--- a/core/src/main/java/hudson/Util.java
+++ b/core/src/main/java/hudson/Util.java
@@ -17,6 +17,7 @@ import java.util.Map;
 import java.util.ResourceBundle;
 import java.util.StringTokenizer;
 import java.util.SimpleTimeZone;
+import java.util.Map.Entry;
 import java.util.logging.Logger;
 import java.util.logging.Level;
 import java.util.regex.Matcher;
@@ -31,6 +32,36 @@ import org.apache.tools.ant.BuildException;
  * @author Kohsuke Kawaguchi
  */
 public class Util {
+
+    /**
+     * Replaces the occurence of '$key' by <tt>properties.get('key')</tt>.
+     *
+     * <p>
+     * This is a rather naive implementation that causes somewhat unexpected
+     * behavior when the expansion contains other macros.
+     */
+    public static String replaceMacro(String s, Map<String,String> properties) {
+        int idx=0;
+        while((idx=s.indexOf('$',idx))>=0) {
+            // identify the key
+            int end=idx+1;
+            while(end<s.length()) {
+                char ch = s.charAt(end++);
+                if(!Character.isJavaIdentifierPart(ch))
+                    break;
+            }
+            String key = s.substring(idx+1,end);
+            String value = properties.get(key);
+            if(value==null) {
+                idx++;  // skip this '$' mark
+            } else {
+                s = s.substring(0,idx)+value+s.substring(end);
+            }
+        }
+        
+        return s;
+    }
+
     /**
      * Loads the contents of a file into a string.
      */
diff --git a/core/src/test/java/hudson/UtilTest.java b/core/src/test/java/hudson/UtilTest.java
new file mode 100644
index 0000000000..eab2f539e9
--- /dev/null
+++ b/core/src/test/java/hudson/UtilTest.java
@@ -0,0 +1,24 @@
+package hudson;
+
+import junit.framework.TestCase;
+
+import java.util.Map;
+import java.util.HashMap;
+
+/**
+ * @author Kohsuke Kawaguchi
+ */
+public class UtilTest extends TestCase {
+    public void testReplaceMacro() {
+        Map<String,String> m = new HashMap<String,String>();
+        m.put("A","a");
+        m.put("AA","aa");
+        m.put("B","B");
+
+        // longest match
+        assertEquals("aa",Util.replaceMacro("$AA",m));
+
+        // invalid keys are ignored
+        assertEquals("$AAB",Util.replaceMacro("$AAB",m));
+    }
+}