D-Net

package eu.dnetlib.repo.manager.controllers;

import com.google.gson.JsonArray;

import com.google.gson.JsonElement;

import eu.dnetlib.repo.manager.domain.dto.Role;

import eu.dnetlib.repo.manager.service.aai.registry.AaiRegistryService;

import eu.dnetlib.repo.manager.service.security.RoleMappingService;

import eu.dnetlib.repo.manager.service.security.AuthoritiesMapper;

import eu.dnetlib.repo.manager.service.security.AuthoritiesUpdater;

import eu.dnetlib.repo.manager.utils.JsonUtils;

import io.swagger.annotations.Api;

import io.swagger.annotations.ApiOperation;

import org.mitre.openid.connect.model.OIDCAuthenticationToken;

import org.mitre.openid.connect.model.UserInfo;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.http.HttpStatus;

import org.springframework.http.ResponseEntity;

import org.springframework.security.access.prepost.PreAuthorize;

import org.springframework.security.core.context.SecurityContextHolder;

import org.springframework.web.bind.annotation.*;

import javax.ws.rs.core.MediaType;

import javax.ws.rs.core.Response;

import java.util.ArrayList;

import java.util.Collection;

import java.util.List;

@RestController

@RequestMapping(value = "/role-management")

@Api(description = "Role Management", value = "role-management")

public class UserRoleController {

    private final AaiRegistryService aaiRegistryService;

    private final AuthoritiesUpdater authoritiesUpdater;

    private final RoleMappingService roleMappingService;

    @Autowired

    UserRoleController(AaiRegistryService aaiRegistryService,

                       AuthoritiesUpdater authoritiesUpdater,

                       RoleMappingService roleMappingService) {

        this.aaiRegistryService = aaiRegistryService;

        this.authoritiesUpdater = authoritiesUpdater;

        this.roleMappingService = roleMappingService;

    }

    /**

     * Get the role with the given id.

     **/

    @RequestMapping(method = RequestMethod.GET, path = "/role/{id}")

//    @PreAuthorize("hasAnyAuthority('REGISTERED_USER', 'SUPER_ADMINISTRATOR', 'CONTENT_PROVIDER_DASHBOARD_ADMINISTRATOR')")

    public Response getRole(@RequestParam(value = "type", defaultValue = "datasource") String type, @PathVariable("id") String id) {

        int roleId = aaiRegistryService.getCouId(type, id);

        return Response.status(HttpStatus.OK.value()).entity(JsonUtils.createResponse("Role id is: " + roleId).toString()).type(MediaType.APPLICATION_JSON).build();

    }

    /**

     * Create a new role with the given name and description.

     **/

    @RequestMapping(method = RequestMethod.POST, path = "/role")

    @PreAuthorize("hasAnyAuthority('SUPER_ADMINISTRATOR')")

    public Response createRole(@RequestBody Role role) {

        aaiRegistryService.createRole(role);

        return Response.status(HttpStatus.OK.value()).entity(JsonUtils.createResponse("Role has been created").toString()).type(MediaType.APPLICATION_JSON).build();

    }

    /**

     * Subscribe to a type(Community, etc.) with id(ee, egi, etc.)

     */

    @ApiOperation(value = "subscribe")

    @RequestMapping(method = RequestMethod.POST, path = "/subscribe/{type}/{id}")

    @PreAuthorize("hasAnyAuthority('SUPER_ADMINISTRATOR', 'CONTENT_PROVIDER_DASHBOARD_ADMINISTRATOR')")

    public Response subscribe(@PathVariable("type") String type, @PathVariable("id") String id) {

        Integer coPersonId = aaiRegistryService.getCoPersonIdByIdentifier();

        if (coPersonId == null) {

            coPersonId = aaiRegistryService.getCoPersonIdByEmail();

        }

        Integer couId = aaiRegistryService.getCouId(type, id);

        if (couId != null) {

            Integer role = aaiRegistryService.getRoleId(coPersonId, couId);

            aaiRegistryService.assignMemberRole(coPersonId, couId, role);

            // Add role to current authorities

            authoritiesUpdater.addRole(roleMappingService.convertRepoIdToAuthority(id));

            return Response.status(HttpStatus.OK.value()).entity(JsonUtils.createResponse("Role has been assigned").toString()).type(MediaType.APPLICATION_JSON).build();

        } else {

            return Response.status(HttpStatus.NOT_FOUND.value()).entity(JsonUtils.createResponse("Role has not been found").toString()).type(MediaType.APPLICATION_JSON).build();

        }

    }

    /////////////////////////////////////////////////////////////////////////////////////////////

    /////////////////////////////////////////////////////////////////////////////////////////////

    @RequestMapping(method = RequestMethod.GET, path = "/users/couid/{id}")

    @PreAuthorize("hasAnyAuthority('SUPER_ADMINISTRATOR', 'CONTENT_PROVIDER_DASHBOARD_ADMINISTRATOR')")

    public ResponseEntity<String> getUsersByCouId(@PathVariable("id") Integer id) {

//        calls.getUserByCoId()

        return ResponseEntity.ok(aaiRegistryService.getUsersByCouId(id).toString());

    }

    @RequestMapping(method = RequestMethod.GET, path = "/users/{email}/roles")

    @PreAuthorize("hasAnyAuthority('SUPER_ADMINISTRATOR', 'CONTENT_PROVIDER_DASHBOARD_ADMINISTRATOR') or hasAuthority('REGISTERED_USER') and authentication.userInfo.email==#email")

    public ResponseEntity<Collection<String>> getRolesByEmail(@PathVariable("email") String email) {

        int coPersonId = aaiRegistryService.getCoPersonIdByEmail(email);

        List<Integer> list = new ArrayList<>();

        for (JsonElement element : aaiRegistryService.getRolesWithStatus(coPersonId, AaiRegistryService.RoleStatus.ACTIVE)) {

            list.add(element.getAsJsonObject().get("CouId").getAsInt());

        }

        return ResponseEntity.ok(aaiRegistryService.getCouNames(list).values());

    }

    @RequestMapping(method = RequestMethod.GET, path = "/user/roles/my")

    @PreAuthorize("hasAuthority('REGISTERED_USER')")

    public ResponseEntity<Collection<String>> getRoleNames() {

        List<String> roles;

        JsonArray entitlements = null;

        UserInfo userInfo = ((OIDCAuthenticationToken) SecurityContextHolder.getContext().getAuthentication()).getUserInfo();

        if (userInfo.getSource().getAsJsonArray("edu_person_entitlements") != null) {

            entitlements = userInfo.getSource().getAsJsonArray("edu_person_entitlements");

        } else if (userInfo.getSource().getAsJsonArray("eduperson_entitlement") != null) {

            entitlements = userInfo.getSource().getAsJsonArray("eduperson_entitlement");

        } else {

            return ResponseEntity.ok(null);

        }

        roles = AuthoritiesMapper.entitlementRoles(entitlements);

        return ResponseEntity.ok(roles);

    }

}