init upload

packed with
vnc managed
- startvnc stable
- stopvnc unstable

Author: SaicharanKandukuri

.gitignore

@@ -0,0 +1 @@
+__pycache__

install.sh

@@ -0,0 +1,61 @@
+#!/bin/bash
+
+_c_magneta="\e[95m"
+_c_green="\e[32m"
+_c_red="\e[31m"
+_c_blue="\e[34m"
+RST="\e[0m"
+
+die()    { echo -e "${_c_red}[E] ${*}${RST}";exit 1;:;}
+warn()   { echo -e "${_c_red}[W] ${*}${RST}";:;}
+shout()  { echo -e "${_c_blue}[-] ${*}${RST}";:;}
+lshout() { echo -e "${_c_blue}-> ${*}${RST}";:;}
+imsg()	 { if [ -n "$UDROID_VERBOSE" ]; then echo -e ": ${*} \e[0m" >&2;fi;:;}
+msg()    { echo -e "${*} \e[0m" >&2;:;}
+
+if ! command -v python3 >/dev/null 2>&1; then
+    die "Python3 is not installed"
+fi
+
+if (( UID != 0 )); then
+    die "You need to be root to run this script"
+fi
+
+shout "Installing udroidmgr"
+mkdir -pv /usr/share/udroid || {
+    die "Failed to create /usr/share/udroid"
+}
+
+if [ -d utils ]; then
+    cp -rv utils /usr/share/udroid/ || {
+        die "Failed to copy utils"
+    }
+fi
+
+if [ -f main.py ]; then
+    cp -v main.py /usr/share/udroid/ || {
+        die "Failed to copy main.py"
+    }
+fi
+
+if [ -f main.sh ]; then
+    cp -v main.sh /usr/share/udroid/ || {
+        die "Failed to copy main.sh"
+    }
+fi
+
+
+if [[ -f /usr/bin/startvnc ]]; then
+    rm -rvf /usr/bin/startvnc
+fi
+
+if [[ -f /usr/bin/stopvnc ]]; then
+    rm -rvf /usr/bin/stopvnc
+fi
+
+ln -sv /usr/share/udroid/main.sh /usr/bin/startvnc || {
+    die "Failed to create symlink"
+}
+ln -sv /usr/share/udroid/main.sh /usr/bin/stopvnc || {
+    die "Failed to create symlink"
+}

main.py

@@ -0,0 +1,38 @@
+import os
+import sys
+import optparse
+from utils.info import *
+from utils.funs import *
+
+def vnc_mode(port=1,mode="start",xstartup=os.getenv('HOME')+"/.vnc/xstartup"):
+    import utils.vnc as vnc
+    
+    if os.path.exists(xstartup):
+        if mode == "start":
+            if vnc.startvnc(port):
+                print(vncokdialog(port))
+            else:
+                print(vncrunningdialog(port))
+        elif mode == "stop":
+            if vnc.stopvnc(port):
+                print(vnckilldialog(port))
+            else:
+                print("No VNC server running at "+str(port))
+    else:
+        print("xstartup file not found")
+
+if __name__ == '__main__':
+    parser =  optparse.OptionParser()
+    parser.add_option("--startvnc",action='store_true', default=False, help="Start VNC on port")
+    parser.add_option("--stopvnc" ,action='store_true', default=False, help="Stop VNC on port")
+    parser.add_option("-p", "--port", help="VNC port", default=1)
+
+    (options, args) = parser.parse_args()
+
+    if options.startvnc:
+        vnc_mode(port=options.port,mode="start")
+        sys.exit(0)
+    if options.stopvnc:
+        vnc_mode(port=options.port,mode="stop")
+        sys.exit(0)
+    

main.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+
+case $(echo "$0" | cut -d "/" -f 4) in
+    "startvnc")
+        python3 /usr/share/udroid/main.py --startvnc $*
+    ;;
+    "stopvnc")
+        python3 /usr/share/udroid/main.py --stopvnc $*
+    ;;
+esac

utils/funs.py

@@ -0,0 +1,22 @@
+def lwarn(msg):
+    print("[W] " + msg)
+
+def nmsg(msg):
+    print("* " + msg)
+
+def die(msg):
+    print("[E] " + msg)
+    exit(1)
+
+## COLORS
+def red(msg):
+    return "\033[1;31m" + msg + "\033[0m"
+
+def blue(msg):
+    return "\033[1;34m" + msg + "\033[0m"
+
+def green(msg):
+    return "\033[1;32m" + msg + "\033[0m"
+
+def magneta(msg):
+    return "\033[1;35m" + msg + "\033[0m"

utils/info.py

@@ -0,0 +1,26 @@
+import os
+from utils.funs import *
+
+######## VNC
+HOSTPUBIP=os.system("hostname -I | cut -d ' ' -f 2")
+LOCALIP="127.0.0.1"
+
+def vncokdialog(port):
+    return "vncserver started on "\
+        +"\n"+magneta(str(LOCALIP)+":"+str(port))\
+        +"\n"+magneta(str(LOCALIP)+":"+str(port))\
+        +"\n"+"To stop it, run:\n"\
+        +green("vncserver -kill :"+str(port))\
+        +" or "+blue("stopvnc")
+
+def vncrunningdialog(port):
+    return "vncserver already started on "\
+        +"\n"+magneta(str(LOCALIP)+":"+str(port))\
+        +"\n"+magneta(str(LOCALIP)+":"+str(port))\
+        +"\n"+"To stop it, run:\n"\
+        +green("vncserver -kill :"+str(port))\
+        +" or "+blue("stopvnc")
+
+def vnckilldialog(port):
+    return "vncserver stopped on "+str(LOCALIP)+" port "+str(port)
+

utils/vnc.py

@@ -0,0 +1,44 @@
+import os
+
+DISPLAY=os.getenv('DISPLAY')
+VNC_LOCK_TYPE="/tmp/X11-unix/X"
+
+def isvncstarted(port:int) -> bool:
+    """
+    Check if VNC is started
+    """
+    if os.path.exists(VNC_LOCK_TYPE+str(port)) \
+        or os.path.exists("/tmp/X"+str(port)+"-lock"):
+        return True
+    else:
+        return False
+
+def startvnc(port:int) -> bool:
+    """
+    Start VNC
+    """
+    if os.getenv('DEFAULT_VNC_PORT') is not None and os.getenv('DEFAULT_VNC_PORT').isnumeric:
+        port=int(os.getenv('DEFAULT_VNC_PORT'))
+
+    if isvncstarted(port):
+        return False
+    else:
+        if os.WEXITSTATUS(os.system("vncserver :"+str(port))) == 0:
+            return True
+        else:
+            return False
+
+def stopvnc(port:int) -> bool:
+    """
+    Stop VNC
+    """
+    if os.getenv('DEFAULT_VNC_PORT') is not None and os.getenv('DEFAULT_VNC_PORT').isnumeric:
+        port = int(os.getenv('DEFAULT_VNC_PORT'))
+
+    if isvncstarted(port):
+        if os.WEXITSTATUS(os.system("vncserver -kill :"+str(port))) == 0:
+            return True
+        else:
+            return False
+    else:
+        return False