import argparse
import os
import subprocess
import sys
import requests
import modules.tui as tui
from modules.console import cls
from modules.virustotal import add_to_cfg
import modules.vars as horsy_vars

# Getting the arguments
parser = argparse.ArgumentParser(description='horsy - the best package manager')
parser.add_argument('option', help='options for horsy (install/i | uninstall/un | source/s | list/l | '
                                   'upload | search | info | like | dislike)',
                    choices=['install', 'i', 'uninstall', 'un', 'source', 's', 'list', 'l', 'upload',
                             'search', 'info', 'like', 'dislike'],
                    nargs='?')
parser.add_argument('app', help='app to do function with', nargs='?')
parser.add_argument('--vt', help='your virustotal api key (account -> api key in VT)', dest='vt_key')

args = parser.parse_args()
option = args.option
app = args.app

# Window title
os.system('title horsy')

# 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('{}')
if not os.path.isfile(horsy_vars.horsypath + 'apps/versions.json'):
    with open(horsy_vars.horsypath + 'apps/versions.json', 'w+') as f:
        f.write('{}')

# Checking version
try:
    with open(horsy_vars.horsypath + 'apps/version', 'r') as f:
        version = int(f.read())
except:
    print('Horsy may be not installed correctly. Please reinstall it or stop other horsy instances. '
          'If you installed it just now, please restart PC.')

if int(requests.get('https://github.com/horsy-ml/horsy/raw/master/web_vars/version').text) > version:
    print('New version available!')
    print('If you see this message again, or horsy doesn\'t launch itself for long time, please type '
          'horsy_updater in your terminal to update it manually.')
    input('Press enter to update...')
    print('Updating...')
    print('Please wait, if process seems closed, its OK, just wait a bit.')
    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)
    subprocess.Popen('horsy_updater.exe horsy', shell=True, close_fds=True)
    sys.exit(0)


isNoArgs = False


# Function to display logo
def log_logo():
    cls()
    print('''
 __   __  _______  ______    _______  __   __ 
|  | |  ||       ||    _ |  |       ||  | |  |
|  |_|  ||   _   ||   | ||  |  _____||  |_|  |
|       ||  | |  ||   |_||_ | |_____ |       |
|       ||  |_|  ||    __  ||_____  ||_     _|
|   _   ||       ||   |  | | _____| |  |   |  
|__| |__||_______||___|  |_||_______|  |___|  
        Search powered by Algolia
    ''')


# Checking if the user has a new VT key
if args.vt_key:
    if args.vt_key != 'disable':
        add_to_cfg(args.vt_key)
        print('VT key updated')
    else:
        print('VT disabled')
        add_to_cfg(None)
    sys.exit()

# Checking if arguments are empty to use in-app CLI
if not args.option:
    log_logo()
    option = ['install', 'uninstall', 'source', 'list', 'upload', 'search', 'info'][
        tui.menu(['install app', 'uninstall app', 'get source', 'list of installed apps',
                  'upload your app', 'search for app', 'get information about app'])]
    isNoArgs = True

if not args.app and option not in ['list', 'upload', 'update']:
    log_logo()
    print('\n')
    app = tui.get(f'Select app to {option}')

# Checking user option
match option:
    case 'upload':
        from modules.uploader import upload
        upload()
    case 'install', 'i':
        from modules.manager import install
        install(app)
    case 'uninstall', 'un':
        from modules.manager import uninstall
        uninstall(app)
    case 'source', 's':
        from modules.source import get_source
        get_source(app)
    case 'search':
        from modules.search import search
        search(app)
    case 'info':
        from modules.search import info
        info(app)
    case 'list', 'l':
        from modules.manager import apps_list
        apps_list()
    case 'like':
        from modules.liker import like
        like(app)
    case 'dislike':
        from modules.liker import dislike
        dislike(app)

if isNoArgs:
    input('[EXIT] Press enter to exit horsy...')