import os
import re
import time
import _winreg
from _winreg import *

check_names = ['Xcitium - Secure Internet Gateway Agent', 'Comodo - Secure Internet Gateway Agent']

def reg(name):

    blacklist=name

    def collectprograms(rtkey,pK,kA):
        try:
            list=[]
            oK=_winreg.OpenKey(rtkey,pK,0,kA)
            i=0
            while True:
                try:
                    bkey=_winreg.EnumKey(oK,i)
                    vkey=os.path.join(pK,bkey)
                    oK1=_winreg.OpenKey(rtkey,vkey,0,kA)
                    try:
                        DN,bla=_winreg.QueryValueEx(oK1,'DisplayName')
                        inlist=[DN.strip(), vkey, pK]
                        list.append(inlist)
                        
                    except:
                        pass
                    i+=1
                except:
                    break
        except:
            pass
        return list

        
    uninstallkey_32='SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Uninstall'

    if 'PROGRAMFILES(X86)' in os.environ.keys():
        
        rklist=[(_winreg.HKEY_LOCAL_MACHINE,uninstallkey_32,_winreg.KEY_WOW64_32KEY | _winreg.KEY_READ),
                (_winreg.HKEY_LOCAL_MACHINE,uninstallkey_32,_winreg.KEY_WOW64_64KEY | _winreg.KEY_READ),
                (_winreg.HKEY_CURRENT_USER,uninstallkey_32,_winreg.KEY_WOW64_32KEY | _winreg.KEY_READ),
                (_winreg.HKEY_CURRENT_USER,uninstallkey_32,_winreg.KEY_WOW64_64KEY | _winreg.KEY_READ)]
    else:
        
        rklist=[(_winreg.HKEY_LOCAL_MACHINE,uninstallkey_32,_winreg.KEY_READ),
                (_winreg.HKEY_CURRENT_USER,uninstallkey_32,_winreg.KEY_READ)]

    bet=[]
    for i in rklist:
        col=collectprograms(i[0], i[1], i[2])
        for c in col:
            if blacklist in c:
                bet.append(c[1])

    if not bet:
        got=0
    else:
        for i in bet:
            j=i.replace(" ", '" "')
            v='\\'
            path="HKEY_LOCAL_MACHINE"+v+i
            path1="HKEY_LOCAL_MACHINE"+v+j
            got=path1
            
    return got
            

def uninstall (find, name):
    os.popen('reg delete "HKEY_LOCAL_MACHINE\SOFTWARE\WOW6432Node\COMODO\SIG Agent" /v pwd-hash /f').read()
    command="MsiExec.exe /X"+find+" /qn CESMCONTEXT=1 REBOOT=REALLYSUPPRESS"
    uninst=os.popen(command).read()
    time.sleep(150)
    fin=reg(name)
    if fin!=0:
        print("%s is still present on the Endpoint. Uninstallation failed" % name)
    else:
        print("%s has been uninstalled successfully" % name)

for name in check_names:
    fin=reg(name)
    if fin!=0:
        fina=fin.split('\\')[-1]
        final1=re.findall('{.*}',fina)[0]
        print("GUID from Registry is : "+final1)
        print("%s is installed on the Endpoint" % name)
        print("Uninstalling has started ")
        uninstall(final1, name)
        break
    else:
        print("%s is not installed on the Endpoint" % name)