Get and update workflow version state

[sdc/sdc-workflow-designer.git] / workflow-designer-be / src / main / java / org / onap / sdc / workflow / api / WorkflowVersionController.java

diff --git a/workflow-designer-be/src/main/java/org/onap/sdc/workflow/api/WorkflowVersionController.java b/workflow-designer-be/src/main/java/org/onap/sdc/workflow/api/WorkflowVersionController.java
index cb1fb1f..6214f1b 100644 (file)
--- a/workflow-designer-be/src/main/java/org/onap/sdc/workflow/api/WorkflowVersionController.java
+++ b/workflow-designer-be/src/main/java/org/onap/sdc/workflow/api/WorkflowVersionController.java
@@ -1,51 +1,117 @@
 package org.onap.sdc.workflow.api;
 
+import static org.onap.sdc.workflow.api.RestConstants.USER_ID_HEADER_PARAM;
 
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.onap.sdc.workflow.api.types.CollectionWrapper;
 import org.onap.sdc.workflow.api.types.VersionRequestDto;
+import org.onap.sdc.workflow.api.types.VersionStateDto;
+import org.onap.sdc.workflow.persistence.types.ArtifactEntity;
 import org.onap.sdc.workflow.persistence.types.WorkflowVersion;
+import org.onap.sdc.workflow.services.WorkflowVersionManager;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.core.io.InputStreamResource;
 import org.springframework.core.io.Resource;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.HttpStatus;
+import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.DeleteMapping;
 import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.PutMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestHeader;
 import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
 import org.springframework.web.multipart.MultipartFile;
 
 @RequestMapping("/workflows/{workflowId}/versions")
 @Api("Workflow versions")
-public interface WorkflowVersionController {
+@RestController("workflowsVersionController")
+public class WorkflowVersionController {
 
+    private final WorkflowVersionManager workflowVersionManager;
+
+    @Autowired
+    public WorkflowVersionController(
+            @Qualifier("workflowVersionManager") WorkflowVersionManager workflowVersionManager) {
+        this.workflowVersionManager = workflowVersionManager;
+    }
 
     @GetMapping
     @ApiOperation("List workflow versions")
-    CollectionWrapper<WorkflowVersion> list(String workflowId,String user);
+    public CollectionWrapper<WorkflowVersion> list(@PathVariable("workflowId") String workflowId,
+            @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        return new CollectionWrapper<>(workflowVersionManager.list(workflowId));
+    }
 
     @PostMapping
     @ApiOperation("Create workflow version")
-    ResponseEntity<WorkflowVersion> create(String workflowId, VersionRequestDto versionRequest, String user);
+    public ResponseEntity<WorkflowVersion> create(@RequestBody VersionRequestDto versionRequest,
+            @PathVariable("workflowId") String workflowId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+
+        WorkflowVersion createdVersion = workflowVersionManager.create(workflowId, versionRequest);
+        return new ResponseEntity<>(createdVersion, HttpStatus.CREATED);
+    }
 
     @GetMapping("/{versionId}")
     @ApiOperation("Get workflow version")
-    WorkflowVersion get(String workflowId,String versionId, String user);
+    public WorkflowVersion get(@PathVariable("workflowId") String workflowId,
+            @PathVariable("versionId") String versionId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        return workflowVersionManager.get(workflowId, versionId);
+    }
 
     @PutMapping("/{versionId}")
     @ApiOperation("Update workflow version")
-    void update(WorkflowVersion version,String workflowId, String versionId, String user);
+    public void update(@RequestBody WorkflowVersion version, @PathVariable("workflowId") String workflowId,
+            @PathVariable("versionId") String versionId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        version.setId(versionId);
+        workflowVersionManager.update(workflowId, version);
+    }
 
+    @GetMapping("/{versionId}/state")
+    @ApiOperation("Get workflow version state")
+    public VersionStateDto getState(@PathVariable("workflowId") String workflowId,
+            @PathVariable("versionId") String versionId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        return new VersionStateDto(workflowVersionManager.getState(workflowId, versionId));
+    }
+
+    @PostMapping("/{versionId}/state")
+    @ApiOperation("Update workflow version state")
+    public VersionStateDto updateState(@RequestBody VersionStateDto state,
+            @PathVariable("workflowId") String workflowId, @PathVariable("versionId") String versionId,
+            @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        workflowVersionManager.updateState(workflowId, versionId, state.getName());
+        return new VersionStateDto(state.getName());
+    }
 
     @PutMapping("/{versionId}/artifact")
     @ApiOperation("Create/update artifact of a version")
-    void createArtifact(MultipartFile fileToUpload, String workflowId, String versionId, String user);
+    public void uploadArtifact(@RequestBody MultipartFile fileToUpload, @PathVariable("workflowId") String workflowId,
+            @PathVariable("versionId") String versionId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        workflowVersionManager.uploadArtifact(workflowId, versionId, fileToUpload);
+    }
 
     @GetMapping("/{versionId}/artifact")
     @ApiOperation("Download workflow version artifact")
-    ResponseEntity<Resource> getArtifact(String workflowId, String versionId, String user);
+    public ResponseEntity<Resource> getArtifact(@PathVariable("workflowId") String workflowId,
+            @PathVariable("versionId") String versionId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        ArtifactEntity artifact = workflowVersionManager.getArtifact(workflowId, versionId);
+
+        return ResponseEntity.ok()
+                             .header(HttpHeaders.CONTENT_DISPOSITION, "attachment;filename=" + artifact.getFileName())
+                             .contentType(MediaType.APPLICATION_OCTET_STREAM)
+                             .body(new InputStreamResource(artifact.getArtifactData()));
+    }
 
     @DeleteMapping("/{versionId}/artifact")
     @ApiOperation("Delete workflow version artifact")
-    void deleteArtifact(String workflowId, String versionId, String user);
+    public void deleteArtifact(@PathVariable("workflowId") String workflowId,
+            @PathVariable("versionId") String versionId, @RequestHeader(USER_ID_HEADER_PARAM) String user) {
+        workflowVersionManager.deleteArtifact(workflowId, versionId);
+    }
 }