first working version of approved usage update

Author: ztarbug

docs/ClassDiagram.drawio.svg

@@ -1,7 +1,10 @@
 package eu.dzhw.fdz.metadatamanagement.datasetmanagement.rest;
 
 import java.net.URI;
+import java.util.ArrayList;
+import java.util.List;
 
+import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.http.ResponseEntity;
 import org.springframework.stereotype.Controller;
 import org.springframework.web.bind.annotation.DeleteMapping;
@@ -15,10 +18,16 @@
 
 import eu.dzhw.fdz.metadatamanagement.common.rest.GenericDomainObjectResourceController;
 import eu.dzhw.fdz.metadatamanagement.common.service.CrudService;
+import eu.dzhw.fdz.metadatamanagement.datapackagemanagement.domain.ApprovedUsageEnum;
+import eu.dzhw.fdz.metadatamanagement.datapackagemanagement.domain.DataPackage;
+import eu.dzhw.fdz.metadatamanagement.datapackagemanagement.repository.DataPackageRepository;
 import eu.dzhw.fdz.metadatamanagement.datasetmanagement.domain.DataSet;
+import eu.dzhw.fdz.metadatamanagement.datasetmanagement.domain.SubDataSet;
+import eu.dzhw.fdz.metadatamanagement.projectmanagement.repository.DataAcquisitionProjectRepository;
 import eu.dzhw.fdz.metadatamanagement.usermanagement.security.UserInformationProvider;
 import io.swagger.v3.oas.annotations.Operation;
 import io.swagger.v3.oas.annotations.tags.Tag;
+import lombok.extern.slf4j.Slf4j;
 
 /**
  * DataSet REST Controller which overrides default spring data rest methods.
@@ -27,9 +36,13 @@
  */
 @Controller
 @Tag(name = "Dataset Resource", description = "Endpoints used by the MDM to manage datasets.")
+@Slf4j
 public class DataSetResourceController
     extends GenericDomainObjectResourceController<DataSet, CrudService<DataSet>> {
 
+  @Autowired
+  DataPackageRepository dataPackageRepository;
+
   public DataSetResourceController(CrudService<DataSet> crudService,
       UserInformationProvider userInformationProvider) {
     super(crudService, userInformationProvider);
@@ -46,16 +59,17 @@ public ResponseEntity<DataSet> getDomainObject(@PathVariable String id) {
     return super.getDomainObject(id);
   }
 
-
   @Override
   @PostMapping(value = "/api/data-sets")
-  public ResponseEntity<?> postDomainObject(@RequestBody DataSet dataSet) {
+  public ResponseEntity<?> postDomainObject(@RequestBody DataSet dataSet) {   
+    addApprovedUsageIfDownloadableContent(dataSet);
     return super.postDomainObject(dataSet);
   }
 
   @Override
   @PutMapping(value = "/api/data-sets/{id:.+}")
   public ResponseEntity<?> putDomainObject(@RequestBody DataSet dataSet) {
+    addApprovedUsageIfDownloadableContent(dataSet);
     return super.putDomainObject(dataSet);
   }
 
@@ -69,4 +83,35 @@ public ResponseEntity<?> deleteDomainObject(@PathVariable String id) {
   protected URI buildLocationHeaderUri(DataSet domainObject) {
     return UriComponentsBuilder.fromPath("/api/data-sets/" + domainObject.getId()).build().toUri();
   }
+
+  private void addApprovedUsageIfDownloadableContent(DataSet dataSet) {
+    boolean hasDownloadableContent = false;
+    var subsets = dataSet.getSubDataSets();
+    for (SubDataSet subDataSet : subsets) {
+      String accessWay = subDataSet.getAccessWay();
+      if (accessWay != null && accessWay.toLowerCase().contains("cuf")) {
+          hasDownloadableContent = true;
+      }
+    }
+
+    log.info("Data set has downloadable content, updating approved usages: " + hasDownloadableContent);
+    if(hasDownloadableContent) {
+      String projectId = dataSet.getDataAcquisitionProjectId();
+      List<DataPackage> dataPackages = dataPackageRepository.findByDataAcquisitionProjectId(projectId);
+      log.info("found " + dataPackages.size() + " data packages for project " + projectId);
+
+      if(dataPackages.size() > 0 ) {
+        log.info("updated approved usages for project " + projectId);
+        DataPackage entity = dataPackages.get(0);
+        List<ApprovedUsageEnum> approvedUsages = entity.getApprovedUsageList();
+        if(approvedUsages == null) {
+          approvedUsages =  new ArrayList<>();
+          entity.setApprovedUsageList(approvedUsages);
+        }
+        approvedUsages.add(ApprovedUsageEnum.TEACHING_PURPOSES);
+        
+        dataPackageRepository.save(entity);
+      }
+    }
+  }
 }