import os
import sys
import math
import threading
import time
import webbrowser
import modules.vars as horsy_vars
from PyQt5 import QtWidgets
import ctypes
import modules.gui as gui
import requests
import subprocess


# Hide console window (does not work on custom terminals like Windows Terminal)
ctypes.windll.user32.ShowWindow(ctypes.windll.kernel32.GetConsoleWindow(), 0)

# Initialize GUI
if __name__ == "__main__":
    app = QtWidgets.QApplication(sys.argv)

UiMainWindow = QtWidgets.QMainWindow()
ui = gui.Ui_MainWindow()
ui.setupUi(UiMainWindow)

UiLoginWindow = QtWidgets.QMainWindow()
login_ui = gui.Ui_LoginWindow()
login_ui.setupUi(UiLoginWindow)

UiDownloadWindow = QtWidgets.QMainWindow()
download_ui = gui.Ui_DownloadWindow()
download_ui.setupUi(UiDownloadWindow)

UiPackageWindow = QtWidgets.QMainWindow()
package_ui = gui.Ui_PackageWindow()
package_ui.setupUi(UiPackageWindow)

if __name__ == "__main__":
    UiMainWindow.show()
    from modules.login import loginload
    ui.username_box.setText(loginload())


# Connected functions
def refresh_gui():
    installed_apps()


