Battle-Cattle/BCUK-Bot-3

package com.expiredminotaur.bcukbot.web.security;

import org.springframework.http.HttpHeaders;
import org.springframework.http.RequestEntity;

public class OAuth2UserAgentUtils
{
    private static final String DISCORD_BOT_USER_AGENT = "BCUK Bot";

    static RequestEntity<?> withUserAgent(RequestEntity<?> request)
    {
        HttpHeaders headers = new HttpHeaders();
        headers.putAll(request.getHeaders());
        headers.add(HttpHeaders.USER_AGENT, DISCORD_BOT_USER_AGENT);

        return new RequestEntity<>(request.getBody(), headers, request.getMethod(), request.getUrl());
    }
}