CurseBreaker.py

import os
import io
import re
import sys
import time
import gzip
import glob
import shutil
import pickle
import zipfile
import requests
import platform
import pyperclip
import subprocess
from csv import reader
from shlex import split
from pathlib import Path
from datetime import datetime
from contextlib import nullcontext
from rich import box
from rich.text import Text
from rich.rule import Rule
from rich.table import Table
from rich.panel import Panel
from rich.console import Console, detect_legacy_windows
from rich.control import Control
from rich.progress import Progress, BarColumn
from rich.traceback import Traceback, install
from multiprocessing import freeze_support
from prompt_toolkit import PromptSession, HTML
from prompt_toolkit.shortcuts import confirm
from prompt_toolkit.completion import WordCompleter, NestedCompleter
from distutils.version import StrictVersion
from CB import HEADERS, HEADLESS_TERMINAL_THEME, __version__
from CB.Core import Core, DependenciesParser
from CB.Compat import pause, timeout, clear, set_terminal_title, set_terminal_size, KBHit
from CB.Wago import WagoUpdater

if platform.system() == 'Windows':
    from ctypes import windll, wintypes

# FIXME - Python bug #39010
import asyncio
import selectors
asyncio.set_event_loop(asyncio.SelectorEventLoop(selectors.SelectSelector()))


