def test_control():
console = Console(file=io.StringIO(), force_terminal=True)
console.control("FOO")
console.print("BAR")
assert console.file.getvalue() == "FOOBAR\n"
return
q.put((-time_password(pad(np, length)), np))
return
def find_password() -> str:
for password in find_password_iter():
pass
return password
if __name__ == "__main__":
from rich.console import Console
import sys
console = Console()
console.show_cursor(False)
for p in find_password_iter():
console.control("\r\033[K")
console.print(f"[red]{p}[/]", end="")
console.control("\r\033[K")
if check_password_vulnerable(p):
console.print(f"[green]{p}[/]")
else:
console.print(f"[red]Not found[/]")
console.show_cursor(True)
def test_control():
console = Console(file=io.StringIO(), force_terminal=True, _environ={})
console.control(Control.clear())
console.print("BAR")
assert console.file.getvalue() == "\x1b[2JBAR\n"
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.tipsDatabase = 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 Classic client
if os.path.basename(os.getcwd()) == '_classic_':
self.core.clientType = 'wow_classic'
set_terminal_title(f'CurseBreaker v{__version__} - Classic')
# 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)
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() == b'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
payload = requests.get('https://api.github.com/repos/AcidWeb/CurseBreaker/releases/latest',
headers=HEADERS, timeout=5).json()
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:
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):
clear()
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:
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 'WINDIR' in os.environ and 'WT_SESSION' not in os.environ and 'ALACRITTY_LOG' not in os.environ:
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}')
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,
'recommendations': 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)},
'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']]:
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 = []
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://github.com/\[username]/\[repository_name] [bold white]|[/bold white] gh:\[use'
'rname]/\[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_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)
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 len(self.core.cfCache) > 0 or len(self.core.wowiCache) > 0:
self.core.cfCache = {}
self.core.wowiCache = {}
self.core.checksumCache = {}
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)
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:
self.core.bulk_check(addons)
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.')
if len(addons) == 0:
self.console.print('Apparently there are no addons installed by CurseBreaker.\n'
'Command [green]import[/green] might be used to detect already installed addons.')
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):
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 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 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'] = '******'
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]')
def c_set(self, 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('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.')
def c_show(self, 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())
self.core.bulk_check(addons)
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)}')
def c_wago_update(self, _, verbose=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()
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(self.core.clientType, force)
statuswa, statusplater = wago.update()
if verbose:
if 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:
self.console.control('\033[A')
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_recommendations(self, _):
if not self.tipsDatabase:
# noinspection PyBroadException
try:
self.tipsDatabase = pickle.load(gzip.open(io.BytesIO(
requests.get('https://storage.googleapis.com/cursebreaker/recommendations.pickle.gz',
headers=HEADERS, timeout=5).content)))
except Exception:
self.tipsDatabase = {}
if len(self.tipsDatabase) > 0:
found = False
for tip in self.tipsDatabase:
breaker = False
for addon, data in tip['Addons'].items():
check = True if self.core.check_if_installed(addon) else False
breaker = check == data['Installed']
if breaker:
found = True
recomendation = tip["Recomendation"].replace('|n', '\n')
self.console.print(f'[bold white underline]{tip["Title"]}[/bold white underline] by [green]'
f'{tip["Author"]}[/green]\n\n{recomendation}\n', highlight=False)
if not found:
self.console.print('Not found any recommendations for you. Good job!')
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]recommendations[/green]\n\tCheck the list of currently installed addons against a co'
'mmunity-driven database of tips.\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]show dependencies[/green]\n\tDisplay a list of dependencies of all installed addons.'
'\n[green]uri_integration[/green]\n\tEnables integration with CurseForge page.\n\t[i]"Instal'
'l"[/i] button will now start this application.\n'
'\n[bold green]Supported URL:[/bold green]\n\thttps://www.curseforge.com/wow/addons/\[addon_'
'name] [bold white]|[/bold white] cf:\[addon_name]\n\thttps://www.wowinterface.com/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.tukui.org/class'
'ic-addons.php?id=\[addon_id] [bold white]|[/bold white] tuc:\[addon_id]\n\thttps://github.c'
'om/\[username]/\[repository_name] [bold white]|[/bold white] gh:\[username]/\[repository_na'
'me]\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)
