n.py

import argparse
import os
from models.alias_metadata import (
    AliasMetadata,
    read_alias_metadata,
    get_param_type_from_str,
    find_alias_metadata,
    save_alias_metadata_for_command_from_file,
    generate_command_alias_file_name,
    quote_command_text
)

from custom_command.gui_alias_editor.editor_app import (
    run_editor
)

from constants import CommandType

# Null parameter value for strings
NULL_PARAM_SYMBOL = '*'

def read_param(param_str):
    if param_str == NULL_PARAM_SYMBOL:
        return None
    else:
        return param_str

parser = argparse.ArgumentParser(description="Shortcut tool")

subparsers = parser.add_subparsers(dest='alias')
subparsers.required = True

# Common paths
root_dir = os.path.dirname(__file__)
command_files_path = os.path.join(root_dir, 'data', 'command_files')
alias_metadata_file_path = os.path.join(root_dir, 'data', 'alias_metadata.json')

alias_collection = read_alias_metadata(alias_metadata_file_path)

for alias_metadata in alias_collection.aliasesMetadata:
    for alias in alias_metadata.aliases:
        alias_parser = subparsers.add_parser(alias, description=alias_metadata.description)
        for param in alias_metadata.params:
            type_value = get_param_type_from_str(param.type)
            nargs = None if param.default is None else '?'
            alias_parser.add_argument(param.name, type=type_value, help=param.description, nargs=nargs, default = param.default)

args = parser.parse_args()

alias_metadata = find_alias_metadata(args.alias, alias_collection)

if alias_metadata is None:
    raise Exception(f'Not found metadata for alias {args.alias}. Make sure this alias is registered')

if alias_metadata.type == CommandType.FINISH_NOTIFICATION.value:
    from complete_notification.done_notification import handle_done_notification
    handle_done_notification(read_param(args.op_name), read_param(args.no_sound))

if alias_metadata.type == CommandType.RETRY.value:
    from custom_command.retry_handler import execute_with_retries

    execute_with_retries(quote_command_text(read_param(args.command)), read_param(args.retry_count), read_param(args.delay_between_retries))

if alias_metadata.type == CommandType.DELAY.value:
    from custom_command.delay_handler import execute_with_delay
    execute_with_delay(quote_command_text(read_param(args.command)), read_param(args.delay_str))

if alias_metadata.type == CommandType.CUSTOM.value:
    passed_params_dict = vars(args)
    if 'alias' in passed_params_dict:
        del passed_params_dict['alias']

    from custom_command.custom_command_handler import handle_custom_command
    handle_custom_command(alias_metadata, passed_params_dict, command_files_path)

if alias_metadata.type == CommandType.REGISTER_ALIAS.value:
    aliases_list = read_param(args.aliases).split(' ')
    command_file_path = read_param(args.command_file)
    metadata_file_path = read_param(args.metadata_file)
    if command_file_path is not None and command_file_path != "":
        command_alias_file_name = generate_command_alias_file_name(aliases_list)
    else:
        command_alias_file_name = None

    alias_metadata = AliasMetadata(
        aliases=aliases_list, 
        type_value='custom',
        description=read_param(args.description),
        params=None,
        command=quote_command_text(read_param(args.command)),
        command_file=command_alias_file_name,
        param_boundary_placeholder=read_param(args.param_boundary_placeholder),
    )
    alias_collection.add_alias_metadata(alias_metadata)

    save_alias_metadata_for_command_from_file(command_file_path, metadata_file_path, command_alias_file_name, alias_metadata, command_files_path)
    alias_collection.save_to_file(alias_metadata_file_path)

    print (f"Aliases {', '.join(aliases_list)} successfully added!")

if alias_metadata.type == CommandType.REMOVE_ALIAS.value:
    aliases_list = read_param(args.aliases).split(' ')
    
    removed_aliases = alias_collection.remove_aliases_metadata(aliases_list, command_files_path)

    alias_collection.save_to_file(alias_metadata_file_path)

    if len(removed_aliases) == 0:
        print('No aliases found to remove')
    else:
        print (f"Aliases {', '.join(removed_aliases)} successfully removed!")

if alias_metadata.type == CommandType.GUI_ALIAS_EDITOR.value:
    run_editor(command_files_path, alias_metadata_file_path, alias_collection)