from logging import getLogger

from defence360agent.contracts.messages import MessageType
from defence360agent.contracts.plugins import MessageSink, expect
from defence360agent.utils import Scope
from im360.plugins.sensor.generic import send_to_agent_socket

logger = getLogger(__name__)


class Repeater(MessageSink):
    SCOPE = Scope.IM360_RESIDENT

    async def create_sink(self, loop):
        self._loop = loop

    @expect(MessageType.MalwareRescanFiles)
    async def resend_to_simple_socket(self, msg):
        send_to_agent_socket(
            ["malware", "rescan"], params={"files": msg.files}
        )

    @expect(MessageType.MalwareSendFiles)
    async def resend_malware_send_files(self, msg):
        send_to_agent_socket(
            ["malware", "send", "files"],
            params={"reason": msg.reason, "files": msg.files},
        )