org.thymeleaf.standard.expression.OGNLExpressionObjectsWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of thymeleaf Show documentation
Modern server-side Java template engine for both web and standalone environments
There is a newer version: 3.1.2.RELEASE
/*
 * =============================================================================
 *
 *   Copyright (c) 2011-2016, The THYMELEAF team (http://www.thymeleaf.org)
 *
 *   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.thymeleaf.standard.expression;

import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.Set;

import org.thymeleaf.expression.IExpressionObjects;

/**
 *
 * @author Daniel Fernández
 *
 * @since 3.0.0
 *
 */
final class OGNLExpressionObjectsWrapper extends HashMap {

    private final IExpressionObjects expressionObjects;
    private final boolean restrictedExpressionExecution;


    OGNLExpressionObjectsWrapper(final IExpressionObjects expressionObjects, final boolean restrictedExpressionExecution) {
        super(5);
        this.expressionObjects = expressionObjects;
        this.restrictedExpressionExecution = restrictedExpressionExecution;
    }


    @Override
    public int size() {
        return super.size() + this.expressionObjects.size();
    }

    @Override
    public boolean isEmpty() {
        return this.expressionObjects.size() == 0 && super.isEmpty();
    }

    @Override
    public Object get(final Object key) {

        if (this.expressionObjects.containsObject(key.toString())) {

            final Object expressionObject = this.expressionObjects.getObject(key.toString());

            // We need to first check if we are in a restricted environment. If so, restrict access to the request.
            if (this.restrictedExpressionExecution &&
                    (StandardExpressionObjectFactory.REQUEST_EXPRESSION_OBJECT_NAME.equals(key) ||
                     StandardExpressionObjectFactory.HTTP_SERVLET_REQUEST_EXPRESSION_OBJECT_NAME.equals(key))) {
                return RestrictedRequestAccessUtils.wrapRequestObject(expressionObject);
            }

            return expressionObject;
        }

        return super.get(key);

    }

    @Override
    public boolean containsKey(final Object key) {
        return this.expressionObjects.containsObject(key.toString()) || super.containsKey(key);
    }

    @Override
    public Object put(final String key, final Object value) {
        if (this.expressionObjects.containsObject(key.toString())) {
            throw new IllegalArgumentException(
                    "Cannot put entry with key \"" + key + "\" into Expression Objects wrapper map: key matches the " +
                    "name of one of the expression objects");
        }
        return super.put(key, value);
    }

    @Override
    public void putAll(final Map m) {
        // This will call put, and therefore perform the key name check
        super.putAll(m);
    }

    @Override
    public Object remove(final Object key) {
        if (this.expressionObjects.containsObject(key.toString())) {
            throw new IllegalArgumentException(
                    "Cannot remove entry with key \"" + key + "\" from Expression Objects wrapper map: key matches the " +
                    "name of one of the expression objects");
        }
        return super.remove(key);
    }

    @Override
    public void clear() {
        throw new UnsupportedOperationException("Cannot clear Expression Objects wrapper map");
    }

    @Override
    public boolean containsValue(final Object value) {
        throw new UnsupportedOperationException("Cannot perform by-value search on Expression Objects wrapper map");
    }

    @Override
    public Object clone() {
        throw new UnsupportedOperationException("Cannot clone Expression Objects wrapper map");
    }

    @Override
    public Set keySet() {
        if (super.isEmpty()) {
            return this.expressionObjects.getObjectNames();
        }
        final Set keys = new LinkedHashSet(this.expressionObjects.getObjectNames());
        keys.addAll(super.keySet());
        return keys;
    }

    @Override
    public Collection