introduce replaceChars(String, String, String)

Author: rbri

src/main/java/org/htmlunit/util/StringUtils.java

@@ -232,6 +232,60 @@ public static boolean containsIgnoreCase(final String s, final String expected)
         return false;
     }
 
+    /**
+     * Replaces multiple characters in a String in one go.
+     * This method can also be used to delete characters.
+     *
+     * @param str          String to replace characters in, may be null.
+     * @param searchChars  a set of characters to search for, may be null.
+     * @param replaceChars a set of characters to replace, may be null.
+     * @return modified String, or the input string if no replace was done.
+     */
+    public static String replaceChars(final String str, final String searchChars, String replaceChars) {
+        if (isEmptyOrNull(str) || isEmptyOrNull(searchChars)) {
+            return str;
+        }
+
+        replaceChars = replaceChars == null ? "" : replaceChars;
+        final int replaceCharsLength = replaceChars.length();
+        final int strLength = str.length();
+
+        StringBuilder buf = null;
+        int i = 0;
+        for ( ; i < strLength; i++) {
+            final char ch = str.charAt(i);
+            final int index = searchChars.indexOf(ch);
+            if (index >= 0) {
+                buf = new StringBuilder(strLength);
+                buf.append(str, 0, i);
+                if (index < replaceCharsLength) {
+                    buf.append(replaceChars.charAt(index));
+                }
+                break;
+            }
+        }
+
+        if (buf == null) {
+            return str;
+        }
+
+        i++;
+        for ( ; i < strLength; i++) {
+            final char ch = str.charAt(i);
+            final int index = searchChars.indexOf(ch);
+            if (index >= 0) {
+                if (index < replaceCharsLength) {
+                    buf.append(replaceChars.charAt(index));
+                }
+            }
+            else {
+                buf.append(ch);
+            }
+        }
+
+        return buf.toString();
+    }
+
     /**
      * Escapes the characters '&lt;', '&gt;' and '&amp;' into their XML entity equivalents.
      *

src/test/java/org/htmlunit/util/StringUtilsTest.java

@@ -334,6 +334,31 @@ public void stringToByteArray() throws Exception {
         assertEquals(104, result[0]);
     }
 
+    /**
+     * Test for method {@link StringUtils#replaceChars(String, String, String)}.
+     */
+    @Test
+    public void replaceChars() {
+        assertEquals(null, StringUtils.replaceChars(null, "", ""));
+        assertEquals("", StringUtils.replaceChars("", "", null));
+        assertEquals("abc", StringUtils.replaceChars("abc", null, null));
+        assertEquals("abc", StringUtils.replaceChars("abc", "", ""));
+        assertEquals("ac", StringUtils.replaceChars("abc", "b", null));
+        assertEquals("ac", StringUtils.replaceChars("abc", "b", ""));
+        assertEquals("ayzya", StringUtils.replaceChars("abcba", "bc", "yz"));
+        assertEquals("ayya", StringUtils.replaceChars("abcba", "bc", "y"));
+        assertEquals("ayzya", StringUtils.replaceChars("abcba", "bc", "yzx"));
+
+        assertEquals("abc", StringUtils.replaceChars("abc", "d", "e"));
+
+        assertEquals("ebc", StringUtils.replaceChars("abc", "a", "e"));
+        assertEquals("bc", StringUtils.replaceChars("abc", "a", ""));
+
+        assertEquals("abe", StringUtils.replaceChars("abc", "c", "e"));
+        assertEquals("ab", StringUtils.replaceChars("abc", "c", null));
+    }
+
+
     /**
      * @throws Exception if the test fails
      */