package etunicorn.controller;


import etunicorn.RestrictedTo;
import etunicorn.entity.Permission;
import etunicorn.entity.Role;
import etunicorn.generated.model.UpdateRoleByIdRequest;
import etunicorn.generated.model.UpdateRoleRequest;
import etunicorn.repository.PermissionRepository;
import etunicorn.repository.RoleRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataIntegrityViolationException;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RestController;

import javax.validation.Valid;
import java.util.List;

/**
 * etunicorn-server
 * Copyright © 2017 Le Club Info Polytech Lille
 * Tous droits réservés
 */
@RestController
public class RoleController extends BaseController implements etunicorn.generated.RoleController {
    @Autowired
    private RoleRepository roleRepository;

    @Autowired
    private PermissionRepository permissionRepository;

    @Override
    public ResponseEntity<?> getRole() {
        return new ResponseEntity<List>((List) roleRepository.findAll(), HttpStatus.OK);
    }

    @Override
    @RestrictedTo("ROLE_ADD")
    public ResponseEntity<?> updateRole(@Valid @RequestBody UpdateRoleRequest updateRoleRequest) {
        Role oldRole = roleRepository.findByNom(updateRoleRequest.getNom());
        Role role = new Role();
        role.setNom(updateRoleRequest.getNom());
        try {
            roleRepository.save(role);
        } catch (DataIntegrityViolationException e) {
            return generateError(HttpStatus.CONFLICT, "Un rôle avec le même nom existe déjà");
        }
        return new ResponseEntity<Object>(role, HttpStatus.CREATED);
    }

    @Override
    @RestrictedTo("ROLE_DELETE")
    public ResponseEntity<?> deleteRoleById(@PathVariable String nomRole) {
        Role role = roleRepository.findByNom(nomRole);
        if (role == null) {
            return generateError(HttpStatus.NOT_FOUND, "Rôle introuvable");
        }
        roleRepository.delete(role);
        return new ResponseEntity<Object>(HttpStatus.NO_CONTENT);
    }

    @Override
    @RestrictedTo("ROLE_PERMISSION_ADD")
    public ResponseEntity<?> updateRoleById(@PathVariable String nomRole, @Valid @RequestBody UpdateRoleByIdRequest updateRoleByIdRequest) {
        Role role = roleRepository.findByNom(nomRole);
        if (role == null) {
            return generateError(HttpStatus.NOT_FOUND, "Rôle introuvable");
        }
        Permission permission = permissionRepository.findByNom(updateRoleByIdRequest.getNom());
        if (permission == null) {
            return generateError(HttpStatus.NOT_FOUND, "Permission introuvable");
        }
        role.addPermission(permission);
        try {
            roleRepository.save(role);
        } catch (DataIntegrityViolationException e) {
            // Si la permission était déjà là, on fait rien
        }
        return new ResponseEntity<Object>(role, HttpStatus.ACCEPTED);
    }

    @Override
    @RestrictedTo("ROLE_PERMISSION_REMOVE")
    public ResponseEntity<?> deleteRoleByNomPermission(@PathVariable String nomPermission, @PathVariable String nomRole) {
        Role role = roleRepository.findByNom(nomRole);
        if (role == null) {
            return generateError(HttpStatus.NOT_FOUND, "Rôle introuvable");
        }
        Permission permission = permissionRepository.findByNom(nomPermission);
        if (permission == null) {
            return new ResponseEntity<Object>("Permission inconnue", HttpStatus.NOT_FOUND);
        }
        role.delPermission(permission);
        roleRepository.save(role);
        return new ResponseEntity<Object>(role, HttpStatus.ACCEPTED);
    }

    @Override
    @RestrictedTo("ROLE_PERMISSION_LIST")
    public ResponseEntity<?> getPermission() {
        return new ResponseEntity<List>((List) permissionRepository.findAll(), HttpStatus.OK);
    }
}