package util;

import java.text.CharacterIterator;
import java.text.StringCharacterIterator;

public class XMLEscaper {
    public static final String[] ESCAPED_CHAR = {"&", "'", "\"", "<", ">"};

    public String escape(String aText) {
        if (aText == null) return null;

        StringBuilder result = new StringBuilder();
        StringCharacterIterator iterator = new StringCharacterIterator(aText);
        char character = iterator.current();
        while (character != CharacterIterator.DONE) {
            if (character == '<') {
                result.append("<");
            } else if (character == '>') {
                result.append(">");
            } else if (character == '&') {
                result.append("&");
            } else if (character == '\"') {
                result.append(""");
            } else if (character == '\'') {
                result.append("&apos;");
            } else {
                result.append(character);
            }
            character = iterator.next();
        }
        return result.toString();
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append("List of escaped characters: ");
        for (String escChar : ESCAPED_CHAR) {
            sb.append(escChar).append(" ");
        }
        return sb.toString();
    }
}