/*
 *  Copyright 2010 Hippo.
 * 
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 * 
 *       http://www.apache.org/licenses/LICENSE-2.0
 * 
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.hippoecm.frontend.editor.plugins;

import java.lang.StringBuffer;
import java.util.regex.Pattern;
import java.util.regex.Matcher;

import org.apache.wicket.model.IModel;
import org.apache.wicket.util.string.Strings;

public class EncodedStringModel implements IModel<String> {
    private static final long serialVersionUID = 1L;

    private static final Pattern DECODE_PATTERN = Pattern.compile("(&amp;|&#039;|&quot;|&lt;|&gt;)");

    private IModel<String> decorated;

    public EncodedStringModel(IModel<String> decorated) {
        this.decorated = decorated;
    }

    public String getObject() {
        String value = decorated.getObject();
        if (value != null) {
            StringBuffer sb = new StringBuffer();
            Matcher matcher = DECODE_PATTERN.matcher(value);
            while (matcher.find()) {
                String match = matcher.group();
                if ("&amp;".equals(match)) {
                    matcher.appendReplacement(sb, "&");
                } else if ("&#039;".equals(match)) {
                    matcher.appendReplacement(sb, "'");
                } else if ("&quot;".equals(match)) {
                    matcher.appendReplacement(sb, "\"");
                } else if ("&lt;".equals(match)) {
                    matcher.appendReplacement(sb, "<");
                } else if ("&gt;".equals(match)) {
                    matcher.appendReplacement(sb, ">");
                }
            }
            matcher.appendTail(sb);
            return sb.toString();
        }
        return null;
    }

    public void setObject(String object) {
        CharSequence escaped = Strings.escapeMarkup(object);
        if (escaped != null) {
            decorated.setObject(escaped.toString());
        } else {
            decorated.setObject(null);
        }
    }

    public void detach() {
        decorated.detach();
    }

}