import os
import subprocess
import ctypes
import ssl
import time
from urllib2 import Request, urlopen

GIMP_INSTALLER_URL = "https://cdn-patchportal-one.comodo.com/portal/packages/spm/GIMP/x64/gimp-3.2.4-setup.exe"

class disable_file_system_redirection:
    _disable = ctypes.windll.kernel32.Wow64DisableWow64FsRedirection
    _revert = ctypes.windll.kernel32.Wow64RevertWow64FsRedirection

    def __enter__(self):
        self.old_value = ctypes.c_long()
        self.success = self._disable(ctypes.byref(self.old_value))

    def __exit__(self, type, value, traceback):
        if self.success:
            self._revert(self.old_value)

def ecmd(command):
    with disable_file_system_redirection():
        process = subprocess.Popen(
            command,
            shell=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE
        )
        out, err = process.communicate()
        return out.strip() + err.strip()

def get_gimp_uninstall_string():

    reg_paths = [
        r'HKLM\SOFTWARE\Microsoft\Windows\CurrentVersion\Uninstall',
        r'HKLM\SOFTWARE\WOW6432Node\Microsoft\Windows\CurrentVersion\Uninstall'
    ]

    for reg_path in reg_paths:

        cmd = 'reg query "%s" /s /f "GIMP" | findstr "UninstallString"' % reg_path

        result = ecmd(cmd)

        lines = result.splitlines()

        for line in lines:

            line = line.strip()

            if "unins" in line.lower():

                parts = line.split("REG_SZ")

                if len(parts) > 1:
                    uninstall_string = parts[1].strip()
                    return uninstall_string

    return None

def spm_uninstall(dir, args):

    uninstall_string = get_gimp_uninstall_string()

    if not uninstall_string:
        return

    if uninstall_string.startswith('"'):
        uninstall_exe = uninstall_string.split('"')[1]
    else:
        uninstall_exe = uninstall_string.split(" ")[0]

    uninstall_cmd = '"%s" /VERYSILENT /SUPPRESSMSGBOXES /NORESTART' % uninstall_exe

    ecmd(uninstall_cmd)

    time.sleep(15)

def download_installer(file_path, url):

    headers = {'User-Agent': 'Mozilla/5.0'}

    request = Request(url, headers=headers)

    try:

        context = ssl._create_unverified_context()

        response = urlopen(request, context=context)

        with open(file_path, 'wb') as f:

            while True:

                chunk = response.read(1024 * 100)

                if not chunk:
                    break

                f.write(chunk)

        return True

    except:
        return False

def spm_install(dir, args):

    temp_path = os.environ.get("TEMP", "C:\\Windows\\Temp")

    file_name = GIMP_INSTALLER_URL.split("/")[-1]

    local_path = os.path.join(temp_path, file_name)

    if not download_installer(local_path, GIMP_INSTALLER_URL):
        return

    install_cmd = '"%s" /VERYSILENT /SUPPRESSMSGBOXES /NORESTART /SP- /ALLUSERS' % local_path

    ecmd(install_cmd)

    try:
        os.remove(local_path)
    except:
        pass

    time.sleep(20)

def spm_update(dir, args):

    spm_uninstall(None, None)

    time.sleep(20)

    spm_install(None, None)