create assetHistory on asset creation to avoid errors

Mixeway · Jul 15, 2024 · a7a2bbd · a7a2bbd
1 parent c984a74
commit a7a2bbd
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/src/main/java/io/mixeway/domain/service/scanmanager/code/CreateOrGetCodeProjectService.java b/src/main/java/io/mixeway/domain/service/scanmanager/code/CreateOrGetCodeProjectService.java
@@ -7,6 +7,7 @@
 import io.mixeway.db.entity.Project;
 import io.mixeway.db.repository.CodeProjectBranchRepository;
 import io.mixeway.db.repository.CodeProjectRepository;
+import io.mixeway.domain.service.assethistory.CreateAssetHistoryService;
 import io.mixeway.domain.service.project.GetOrCreateProjectService;
 import io.mixeway.scanmanager.model.CodeScanRequestModel;
 import io.mixeway.utils.CodeGroupPutModel;
@@ -38,6 +39,7 @@ public class CreateOrGetCodeProjectService {
     private final VaultHelper vaultHelper;
     private final GetOrCreateProjectService getOrCreateProjectService;
     private final GetOrCreateCodeProjectBranchService getOrCreateCodeProjectBranchService;
+    private final CreateAssetHistoryService createAssetHistoryService;
 
 
     public CodeProject getOrCreateCodeProject(Project project, String projectName, String codeDefaultBranch) {
@@ -68,6 +70,7 @@ public CodeProject createCodeProject(String repoUrl, String repoName, String bra
         CodeProject codeProject = new CodeProject(project, repoName, (branch == null || branch.isEmpty()) ? "master" : branch, null,repoUrl,null,null);
 
         codeProject = codeProjectRepository.saveAndFlush(codeProject);
+        createAssetHistoryService.create(codeProject,0,0,0,0,0,0,0,0,0,0);
         getOrCreateCodeProjectBranchService.getOrCreateCodeProjectBranch(codeProject, codeProject.getBranch());
         return codeProject;
     }

diff --git a/src/main/java/io/mixeway/domain/service/scanmanager/webapp/GetOrCreateWebAppService.java b/src/main/java/io/mixeway/domain/service/scanmanager/webapp/GetOrCreateWebAppService.java
@@ -4,6 +4,7 @@
 import io.mixeway.config.Constants;
 import io.mixeway.db.entity.*;
 import io.mixeway.db.repository.*;
+import io.mixeway.domain.service.assethistory.CreateAssetHistoryService;
 import io.mixeway.scanmanager.model.CustomCookie;
 import io.mixeway.scanmanager.model.RequestHeaders;
 import io.mixeway.scanmanager.model.WebAppScanHelper;
@@ -32,6 +33,7 @@ public class GetOrCreateWebAppService {
     private final VaultHelper vaultHelper;
     private final WebAppCookieRepository webAppCookieRepository;
     private final WebAppHeaderRepository webAppHeaderRepository;
+    private final CreateAssetHistoryService createAssetHistoryService;
 
     private SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
 
@@ -243,6 +245,8 @@ public WebApp createWebApp(Project project, String appUrl, String appName, Routi
         webApp.setApikey(apiKeyName);
         webApp.setProject(project);
         webApp = webAppRepository.save(webApp);
+        createAssetHistoryService.create(webApp,0,0,0,0,0,0,0,0,0,0);
+
         return webApp;
     }
 }