Utility

PVEDiscordDark.py

@@ -0,0 +1,170 @@
+import sys,os
+import subprocess
+import urllib.request
+import os.path
+import shutil
+import os
+import time
+
+images = [
+    'dd_cephblurp.png', 'dd_cephwhite.png',
+    'dd_icon-cpu.png', 'dd_icon-hdd.png',
+    'dd_icon-ram.png', 'dd_icon-swap.png',
+    'dd_icon-cd.png', 'dd_icon-display.png',
+    'dd_icon-network.png', 'dd_icon-cloud.png',
+    'dd_icon-serial.png', 'dd_icon-usb.png',
+    'dd_icon-pci.png', 'dd_logo.png',
+    'dd_mini-bottom.png', 'dd_mini-top.png',
+    'dd_readme', '/dd_tool-sprites.png',
+    'dd_trigger.png', 'dd_loading.svg']
+
+class colors:
+    HEADER = '\033[95m'
+    OKBLUE = '\033[94m'
+    OKGREEN = '\033[92m'
+    WARNING = '\033[93m'
+    FAIL = '\033[91m'
+    NORMAL = '\033[0m'
+    BOLD = '\033[1m'
+    UNDERLINE = '\033[4m'
+
+def clear():
+    print("\033c", end="")
+
+def cprint(color, text, bold=False, inline=False):
+    endc = '\n'
+    if inline:
+        endc = ''
+    if bold:
+        print(colors.BOLD)
+    print(f'{color}{text}{colors.NORMAL}', end=endc)
+
+def getPVEVersion():
+    pv = subprocess.check_output('pveversion --verbose | grep pve-manager | cut -c 14- | cut -c -6', shell=True, stderr=open(os.devnull, 'w'))
+    pv = pv.decode('ascii').replace(" ", "").replace("\n","")
+    if (('/bin/' in pv) or (len(pv) > 7)):
+        return ""
+    else:
+        return pv
+
+def checkPVE():
+    ver = getPVEVersion()
+    if (len(ver) < 1):
+        cprint(colors.FAIL, 'Unable to detect Proxmox.', True)
+        cprint(colors.NORMAL, 'A Proxmox installation could not be detected.')
+        exit(1)
+    else:
+        cprint(colors.OKGREEN, f'\nPVE {ver} detected.')
+
+def checkConn():
+    try:
+        urllib.request.urlopen('http://google.com')
+        return
+    except:
+        cprint(colors.FAIL, 'An Internet connection is required to install PVEDiscordDark.', True)
+        cprint(colors.NORMAL, 'Connect to the Internet and try again.')
+        exit(1)
+
+def themeIsInstalled():
+    tplUsesTheme = False
+    tplFile = open('/usr/share/pve-manager/index.html.tpl')
+    tplLines = tplFile.readlines()
+    for line in tplLines:
+        if "<link rel='stylesheet' type='text/css' href='/pve2/css/dd_style.css'>" in line:
+            tplUsesTheme = True
+    if (os.path.isfile('/usr/share/pve-manager/css/dd_style.css') or tplUsesTheme):
+        return True
+    return False
+
+def installTheme():
+    clear()
+    doHeader()
+    baseURL = os.getenv('BASEURL', f'https://github.com/Weilbyte/PVEDiscordDark/raw/{os.getenv("BRANCH", "master")}')
+    cprint(colors.NORMAL, '\nBacking up index template file..')
+    shutil.copyfile('/usr/share/pve-manager/index.html.tpl', '/usr/share/pve-manager/index.html.tpl.bak')
+    cprint(colors.NORMAL, 'Downloading stylesheet..')
+    urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/sass/PVEDiscordDark.css', '/usr/share/pve-manager/css/dd_style.css')
+    cprint(colors.NORMAL, 'Downloading patcher..')
+    urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/js/PVEDiscordDark.js', '/usr/share/pve-manager/js/dd_patcher.js')
+    cprint(colors.NORMAL, 'Applying stylesheet and patcher..')
+    with open('/usr/share/pve-manager/index.html.tpl', 'a') as tplFile:
+        tplFile.write("<link rel='stylesheet' type='text/css' href='/pve2/css/dd_style.css'>")
+        tplFile.write("<script type='text/javascript' src='/pve2/js/dd_patcher.js'></script>")
+    cprint(colors.NORMAL, 'Replacing charts.js..')
+    if os.path.exists('/usr/share/javascript/extjs/charts.js'):
+        shutil.copyfile('/usr/share/javascript/extjs/charts.js', '/usr/share/javascript/extjs/charts.js.bak')
+        os.remove('/usr/share/javascript/extjs/charts.js')
+    urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/js/charts.js', '/usr/share/javascript/extjs/charts.js')
+    for index, image in enumerate(images):
+        cprint(colors.NORMAL, f'Downloading images [{index + 1}/{len(images)}]..\r', False, True)
+        urllib.request.urlretrieve(f'{baseURL}/PVEDiscordDark/images/{image}', f'/usr/share/pve-manager/images/{image}')
+    cprint(colors.OKGREEN, '\nTheme installed successfully!', True)
+    cprint(colors.NORMAL, 'Press [ENTER] to go back.')
+    input('')
+    doMainMenu()
+
+def uninstallTheme():
+    clear()
+    doHeader()
+    cprint(colors.NORMAL, '\nCleaning up index template file..')
+    with open('/usr/share/pve-manager/index.html.tpl', 'r+') as tplFile:
+        tplLines = tplFile.readlines()
+        tplFile.seek(0)
+        for line in tplLines:
+            if ("<link rel='stylesheet' type='text/css' href='/pve2/css/dd_style.css'>" not in line or "<script type='text/javascript' src='/pve2/js/dd_patcher.js'></script>" not in line):
+                tplFile.write(line)
+        tplFile.truncate()
+    if os.path.exists('/usr/share/pve-manager/css/dd_style.css'):
+        cprint(colors.NORMAL, 'Removing stylesheet..')
+        os.remove('/usr/share/pve-manager/css/dd_style.css')
+    if os.path.exists('/usr/share/pve-manager/js/dd_patcher.js'):
+        cprint(colors.NORMAL, 'Removing patcher..')
+        os.remove('/usr/share/pve-manager/js/dd_patcher.js')
+    cprint(colors.NORMAL, 'Removing images..')
+    for asset in os.listdir('/usr/share/pve-manager/images/'):
+        if asset.startswith('dd_'):
+            os.remove(f'/usr/share/pve-manager/images/{asset}')
+    cprint(colors.OKGREEN, '\n\nTheme uninstalled successfully!', True)
+    cprint(colors.NORMAL, 'Press [ENTER] to go back.')
+    input('')
+    doMainMenu()
+
+def doHeader():
+    cprint(colors.HEADER, '[~]', True, True)
+    cprint(colors.NORMAL, ' PVEDiscordDark Utility\n', False, True)
+
+def doMainMenu():
+    clear()
+    doHeader()
+    isInstalled = themeIsInstalled()
+    cprint(colors.NORMAL, '[I]', True, True)
+    cprint(colors.NORMAL, ' Install theme', False, True)
+    if isInstalled:
+        cprint(colors.NORMAL, '[U]', True, True)
+        cprint(colors.NORMAL, ' Uninstall theme', False, True)
+    cprint(colors.NORMAL, '[Q]', True, True)
+    cprint(colors.NORMAL, ' Exit', False, True)
+    choice = input('\n\n>? ')
+    choice = choice.upper().replace(" ", "")
+    if choice == 'I':
+        installTheme()
+    elif (choice == 'U' and isInstalled):
+        uninstallTheme()
+    elif choice == 'Q':
+        exit(0)
+    else:
+        doMainMenu()
+
+def main():
+    checkPVE()
+    checkConn()
+    time.sleep(0.5)
+    try:
+        doMainMenu()
+    except KeyboardInterrupt:
+        print('\n')
+        exit(0)
+
+
+if __name__ == "__main__":
+    main()