p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Register FileSimulationRepository with the DIContainer

This commit is contained in:
Mike Salvatore 2022-07-01 10:59:05 -04:00
parent d2906ef23b
commit 350ac6957f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
monkey/monkey_island/cc/services/initialize.py
View File

@ -8,9 +8,11 @@ from common.utils.file_utils import get_binary_io_sha256_hash
from monkey_island.cc.repository import ( from monkey_island.cc.repository import (
AgentBinaryRepository, AgentBinaryRepository,
FileAgentConfigurationRepository, FileAgentConfigurationRepository,
FileSimulationRepository,
IAgentBinaryRepository, IAgentBinaryRepository,
IAgentConfigurationRepository, IAgentConfigurationRepository,
IFileRepository, IFileRepository,
ISimulationRepository,
LocalStorageFileRepository, LocalStorageFileRepository,
RetrievalError, RetrievalError,
) )
@ -45,6 +47,7 @@ def initialize_services(data_dir: Path) -> DIContainer:
container.register_instance( container.register_instance(
IAgentConfigurationRepository, container.resolve(FileAgentConfigurationRepository) IAgentConfigurationRepository, container.resolve(FileAgentConfigurationRepository)
) )
container.register_instance(ISimulationRepository, container.resolve(FileSimulationRepository))
# This is temporary until we get DI all worked out. # This is temporary until we get DI all worked out.
PostBreachFilesService.initialize(container.resolve(IFileRepository)) PostBreachFilesService.initialize(container.resolve(IFileRepository))