Weilbyte
3 years ago
committed by
GitHub
12 changed files with 88 additions and 274 deletions
@ -1,191 +0,0 @@ |
|||
#!/usr/bin/python3 |
|||
|
|||
import os |
|||
import sys |
|||
import time |
|||
import subprocess |
|||
import urllib.request |
|||
import os.path |
|||
import shutil |
|||
import argparse |
|||
|
|||
ACTION = None |
|||
|
|||
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', |
|||
'dd_icon-die.svg', 'dd_clear-trigger.png'] |
|||
|
|||
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(): |
|||
if ACTION == None: |
|||
print("\033c", end="") |
|||
|
|||
def cprint(color, text, bold=False, inline=False): |
|||
endc = '\n' |
|||
if inline: |
|||
endc = '' |
|||
if bold: |
|||
print(colors.BOLD) |
|||
print(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, '\nPVE ' + ver + ' detected.') |
|||
|
|||
def checkConn(): |
|||
try: |
|||
urllib.request.urlopen('http://github.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 or "<script type='text/javascript' src='/pve2/js/dd_patcher.js'></script>" 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', '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(baseURL + '/PVEDiscordDark/sass/PVEDiscordDark.css', '/usr/share/pve-manager/css/dd_style.css') |
|||
cprint(colors.NORMAL, 'Downloading patcher..') |
|||
urllib.request.urlretrieve(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>") |
|||
for index, image in enumerate(images): |
|||
imageCurrent = index + 1 |
|||
cprint(colors.NORMAL, 'Downloading images [' + str(imageCurrent) + '/' + str(len(images)) + ']..\r', False, True) |
|||
urllib.request.urlretrieve(baseURL + '/PVEDiscordDark/images/' + image, '/usr/share/pve-manager/images/' + image) |
|||
cprint(colors.OKGREEN, '\nTheme installed successfully!', True) |
|||
if ACTION == None: |
|||
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/javascript/extjs/charts.js.bak'): |
|||
cprint(colors.NORMAL, 'Reverting charts.js replacement..') |
|||
os.remove('/usr/share/javascript/extjs/charts.js') |
|||
shutil.copyfile('/usr/share/javascript/extjs/charts.js.bak', '/usr/share/javascript/extjs/charts.js') |
|||
os.remove('/usr/share/javascript/extjs/charts.js.bak') |
|||
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('/usr/share/pve-manager/images/' + asset) |
|||
cprint(colors.OKGREEN, '\n\nTheme uninstalled successfully!', True) |
|||
if ACTION == None: |
|||
cprint(colors.NORMAL, 'Press [ENTER] to go back.') |
|||
input('') |
|||
doMainMenu() |
|||
|
|||
def doHeader(): |
|||
cprint(colors.HEADER, '[~]', True, True) |
|||
cprint(colors.NORMAL, ' PVEDiscordDark Utility (DEPRECATED)\nThis installer is now deprecated, please use PVEDiscordDark.sh\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(): |
|||
parser = argparse.ArgumentParser(description='PVEDiscordDark Theme Utility (DEPRECATED)') |
|||
parser.add_argument('--action', '-a', choices=['install', 'uninstall'], help='action for unattended mode') |
|||
args = parser.parse_args() |
|||
global ACTION |
|||
ACTION = args.action |
|||
checkPVE() |
|||
checkConn() |
|||
time.sleep(0.5) |
|||
if ACTION == None: |
|||
try: |
|||
doMainMenu() |
|||
except KeyboardInterrupt: |
|||
print('\n') |
|||
exit(0) |
|||
else: |
|||
if ACTION == 'install': |
|||
installTheme() |
|||
else: |
|||
if themeIsInstalled(): |
|||
uninstallTheme() |
|||
|
|||
if __name__ == "__main__": |
|||
main() |
After Width: | Height: | Size: 1.6 KiB |
File diff suppressed because one or more lines are too long
@ -0,0 +1,61 @@ |
|||
# Script to assist with PVEDiscordDark development |
|||
# |
|||
# By default serves HTTP on port 3000, any *.js request gets the JS script, any *.css request gets the CSS file and any image request gets corresponding image |
|||
# Meant to be used with the "Requestly" browser extension to redirect PVEDD requests from PVE server to localhost:3000 |
|||
# |
|||
|
|||
from http.server import HTTPServer, BaseHTTPRequestHandler |
|||
import json |
|||
import os |
|||
|
|||
PORT = 3000 |
|||
DIR_SASS = os.path.join(os.path.dirname(__file__), "sass") |
|||
DIR_IMAGES = os.path.join(os.path.dirname(__file__), "images") |
|||
DIR_JS = os.path.join(os.path.dirname(__file__), "js") |
|||
|
|||
|
|||
class Server(BaseHTTPRequestHandler): |
|||
def log_message(self, format, *args): |
|||
return |
|||
|
|||
def _set_headers(self, status, type): |
|||
self.send_response(status) |
|||
self.send_header("Content-type", type) |
|||
self.end_headers() |
|||
|
|||
def do_GET(self): |
|||
status = 200 |
|||
type = "application/json" |
|||
data = None |
|||
|
|||
file = self.path.rpartition("/")[2] |
|||
ext = file.rpartition(".")[2] |
|||
|
|||
if ext == "css": |
|||
data = open(os.path.join(DIR_SASS, "PVEDiscordDark.css"), "rb").read() |
|||
type = "text/css" |
|||
elif ext == "js": |
|||
data = open(os.path.join(DIR_JS, "PVEDiscordDark.js"), "rb").read() |
|||
type = "application/javascript" |
|||
elif ext == "png" or ext == "jpg" or ext == "jpeg": |
|||
try: |
|||
data = open(os.path.join(DIR_IMAGES, file), "rb").read() |
|||
type = f"image/{ext}" |
|||
except FileNotFoundError: |
|||
status = 404 |
|||
else: |
|||
status = 400 |
|||
self._set_headers(status, type) |
|||
if status == 200: |
|||
self.wfile.write(data) |
|||
else: |
|||
self.wfile.write(json.dumps({"error": status}).encode()) |
|||
|
|||
|
|||
if __name__ == "__main__": |
|||
print(f"Serving on localhost:{PORT}") |
|||
server = HTTPServer(server_address=("", PORT), RequestHandlerClass=Server) |
|||
try: |
|||
server.serve_forever() |
|||
except KeyboardInterrupt: |
|||
quit() |
@ -1,73 +0,0 @@ |
|||
#!/bin/bash |
|||
#Script by RisedSky to allow users to update their python script of ProxmoxDiscordDark |
|||
#Created : 05/12/2020 |
|||
|
|||
|
|||
currentuser=`whoami | grep 'root'` |
|||
default="\e[39m" |
|||
green="\e[32m" |
|||
red="\e[91m" |
|||
yellow="\e[93m" |
|||
|
|||
clear |
|||
echo -ne "Verifying that you're root to prevent perm errors \r" |
|||
sleep 1 |
|||
if [ -z "$currentuser" ] |
|||
then |
|||
echo -ne "Verifying that you're root to prevent perm errors$red [ERROR] $default \r" |
|||
echo -ne "\nYou are not root !" |
|||
exit 1 |
|||
fi |
|||
|
|||
echo -ne "Verifying that you're root to prevent perm errors$green [OK] $default \r" |
|||
echo -ne '\n' |
|||
|
|||
sleep 1 |
|||
echo -ne '\n' |
|||
|
|||
sleep 1 |
|||
echo -ne '# (0%) Checking the latestversion\r' |
|||
latestversion=`curl -s "https://api.github.com/repos/Weilbyte/PVEDiscordDark/releases/latest" | awk -F '"' '/tag_name/{print $4}'` |
|||
echo -ne '####################### (100%) Checking the latestversion\r' |
|||
sleep 1 |
|||
echo -ne '\n' |
|||
|
|||
file="./currentversion.txt" |
|||
currentversion="" |
|||
if [ ! -f "$file" ] |
|||
then |
|||
echo "$latestversion" > "$file" |
|||
currentversion="$latestversion" |
|||
else |
|||
currentversion=`cat $file` |
|||
fi |
|||
|
|||
echo "Your version : $currentversion" |
|||
echo "Latest version : $latestversion" |
|||
|
|||
if [ ! "$currentversion" = "$latestversion" ] |
|||
then |
|||
#Not up tp date |
|||
read -p "$(echo -e $yellow)[Warning] Your version is different, would you like to update ? $(echo -e $default)[y/n]" -n 1 -r |
|||
echo "" # (optional) move to a new line |
|||
if [[ $REPLY =~ ^[YyOo]$ ]] |
|||
then |
|||
echo -ne "Downloading...\n" |
|||
wget -O "PVEDiscordDark.py" "https://raw.githubusercontent.com/Weilbyte/PVEDiscordDark/master/PVEDiscordDark.py" |
|||
echo "$latestversion" > $file |
|||
read -p "$(echo -e $yellow)[Warning] Start the updated script ? $(echo -e $default)[y/n]" -n 1 -r |
|||
echo "" # (optional) move to a new line |
|||
if [[ $REPLY =~ ^[YyOo]$ ]] |
|||
then |
|||
python3 PVEDiscordDark.py |
|||
exit 0 |
|||
fi |
|||
else |
|||
echo -ne "Exiting...\n" |
|||
exit 0 |
|||
fi |
|||
else |
|||
#Up to date |
|||
echo -e "$green""You have the latest version of the script ! $default" |
|||
exit 0 |
|||
fi |
@ -1 +1 @@ |
|||
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_tool-sprites.png dd_trigger.png dd_loading.svg dd_icon-die.svg dd_arrows.png dd_default-toolbar-small-arrow.png dd_default-toolbar-small-s-arrow.png dd_icon-cdblurp.png |
|||
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_tool-sprites.png dd_trigger.png dd_loading.svg dd_icon-die.svg dd_arrows.png dd_default-toolbar-small-arrow.png dd_default-toolbar-small-s-arrow.png dd_icon-cdblurp.png dd_date-trigger.png |
Loading…
Reference in new issue