Blame view

src/main/java/etunicorn/controller/RoleController.java 4.14 KB
2f1f76a9   Geoffrey PREUD'HOMME   Mise sous packages
1
  package etunicorn.controller;
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
2
3
  
  
2f1f76a9   Geoffrey PREUD'HOMME   Mise sous packages
4
5
6
  import etunicorn.RestrictedTo;
  import etunicorn.entity.Permission;
  import etunicorn.entity.Role;
5596f9d9   Geoffrey PREUD'HOMME   Début de l'implém...
7
8
  import etunicorn.generated.model.UpdateRoleByIdRequest;
  import etunicorn.generated.model.UpdateRoleRequest;
2f1f76a9   Geoffrey PREUD'HOMME   Mise sous packages
9
10
  import etunicorn.repository.PermissionRepository;
  import etunicorn.repository.RoleRepository;
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
11
12
13
14
15
  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;
5596f9d9   Geoffrey PREUD'HOMME   Début de l'implém...
16
  import org.springframework.web.bind.annotation.RequestBody;
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
17
18
  import org.springframework.web.bind.annotation.RestController;
  
5596f9d9   Geoffrey PREUD'HOMME   Début de l'implém...
19
  import javax.validation.Valid;
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
20
21
22
  import java.util.List;
  
  /**
3064f583   Geoffrey PREUD'HOMME   Copyright
23
24
25
   * etunicorn-server
   * Copyright © 2017 Le Club Info Polytech Lille
   * Tous droits réservés
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
26
27
   */
  @RestController
8f35fffd   Geoffrey PREUD'HOMME   Ajout de la sécurité
28
  public class RoleController extends BaseController implements etunicorn.generated.RoleController {
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
29
30
31
32
33
34
35
36
37
38
39
40
      @Autowired
      private RoleRepository roleRepository;
  
      @Autowired
      private PermissionRepository permissionRepository;
  
      @Override
      public ResponseEntity<?> getRole() {
          return new ResponseEntity<List>((List) roleRepository.findAll(), HttpStatus.OK);
      }
  
      @Override
8f35fffd   Geoffrey PREUD'HOMME   Ajout de la sécurité
41
      @RestrictedTo("ROLE_ADD")
5596f9d9   Geoffrey PREUD'HOMME   Début de l'implém...
42
      public ResponseEntity<?> updateRole(@Valid @RequestBody UpdateRoleRequest updateRoleRequest) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
43
          Role oldRole = roleRepository.findByNom(updateRoleRequest.getNom());
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
44
          if (oldRole != null) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
45
              return generateError(HttpStatus.CONFLICT, "Un rôle avec le même nom existe déjà");
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
46
47
          }
          Role role = new Role();
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
48
          role.setNom(updateRoleRequest.getNom());
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
49
50
51
          try {
              roleRepository.save(role);
          } catch (DataIntegrityViolationException e) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
52
              return generateError(HttpStatus.CONFLICT, "Un rôle avec le même nom existe déjà");
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
53
54
55
56
57
          }
          return new ResponseEntity<Object>(role, HttpStatus.CREATED);
      }
  
      @Override
8f35fffd   Geoffrey PREUD'HOMME   Ajout de la sécurité
58
      @RestrictedTo("ROLE_DELETE")
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
59
60
61
      public ResponseEntity<?> deleteRoleById(@PathVariable String nomRole) {
          Role role = roleRepository.findByNom(nomRole);
          if (role == null) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
62
              return generateError(HttpStatus.NOT_FOUND, "Rôle introuvable");
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
63
64
65
66
67
68
          }
          roleRepository.delete(role);
          return new ResponseEntity<Object>(HttpStatus.NO_CONTENT);
      }
  
      @Override
8f35fffd   Geoffrey PREUD'HOMME   Ajout de la sécurité
69
      @RestrictedTo("ROLE_PERMISSION_ADD")
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
70
      public ResponseEntity<?> updateRoleById(@PathVariable String nomRole, @Valid @RequestBody UpdateRoleByIdRequest updateRoleByIdRequest) {
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
71
72
          Role role = roleRepository.findByNom(nomRole);
          if (role == null) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
73
              return generateError(HttpStatus.NOT_FOUND, "Rôle introuvable");
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
74
          }
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
75
          Permission permission = permissionRepository.findByNom(updateRoleByIdRequest.getNom());
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
76
          if (permission == null) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
77
              return generateError(HttpStatus.NOT_FOUND, "Permission introuvable");
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
78
79
80
81
82
83
84
85
86
87
88
          }
          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
8f35fffd   Geoffrey PREUD'HOMME   Ajout de la sécurité
89
      @RestrictedTo("ROLE_PERMISSION_REMOVE")
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
90
91
92
      public ResponseEntity<?> deleteRoleByNomPermission(@PathVariable String nomPermission, @PathVariable String nomRole) {
          Role role = roleRepository.findByNom(nomRole);
          if (role == null) {
474776a8   Geoffrey PREUD'HOMME   Implémentation JS...
93
              return generateError(HttpStatus.NOT_FOUND, "Rôle introuvable");
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
94
95
96
97
98
99
100
101
102
103
104
          }
          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
8f35fffd   Geoffrey PREUD'HOMME   Ajout de la sécurité
105
      @RestrictedTo("ROLE_PERMISSION_LIST")
8f05ee77   Geoffrey PREUD'HOMME   API /role et amél...
106
107
108
109
      public ResponseEntity<?> getPermission() {
          return new ResponseEntity<List>((List) permissionRepository.findAll(), HttpStatus.OK);
      }
  }