org.restheart.handlers.QueryStringRebuilder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of restheart-commons Show documentation
RESTHeart Commons - Common classes for core components and plugins.
There is a newer version: 8.1.4
/*-
 * ========================LICENSE_START=================================
 * restheart-commons
 * %%
 * Copyright (C) 2019 - 2024 SoftInstigate
 * %%
 * 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.
 * =========================LICENSE_END==================================
 */
package org.restheart.handlers;

import io.undertow.server.HttpServerExchange;
import io.undertow.util.AttachmentKey;
import io.undertow.util.QueryParameterUtils;
import java.net.URLEncoder;
import java.util.ArrayDeque;
import java.util.Deque;
import java.util.Map;
import java.util.TreeMap;

/**
 *
 * rebuild the query string from the exchange.getQueryParameters() that might
 * have been updated by request interceptors. it also encodes values
 *
 * @author Andrea Di Cesare {@literal }
 *
 */
public class QueryStringRebuilder extends PipelinedHandler {

    static final AttachmentKey ORIGINAL_QUERY_STRING = AttachmentKey.create(String.class);

    /**
     * Creates a new instance of QueryStringRebuiler
     *
     * @param next
     */
    public QueryStringRebuilder(PipelinedHandler next) {
        super(next);
    }

    /**
     * Creates a new instance of QueryStringRebuiler
     *
     */
    public QueryStringRebuilder() {
        super(null);
    }

    /**
     *
     * @param exchange
     * @throws Exception
     */
    @Override
    public void handleRequest(HttpServerExchange exchange) throws Exception {
        // save the original request URI
        setOriginalQueryString(exchange);

        Map> qps = exchange.getQueryParameters();

        var decodedQueryParameters = new TreeMap>();

        final var encoding = QueryParameterUtils.getQueryParamEncoding(exchange);

        for (var k : qps.keySet()) {
            var values = qps.get(k);

            var nvalues = new ArrayDeque(values.size());

            for (var value : values) {
                nvalues.add(URLEncoder.encode(value, encoding));
            }

            decodedQueryParameters.put(k, nvalues);
        }

        var newqs = QueryParameterUtils.buildQueryString(decodedQueryParameters);

        exchange.setQueryString(newqs);

        next(exchange);
    }

    private void setOriginalQueryString(HttpServerExchange exchange) {
        if (exchange.getAttachment(ORIGINAL_QUERY_STRING) == null) {
            exchange.putAttachment(ORIGINAL_QUERY_STRING,
                    exchange.getQueryString());
        }
    }

    public static String getOriginalQueryString(HttpServerExchange exchange) {
        var oqs = exchange.getAttachment(ORIGINAL_QUERY_STRING);

        return oqs == null ? exchange.getQueryString() : oqs;
    }
}