add upgrade feature

Author: SaicharanKandukuri

install.sh

@@ -53,9 +53,23 @@ if [[ -f /usr/bin/stopvnc ]]; then
     rm -rvf /usr/bin/stopvnc
 fi
 
+if [[ -f /usr/bin/udroid-upgrade-check ]]; then
+    rm -rvf /usr/bin/udroid-upgrade-check
+fi
+
+if [[ -f /usr/bin/udroid-upgrade ]]; then
+    rm -rvf /usr/bin/udroid-upgrade
+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"
 }
+ln -sv /usr/share/udroid/main.sh /usr/bin/udroid-upgrade-check || {
+    die "Failed to create symlink"
+}
+ln -sv /usr/share/udroid/main.sh /usr/bin/udroid-upgrade || {
+    die "Failed to create symlink"
+}

main.py

@@ -23,9 +23,11 @@ def vnc_mode(port=1,mode="start",xstartup=os.getenv('HOME')+"/.vnc/xstartup"):
 
 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)
+    parser.add_option("--upgrade"   ,action="store_true", default=False, help="Installs update if found updates")
+    parser.add_option("--upgrade-check",action="store_true", default=False, help="check for updates")
+    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()
 
@@ -35,4 +37,12 @@ def vnc_mode(port=1,mode="start",xstartup=os.getenv('HOME')+"/.vnc/xstartup"):
     if options.stopvnc:
         vnc_mode(port=options.port,mode="stop")
         sys.exit(0)
+    if options.upgrade:
+        from utils.upgrade import *
+        do_upgrade()
+        sys.exit(0)
+    if options.upgrade_check:
+        from utils.upgrade import *
+        upgrade_check()
+        sys.exit(0)
 

main.sh

@@ -8,4 +8,10 @@ case $(echo "$0" | cut -d "/" -f 4) in
     "stopvnc")
         python3 /usr/share/udroid/main.py --stopvnc $*
     ;;
+    "udroid-upgrade")
+        python3 /usr/share/udroid/main.py --upgrade
+    ;;
+    "udroid-upgrade-check")
+        python3 /usr/share/udroid/main.py --upgrade-check
+    ;;
 esac

utils/upgrade.py

@@ -0,0 +1,41 @@
+import os
+from utils.funs import *
+
+CURRENT_VERSION = "v0.1"
+REMOTE_VERISON = os.popen("git -c 'versionsort.suffix=-' ls-remote --tags --sort='v:refname' https://github.com/RandomCoderOrg/udroid-extra-tool-proot | tail -n1 | cut -d \"/\" -f 3").read().strip()
+
+def isold():
+    if CURRENT_VERSION == REMOTE_VERISON:
+        return False
+    else:
+        return True
+
+def isconnected():
+    if os.system("ping -W 1 -c 1 github.com >> /dev/null") == 0:
+        return True
+    else:
+        return False
+
+def upgrade_check():
+    if isconnected():
+        if isold():
+            lwarn("You are using an old version of udroid-extra-tool-proot. Please update to the latest version.")
+            print("Current version: " + str(CURRENT_VERSION))
+            print("Latest version: " + str(REMOTE_VERISON))
+            print("[W] You can update by running: "+green("udroid-upgrade"))
+        else:
+            nmsg("udroid-tools are already on the latest version.")
+    else:
+        lwarn("You are not connected to the internet. Please connect to the internet and try again.")
+
+def do_upgrade():
+    if isconnected():
+        if isold():
+            print("Upgrading...")
+            if os.path.exists("/usr/share/udroid/utils/upgrade.sh"):
+                os.system("/usr/share/udroid/utils/upgrade.sh")
+            else:
+                lwarn("Could not find upgrade script.")
+                die  ("Try installing from the official repository.")
+        else:
+            nmsg("udroid-tools are already on the latest version.")

utils/upgrade.sh

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+repo_link="https://github.com/RandomCoderOrg/udroid-extra-tool-proot"
+_basename="$(basnemae $repo_link)"
+if [ -d "$_basename" ]; then
+    rm -rf "$_basename"
+fi
+
+git clone "$repo_link"
+cd "$_basename" || exit 1
+bash install.sh