from model.plugin_model import Plugin
from model.model_base import Scanner, OutflankPatch, ScanInfo
from model.model_data import Match
from model.model_verification import MatchConclusion
from plugins.pe.file_pe import FilePe
from typing import List, Tuple, Set

from plugins.pe.analyzer_pe import analyzeFilePe
from plugins.pe.augment_pe import augmentFilePe
from plugins.pe.outflank_pe import outflankPe
from plugins.pe.file_pe import FilePe
from reducer import Reducer
from model.file_model import BaseFile


class PluginPe(Plugin):
    
    def makeFile(self, filepath: str):
        file = FilePe()
        file.loadFromFile(filepath)
        return file

    
    def analyzeFile(self, file: BaseFile, scanner: Scanner, reducer: Reducer, analyzerOptions={}) -> Tuple[Match, ScanInfo]:
        # We use the simple PE analyzer
        return analyzeFilePe(file, scanner, reducer, analyzerOptions)
    
    
    def augmentFile(self, file: BaseFile, matches: List[Match]) -> str:
        return augmentFilePe(file, matches)

    
    def outflankFile(
        self, file: BaseFile, matches: List[Match], matchConclusion: MatchConclusion, scanner: Scanner = None
    ) -> List[OutflankPatch]:
        return outflankPe(file, matches, matchConclusion, scanner)