[backend] fix setters

OpenBAS-Platform · Sep 30, 2024 · 729a7ce · 729a7ce
1 parent 183d140
commit 729a7ce
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 14 deletions.
diff --git a/openbas-framework/src/main/java/io/openbas/integrations/PayloadService.java b/openbas-framework/src/main/java/io/openbas/integrations/PayloadService.java
@@ -183,16 +183,13 @@ public Payload duplicate(@NotBlank final String payloadId) {
     return duplicate;
   }
 
-  private <T extends Payload> void duplicateCommonProperties(@org.jetbrains.annotations.NotNull final T origin,
-      @org.jetbrains.annotations.NotNull T duplicate) {
-    BeanUtils.copyProperties(origin, duplicate);
-    duplicate.setId(null);
-    duplicate.setName(StringUtils.duplicateString(origin.getName()));
-    duplicate.setAttackPatterns(new ArrayList<>(origin.getAttackPatterns()));
-    duplicate.setTags(new HashSet<>(origin.getTags()));
-    duplicate.setExternalId(null);
-    duplicate.setSource(MANUAL);
-    duplicate.setStatus(VERIFIED);
-    duplicate.setCollector(null);
-  }
+    private <T extends Payload> void duplicateCommonProperties(@org.jetbrains.annotations.NotNull final T origin, @org.jetbrains.annotations.NotNull T duplicate) {
+        BeanUtils.copyProperties(origin, duplicate);
+        duplicate.setId(null);
+        duplicate.setName(StringUtils.duplicateString(origin.getName()));
+        duplicate.setAttackPatterns(new ArrayList<>(origin.getAttackPatterns()));
+        duplicate.setTags(new HashSet<>(origin.getTags()));
+        duplicate.setExternalId(null);
+        duplicate.setCollector(null);
+    }
 }
diff --git a/openbas-model/src/main/java/io/openbas/database/model/Payload.java b/openbas-model/src/main/java/io/openbas/database/model/Payload.java
@@ -118,14 +118,12 @@ public enum PAYLOAD_STATUS {
   private String externalId;
 
   @Queryable(filterable = true, sortable = true)
-  @Setter
   @Column(name = "payload_source")
   @Enumerated(EnumType.STRING)
   @JsonProperty("payload_source")
   private PAYLOAD_SOURCE source;
 
   @Queryable(filterable = true)
-  @Setter
   @Column(name = "payload_status")
   @Enumerated(EnumType.STRING)
   @JsonProperty("payload_status")
@@ -168,6 +166,14 @@ private String getCollectorType() {
     return this.collector != null ? this.collector.getType() : null;
   }
 
+  public void setStatus(String status) {
+    this.status = PAYLOAD_STATUS.valueOf(status.toUpperCase());
+  }
+
+  public void setSource(String source) {
+    this.source = PAYLOAD_SOURCE.valueOf(source.toUpperCase());
+  }
+
   @Override
   public int hashCode() {
     return Objects.hash(id);