def installed_apps():
    from modules.manager import apps_list
    ui.installed_table.clear()
    apps = apps_list(True)
    ui.installed_table.setColumnCount(4)
    ui.installed_table.setRowCount(math.ceil(len(apps) / 4))
    for i in range(len(apps)):
        ui.installed_table.setItem(i // 4, i % 4, QtWidgets.QTableWidgetItem(str(apps[i])))


def update_app():
    try:
        app_name = ui.installed_table.currentItem().text()
        if app_name == "":
            return
        else:
            from modules.gui_manager import install
            install(app_name)
    except:
        return


def uninstall_app():
    try:
        app_name = ui.installed_table.currentItem().text()
        if app_name == "":
            return
        else:
            from modules.gui_manager import uninstall
            uninstall(app_name)
            installed_apps()
    except:
        return


def search_gui():
    from modules.search import search
    search_query = ui.search_box.toPlainText()
    if search_query == "":
        return
    else:
        found = search(search_query, True)
        ui.search_table.clear()
        ui.search_table.setColumnCount(4)
        ui.search_table.setRowCount(math.ceil(len(found) / 4))
        for i in range(len(found)):
            ui.search_table.setItem(i // 4, i % 4, QtWidgets.QTableWidgetItem(str(found[i])))


def install_app():
    from modules.gui_manager import install
    try:
        app_name = ui.search_table.currentItem().text()
        if app_name == "":
            return
        else:
            install(app_name)
    except:
        return


def get_source_gui():
    from modules.source import get_source
    try:
        app_name = ui.search_table.currentItem().text()
        if app_name == "":
            return
        else:
            source = get_source(app_name)
            if source is not None:
                gui.popup("Error", source)
    except:
        return


def info_gui():
    from modules.search import info
    try:
        app_name = ui.search_table.currentItem().text()
        if app_name == "":
            return
        else:
            info = info(app_name, download_ui, UiDownloadWindow)
            if info is not None:
                gui.popup("Error", info)
    except:
        return


def like_gui():
    from modules.liker import like
    try:
        app_name = ui.search_table.currentItem().text()
        if app_name == "":
            return
        else:
            like = like(app_name, login_ui, UiLoginWindow)
            if 'Success' not in like:
                gui.popup("Like", like)
    except:
        return


def dislike_gui():
    from modules.liker import dislike
    try:
        app_name = ui.search_table.currentItem().text()
        if app_name == "":
            return
        else:
            dislike = dislike(app_name, login_ui, UiLoginWindow)
            if 'Success' not in dislike:
                gui.popup("Dislike", dislike)
    except:
        return


def upload_gui():
    from modules.uploader import upload
    gui.popup('Upload', str(upload(True, ui)))


def change_password_gui():
    from modules.change_password import change
    change(ui.oldpass_box.toPlainText(), ui.newpass_box.toPlainText())


def change_email_gui():
    from modules.change_email import change
    change(ui.email_box.toPlainText())


def login_logout_gui():
    from modules.login import login
    ui.username_box.setText(login())


def get_users_apps():
    def get_threaded():
        apps = requests.get(f"{horsy_vars.protocol}{horsy_vars.server_url}/users/public"
                            f"/{ui.username_box.text()}").json()
        try:
            if apps['message'] == 'Too many requests':
                while apps['message'] == 'Too many requests':
                    time.sleep(0.5)
                    apps = requests.get(f"{horsy_vars.protocol}{horsy_vars.server_url}/users/public"
                                        f"/{ui.username_box.text()}").json()
        except:
            pass
        try:
            apps = apps['packages']
            ui.manage_packages_table.clear()
            ui.manage_packages_table.setColumnCount(4)
            ui.manage_packages_table.setRowCount(math.ceil(len(apps) / 4))
            for i in range(len(apps)):
                ui.manage_packages_table.setItem(i // 4, i % 4, QtWidgets.QTableWidgetItem(str(apps[i])))
        except:
            pass
    threading.Thread(target=get_threaded).start()


def gui_package_edit():
    from modules.package_edit import edit
    try:
        app_name = ui.manage_packages_table.currentItem().text()
        if app_name == "":
            return
        else:
            UiPackageWindow.show()
            edit(app_name, UiPackageWindow)
    except:
        return


# Run functions on startup
if __name__ == "__main__":
    # Checking directories and files
    if not os.path.exists(horsy_vars.horsypath + 'apps'):
        os.makedirs(horsy_vars.horsypath + 'apps')
    if not os.path.isfile(horsy_vars.horsypath + 'config.cfg'):
        with open(horsy_vars.horsypath + 'config.cfg', 'w') as f:
            f.write('{}')

    # Checking version
    try:
        f = open(horsy_vars.horsypath + 'apps/version', 'r')
    except:
        gui.popup('Error', 'Horsy may be not installed correctly. Please reinstall it or stop another instances if '
                           'running. If you installed it just now, please restart PC.')
    version = int(f.read())
    if int(requests.get('https://github.com/horsy-ml/horsy/raw/master/web_vars/version').text) > version:
        gui.popup('Update', 'New version available! \nWe appreciate your safety, so you need to update horsy.'
                            '\nPress OK and updater will download the latest version.\n'
                            'If you see this message again, or horsy doesn\'t launch \n'
                            'itself for long time, please type horsy_updater in your terminal.')
        try:
            with open(os.path.join(horsy_vars.horsypath) + '/horsy_updater.exe', 'wb') as f:
                f.write(requests.get('https://github.com/horsy-ml/horsy/raw/master/bin/horsy_updater.exe').content)
        except:
            gui.popup('Error', 'Could not download updater. \nMaybe installation folder is not writable '
                               '(only for admins).\n Please reinstall horsy or update it manually. \n'
                               'Click OK, download file that will open browser and launch it.\n'
                               'Afterwards, delete updater file and launch horsy again.')
            webbrowser.open('https://github.com/horsy-ml/horsy/raw/master/bin/horsy_updater.exe')
        UiMainWindow.close()
        subprocess.Popen('horsy_updater.exe horsygui', shell=True, close_fds=True)
        sys.exit()

    get_users_apps()
    installed_apps()

    # Binds
    ui.tabWidget.currentChanged.connect(refresh_gui)
    ui.update_button.clicked.connect(update_app)
    ui.delete_button.clicked.connect(uninstall_app)
    ui.search_button.clicked.connect(search_gui)
    ui.install_button.clicked.connect(install_app)
    ui.source_button.clicked.connect(get_source_gui)
    ui.info_button.clicked.connect(info_gui)
    ui.search_table.itemDoubleClicked.connect(info_gui)
    ui.like_button.clicked.connect(like_gui)
    ui.dislike_button.clicked.connect(dislike_gui)
    ui.upload_button.clicked.connect(upload_gui)
    ui.regmessage_button.clicked.connect(lambda: webbrowser.open(f"{horsy_vars.protocol}{horsy_vars.server_url}"
                                                                 f"/registration"))
    ui.changepass_button.clicked.connect(change_password_gui)
    ui.changeemail_button.clicked.connect(change_email_gui)
    ui.loginlogout_button.clicked.connect(login_logout_gui)
    ui.manage_packages_table.itemDoubleClicked.connect(gui_package_edit)

    # Handle GUI exiting to exit whole program
    sys.exit(app.exec_())