001/*
002 * (C) Copyright 2015 Nuxeo SA (http://nuxeo.com/) and contributors.
003 *
004 * All rights reserved. This program and the accompanying materials
005 * are made available under the terms of the GNU Lesser General Public License
006 * (LGPL) version 2.1 which accompanies this distribution, and is available at
007 * http://www.gnu.org/licenses/lgpl-2.1.html
008 *
009 * This library is distributed in the hope that it will be useful,
010 * but WITHOUT ANY WARRANTY; without even the implied warranty of
011 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
012 * Lesser General Public License for more details.
013 *
014 * Contributors:
015 *     Andre Justo
016 */
017package org.nuxeo.ecm.platform.ui.web.component.message;
018
019import com.sun.faces.renderkit.html_basic.MessageRenderer;
020
021import javax.faces.component.UIComponent;
022import javax.faces.context.FacesContext;
023import javax.faces.context.ResponseWriter;
024import javax.faces.context.ResponseWriterWrapper;
025import javax.faces.render.FacesRenderer;
026import java.io.IOException;
027
028/**
029 * Custom MessageRenderer that allows the use of HTML in <h:message>
030 *
031 * @since 7.3
032 */
033@FacesRenderer(componentFamily="javax.faces.Message", rendererType="javax.faces.Message")
034public class EscapableMessageRenderer extends MessageRenderer {
035
036    @Override
037    public void encodeEnd(FacesContext context, UIComponent component) throws IOException {
038        final ResponseWriter originalResponseWriter = context.getResponseWriter();
039
040        try {
041            context.setResponseWriter(new ResponseWriterWrapper() {
042
043                @Override
044                public ResponseWriter getWrapped() {
045                    return originalResponseWriter;
046                }
047
048                @Override
049                public void writeText(Object text, UIComponent component, String property) throws IOException {
050                    String string = String.valueOf(text);
051                    String escape = (String) component.getAttributes().get("escape");
052                    if (escape != null && !Boolean.valueOf(escape)) {
053                        super.write(string);
054                    } else {
055                        super.writeText(string, component, property);
056                    }
057                }
058            });
059
060            super.encodeEnd(context, component); // Now, render it!
061        } finally {
062            context.setResponseWriter(originalResponseWriter); // Restore original writer.
063        }
064    }
065}