import requests, zipfile, os, sys, shutil
from io import BytesIO

root_dir = __file__.replace("update.py", "")
filename = root_dir + "Vocard.zip"
__version__ = "v2.5.2"

def checkVersion(withMsg = False):
    resp = requests.get("https://api.github.com/repos/ChocoMeow/Vocard/releases/latest")
    version = resp.json().get("name", __version__)
    if withMsg:
        if version == __version__:
            print(f"Your bot is up-to-date! - {version}")
        else:
            print(f"Your bot is not up-to-date! This latest version is {version} and you are currently running version {__version__}\n. Run `python update --start` to update your bot!")
    return version
    
def downloadFile(version:str = None):
    if not version:
        version = checkVersion()
    print("Downloading Vocard version: " + version)
    response = requests.get("https://github.com/ChocoMeow/Vocard/archive/" + version + ".zip")
    print("Download Completed")
    unZip(response, version)

def unZip(response, version: str):
    print("Installing ...")
    zfile = zipfile.ZipFile(BytesIO(response.content))
    zfile.extractall(root_dir)

    version = version.replace("v", "")
    source_dir = root_dir + "Vocard-" + version
    if os.path.exists(source_dir):
        for filename in os.listdir(root_dir):
            if filename in ["settings.json", ".env", "Vocard-" + version]:
                continue
            filename = os.path.join(root_dir, filename)
            if os.path.isdir(os.path.join(root_dir, filename)):
                shutil.rmtree(filename)
            else:
                os.remove(filename)
        for filename in os.listdir(source_dir):
            shutil.move(os.path.join(source_dir, filename), os.path.join(root_dir, filename))
        os.rmdir(source_dir)

def start():
    try:
        downloadFile()
        if os.path.exists(filename):
            os.remove(filename)
        print("Update Successfully! Run `python main.py` to start your bot")
    except Exception as e:
        print(f"Error: {e}")

if "--start" in sys.argv:
    start()

if "--check" in sys.argv:
    checkVersion(withMsg = True)