From b3e9bbe05f26aa6898c4ea3a3e5859d6e6535b18 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Wed, 14 Sep 2022 13:54:38 -0400
Subject: [PATCH] Island: Subscribe save_stolen_credentials_to_repository()

---
 .../monkey_island/cc/setup/agent_event_handlers.py  | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/monkey/monkey_island/cc/setup/agent_event_handlers.py b/monkey/monkey_island/cc/setup/agent_event_handlers.py
index 9fd0a2ea7..af37e6740 100644
--- a/monkey/monkey_island/cc/setup/agent_event_handlers.py
+++ b/monkey/monkey_island/cc/setup/agent_event_handlers.py
@@ -1,7 +1,11 @@
 from common import DIContainer
 from common.event_queue import IAgentEventQueue
-from monkey_island.cc.agent_event_subscribers import save_event_to_event_repository
-from monkey_island.cc.repository import IEventRepository
+from common.events import CredentialsStolenEvent
+from monkey_island.cc.agent_event_subscribers import (
+    save_event_to_event_repository,
+    save_stolen_credentials_to_repository,
+)
+from monkey_island.cc.repository import ICredentialsRepository, IEventRepository
 
 
 def setup_agent_event_handlers(container: DIContainer):
@@ -14,3 +18,8 @@ def _subscribe_and_store_to_event_repository(container: DIContainer):
     event_repository = container.resolve(IEventRepository)
     save_event_subscriber = save_event_to_event_repository(event_repository)
     agent_event_queue.subscribe_all_events(save_event_subscriber)
+
+    save_stolen_credentials_subscriber = save_stolen_credentials_to_repository(
+        container.resolve(ICredentialsRepository)
+    )
+    agent_event_queue.subscribe_type(CredentialsStolenEvent, save_stolen_credentials_subscriber)