#Packages
import os
import subprocess
import shutil
import time
import re
#----------------------------------------------------------------------------------------------------
start_time = time.time()
#----------------------------------------------------------------------------------------------------
#Functions
def remove(path):
    #Remove the file or directory
    if os.path.isdir(path):
        try:
            shutil.rmtree(path)
        except OSError:
            print "Unable to remove folder: %s" % path
    else:
        try:
            if os.path.exists(path):
                os.remove(path)
        except OSError:
            print "Unable to remove file: %s" % path


def chrome_cleanup(dirpath,folder_to_exclude):
    for root, dirs, files in os.walk(dirpath, topdown=True):
        for file_ in files:
            full_path = os.path.join(root, file_)
            if folder_to_exclude not in full_path:
                print 'removing -> ' + full_path
                remove(full_path)
        for folder in dirs:
            full_path = os.path.join(root, folder)
            if folder_to_exclude not in full_path:
                remove(full_path)

def firefox_cleanup(dirpath,folder_to_exclude):
    for root, dirs, files in os.walk(dirpath, topdown=True):
        for file_ in files:
            full_path = os.path.join(root, file_)
            if folder_to_exclude[0] not in full_path and folder_to_exclude[1] not in full_path:
                print 'removing -> ' + full_path
                remove(full_path)
        for folder in dirs:
            full_path = os.path.join(root, folder)
            if folder_to_exclude[0] not in full_path and folder_to_exclude[1] not in full_path:
                remove(full_path)
                
def permissions(dirpath):
    mode=0o777
    if os.path.isdir(dirpath):
        try:
            for root,dirs,files in os.walk(dirpath,topdown=False):
                for dircs in [os.path.join(root,d) for d in dirs]:
                    os.chmod(dircs,mode)
                for s_file in [os.path.join(root,f) for f in files]:
                    os.chmod(s_file,mode)
        except Exception as E:
            print E

#path_creation
def path_creation(fil_users):
    fir_ls=[]
    chr_ls=[]
    ls=[]
    #firefox
    for i in fil_users:
        try:
            path="C:\Users\%s\AppData\Roaming\Mozilla\Firefox\Profiles"%i
            if os.path.exists(path):
                os.chdir(path)
                ls.append(os.popen('cd').read().rstrip())
        except:
            print("No Path")
    
    for i in ls:
        os.chdir(i)
        out=os.popen('dir').read()
        fir_ls.append(re.findall('\w+.default-release',out))
    fir_ls_n=[]
    for i in fir_ls:
        if i[0]:
            fir_ls_n.append(i[0])
        else:pass
    l=[]
    for i,j in zip(fir_ls_n,ls):
        l.append(os.path.join(j,i))
    #chrome
    for i in fil_users:
        chr_path='C:\Users\%s\AppData\Local\Google\Chrome\User Data\Default'%i
        if os.path.exists(chr_path):
            chr_ls.append(chr_path)
        else:
            pass
    return [l,chr_ls]
#-----------------------------------------------------------------------------------------------------
#codes
users=os.popen("net user").read().split()[5:-5]
filtered_users=[i for i in users if i!='Guest']
#print filtered_users

paths=path_creation(filtered_users)
#print paths[1]

# save_path_chrome=[]
# for i in paths[1]:
#     save_path_chrome.append(i+"\Login Data")
# tot_chrome=zip(paths[1],save_path_chrome)

save_path_fire=[]
for i in paths[0]:
    save_path_fire.append([i+"\logins.json",i+"\key4.db"])
tot_fire=zip(paths[0],save_path_fire)
    
#print paths[0]
#print paths[1]
#print save_path_chrome
#print tot_chrome
#print tot_fire[0][1]
try:
    #chrome
    for i in paths[1]:
        if os.path.exists(i):
            print("Starting Google Chrome...\n")
            print("Chrome Permission fixing...")
            permissions(i)
            print("Cleaning Up Chrome....\n")
            chrome_cleanup(i,'Login Data')
            print("Chrome Completed!!!\n")
    #firefox
    for i,j in tot_fire:
        if os.path.exists(i):
            print("Starting Mozilla Firefox...\n")
            print("Firefox Permission fixing...")
            permissions(i)
            print("Cleaning Up Firefox....\n")
            firefox_cleanup(i,j)
            print("Firefox Completed!!!\n")
except Exception as E:
    print E
finally:
    print "Code Executed"
    print("--- %s seconds ---" % (time.time() - start_time))