Changeset 33d4f5d for src/main/java/it/finki/tinki/web/controller/WorkRegisterController.java

Timestamp:

01/12/21 02:42:18 (3 years ago)

Author:

Vzdra <vladko.zdravkovski@…>

Branches:

master

Children:

a70db1a

Parents:

8f1f460

Message:

bugfixes and refactoring

File:

• src/main/java/it/finki/tinki/web/controller/WorkRegisterController.java (modified) (3 diffs)

src/main/java/it/finki/tinki/web/controller/WorkRegisterController.java

@@ -1 +1 @@
 package it.finki.tinki.web.controller;
 
+import it.finki.tinki.model.Users.Account;
 import it.finki.tinki.model.Work.Internship;
 import it.finki.tinki.model.Work.Job;
@@ -10 +11 @@
 import it.finki.tinki.model.dto.response.work.JobResponseDTO;
 import it.finki.tinki.model.dto.response.work.ProjectResponseDTO;
+import it.finki.tinki.model.enumerator.AccountType;
 import it.finki.tinki.service.WorkService;
 import org.springframework.web.bind.annotation.*;
@@ -26 +28 @@
     @PostMapping("/job")
     public JobResponseDTO registerJob(@RequestBody JobRegisterDTO body){
-        Job j = this.workService.insertJob(body.getTitle(),
-                body.getDescription(), body.getAccountId(), body.getSalary(), body.getSkillsRequired(), body.getType());
-        return new JobResponseDTO(j);
+
+        if(body.getType() != AccountType.USER){
+            Job j = this.workService.insertJob(body.getTitle(),
+                    body.getDescription(), body.getAccountId(), body.getSalary(), body.getSkillsRequired(), body.getType());
+            return new JobResponseDTO(j);
+        }
+
+        return new JobResponseDTO();
     }
 
     @PostMapping("/internship")
     public InternshipResponseDTO registerInternship(@RequestBody InternshipRegisterDTO body){
-        Internship j = this.workService.insertInternship(body.getTitle(),
-                body.getDescription(), body.getAccountId(), body.getSalary(), body.getSkillsTrained(), body.getOpenSpots(), body.getType());
-        return new InternshipResponseDTO(j);
+        if(body.getType() == AccountType.COMPANY){
+            Internship j = this.workService.insertInternship(body.getTitle(),
+                    body.getDescription(), body.getAccountId(), body.getSalary(), body.getSkillsTrained(), body.getOpenSpots(), body.getType());
+            return new InternshipResponseDTO(j);
+        }
+
+        return new InternshipResponseDTO();
     }
 
     @PostMapping("/project")
     public ProjectResponseDTO registerProject(@RequestBody ProjectRegisterDTO body){
-        Project j = this.workService.insertProject(body.getTitle(),
-                body.getDescription(), body.getAccountId(), body.getSalary(), body.getSkillsRequired(), body.getValidUntil(), body.getType());
-        return new ProjectResponseDTO(j);
+
+        if(body.getType() == AccountType.TEAM){
+            Project j = this.workService.insertProject(body.getTitle(),
+                    body.getDescription(), body.getAccountId(), body.getSalary(), body.getSkillsRequired(), body.getValidUntil(), body.getType());
+            return new ProjectResponseDTO(j);
+        }
+
+        return new ProjectResponseDTO();
     }
 }