package com.example.resourceserver.security;

import java.security.Principal;
import java.util.Collection;
import java.util.Map;
import java.util.Optional;
import java.util.Set;
import java.util.stream.Collectors;

import org.springframework.http.ResponseEntity;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.GrantedAuthority;
import org.springframework.security.core.annotation.AuthenticationPrincipal;
import org.springframework.security.oauth2.jwt.Jwt;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/api/security/users")
public class UserRestController {
    private UserService userService;

    public UserRestController(UserService userService) {
        this.userService = userService;
    }

    @GetMapping("/test")
    public String test(Principal principal) {
        return principal.toString();
    }

    @GetMapping("/test-username")
    public String testName(Principal principal) {
        return principal.getName();
    }

    // Get roles using the Authentication object
    @GetMapping("/test-roles-auth")
    public Collection<String> getRoles(Authentication authentication) {
        return authentication.getAuthorities().stream()
                .map(GrantedAuthority::getAuthority)
                .collect(Collectors.toList());
    }

    // Inspect raw JWT claims directly
    @GetMapping("/test-roles-jwt")
    public Map<String, Object> getClaims(@AuthenticationPrincipal Jwt jwt) {
        return jwt.getClaims(); // Extract custom JSON fields containing roles
    }

    /* Not very secure because unverifiable headers can be introduced by unwanted parties */
    @GetMapping("/test-roles-headers")
    public String getRolesFromHeaders(@RequestHeader("X-User-Name") String username,
            @RequestHeader("X-User-Roles") String roles) {
        return "Roles: " + roles + ", username: " + username ;
    }
}