fixed some servlets

Author: knysh

src/main/java/main/controllers/AuditController.java

@@ -46,7 +46,15 @@ public AuditController(UserDto user) {
     @Override
     public List<AuditDto> get(AuditDto searchTemplate) throws AqualityException {
         List<AuditDto> audits = auditDao.searchAll(searchTemplate);
-        if (baseUser.isFromGlobalManagement() || baseUser.getProjectUser(searchTemplate.getProject_id()).isViewer()) {
+        Integer projectId;
+        try {
+            projectId = searchTemplate.getId() == null
+                    ? searchTemplate.getProject_id()
+                    : audits.get(0).getProject_id();
+        } catch (IndexOutOfBoundsException e) {
+            throw new AqualityException("The Audit you trying to access is not present!");
+        }
+        if (baseUser.isFromGlobalManagement() || baseUser.getProjectUser(projectId).isViewer()) {
             return completeAudits(audits);
         } else {
             throw new AqualityPermissionsException("Account is not allowed to view Audits.", baseUser);

src/main/java/main/view/Audits/GetAuditsServlet.java

@@ -20,7 +20,6 @@ public void doPost(HttpServletRequest req, HttpServletResponse resp){
         setEncoding(resp);
 
         try {
-            validateAndGetProjectId(req);
             Session session = createSession(req);
             String requestedJson = getRequestJson(req);
             AuditDto audit = mapper.mapObject(AuditDto.class, requestedJson);

src/main/java/main/view/Project/ProjectUsersServlet.java

@@ -20,10 +20,10 @@ public void doGet(HttpServletRequest req, HttpServletResponse resp) {
         setEncoding(resp);
 
         try {
-            Integer projectId = validateAndGetProjectId(req);
             Session session = createSession(req);
             ProjectUserDto projectUserDto = new ProjectUserDto();
             Integer userId = getIntegerQueryParameter(req, "userId");
+            Integer projectId = getProjectId(req, false);
             projectUserDto.setProject_id(projectId);
             projectUserDto.setUser_id(userId);
             List<ProjectUserDto> projectUsers = session.controllerFactory.getHandler(projectUserDto).get(projectUserDto);