/* * Copyright (C) 2008-2020, Juick * * 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 <http://www.gnu.org/licenses/>. */ package com.juick.www; import com.juick.model.AnonymousUser; import com.juick.model.User; import com.juick.service.UserService; import com.juick.service.security.entities.JuickUser; import org.springframework.security.core.Authentication; import org.springframework.security.core.annotation.AuthenticationPrincipal; import org.springframework.security.oauth2.jwt.Jwt; import org.springframework.security.oauth2.server.resource.authentication.JwtAuthenticationToken; import org.springframework.ui.Model; import org.springframework.web.bind.annotation.ControllerAdvice; import org.springframework.web.bind.annotation.ModelAttribute; import org.springframework.web.servlet.support.ServletUriComponentsBuilder; import jakarta.servlet.http.HttpServletResponse; import javax.inject.Inject; @ControllerAdvice public class SiteAttributesHandler { @Inject private UserService userService; @ModelAttribute public void setVaryResponseHeader(HttpServletResponse response) { response.setHeader("Vary", "Accept-Language"); } @ModelAttribute public void setReturnPathAttribute(Model model) { model.addAttribute("retpath", ServletUriComponentsBuilder.fromCurrentRequestUri().toUriString()); } @ModelAttribute public User visitor(Model model, @AuthenticationPrincipal Object principal) { if (principal != null) { if (principal instanceof JuickUser) { return ((JuickUser) principal).getUser(); } if (principal instanceof Jwt) { var uname = (String) ((Jwt) principal).getClaims().get("sub"); return userService.getUserByName(uname); } } return AnonymousUser.INSTANCE; } }