Add script that installs or uninstalls openasar on linux (tested only on mint)

OpenAsarInjector/main.py

@@ -0,0 +1,120 @@
+ASAR = "app.asar"
+ASAR_BACKUP = ASAR + ".bak"
+SYMLINK_PATH = ""
+REAL_PATH = ""
+DISCORD_FOLDER = ""
+RESOURCES_PATH = ""
+
+def root():
+    import os
+    if os.geteuid() != 0:
+        print("Script not running as root")
+        exit()
+
+def backup():
+    import os
+    # Check for existing backup
+    if os.path.isfile(RESOURCES_PATH + ASAR_BACKUP):
+        print("Backup already exists, openasar is already installed")
+        exit()
+    try:
+        print(RESOURCES_PATH + ASAR)
+        os.rename(RESOURCES_PATH + ASAR, RESOURCES_PATH + ASAR_BACKUP)
+    except Exception as e:
+        print("Couldn't rename file: " + str(e)) 
+
+def find():
+    import os
+    from shutil import which
+
+    # Get full path of discord symlink on path
+    global SYMLINK_PATH
+    SYMLINK_PATH = which("discord")
+    # Parse real path of discord symlink
+    global REAL_PATH
+    REAL_PATH = os.path.realpath(SYMLINK_PATH)
+    # Get containing folder
+    global DISCORD_FOLDER
+    DISCORD_FOLDER = os.path.dirname(REAL_PATH)
+    # Add /resources/ to the real path
+    global RESOURCES_PATH
+    RESOURCES_PATH = DISCORD_FOLDER + "/resources/"
+
+
+def download():
+    import urllib.request
+    import json
+    # Get nightly release from github
+    # url: https://github.com/GooseMod/OpenAsar/
+    # release: https://github.com/GooseMod/OpenAsar/releases/tag/nightly
+    API_URL = "https://api.github.com/repos/GooseMod/OpenAsar/releases/latest"
+    DOWNLOAD_URL = ""
+
+    with urllib.request.urlopen(API_URL) as url:
+        data = json.loads(url.read().decode())
+        DOWNLOAD_URL = data['assets'][0]['browser_download_url']
+
+    # Download the file
+    urllib.request.urlretrieve(DOWNLOAD_URL, RESOURCES_PATH + ASAR)
+
+def kill():
+    import os
+    import signal
+    import psutil
+
+    # Get discord process
+    for proc in psutil.process_iter():
+        if proc.name() == "Discord":
+            # Kill discord
+            os.kill(proc.pid, signal.SIGTERM)
+
+def uninstall():
+    import os
+    if os.path.isfile(RESOURCES_PATH + ASAR_BACKUP):
+        try:
+            os.remove(RESOURCES_PATH + ASAR)
+            os.rename(RESOURCES_PATH + ASAR_BACKUP, RESOURCES_PATH + ASAR)
+        except Exception as e:
+            print("Couldn't remove/rename file: " + str(e))
+    else:
+        print("No backup found, openasar is not installed")
+        exit()
+
+
+# Should check if the package has been updated, as discords asar is much larger
+def sanity_check():
+    import os
+    # Get file size of asar
+    asar_size = os.path.getsize(RESOURCES_PATH + ASAR)
+    # If the asar is larger than 1MB, discord has probably updated
+    if asar_size > 1000000:
+        print("Discord has probably updated, please clear backups and reinstall")
+        print("Would you like to clear backups? (y/n)")
+        answer = input().lower()
+        if answer == "y":
+            os.remove(RESOURCES_PATH + ASAR_BACKUP)
+            print("Backups cleared, installation will continue as normal")
+        else:
+            print("Backups not cleared, please clean manually")
+            exit()
+
+
+def main():
+    import sys
+    root()
+    kill()
+    find()
+    sanity_check()
+    if len(sys.argv) > 1:
+        if sys.argv[1] == "uninstall":
+            uninstall()
+            print("Uninstalled openasar")
+        else:
+            print("Unknown argument")
+    else:
+        backup()
+        download()
+        print("Installed openasar")
+
+if __name__ == "__main__":
+    main()