# pylint: disable=multiple-statements
# Define Files
# Copyright (C) 2020-2021, Dhinak G, Mykola Grymalyuk

from __future__ import print_function

from pathlib import Path
from typing import Optional

from resources import device_probe


class Constants:
    def __init__(self):
        # Patcher Versioning
        self.patcher_version = "0.3.0"  #              OpenCore-Legacy-Patcher
        self.patcher_support_pkg_version = "0.1.5"  #  PatcherSupportPkg

        # OpenCore Versioning
        # https://github.com/acidanthera/OpenCorePkg
        self.opencore_commit = "ff4b099 - 09-06-2021"
        self.opencore_version = "0.7.3"

        # Kext Versioning
        ## Acidanthera
        ## https://github.com/acidanthera
        self.lilu_version = "1.5.6"  #               Lilu
        self.whatevergreen_version = "1.5.3"  #      WhateverGreen
        self.airportbcrmfixup_version = "2.1.3"  #   AirPortBrcmFixup
        self.nvmefix_version = "1.0.9"  #            NVMeFix
        self.applealc_version = "1.6.3"  #           AppleALC
        self.restrictevents_version = "1.0.5"  #     RestrictEvents
        self.restrictevents_mbp_version = "1.0.5"  # RestrictEvents blocking displaypolicyd (see RestrictEvents-MBP91.patch)
        self.featureunlock_version = "1.0.3"  #      FeatureUnlock
        self.debugenhancer_version = "1.0.4"  #      DebugEnhancer
        self.cpufriend_version = "1.2.4"  #          CPUFriend
        self.bluetool_version = "2.6.1"  #           BlueToolFixup
        self.cslvfixup_version = "2.6.1"  #          CSLVFixup

        ## Apple
        ## https://www.apple.com
        self.marvel_version = "1.0.1"  #  MarvelYukonEthernet
        self.nforce_version = "1.0.1"  #  nForceEthernet
        self.piixata_version = "1.0.1"  # AppleIntelPIIXATA
        self.fw_kext = "1.0.1"  #         IOFireWireFamily
        self.apple_trackpad = "1.0.1"  #  AppleUSBTrackpad

        ## Apple - Dortania Modified
        self.bcm570_version = "1.0.2"  #             CatalinaBCM5701Ethernet
        self.corecaptureelcap_version = "1.0.1"  #   corecaptureElCap
        self.io80211elcap_version = "2.0.0"  #       IO80211ElCap
        self.io80211high_sierra_version = "1.0.1"  # IO80211HighSierra
        self.io80211mojave_version = "1.0.1"  #      IO80211Mojave

        ## Dortania
        ## https://github.com/dortania
        self.backlight_injector_version = "1.0.0"  # BacklightInjector
        self.smcspoof_version = "1.0.0"  #           SMC-Spoof
        self.mce_version = "1.0.0"  #                AppleMCEReporterDisabler
        self.btspoof_version = "1.0.0"  #            Bluetooth-Spoof

        ## Syncretic
        ## https://forums.macrumors.com/members/syncretic.1173816/
        ## https://github.com/reenigneorcim/latebloom
        self.latebloom_version = "0.22"  #   Latebloom
        self.mousse_version = "0.95"  #      MouSSE
        self.telemetrap_version = "1.0.0"  # telemetrap

        ## cdf
        ## https://github.com/cdf/Innie
        self.innie_version = "1.3.0"  # Innie

        # Get resource path
        self.current_path = Path(__file__).parent.parent.resolve()
        self.payload_path = self.current_path / Path("payloads")

        # Patcher Settings
        self.allow_oc_everywhere = False  # Set whether Patcher can be run on unsupported Macs
        self.gui_mode = False  #            Determine whether running in a GUI or TUI
        self.disk = ""  #                   Set installation ESP
        self.patch_disk = ""  #             Set Root Volume to patch
        self.validate = False  #            Enable validation testing for CI
        self.recovery_status = False  #     Detect if booted into RecoveryOS

        ## Hardware
        self.computer: device_probe.Computer = None  # type: ignore
        self.custom_model: Optional[str] = None

        ## OpenCore Settings
        self.opencore_debug = False
        self.opencore_build = "RELEASE"
        self.showpicker = True  # Show or Hide OpenCore's Boot Picker
        self.boot_efi = False  #  Use EFI/BOOT/BOOTx64.efi bootstrap

        ## Kext Settings
        self.kext_debug = False  # Enables Lilu debug and DebugEnhancer

        ## NVRAM Settings
        self.verbose_debug = False  # -v

        ## SMBIOS Settings
        self.custom_cpu_model = 2  #        Patch type value
        self.custom_cpu_model_value = ""  # New CPU name within About This Mac
        self.serial_settings = "Minimal"  # Set SMBIOS level used
        self.override_smbios = "Default"  # Set SMBIOS model used

        ## Latebloom Settings
        self.latebloom_status = False  # Latebloom Enabled
        self.latebloom_delay = 0  #      Delay between each PCIe Probe
        self.latebloom_range = 0  #      Range each delay can differ
        self.latebloom_debug = 0  #      Debug Setting

        ## Security Settings
        self.apecid_support = False  # ApECID
        self.amfi_status = True  #     Apple Mobile File Integrity
        self.sip_status = True  #      System Integrity Protection
        self.secure_status = False  #  Secure Boot Model
        self.vault = False  #          EFI Vault
        self.disable_cs_lv = False  #  Disable Library validation

        ## OS Settings
        self.os_support = 12.0
        self.detected_os = 0  #        Major Kernel Version
        self.detected_os_minor = 0  #  Minor Kernel Version
        self.detected_os_build = ""  # OS Build
        self.allow_fv_root = False  #  Allow FileVault on broken sealed snapshots

        ## Boot Volume Settings
        self.firewire_boot = False  # Allow macOS FireWire Boot
        self.nvme_boot = False  #     Allow UEFI NVMe Boot

        ## Graphics Settings
        self.metal_build = False  #    Set MXM Build support
        self.imac_vendor = "None"  #   Set MXM GPU vendor
        self.drm_support = False  #    Set iMac14,x DRM support
        self.allow_ivy_igpu = False  # Set iMac13,x iGPU support
        self.moj_cat_accel = False  #  Set Mojave/Catalina Acceleration support
        self.allow_ts2_accel = True  # Set TeraScale 2 Acceleration support

        ## Miscellaneous
        self.disallow_cpufriend = False  #   Disable CPUFriend
        self.enable_wake_on_wlan = False  #  Allow Wake on WLAN for modern Broadcom
        self.disable_tb = False  #  Disable Thunderbolt Controller
        self.set_alc_usage = True  #         Set AppleALC usage
        self.dGPU_switch = True  #           Set Display GPU Switching for Windows
        self.force_surplus = False  #        Force SurPlus patch in newer OSes

        # OS Versions
        ## Based off Major Kernel Version
        self.tiger = 8
        self.leopard = 9
        self.snow_leopard = 10
        self.lion = 11
        self.mountain_lion = 12
        self.mavericks = 13
        self.yosemite = 14
        self.el_capitan = 15
        self.sierra = 16
        self.high_sierra = 17
        self.mojave = 18
        self.catalina = 19
        self.big_sur = 20
        self.monterey = 21

        # Vendor IDs
        self.pci_nvidia = "10DE"
        self.pci_amd_ati = "1002"
        self.pci_intel = "8086"
        self.pci_broadcom = "14E4"
        self.pci_atheros = "168C"
        self.pci_apple = "106B"
        self.pci_aquantia = "1D6A"
        self.pci_marvell = "11AB"
        self.pci_syskonnect = "1148"

        # Class Codes
        ## https://pci-ids.ucw.cz/read/PD
        self.classcode_sata = "01060100"
        self.classcode_nvme = "02080100"
        self.classcode_nvme_generic = "02800100"
        self.classcode_wifi = "00800200"
        self.classcode_gpu = "00000300"
        self.classcode_gpu_variant = "00800300"
        self.classcode_xhci = "30030C00"
        self.classcode_ethernet = "00000200"

        # Nvidia GPU Architecture
        self.arch_tesla = "NV50"
        self.arch_fermi = "GF100"
        self.arch_kepler = "GK100"

        # External Files
        self.url_patcher_support_pkg = "https://github.com/dortania/PatcherSupportPkg/releases/download/"

        self.legacy_accel_support = [
            self.mojave,
            self.catalina,
            self.big_sur,
            self.monterey,
        ]

    # Payload Location
    # OpenCore
    @property
    def opencore_zip_source(self):
        return self.payload_path / Path(f"OpenCore/OpenCore-{self.opencore_build}.zip")

    @property
    def plist_template(self):
        return self.payload_path / Path("Config/config.plist")

    # Mount Location
    @property
    def payload_mnt1_path(self):
        return self.payload_path / Path("mnt1")

    # ACPI
    @property
    def pci_ssdt_path(self):
        return self.payload_path / Path("ACPI/SSDT-CPBG.aml")

    @property
    def windows_ssdt_path(self):
        return self.payload_path / Path("ACPI/SSDT-PCI.aml")

    # Drivers
    @property
    def nvme_driver_path(self):
        return self.payload_path / Path("Drivers/NvmExpressDxe.efi")

    @property
    def exfat_legacy_driver_path(self):
        return self.payload_path / Path("Drivers/ExFatDxeLegacy.efi")

    @property
    def xhci_driver_path(self):
        return self.payload_path / Path("Drivers/XhciDxe.efi")

    # Kexts
    @property
    def payload_kexts_path(self):
        return self.payload_path / Path("Kexts")

    @property
    def lilu_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/Lilu-v{self.lilu_version}.zip")

    @property
    def whatevergreen_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/WhateverGreen-v{self.whatevergreen_version}.zip")

    @property
    def airportbcrmfixup_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/AirportBrcmFixup-v{self.airportbcrmfixup_version}.zip")

    @property
    def restrictevents_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/RestrictEvents-v{self.restrictevents_version}.zip")

    @property
    def restrictevents_mbp_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/RestrictEvents-MBP91-v{self.restrictevents_mbp_version}.zip")

    @property
    def bcm570_path(self):
        return self.payload_kexts_path / Path(f"Ethernet/CatalinaBCM5701Ethernet-v{self.bcm570_version}.zip")

    @property
    def marvel_path(self):
        return self.payload_kexts_path / Path(f"Ethernet/MarvelYukonEthernet-v{self.marvel_version}.zip")

    @property
    def nforce_path(self):
        return self.payload_kexts_path / Path(f"Ethernet/nForceEthernet-v{self.nforce_version}.zip")

    @property
    def mce_path(self):
        return self.payload_kexts_path / Path(f"Misc/AppleMCEReporterDisabler-v{self.mce_version}.zip")

    @property
    def mousse_path(self):
        return self.payload_kexts_path / Path(f"SSE/AAAMouSSE-v{self.mousse_version}.zip")

    @property
    def telemetrap_path(self):
        return self.payload_kexts_path / Path(f"SSE/telemetrap-v{self.telemetrap_version}.zip")

    @property
    def corecaptureelcap_path(self):
        return self.payload_kexts_path / Path(f"Wifi/corecaptureElCap-v{self.corecaptureelcap_version}.zip")

    @property
    def io80211elcap_path(self):
        return self.payload_kexts_path / Path(f"Wifi/IO80211ElCap-v{self.io80211elcap_version}.zip")

    @property
    def io80211high_sierra_path(self):
        return self.payload_kexts_path / Path(f"Wifi/IO80211HighSierra-v{self.io80211high_sierra_version}.zip")

    @property
    def io80211mojave_path(self):
        return self.payload_kexts_path / Path(f"Wifi/IO80211Mojave-v{self.io80211mojave_version}.zip")

    @property
    def applealc_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/AppleALC-v{self.applealc_version}.zip")

    @property
    def piixata_path(self):
        return self.payload_kexts_path / Path(f"Misc/AppleIntelPIIXATA-v{self.piixata_version}.zip")

    @property
    def backlight_injector_path(self):
        return self.payload_kexts_path / Path(f"Misc/BacklightInjector-v{self.backlight_injector_version}.zip")

    @property
    def cpufriend_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/CPUFriend-v{self.cpufriend_version}.zip")

    @property
    def smcspoof_path(self):
        return self.payload_kexts_path / Path(f"Misc/SMC-Spoof-v{self.smcspoof_version}.zip")

    @property
    def btspoof_path(self):
        return self.payload_kexts_path / Path(f"Misc/Bluetooth-Spoof-v{self.btspoof_version}.zip")

    @property
    def nvmefix_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/NVMeFix-v{self.nvmefix_version}.zip")

    @property
    def featureunlock_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/FeatureUnlock-v{self.featureunlock_version}.zip")

    @property
    def debugenhancer_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/DebugEnhancer-v{self.debugenhancer_version}.zip")

    @property
    def bluetool_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/BlueToolFixup-v{self.bluetool_version}.zip")

    @property
    def cslvfixup_path(self):
        return self.payload_kexts_path / Path(f"Acidanthera/CSLVFixup-v{self.cslvfixup_version}.zip")

    @property
    def innie_path(self):
        return self.payload_kexts_path / Path(f"Misc/Innie-v{self.innie_version}.zip")

    @property
    def latebloom_path(self):
        return self.payload_kexts_path / Path(f"Misc/latebloom-v{self.latebloom_version}.zip")

    @property
    def apple_trackpad_path(self):
        return self.payload_kexts_path / Path(f"Misc/AppleUSBTrackpad-v{self.apple_trackpad}.zip")

    @property
    def plist_folder_path(self):
        return self.payload_kexts_path / Path("Plists")

    @property
    def platform_plugin_plist_path(self):
        return self.plist_folder_path / Path("PlatformPlugin")

    @property
    def fw_family_path(self):
        return self.payload_kexts_path / Path(f"FireWire/IOFireWireFamily-v{self.fw_kext}.zip")

    @property
    def fw_sbp2_path(self):
        return self.payload_kexts_path / Path(f"FireWire/IOFireWireSBP2-v{self.fw_kext}.zip")

    @property
    def fw_bus_path(self):
        return self.payload_kexts_path / Path(f"FireWire/IOFireWireSerialBusProtocolTransport-v{self.fw_kext}.zip")

    # Build Location
    @property
    def build_path(self):
        return self.current_path / Path("Build-Folder/")

    @property
    def opencore_release_folder(self):
        return self.build_path / Path(f"OpenCore-Build")

    @property
    def opencore_zip_copied(self):
        return self.build_path / Path(f"OpenCore-{self.opencore_build}.zip")

    @property
    def oc_folder(self):
        return self.opencore_release_folder / Path("EFI/OC/")

    @property
    def plist_path(self):
        return self.oc_folder / Path("config.plist")

    @property
    def acpi_path(self):
        return self.oc_folder / Path("ACPI")

    @property
    def drivers_path(self):
        return self.oc_folder / Path("Drivers")

    @property
    def kexts_path(self):
        return self.oc_folder / Path("Kexts")

    @property
    def resources_path(self):
        return self.oc_folder / Path("Resources")

    @property
    def map_kext_folder(self):
        return self.kexts_path / Path("USB-Map.kext")

    @property
    def map_contents_folder(self):
        return self.map_kext_folder / Path("Contents")

    @property
    def pp_kext_folder(self):
        return self.kexts_path / Path("CPUFriendDataProvider.kext")

    @property
    def pp_contents_folder(self):
        return self.pp_kext_folder / Path("Contents")

    @property
    def agdp_kext_folder(self):
        return self.kexts_path / Path("AGDP-Override.kext")

    @property
    def agdp_contents_folder(self):
        return self.agdp_kext_folder / Path("Contents")

    @property
    def agpm_kext_folder(self):
        return self.kexts_path / Path("AGPM-Override.kext")

    @property
    def agpm_contents_folder(self):
        return self.agpm_kext_folder / Path("Contents")

    @property
    def amc_kext_folder(self):
        return self.kexts_path / Path("AMC-Override.kext")

    @property
    def amc_contents_folder(self):
        return self.amc_kext_folder / Path("Contents")

    # Tools
    @property
    def macserial_path(self):
        return self.payload_path / Path("Tools/macserial")

    @property
    def gfxutil_path(self):
        return self.payload_path / Path("Tools/gfxutil")

    @property
    def vault_path(self):
        return self.payload_path / Path("Tools/CreateVault/sign.command")

    @property
    def ocvalidate_path(self):
        return self.payload_path / Path("Tools/ocvalidate")

    # Icons
    @property
    def app_icon_path(self):
        return self.payload_path / Path("OC-Patcher.icns")

    @property
    def icon_path_external(self):
        return self.payload_path / Path("Icon/External/.VolumeIcon.icns")

    @property
    def icon_path_internal(self):
        return self.payload_path / Path("Icon/Internal/.VolumeIcon.icns")

    @property
    def icon_path_sd(self):
        return self.payload_path / Path("Icon/SD-Card/.VolumeIcon.icns")

    @property
    def icon_path_ssd(self):
        return self.payload_path / Path("Icon/SSD/.VolumeIcon.icns")

    @property
    def gui_path(self):
        return self.payload_path / Path("Icon/Resources.zip")

    # Apple Payloads Paths

    @property
    def payload_apple_root_path_zip(self):
        return self.payload_path / Path("Apple.zip")

    @property
    def payload_apple_root_path(self):
        return self.payload_path / Path("Apple")

    @property
    def payload_apple_kexts_path(self):
        return self.payload_apple_root_path / Path("Extensions")

    @property
    def payload_apple_coreservices_path(self):
        return self.payload_apple_root_path / Path("CoreServices")

    @property
    def payload_apple_usr_path(self):
        return self.payload_apple_root_path / Path("usr")

    @property
    def payload_apple_libexec_path(self):
        return self.payload_apple_usr_path / Path("libexec")

    @property
    def payload_apple_frameworks_path(self):
        return self.payload_apple_root_path / Path("Frameworks")

    @property
    def payload_apple_frameworks_path_accel(self):
        return self.payload_apple_frameworks_path / Path("Graphics-Acceleration")

    @property
    def payload_apple_frameworks_path_accel_ts2(self):
        return self.payload_apple_frameworks_path / Path("Graphics-Acceleration-TeraScale-2")

    @property
    def payload_apple_frameworks_path_accel_ivy(self):
        return self.payload_apple_frameworks_path / Path("Graphics-Acceleration-Ivy-Bridge")

    @property
    def payload_apple_lauchd_path(self):
        return self.payload_apple_root_path / Path("LaunchDaemons")

    @property
    def payload_apple_lauchd_path_accel(self):
        return self.payload_apple_lauchd_path / Path("Graphics-Acceleration")

    @property
    def payload_apple_private_frameworks_path(self):
        return self.payload_apple_root_path / Path("PrivateFrameworks")

    @property
    def payload_apple_private_frameworks_path_accel(self):
        return self.payload_apple_private_frameworks_path / Path("Graphics-Acceleration")

    @property
    def payload_apple_private_frameworks_path_accel_ts2(self):
        return self.payload_apple_private_frameworks_path / Path("Graphics-Acceleration-TeraScale-2")

    @property
    def payload_apple_private_frameworks_path_accel_ivy(self):
        return self.payload_apple_private_frameworks_path / Path("Graphics-Acceleration-Ivy-Bridge")

    @property
    def payload_apple_private_frameworks_path_brightness(self):
        return self.payload_apple_private_frameworks_path / Path("Brightness-Control")

    # Apple Extensions
    # El Capitan Extensions
    @property
    def audio_path(self):
        return self.payload_apple_kexts_path / Path("Audio")

    # High Sierra Extensions
    @property
    def audio_v2_path(self):
        return self.payload_apple_kexts_path / Path("Audio-v2")

    # GPU Kexts and Bundles

    @property
    def legacy_graphics(self):
        return self.payload_apple_kexts_path / Path("Graphics-Acceleration")

    @property
    def legacy_nvidia_path(self):
        return self.legacy_graphics / Path("Nvidia-Tesla")

    @property
    def legacy_nvidia_kepler_path(self):
        return self.legacy_graphics / Path("Nvidia-Kepler")

    @property
    def legacy_amd_path(self):
        return self.legacy_graphics / Path("AMD-TeraScale")

    @property
    def legacy_amd_path_ts2(self):
        return self.legacy_graphics / Path("AMD-TeraScale-2")

    @property
    def legacy_intel_gen1_path(self):
        return self.legacy_graphics / Path("Intel-Gen5-Ironlake")

    @property
    def legacy_intel_gen2_path(self):
        return self.legacy_graphics / Path("Intel-Gen6-SandyBridge")

    @property
    def legacy_intel_gen3_path(self):
        return self.legacy_graphics / Path("Intel-Gen7-IvyBridge")

    @property
    def legacy_general_path(self):
        return self.legacy_graphics / Path("General-Patches")

    @property
    def legacy_brightness(self):
        return self.payload_apple_kexts_path / Path("Brightness-Control")
    
    @property
    def legacy_mux_path(self):
        return self.payload_apple_kexts_path / Path("Legacy-Mux")

    @property
    def legacy_wifi_coreservices(self):
        return self.payload_apple_coreservices_path / Path("Legacy-Wifi")

    @property
    def legacy_wifi_libexec(self):
        return self.payload_apple_libexec_path / Path("Legacy-Wifi")

    sbm_values = [
        "j137ap",  #  iMacPro1,1
        "j680ap",  #  MacBookPro15,1
        "j132ap",  #  MacBookPro15,2
        "j174ap",  #  Macmini8,1
        "j140kap",  # MacBookAir8,1
        "j780ap",  #  MacBookPro15,3
        "j213ap",  #  MacBookPro15,4
        "j140aap",  # MacBookAir8,2
        "j152fap",  # MacBookPro16,1
        "j160ap",  #  MacPro7,1
        "j230kap",  # MacBookAir9,1
        "j214kap",  # MacBookPro16,2
        "j223ap",  #  MacBookPro16,3
        "j215ap",  #  MacBookPro16,4
        "j185ap",  #  iMac20,1
        "j185fap",  # iMac20,2
        # "x86legacy",  # non-T2 Macs/VMs, Monterey's boot.efi enforces this on all Macs
    ]

    sandy_board_id = [
        "Mac-E43C1C25D4880AD6",  # MacBookPro12,1
        "Mac-06F11F11946D27C5",  # MacBookPro11,5
        "Mac-9F18E312C5C2BF0B",  # MacBookAir7,1
        "Mac-937CB26E2E02BB01",  # MacBookAir7,2
        "Mac-35C5E08120C7EEAF",  # Macmini7,1
        "Mac-7BA5B2D9E42DDD94",  # iMacPro1,1
    ]