org.ssssssss.magicapi.utils.WebUtils Maven / Gradle / Ivy

Go to download
package org.ssssssss.magicapi.utils;

import org.springframework.web.context.request.RequestAttributes;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import org.ssssssss.magicapi.interceptor.MagicUser;
import org.ssssssss.magicapi.model.Constants;

import javax.servlet.http.HttpServletRequest;
import java.security.Principal;
import java.util.Optional;

public class WebUtils {

	public static Optional getRequest(){
		RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
		if (requestAttributes instanceof ServletRequestAttributes) {
			return Optional.of(((ServletRequestAttributes) requestAttributes).getRequest());
		}
		return Optional.empty();
	}

	public static String currentUserName(){
		Optional request = getRequest();
		return request.map(r -> (MagicUser)r.getAttribute(Constants.ATTRIBUTE_MAGIC_USER))
				.map(MagicUser::getUsername)
				.orElseGet(() -> request.map(HttpServletRequest::getUserPrincipal)
						.map(Principal::getName)
						.orElse(null)
				);
	}
}