# Mobile Verification Toolkit (MVT)
# Copyright (c) 2021-2023 The MVT Authors.
# Use of this software is governed by the MVT License 1.1 that can be found at
#   https://license.mvt.re/1.1/

import logging
from typing import Optional

from mvt.android.artifacts.dumpsys_receivers import DumpsysReceiversArtifact
from mvt.common.module_types import ModuleResults

from .base import BugReportModule


class DumpsysReceivers(DumpsysReceiversArtifact, BugReportModule):
    """This module extracts details on receivers for risky activities."""

    def __init__(
        self,
        file_path: Optional[str] = None,
        target_path: Optional[str] = None,
        results_path: Optional[str] = None,
        module_options: Optional[dict] = None,
        log: logging.Logger = logging.getLogger(__name__),
        results: ModuleResults = [],
    ) -> None:
        super().__init__(
            file_path=file_path,
            target_path=target_path,
            results_path=results_path,
            module_options=module_options,
            log=log,
            results=results,
        )

        self.results = results if results else {}

    def check_indicators(self) -> None:
        for result in self.results:
            if self.indicators:
                receiver_name = self.results[result][0]["receiver"]

                # return IoC if the stix2 process name a substring of the receiver name
                ioc_match = self.indicators.check_receiver_prefix(receiver_name)
                if ioc_match:
                    self.alertstore.critical(
                        ioc_match.message,
                        "",
                        self.results[result][0],
                        matched_indicator=ioc_match.ioc,
                    )
                    continue



    def run(self) -> None:
        content = self._get_dumpstate_file()
        if not content:
            self.log.error(
                "Unable to find dumpstate file. "
                "Did you provide a valid bug report archive?"
            )
            return

        dumpsys_section = self.extract_dumpsys_section(
            content.decode("utf-8", errors="replace"), "DUMP OF SERVICE package:"
        )

        self.parse(dumpsys_section)
        self.log.info("Extracted receivers for %d intents", len(self.results))
