# Mobile Verification Toolkit (MVT)
# Copyright (c) 2021-2022 Claudio Guarnieri.
# See the file 'LICENSE' for usage and copying permissions, or find a copy at
#   https://github.com/mvt-project/mvt/blob/main/LICENSE

import fnmatch
import logging
import os

from mvt.common.module import MVTModule

log = logging.getLogger(__name__)


class BugReportModule(MVTModule):
    """This class provides a base for all Android Bug Report modules."""

    zip_archive = None

    def from_folder(self, extract_path, extract_files):
        self.extract_path = extract_path
        self.extract_files = extract_files

    def from_zip(self, zip_archive, zip_files):
        self.zip_archive = zip_archive
        self.zip_files = zip_files

    def _get_files_by_pattern(self, pattern):
        file_names = []
        if self.zip_archive:
            for zip_file in self.zip_files:
                file_names.append(zip_file)
        else:
            file_names = self.extract_files

        return fnmatch.filter(file_names, pattern)

    def _get_file_content(self, file_path):
        if self.zip_archive:
            handle = self.zip_archive.open(file_path)
        else:
            handle = open(os.path.join(self.extract_path, file_path), "rb")

        data = handle.read()
        handle.close()

        return data