Merge pull request #3747 from akto-api-security/agent-traffic-detector-cronjob

Added AgentBasePromptDetectionCron class and integrated it into Initi…

Author: krngrover94

apps/dashboard/src/main/java/com/akto/listener/InitializerListener.java

@@ -225,6 +225,7 @@
 import com.akto.utils.crons.SyncCron;
 import com.akto.utils.crons.TokenGeneratorCron;
 import com.akto.utils.crons.UpdateSensitiveInfoInApiInfo;
+import com.akto.utils.crons.AgentBasePromptDetectionCron;
 import com.akto.utils.jobs.CleanInventory;
 import com.akto.utils.jobs.DeactivateCollections;
 import com.akto.utils.jobs.JobUtils;
@@ -278,6 +279,7 @@ public class InitializerListener implements ServletContextListener {
     SyncCron syncCronInfo = new SyncCron();
     TokenGeneratorCron tokenGeneratorCron = new TokenGeneratorCron();
     UpdateSensitiveInfoInApiInfo updateSensitiveInfoInApiInfo = new UpdateSensitiveInfoInApiInfo();
+    AgentBasePromptDetectionCron agentBasePromptDetectionCron = new AgentBasePromptDetectionCron();
 
     private static String domain = null;
     public static String subdomain = "https://app.akto.io";
@@ -2627,6 +2629,7 @@ public void accept(Account account) {
                     updateSensitiveInfoInApiInfo.setUpSensitiveMapInApiInfoScheduler();
                     syncCronInfo.setUpUpdateCronScheduler();
                     syncCronInfo.setUpMcpMaliciousnessCronScheduler();
+                    agentBasePromptDetectionCron.setUpAgentBasePromptDetectionScheduler();
                     setUpTestEditorTemplatesScheduler();
                     JobsCron.instance.jobsScheduler(JobExecutorType.DASHBOARD);
                     updateApiGroupsForAccounts(); 

apps/dashboard/src/main/java/com/akto/utils/crons/AgentBasePromptDetectionCron.java

@@ -0,0 +1,41 @@
+package com.akto.utils.crons;
+
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+import java.util.function.Consumer;
+
+import com.akto.dto.Account;
+import com.akto.log.LoggerMaker;
+import com.akto.log.LoggerMaker.LogDb;
+import com.akto.runtime.AgentBasePromptDetectionService;
+import com.akto.util.AccountTask;
+
+public class AgentBasePromptDetectionCron {
+    private static final LoggerMaker loggerMaker = new LoggerMaker(AgentBasePromptDetectionCron.class, LogDb.DASHBOARD);
+    ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
+
+    public void setUpAgentBasePromptDetectionScheduler() {
+        scheduler.scheduleAtFixedRate(new Runnable() {
+            public void run() {
+                AccountTask.instance.executeTask(new Consumer<Account>() {
+                    @Override
+                    public void accept(Account t) {
+                        try {
+                            loggerMaker.debugAndAddToDb("Starting agent base prompt detection for account: " + t.getId(), LogDb.DASHBOARD);
+                            AgentBasePromptDetectionService service = new AgentBasePromptDetectionService();
+                            service.runJob();
+                            loggerMaker.debugAndAddToDb("Completed agent base prompt detection for account: " + t.getId(), LogDb.DASHBOARD);
+                        } catch (Exception e) {
+                            loggerMaker.errorAndAddToDb(e,
+                                    String.format("Error while running agent base prompt detection for account %d: %s", 
+                                            t.getId(), e.toString()),
+                                    LogDb.DASHBOARD);
+                        }
+                    }
+                }, "agent-base-prompt-detection");
+            }
+        }, 0, 24, TimeUnit.HOURS);
+    }
+}
+