io.getlime.security.powerauth.rest.api.spring.service.HttpCustomizationService Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of powerauth-restful-security-spring-annotation Show documentation

PowerAuth RESTful API Security Annotations for Spring

There is a newer version: 1.9.0

Show newest version

/*
 * PowerAuth Enrollment Server
 * Copyright (C) 2022 Wultra s.r.o.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see .
 */
package io.getlime.security.powerauth.rest.api.spring.service;

import io.getlime.security.powerauth.rest.api.spring.configuration.CorrelationHeaderConfiguration;
import org.slf4j.MDC;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;

import java.util.Collections;

/**
 * Service for configuration of HTTP headers in requests.
 *
 * @author Roman Strobl, [email protected]
 */
@Service
public class HttpCustomizationService {

    private static final MultiValueMap EMPTY_MULTI_MAP = new LinkedMultiValueMap<>();

    private CorrelationHeaderConfiguration correlationHeaderConfig;

    /**
     * Autowire correlation header configuration
     * @param correlationHeaderConfig Correlation header configuration.
     */
    @Autowired(required = false)
    public void setCorrelationHeaderConfig(CorrelationHeaderConfiguration correlationHeaderConfig) {
        this.correlationHeaderConfig = correlationHeaderConfig;
    }

    /**
     * Get HTTP headers.
     * @return HTTP headers.
     */
    public MultiValueMap getHttpHeaders() {
        if (correlationHeaderConfig == null) {
            // By default, no HTTP header customization is done
            return EMPTY_MULTI_MAP;
        }
        // The correlation header is added when the interceptor configuration is enabled
        final String correlationHeaderName = correlationHeaderConfig.getCorrelationHeaderName();
        final MultiValueMap headerMap = new LinkedMultiValueMap<>();
        headerMap.put(correlationHeaderName, Collections.singletonList(MDC.get(correlationHeaderName)));
        return headerMap;
    }

    /**
     * Get HTTP query parameters.
     * @return HTTP query parameters.
     */
    public MultiValueMap getQueryParams() {
        // By default, no HTTP query parameter customization is done
        return EMPTY_MULTI_MAP;
    }
}