class TUI:
    def __init__(self):
        self.core = Core()
        self.session = PromptSession(reserve_space_for_menu=6, complete_in_thread=True)
        self.headless = False
        self.console = None
        self.table = None
        self.slugs = None
        self.completer = None
        self.os = platform.system()
        install()

    def start(self):
        # Check if headless mode was requested
        if len(sys.argv) == 2 and sys.argv[1].lower() == 'headless':
            self.headless = True
        self.setup_console()
        self.print_header()
        self.core.init_master_config()
        # Check if executable is in good location
        if not glob.glob('World*.app') and not glob.glob('Wow*.exe') or \
                not os.path.isdir(Path('Interface/AddOns')) or not os.path.isdir('WTF'):
            self.console.print('[bold red]This executable should be placed in the same directory where Wow.exe, '
                               'WowClassic.exe or World of Warcraft.app is located. Additionally, make sure that '
                               'this WoW installation was started at least once.[/bold red]\n')
            pause(self.headless)
            sys.exit(1)
        # Detect client flavor
        if 'CURSEBREAKER_FLAVOR' in os.environ:
            flavor = os.environ.get('CURSEBREAKER_FLAVOR')
        else:
            flavor = os.path.basename(os.getcwd())
        if flavor in {'_retail_', '_ptr_'}:
            self.core.clientType = 'wow_retail'
        elif flavor in {'_classic_', '_classic_ptr_'}:
            self.core.clientType = 'wow_burning_crusade'
            set_terminal_title(f'CurseBreaker v{__version__} - Burning Crusade')
        elif flavor in {'_classic_era_', '_classic_era_ptr_'}:
            self.core.clientType = 'wow_classic'
            set_terminal_title(f'CurseBreaker v{__version__} - Classic')
        else:
            self.console.print('[bold red]This client release is currently unsupported by CurseBreaker.[/bold red]\n')
            pause(self.headless)
            sys.exit(1)
        # Check if client have write access
        try:
            with open('PermissionTest', 'w') as _:
                pass
            os.remove('PermissionTest')
        except IOError:
            self.console.print('[bold red]CurseBreaker doesn\'t have write rights for the current directory.\n'
                               'Try starting it with administrative privileges.[/bold red]\n')
            pause(self.headless)
            sys.exit(1)
        self.auto_update()
        try:
            self.core.init_config()
        except RuntimeError:
            self.console.print('[bold red]The config file is corrupted. Restore the earlier version from backup.'
                               '[/bold red]\n')
            pause(self.headless)
            sys.exit(1)
        self.setup_table()
        # CurseForge URI Support
        if len(sys.argv) == 2 and any(x in sys.argv[1] for x in {'twitch://', 'curseforge://'}):
            try:
                self.c_install(sys.argv[1].strip())
            except Exception as e:
                self.handle_exception(e)
            timeout(self.headless)
            sys.exit(0)
        # Wago URI Support
        if len(sys.argv) == 2 and sys.argv[1].startswith('weakauras-companion://wago/push/'):
            try:
                self.core.config['WAStash'].append(sys.argv[1].strip().replace('weakauras-companion://wago/push/', ''))
                self.core.config['WAStash'] = list(set(self.core.config['WAStash']))
                self.core.save_config()
                self.c_wago_update(_, flush=False)
            except Exception as e:
                self.handle_exception(e)
            timeout(self.headless)
            sys.exit(0)
        if len(sys.argv) == 2 and '.ccip' in sys.argv[1]:
            try:
                path = sys.argv[1].strip()
                self.c_install(self.core.parse_cf_payload(path))
                if os.path.exists(path):
                    os.remove(path)
            except Exception as e:
                self.handle_exception(e)
            timeout(self.headless)
            sys.exit(0)
        # CLI command
        if len(sys.argv) >= 2:
            command = ' '.join(sys.argv[1:]).split(' ', 1)
            if command[0].lower() == 'headless':
                pass
            elif getattr(self, f'c_{command[0].lower()}', False):
                try:
                    getattr(self, f'c_{command[0].lower()}')(command[1].strip() if len(command) > 1 else False)
                except Exception as e:
                    self.handle_exception(e)
                sys.exit(0)
            else:
                self.console.print('Command not found.')
                sys.exit(0)
        # Addons auto update
        if len(self.core.config['Addons']) > 0 and self.core.config['AutoUpdate']:
            if not self.headless:
                self.console.print('Automatic update of all addons will start in 5 seconds.\n'
                                   'Press any button to enter interactive mode.', highlight=False)
            keypress = self.handle_keypress(5)
            if not keypress:
                if not self.headless:
                    self.print_header()
                try:
                    self.motd_parser()
                    self.c_update(None, True)
                    if self.core.backup_check():
                        self.setup_table()
                        self.console.print(f'\n[green]Backing up WTF directory{"!" if self.headless else ":"}[/green]')
                        self.core.backup_wtf(None if self.headless else self.console)
                    if self.core.config['WAUsername'] != 'DISABLED':
                        self.console.print('')
                        with nullcontext() if self.headless else self.console.status("Processing Wago data"):
                            self.setup_table()
                            self.c_wago_update(None, False)
                except Exception as e:
                    self.handle_exception(e)
                self.console.print('')
                self.print_log()
                if self.headless:
                    sys.exit(0)
                else:
                    self.console.print('Press [bold]I[/bold] to enter interactive mode or any other button to close'
                                       ' the application.')
                    keypress = self.handle_keypress(0)
                    if keypress and keypress.lower() in [b'i', 'i']:
                        pass
                    else:
                        sys.exit(0)
        if self.headless:
            sys.exit(1)
        self.setup_completer()
        self.print_header()
        self.console.print('Use command [green]help[/green] or press [green]TAB[/green] to see a list of available comm'
                           'ands.\nCommand [green]exit[/green] or pressing [green]CTRL+D[/green] will close the applica'
                           'tion.\n')
        if len(self.core.config['Addons']) == 0:
            self.console.print('Command [green]import[/green] might be used to detect already installed addons.\n')
        self.motd_parser()
        if self.core.backup_check():
            self.console.print(f'[green]Backing up WTF directory:[/green]')
            self.core.backup_wtf(self.console)
            self.console.print('')
        # Prompt session
        while True:
            try:
                command = self.session.prompt(HTML('<ansibrightgreen>CB></ansibrightgreen> '), completer=self.completer)
            except KeyboardInterrupt:
                continue
            except EOFError:
                break
            else:
                command = command.split(' ', 1)
                if getattr(self, f'c_{command[0].lower()}', False):
                    try:
                        self.setup_table()
                        getattr(self, f'c_{command[0].lower()}')(command[1].strip() if len(command) > 1 else False)
                        self.setup_completer()
                    except Exception as e:
                        self.handle_exception(e)
                else:
                    self.console.print('Command not found.')

    def auto_update(self):
        if getattr(sys, 'frozen', False) and 'CURSEBREAKER_VARDEXMODE' not in os.environ:
            try:
                if os.path.isfile(sys.executable + '.old'):
                    try:
                        os.remove(sys.executable + '.old')
                    except PermissionError:
                        pass
                try:
                    payload = requests.get('https://api.github.com/repos/AcidWeb/CurseBreaker/releases/latest', headers=HEADERS, timeout=5).json()
                except (requests.exceptions.ConnectionError, requests.exceptions.Timeout):
                    return
                if 'name' in payload and 'body' in payload and 'assets' in payload:
                    remoteversion = payload['name']
                    changelog = payload['body']
                    url = None
                    for binary in payload['assets']:
                        if (self.os == 'Windows' and '.exe' in binary['name'])\
                                or (self.os == 'Darwin' and '.zip' in binary['name'])\
                                or (self.os == 'Linux' and '.gz' in binary['name']):
                            url = binary['browser_download_url']
                            break
                    if url and StrictVersion(remoteversion[1:]) > StrictVersion(__version__):
                        self.console.print('[green]Updating CurseBreaker...[/green]')
                        shutil.move(sys.executable, sys.executable + '.old')
                        payload = requests.get(url, headers=HEADERS, timeout=5)
                        if self.os == 'Darwin':
                            zipfile.ZipFile(io.BytesIO(payload.content)).extractall(path=os.path.dirname(
                                os.path.abspath(sys.executable)))
                        else:
                            with open(sys.executable, 'wb') as f:
                                if self.os == 'Windows':
                                    f.write(payload.content)
                                elif self.os == 'Linux':
                                    f.write(gzip.decompress(payload.content))
                        os.chmod(sys.executable, 0o775)
                        self.console.print(f'[bold green]Update complete! The application will be restarted now.'
                                           f'[/bold green]\n\n[green]Changelog:[/green]\n{changelog}\n')
                        self.print_log()
                        pause(self.headless)
                        subprocess.call([sys.executable] + sys.argv[1:])
                        sys.exit(0)
            except Exception as e:
                if os.path.isfile(sys.executable + '.old'):
                    shutil.move(sys.executable + '.old', sys.executable)
                self.console.print(f'[bold red]Update failed!\n\nReason: {str(e)}[/bold red]\n')
                self.print_log()
                pause(self.headless)
                sys.exit(1)

    def motd_parser(self):
        payload = requests.get('https://storage.googleapis.com/cursebreaker/motd', headers=HEADERS, timeout=5)
        if payload.status_code == 200:
            self.console.print(Panel(payload.content.decode('UTF-8'), title='MOTD', border_style='red'))
            self.console.print('')

    def handle_exception(self, e, table=True):
        if self.table.row_count > 1 and table:
            self.console.print(self.table)
        if getattr(sys, 'frozen', False) and 'CURSEBREAKER_DEBUG' not in os.environ:
            sys.tracebacklimit = 0
            width = 0
        else:
            width = 100
        if isinstance(e, list):
            for es in e:
                self.console.print(Traceback.from_exception(exc_type=es.__class__, exc_value=es,
                                                            traceback=es.__traceback__, width=width))
        else:
            self.console.print(Traceback.from_exception(exc_type=e.__class__, exc_value=e,
                                                        traceback=e.__traceback__, width=width))

    def handle_keypress(self, wait):
        if not self.headless:
            kb = KBHit()
        starttime = time.time()
        keypress = None
        while True:
            # noinspection PyUnboundLocalVariable
            if self.headless:
                break
            elif kb.kbhit():
                keypress = kb.getch()
                break
            elif wait and time.time() - starttime > wait:
                break
            time.sleep(0.01)
        if not self.headless:
            kb.set_normal_term()
        return keypress

    def print_header(self):
        if self.headless:
            self.console.print(f'[bold green]CurseBreaker[/bold green] [bold red]v{__version__}[/bold red] | '
                               f'[yellow]{datetime.now()}[/yellow]', highlight=False)
        else:
            clear()
            self.console.print(Rule(f'[bold green]CurseBreaker[/bold green] [bold red]v{__version__}[/bold red]'))
            self.console.print('')

    def print_log(self):
        if self.headless:
            html = self.console.export_html(inline_styles=True, theme=HEADLESS_TERMINAL_THEME)
            with open('CurseBreaker.html', 'a+', encoding='utf-8') as log:
                log.write(html)

    def setup_console(self):
        if self.headless:
            self.console = Console(record=True)
            if self.os == 'Windows':
                window = windll.kernel32.GetConsoleWindow()
                if window:
                    windll.user32.ShowWindow(window, 0)
        elif detect_legacy_windows():
            set_terminal_size(100, 50)
            windll.kernel32.SetConsoleScreenBufferSize(windll.kernel32.GetStdHandle(-11), wintypes._COORD(100, 200))
            self.console = Console(width=97)
        else:
            self.console = Console()

    def setup_completer(self):
        if not self.slugs:
            # noinspection PyBroadException
            try:
                self.slugs = pickle.load(gzip.open(io.BytesIO(
                    requests.get('https://storage.googleapis.com/cursebreaker/slugs.pickle.gz',
                                 headers=HEADERS, timeout=5).content)))
            except Exception:
                self.slugs = {'cf': [], 'wowi': [], 'tukui': []}
        addons = []
        for addon in sorted(self.core.config['Addons'], key=lambda k: k['Name'].lower()):
            addons.append(addon['Name'])
        slugs = ['ElvUI', 'Tukui']
        for item in self.slugs['cf']:
            slugs.append(f'cf:{item}')
        for item in self.slugs['wowi']:
            slugs.append(f'wowi:{item}')
        for item in self.slugs['ty']:
            slugs.append(f'ty:{item}')
        slugs.extend(['ElvUI:Dev', 'Tukui:Dev', 'Shadow&Light:Dev'])
        accounts = []
        for account in self.core.detect_accounts():
            accounts.append(account)
        self.completer = NestedCompleter.from_nested_dict({
            'install': WordCompleter(slugs, ignore_case=True, match_middle=True, WORD=True),
            'uninstall': WordCompleter(addons, ignore_case=True),
            'update': WordCompleter(addons, ignore_case=True),
            'force_update': WordCompleter(addons, ignore_case=True),
            'wago_update': None,
            'status': WordCompleter(addons, ignore_case=True),
            'orphans': None,
            'search': None,
            'backup': None,
            'import': {'install': None},
            'export': None,
            'toggle': {'authors': None,
                       'autoupdate': None,
                       'backup': None,
                       'channel': WordCompleter(addons + ['global'], ignore_case=True, sentence=True),
                       'compact_mode': None,
                       'pinning': WordCompleter(addons, ignore_case=True, sentence=True),
                       'wago': None},
            'set': {'wago_api': None,
                    'wago_wow_account': WordCompleter(accounts, ignore_case=True, sentence=True),
                    'gh_api': None},
            'show': {'dependencies': None},
            'uri_integration': None,
            'help': None,
            'exit': None
        })

    def setup_table(self):
        self.table = Table(box=box.SQUARE)
        self.table.add_column('Status', header_style='bold white', no_wrap=True, justify='center')
        self.table.add_column('Name / Author' if self.core.config['ShowAuthors'] else 'Name', header_style='bold white')
        self.table.add_column('Version', header_style='bold white')

    def parse_args(self, args):
        parsed = []
        for addon in sorted(self.core.config['Addons'], key=lambda k: len(k['Name']), reverse=True):
            if addon['Name'] in args or addon['URL'] in args:
                parsed.append(addon['Name'])
                args = args.replace(addon['Name'], '', 1)
        return sorted(parsed)

    def parse_link(self, text, link, dev=None, authors=None, uiversion=None):
        if dev == 1:
            dev = ' [bold][B][/bold]'
        elif dev == 2:
            dev = ' [bold][A][/bold]'
        else:
            dev = ''
        if authors and self.core.config['ShowAuthors']:
            authors.sort()
            authors = f' [bold black]by {", ".join(authors)}[/bold black]'
        else:
            authors = ''
        if uiversion and uiversion not in [self.core.masterConfig['RetailVersion'],
                                           self.core.masterConfig['ClassicVersion'],
                                           self.core.masterConfig['BurningCrusadeVersion']]:
            uiversion = ' [bold yellow][!][/bold yellow]'
        else:
            uiversion = ''
        if link:
            obj = Text.from_markup(f'[link={link}]{text}[/link]{dev}{authors}{uiversion}')
        else:
            obj = Text.from_markup(f'{text}{dev}{authors}{uiversion}')
        obj.no_wrap = True
        return obj

    def c_install(self, args, recursion=False):
        if args:
            optignore = False
            optnodeps = False
            pargs = split(args.replace("'", "\\'"))
            if '-i' in pargs:
                optignore = True
                args = args.replace('-i', '', 1)
            if '-d' in pargs:
                optnodeps = True
                args = args.replace('-d', '', 1)
            dependencies = DependenciesParser(self.core)
            args = re.sub(r'([a-zA-Z0-9_:])([ ]+)([a-zA-Z0-9_:])', r'\1,\3', args)
            addons = [addon.strip() for addon in list(reader([args], skipinitialspace=True))[0]]
            exceptions = []
            if len(addons) > 0:
                with Progress('{task.completed}/{task.total}', '|', BarColumn(bar_width=None), '|',
                              auto_refresh=False, console=self.console) as progress:
                    task = progress.add_task('', total=len(addons))
                    while not progress.finished:
                        for addon in addons:
                            try:
                                installed, name, version, deps = self.core.add_addon(addon, optignore, optnodeps)
                                if installed:
                                    self.table.add_row('[green]Installed[/green]', Text(name, no_wrap=True),
                                                       Text(version, no_wrap=True))
                                    if not recursion:
                                        dependencies.add_dependency(deps)
                                else:
                                    self.table.add_row('[bold black]Already installed[/bold black]',
                                                       Text(name, no_wrap=True), Text(version, no_wrap=True))
                            except Exception as e:
                                exceptions.append(e)
                            progress.update(task, advance=1, refresh=True)
                self.console.print(self.table)
            dependencies = dependencies.parse_dependency()
            if dependencies:
                self.setup_table()
                self.console.print('Installing dependencies:')
                self.c_install(dependencies, recursion=True)
            if len(exceptions) > 0:
                self.handle_exception(exceptions, False)
        else:
            self.console.print('[green]Usage:[/green]\n\tThis command accepts a space-separated list of links as an arg'
                               'ument.[bold white]\n\tFlags:[/bold white]\n\t\t[bold white]-d[/bold white] - Disable de'
                               'pendency parser.\n\t\t[bold white]-i[/bold white] - Disable the client version check.\n'
                               '[bold green]Supported URL:[/bold green]\n\thttps://www.curseforge.com/wow/addons/\[addo'
                               'n_name] [bold white]|[/bold white] cf:\[addon_name]\n\thttps://www.wowinterface.com/dow'
                               'nloads/\[addon_name] [bold white]|[/bold white] wowi:\[addon_id]\n\thttps://www.tukui.o'
                               'rg/addons.php?id=\[addon_id] [bold white]|[/bold white] tu:\[addon_id]\n\thttps://www.t'
                               'ukui.org/classic-addons.php?id=\[addon_id] [bold white]|[/bold white] tuc:\[addon_id]\n'
                               '\thttps://www.tukui.org/classic-tbc-addons.php?id=\[addon_id] [bold white]|[/bold white'
                               '] tubc:\[addon_id]\n\thttps://www.townlong-yak.com/addons/\[addon_name] [bold white]|[/'
                               'bold white] ty:\[addon_name]\n\thttps://github.com/\[username]/\[repository_name] [bold'
                               ' white]|[/bold white] gh:\[username]/\[repository_name]\n\tElvUI [bold white]|[/bold wh'
                               'ite] ElvUI:Dev\n\tTukui [bold white]|[/bold white] Tukui:Dev\n\tShadow&Light:Dev',
                               highlight=False)

    def c_uninstall(self, args):
        if args:
            optkeep = False
            pargs = split(args.replace("'", "\\'"))
            if '-k' in pargs:
                optkeep = True
                args = args.replace('-k', '', 1)
            addons = self.parse_args(args)
            if len(addons) > 0:
                with Progress('{task.completed}/{task.total}', '|', BarColumn(bar_width=None), '|',
                              auto_refresh=False, console=self.console) as progress:
                    task = progress.add_task('', total=len(addons))
                    while not progress.finished:
                        for addon in addons:
                            name, version = self.core.del_addon(addon, optkeep)
                            if name:
                                self.table.add_row(f'[bold red]Uninstalled[/bold red]',
                                                   Text(name, no_wrap=True), Text(version, no_wrap=True))
                            else:
                                self.table.add_row(f'[bold black]Not installed[/bold black]',
                                                   Text(addon, no_wrap=True), Text('', no_wrap=True))
                            progress.update(task, advance=1, refresh=True)
                self.console.print(self.table)
        else:
            self.console.print('[green]Usage:[/green]\n\tThis command accepts a space-separated list of addon names or '
                               'full links as an argument.\n\t[bold white]Flags:[/bold white]\n\t\t[bold white]-k[/bold'
                               ' white] - Keep the addon files after uninstalling.', highlight=False)

    def c_update(self, args, addline=False, update=True, force=False, provider=False, reversecompact=False):
        compact = not self.core.config['CompactMode'] if reversecompact else self.core.config['CompactMode']
        if args:
            addons = self.parse_args(args)
            compacted = -1
        else:
            addons = sorted(self.core.config['Addons'], key=lambda k: k['Name'].lower())
            compacted = 0
        exceptions = []
        dependencies = DependenciesParser(self.core)
        if len(addons) > 0:
            with Progress('{task.completed:.0f}/{task.total}', '|', BarColumn(bar_width=None), '|',
                          auto_refresh=False, console=None if self.headless else self.console) as progress:
                task = progress.add_task('', total=len(addons))
                if not args:
                    try:
                        self.core.bulk_check(addons)
                    except RuntimeError:
                        pass
                    self.core.bulk_check_checksum(addons, progress)
                while not progress.finished:
                    for addon in addons:
                        try:
                            name, authors, versionnew, versionold, uiversion, modified, blocked, source, sourceurl,\
                                changelog, deps, dstate = self.core.update_addon(
                                    addon if isinstance(addon, str) else addon['URL'], update, force)
                            dependencies.add_dependency(deps)
                            if provider:
                                source = f' [bold white]{source}[/bold white]'
                            else:
                                source = ''
                            if versionold:
                                if versionold == versionnew:
                                    if modified:
                                        self.table.add_row(f'[bold red]Modified[/bold red]{source}',
                                                           self.parse_link(name, sourceurl, authors=authors),
                                                           self.parse_link(versionold, changelog, dstate,
                                                                           uiversion=uiversion))
                                    else:
                                        if compact and compacted > -1:
                                            compacted += 1
                                        else:
                                            self.table.add_row(f'[green]Up-to-date[/green]{source}',
                                                               self.parse_link(name, sourceurl, authors=authors),
                                                               self.parse_link(versionold, changelog, dstate,
                                                                               uiversion=uiversion))
                                else:
                                    if modified or blocked:
                                        self.table.add_row(f'[bold red]Update suppressed[/bold red]{source}',
                                                           self.parse_link(name, sourceurl, authors=authors),
                                                           self.parse_link(versionold, changelog, dstate,
                                                                           uiversion=uiversion))
                                    else:
                                        version = self.parse_link(versionnew, changelog, dstate, uiversion=uiversion)
                                        version.stylize('yellow')
                                        self.table.add_row(
                                            f'[yellow]{"Updated" if update else "Update available"}[/yellow]{source}',
                                            self.parse_link(name, sourceurl, authors=authors),
                                            version)
                            else:
                                self.table.add_row(f'[bold black]Not installed[/bold black]{source}',
                                                   Text(addon, no_wrap=True),
                                                   Text('', no_wrap=True))
                        except Exception as e:
                            exceptions.append(e)
                        progress.update(task, advance=1 if args else 0.5, refresh=True)
            if addline:
                self.console.print('')
            self.console.print(self.table)
            dependencies = dependencies.parse_dependency()
            if dependencies and update:
                self.setup_table()
                self.console.print('Installing dependencies:')
                self.c_install(dependencies, recursion=True)
            if compacted > 0:
                self.console.print(f'Additionally [green]{compacted}[/green] addons are up-to-date.')
        else:
            self.console.print('Apparently there are no addons installed by CurseBreaker (or you provided incorrect add'
                               'on name).\nCommand [green]import[/green] might be used to detect already installed addo'
                               'ns.', highlight=False)
        if len(exceptions) > 0:
            self.handle_exception(exceptions, False)

    def c_force_update(self, args):
        if args:
            self.c_update(args, False, True, True)
        else:
            # noinspection PyTypeChecker
            answer = confirm(HTML('<ansibrightred>Execute a forced update of all addons and overwrite ALL local '
                                  'changes?</ansibrightred>'))
            if answer:
                self.c_update(False, False, True, True)

    def c_status(self, args):
        optsource = False
        optcompact = False
        if args:
            pargs = split(args.replace("'", "\\'"))
            if '-s' in pargs:
                optsource = True
                args = args.replace('-s', '', 1)
            if '-a' in pargs:
                optcompact = True
                args = args.replace('-a', '', 1)
            args = args.strip()
        self.c_update(args, False, False, False, optsource, optcompact)

    def c_orphans(self, _):
        orphansd, orphansf = self.core.find_orphans()
        self.console.print('[green]Directories that are not part of any installed addon:[/green]')
        for orphan in sorted(orphansd):
            self.console.print(orphan.replace('[GIT]', '[yellow][GIT][/yellow]'), highlight=False)
        self.console.print('\n[green]Files that are leftovers after no longer installed addons:[/green]')
        for orphan in sorted(orphansf):
            self.console.print(orphan, highlight=False)

    def c_uri_integration(self, _):
        if self.os == 'Windows':
            self.core.create_reg()
            self.console.print('CurseBreaker.reg file was created. Attempting to import...')
            out = os.system('"' + str(Path(os.path.dirname(sys.executable), 'CurseBreaker.reg')) + '"')
            if out != 0:
                self.console.print('Import failed. Please try to import REG file manually.')
            else:
                os.remove('CurseBreaker.reg')
        else:
            self.console.print('This feature is available only on Windows.')

    def c_toggle(self, args):
        if args:
            args = args.strip()
            if args.startswith('channel'):
                args = args[8:]
                if args:
                    status = self.core.dev_toggle(args)
                    if status is None:
                        self.console.print('[bold red]This addon doesn\'t exist or it is not installed yet.[/bold red]')
                    elif status == -1:
                        self.console.print('[bold red]This feature can be only used with CurseForge addons.[/bold red]')
                    elif status == 0:
                        self.console.print(
                            'All CurseForge addons are now switched' if args == 'global' else 'Addon switched',
                            'to the [yellow]beta[/yellow] channel.')
                    elif status == 1:
                        self.console.print(
                            'All CurseForge addons are now switched' if args == 'global' else 'Addon switched',
                            'to the [red]alpha[/red] channel.')
                    elif status == 2:
                        self.console.print(
                            'All CurseForge addons are now switched' if args == 'global' else 'Addon switched',
                            'to the [green]stable[/green] channel.')
                else:
                    self.console.print('[green]Usage:[/green]\n\tThis command accepts an addon name (or "global") as an'
                                       ' argument.', highlight=False)
            elif args.startswith('pinning'):
                args = args[8:]
                if args:
                    status = self.core.block_toggle(args)
                    if status is None:
                        self.console.print('[bold red]This addon does not exist or it is not installed yet.[/bold red]')
                    elif status:
                        self.console.print('Updates for this addon are now [red]suppressed[/red].')
                    else:
                        self.console.print('Updates for this addon are [green]no longer suppressed[/green].')
                else:
                    self.console.print('[green]Usage:[/green]\n\tThis command accepts an addon name as an argument.')
            elif args.startswith('wago'):
                args = args[5:]
                if args:
                    if args == self.core.config['WAUsername']:
                        self.console.print(f'Wago version check is now: [green]ENABLED[/green]\nEntries created by '
                                           f'[bold white]{self.core.config["WAUsername"]}[/bold white] are now '
                                           f'included.')
                        self.core.config['WAUsername'] = ''
                    else:
                        self.core.config['WAUsername'] = args.strip()
                        self.console.print(f'Wago version check is now: [green]ENABLED[/green]\nEntries created by '
                                           f'[bold white]{self.core.config["WAUsername"]}[/bold white] are now '
                                           f'ignored.')
                else:
                    if self.core.config['WAUsername'] == 'DISABLED':
                        self.core.config['WAUsername'] = ''
                        self.console.print('Wago version check is now: [green]ENABLED[/green]')
                    else:
                        self.core.config['WAUsername'] = 'DISABLED'
                        shutil.rmtree(Path('Interface/AddOns/WeakAurasCompanion'), ignore_errors=True)
                        self.console.print('Wago version check is now: [red]DISABLED[/red]')
                self.core.save_config()
            elif args.startswith('authors'):
                status = self.core.generic_toggle('ShowAuthors')
                self.console.print('The authors listing is on now:',
                                   '[green]ENABLED[/green]' if status else '[red]DISABLED[/red]')
            elif args.startswith('autoupdate'):
                status = self.core.generic_toggle('AutoUpdate')
                self.console.print('The automatic addon update on startup is now:',
                                   '[green]ENABLED[/green]' if status else '[red]DISABLED[/red]')
            elif args.startswith('backup'):
                status = self.core.generic_toggle('Backup', 'Enabled')
                self.console.print('Backup of WTF directory is now:',
                                   '[green]ENABLED[/green]' if status else '[red]DISABLED[/red]')
            elif args.startswith('compact_mode'):
                status = self.core.generic_toggle('CompactMode')
                self.console.print('Table compact mode is now:',
                                   '[green]ENABLED[/green]' if status else '[red]DISABLED[/red]')
            else:
                self.console.print('Unknown option.')
        else:
            self.console.print('[green]Usage:[/green]\n\t[green]toggle authors[/green]\n\t\tEnables/disables the displa'
                               'y of addon author names in the table.\n\t[green]toggle autoupdate[/green]\n\t\tEnables/'
                               'disables the automatic addon update on startup.\n\t[green]toggle backup[/green]\n\t\tEn'
                               'ables/disables automatic daily backup of WTF directory.\n\t[green]toggle channel [Name]'
                               '[/green]\n\t\tCommand accepts an addon name (or "global") as argument.\n\t\tPrioritizes'
                               ' alpha/beta versions for the provided addon.\n\t[green]toggle compact_mode [/green]\n\t'
                               '\tEnables/disables compact table mode that hides entries of up-to-date addons.\n\t[gree'
                               'n]toggle pinning [Name][/green]\n\t\tCommand accepts an addon name as argument.\n\t\tBl'
                               'ocks/unblocks updating of the provided addon.\n\t[green]toggle wago [Username][/green]'
                               '\n\t\tEnables/disables automatic Wago updates.\n\t\tIf a username is provided check wil'
                               'l start to ignore the specified author.', highlight=False)

    def c_set(self, args):
        if args:
            args = args.strip()
            if args.startswith('wago_api'):
                args = args[9:]
                if args:
                    self.console.print('Wago API key is now set.')
                    self.core.config['WAAPIKey'] = args.strip()
                    self.core.save_config()
                elif self.core.config['WAAPIKey'] != '':
                    self.console.print('Wago API key is now removed.')
                    self.core.config['WAAPIKey'] = ''
                    self.core.save_config()
                else:
                    self.console.print('[green]Usage:[/green]\n\tThis command accepts API key as an argument.')
            elif args.startswith('gh_api'):
                args = args[7:]
                if args:
                    self.console.print('GitHub API key is now set.')
                    self.core.config['GHAPIKey'] = args.strip()
                    self.core.save_config()
                elif self.core.config['GHAPIKey'] != '':
                    self.console.print('GitHub API key is now removed.')
                    self.core.config['GHAPIKey'] = ''
                    self.core.save_config()
                else:
                    self.console.print('[green]Usage:[/green]\n\tThis command accepts API key as an argument.')
            elif args.startswith('wago_wow_account'):
                args = args[17:]
                if args:
                    args = args.strip()
                    if os.path.isfile(Path(f'WTF/Account/{args}/SavedVariables/WeakAuras.lua')) or \
                            os.path.isfile(Path(f'WTF/Account/{args}/SavedVariables/Plater.lua')):
                        self.console.print(f'WoW account name set to: [bold white]{args}[/bold white]')
                        self.core.config['WAAccountName'] = args
                        self.core.save_config()
                    else:
                        self.console.print('Incorrect WoW account name.')
                else:
                    self.console.print('[green]Usage:[/green]\n\tThis command accepts the WoW account name as an'
                                       ' argument.')
            else:
                self.console.print('Unknown option.')
        else:
            self.console.print('[green]Usage:[/green]\n\t[green]set wago_api [API key][/green]\n\t\tSets Wago API key r'
                               'equired to access private entries.\n\t\tIt can be procured here: [link=https://wago.io/'
                               'account]https://wago.io/account[/link]\n\t[green]set wago_wow_account [Account name][/g'
                               'reen]\n\t\tSets WoW account used by Wago updater.\n\t\tNeeded only if compatibile addon'
                               's are used on more than one WoW account.\n\t[green]set gh_api [API key][/green]\n\t\tSe'
                               'ts GitHub API key. Might be needed to get around API rate limits.', highlight=False)

    def c_show(self, args):
        if args:
            args = args.strip()
            if args.startswith('dependencies'):
                addons = sorted(list(filter(lambda k: k['URL'].startswith('https://www.curseforge.com/wow/addons/'),
                                            self.core.config['Addons'])), key=lambda k: k['Name'].lower())
                try:
                    self.core.bulk_check(addons)
                except RuntimeError:
                    pass
                for addon in addons:
                    dependencies = DependenciesParser(self.core)
                    name, _, _, _, _, _, _, _, _, _, deps, _ = self.core.update_addon(addon['URL'], False, False)
                    dependencies.add_dependency(deps)
                    deps = dependencies.parse_dependency(output=True)
                    if len(deps) > 0:
                        self.console.print(f'[green]{name}[/green]\n{", ".join(deps)}')
            else:
                self.console.print('Unknown option.')
        else:
            self.console.print('[green]Usage:[/green]\n\t[green]show dependencies[/green]\n\t\tDisplay a list of depend'
                               'encies of all installed addons.', highlight=False)

    def c_wago_update(self, _, verbose=True, flush=True):
        if os.path.isdir(Path('Interface/AddOns/WeakAuras')) or os.path.isdir(Path('Interface/AddOns/Plater')):
            accounts = self.core.detect_accounts()
            if self.core.config['WAAccountName'] != '' and self.core.config['WAAccountName'] not in accounts:
                self.core.config['WAAccountName'] = ''
            if len(accounts) == 0:
                return
            elif len(accounts) > 1 and self.core.config['WAAccountName'] == '':
                if verbose:
                    self.console.print('More than one WoW account detected.\nPlease use [bold white]set wago_wow_accoun'
                                       't[''/bold white] command to set the correct account name.')
                else:
                    self.console.print('\n[green]More than one WoW account detected.[/green]\nPlease use [bold white]se'
                                       't wago_wow_account[/bold white] command to set the correct account name.')
                return
            elif len(accounts) == 1 and self.core.config['WAAccountName'] == '':
                self.core.config['WAAccountName'] = accounts[0]
                self.core.save_config()
            if flush and len(self.core.config['WAStash']) > 0:
                self.core.config['WAStash'] = []
                self.core.save_config()
            wago = WagoUpdater(self.core.config, self.core.masterConfig)
            if self.core.masterConfig['WagoVersion'] != self.core.config['WACompanionVersion']:
                self.core.config['WACompanionVersion'] = self.core.masterConfig['WagoVersion']
                self.core.save_config()
                force = True
            else:
                force = False
            wago.install_companion(force)
            statuswa, statusplater, statusstash = wago.update()
            if verbose:
                if len(statusstash) > 0:
                    self.console.print('[green]WeakAuras ready to install:[/green]')
                    for aura in statusstash:
                        self.console.print(aura)
                    self.console.print('\nReload the interface in the WoW client to access them.')
                elif len(statuswa[0]) > 0 or len(statuswa[1]) > 0:
                    self.console.print('[green]Outdated WeakAuras:[/green]')
                    for aura in statuswa[0]:
                        self.console.print(f'[link={aura[1]}]{aura[0]}[/link]', highlight=False)
                    self.console.print('\n[green]Detected WeakAuras:[/green]')
                    for aura in statuswa[1]:
                        self.console.print(f'[link={aura[1]}]{aura[0]}[/link]', highlight=False)
                if len(statusplater[0]) > 0 or len(statusplater[1]) > 0:
                    if len(statuswa[0]) != 0 or len(statuswa[1]) != 0:
                        self.console.print('')
                    self.console.print('[green]Outdated Plater profiles/scripts:[/green]')
                    for aura in statusplater[0]:
                        self.console.print(f'[link={aura[1]}]{aura[0]}[/link]', highlight=False)
                    self.console.print('\n[green]Detected Plater profiles/scripts:[/green]')
                    for aura in statusplater[1]:
                        self.console.print(f'[link={aura[1]}]{aura[0]}[/link]', highlight=False)
            else:
                if not self.headless:
                    self.console.control(Control.move(x=0, y=-1))
                if len(statuswa[0]) > 0:
                    self.console.print(f'\n[green]The number of outdated WeakAuras:[/green] '
                                       f'{len(statuswa[0])}', highlight=False)
                if len(statusplater[0]) > 0:
                    self.console.print(f'\n[green]The number of outdated Plater profiles/scripts:[/green] '
                                       f'{len(statusplater[0])}', highlight=False)
        elif verbose:
            self.console.print('No compatible addon is installed.')

    def c_search(self, args):
        if args:
            results = self.core.search(args)
            self.console.print('[green]Top results of your search:[/green]')
            for url in results:
                if self.core.check_if_installed(url):
                    self.console.print(f'[link={url}]{url}[/link] [yellow][Installed][/yellow]', highlight=False)
                else:
                    self.console.print(f'[link={url}]{url}[/link]', highlight=False)
        else:
            self.console.print('[green]Usage:[/green]\n\tThis command accepts a search query as an argument.')

    def c_backup(self, _):
        self.core.backup_wtf(None if self.headless else self.console)

    def c_import(self, args):
        hit, partial_hit, miss = self.core.detect_addons()
        if args == 'install' and len(hit) > 0:
            self.c_install(','.join(hit))
        else:
            self.console.print(f'[green]Addons found:[/green]')
            for addon in hit:
                self.console.print(addon, highlight=False)
            self.console.print(f'\n[yellow]Possible matches:[/yellow]')
            for addon in partial_hit:
                self.console.print(' [bold white]or[/bold white] '.join(addon), highlight=False)
            self.console.print(f'\n[red]Unknown directories:[/red]')
            for addon in miss:
                self.console.print(f'{addon}', highlight=False)
            self.console.print(f'\nExecute [bold white]import install[/bold white] command to install all detected addo'
                               f'ns.\nPossible matches need to be installed manually with the [bold white]install[/bold'
                               f' white] command.\nAddons that are available only on WoWInterface and/or Tukui are not '
                               f'detected by this process.')

    def c_export(self, _):
        payload = self.core.export_addons()
        pyperclip.copy(payload)
        self.console.print(f'{payload}\n\nThe command above was copied to the clipboard.', highlight=False)

    def c_help(self, _):
        self.console.print('[green]install [URL][/green]\n\tCommand accepts a space-separated list of links.\n\t[bold w'
                           'hite]Flags:[/bold white]\n\t\t[bold white]-d[/bold white] - Disable dependency parser.\n\t'
                           '\t[bold white]-i[/bold white] - Disable the client version check.\n'
                           '[green]uninstall [URL/Name][/green]\n\tCommand accepts a space-separated list of addon name'
                           's or full links.\n\t[bold white]Flags:[/bold white]\n\t\t[bold white]-k[/bold white] - Keep'
                           ' the addon files after uninstalling.\n'
                           '[green]update [URL/Name][/green]\n\tCommand accepts a space-separated list of addon names o'
                           'r full links.\n\tIf no argument is provided all non-modified addons will be updated.\n'
                           '[green]force_update [URL/Name][/green]\n\tCommand accepts a space-separated list of addon n'
                           'ames or full links.\n\tSelected addons will be reinstalled or updated regardless of their c'
                           'urrent state.\n\tIf no argument is provided all addons will be forcefully updated.\n'
                           '[green]wago_update[/green]\n\tCommand detects all installed WeakAuras and Plater profiles/s'
                           'cripts.\n\tAnd then generate WeakAuras Companion payload.\n'
                           '[green]status[/green]\n\tPrints the current state of all installed addons.\n\t[bold yellow]'
                           '[!][/bold yellow] mark means that the latest release is not updated yet for the current WoW'
                           ' version.\n\t[bold white]Flags:[/bold white]\n\t\t[bold white]-a[/bold white] - Temporary r'
                           'everse the table compacting option.\n\t\t[bold white]-s[/bold white] - Display the source o'
                           'f the addons.\n'
                           '[green]orphans[/green]\n\tPrints list of orphaned directories and files.\n'
                           '[green]search [Keyword][/green]\n\tExecutes addon search on CurseForge.\n'
                           '[green]backup[/green]\n\tCommand creates a backup of WTF directory.\n'
                           '[green]import[/green]\n\tCommand attempts to import already installed addons.\n'
                           '[green]export[/green]\n\tCommand prints list of all installed addons in a form suitable f'
                           'or sharing.\n'
                           '[green]toggle authors[/green]\n\tEnables/disables the display of addon author names in the '
                           'table.\n'
                           '[green]toggle autoupdate[/green]\n\tEnables/disables the automatic addon update on startup'
                           '.\n'
                           '[green]toggle backup[/green]\n\tEnables/disables automatic daily backup of WTF directory.\n'
                           '[green]toggle channel [Name][/green]\n\tCommand accepts an addon name (or "global") as argu'
                           'ment.\n\tPrioritizes alpha/beta versions for the provided addon.\n'
                           '[green]toggle compact_mode [/green]\n\tEnables/disables compact table mode that hides entri'
                           'es of up-to-date addons.\n'
                           '[green]toggle pinning [Name][/green]\n\tCommand accepts an addon name as argument.\n\tBlock'
                           's/unblocks updating of the provided addon.\n'
                           '[green]toggle wago [Username][/green]\n\tEnables/disables automatic Wago updates.\n\tIf a u'
                           'sername is provided check will start to ignore the specified author.\n'
                           '[green]set wago_api [API key][/green]\n\tSets Wago API key required to access private entri'
                           'es.\n\tIt can be procured here:'
                           ' [link=https://wago.io/account]https://wago.io/account[/link]\n'
                           '[green]set wago_wow_account [Account name][/green]\n\tSets WoW account used by Wago updater'
                           '.\n\tNeeded only if compatibile addons are used on more than one WoW account.\n[green]set g'
                           'h_api [API key][/green]\n\tSets GitHub API key. Might be needed to get around API rate limi'
                           'ts.\n'
                           '[green]show dependencies[/green]\n\tDisplay a list of dependencies of all installed addons.'
                           '\n[green]uri_integration[/green]\n\tEnables integration with CurseForge and Wago page.\n\t['
                           'i]"Install"[/i] and [i]"Send to WeakAura Companion App"[/i] buttons will now start this app'
                           'lication.\n\n[bold green]Supported URL:[/bold green]\n\thttps://www.curseforge.com/wow/addo'
                           'ns/\[addon_name] [bold white]|[/bold white] cf:\[addon_name]\n\thttps://www.wowinterface.co'
                           'm/downloads/\[addon_name] [bold white]|[/bold white] wowi:\[addon_id]\n\thttps://www.tukui.'
                           'org/addons.php?id=\[addon_id] [bold white]|[/bold white] tu:\[addon_id]\n\thttps://www.tuku'
                           'i.org/classic-addons.php?id=\[addon_id] [bold white]|[/bold white] tuc:\[addon_id]\n\thttps'
                           '://www.tukui.org/classic-tbc-addons.php?id=\[addon_id] [bold white]|[/bold white] tubc:\[ad'
                           'don_id]\n\thttps://www.townlong-yak.com/addons/\[addon_name] [bold white]|[/bold white] ty:'
                           '\[addon_name]\n\thttps://github.com/\[username]/\[repository_name] [bold white]|[/bold whit'
                           'e] gh:\[username]/\[repository_name]\n\tElvUI [bold white]|[/bold white] ElvUI:Dev\n\tTukui'
                           ' [bold white]|[/bold white] Tukui:Dev\n\tShadow&Light:Dev', highlight=False)

    def c_exit(self, _):
        sys.exit(0)


if __name__ == '__main__':
    freeze_support()
    clientpath = os.environ.get('CURSEBREAKER_PATH')
    if clientpath:
        os.chdir(clientpath)
    elif getattr(sys, 'frozen', False):
        os.chdir(os.path.dirname(os.path.abspath(sys.executable)))
    set_terminal_title(f'CurseBreaker v{__version__}')
    app = TUI()
    app.start()