Chenged newest plan response

restservice/src/main/java/com/plannaplan/App.java

@@ -32,7 +32,7 @@ public class App {
                 filip.setEmail("filizy@st.amu.edu.pl");
                 filip.setName("Filip");
                 filip.setSurname("Izydorczyk");
-                filip.setRole(UserRoles.DEANERY);
+                filip.setRole(UserRoles.ADMIN);
                 this.userService.save(filip);
 
                 User hub = new User();

restservice/src/main/java/com/plannaplan/responses/mappers/AssignmentResponseMappers.java

@@ -1,16 +1,45 @@
 package com.plannaplan.responses.mappers;
 
 import java.util.List;
-import java.util.Objects;
-import java.util.stream.Collectors;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
 
 import com.plannaplan.entities.Assignment;
+import com.plannaplan.entities.Course;
+import com.plannaplan.entities.Groups;
 import com.plannaplan.responses.models.GetCurrentAssignmentsResponse;
+import com.plannaplan.types.GroupType;
 
 public class AssignmentResponseMappers {
 
     public static final List<GetCurrentAssignmentsResponse> mapToResponse(List<Assignment> assignments) {
-        return assignments.stream().filter(Objects::nonNull).map(GetCurrentAssignmentsResponse::new)
-                .collect(Collectors.toList());
+        List<GetCurrentAssignmentsResponse> response = new ArrayList<>();
+        HashMap<Course, List<Groups>> courses = new HashMap<>();
+        assignments.stream().forEach((Assignment assignment) -> {
+            final Groups group = assignment.getGroup();
+            final Course course = group.getCourseId();
+            if (courses.get(course) == null) {
+                courses.put(course, new ArrayList<>());
+            }
+            courses.get(course).add(group);
+        });
+
+        for (Map.Entry<Course, List<Groups>> entry : courses.entrySet()) {
+            final Course course = entry.getKey();
+            final List<Groups> courseGroups = entry.getValue();
+            if (courseGroups.size() == 1) {
+                response.add(new GetCurrentAssignmentsResponse(course, courseGroups.get(0)));
+            }
+            if (courseGroups.size() == 2) {
+                final Groups lecture = courseGroups.stream().filter(o -> o.getType() == GroupType.LECTURE).findFirst()
+                        .get();
+                final Groups classes = courseGroups.stream().filter(o -> o.getType() == GroupType.CLASS).findFirst()
+                        .get();
+                response.add(new GetCurrentAssignmentsResponse(course, lecture, classes));
+
+            }
+        }
+        return response;
     }
 }

restservice/src/main/java/com/plannaplan/responses/models/GetCurrentAssignmentsResponse.java

@@ -1,11 +1,41 @@
 package com.plannaplan.responses.models;
 
-import com.plannaplan.entities.Assignment;
+import com.plannaplan.entities.Course;
+import com.plannaplan.entities.Groups;
+import com.plannaplan.types.GroupType;
 
-public class GetCurrentAssignmentsResponse extends WithCapacityGroupResponse {
+public class GetCurrentAssignmentsResponse {
+    private Long id;
+    private String name;
+    private WithCapacityGroupResponse classes;
+    private WithCapacityGroupResponse lecture;
 
-    public GetCurrentAssignmentsResponse(Assignment assignment) {
-        super(assignment);
+    public GetCurrentAssignmentsResponse(Course course, Groups lecture, Groups classes) {
+        this.id = course.getId();
+        this.name = course.getName();
+        this.lecture = lecture == null ? null : new WithCapacityGroupResponse(lecture);
+        this.classes = classes == null ? null : new WithCapacityGroupResponse(classes);
+    }
+
+    public GetCurrentAssignmentsResponse(Course course, Groups group) {
+        this(course, group.getType() == GroupType.LECTURE ? group : null,
+                group.getType() == GroupType.CLASS ? group : null);
+    }
+
+    public WithCapacityGroupResponse getLecture() {
+        return this.lecture;
+    }
+
+    public WithCapacityGroupResponse getClasses() {
+        return this.classes;
+    }
+
+    public String getName() {
+        return this.name;
+    }
+
+    public Long getId() {
+        return this.id;
     }
 
 }

restservice/src/main/resources/application.properties

@@ -7,5 +7,6 @@ spring.jpa.open-in-view=true
 spring.jpa.hibernate.ddl-auto=create-drop
 spring.jackson.serialization.fail-on-empty-beans=false
 spring.main.allow-bean-definition-overriding=true
+spring.jackson.default-property-inclusion = NON_NULL
 
 server.port=1285