Elep

# -*- coding: utf-8 -*-

# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2021 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENSE.TXT

"""
This module contains an argparse command line parser
for cloudlinux-xray-manager
"""
import argparse
import logging

from xray import gettext as _
from xray.internal.utils import read_sys_id
from xray.internal.constants import advice_action_sources

def get_system_id():
    try:
        return read_sys_id()
    except Exception as e:
        logging.error('Unable to get system_id, error: %s', str(e))
        return ''

def cmd_parser_manager() -> 'argparse.ArgumentParser':
    """
    Command line parser for X Ray Manager using built-in argparse module
    :return: parser
    """
    parser = argparse.ArgumentParser(prog='cloudlinux-xray-manager',
                                     description='Utility to manage X-Ray Client')
    subparsers = parser.add_subparsers(title='Commands', dest='command',
                                       required=True)

start_parser = subparsers.add_parser('start',
                                         help=_('Start monitoring of URL by IP for COUNT of minutes or requests'),
                                         argument_default=argparse.SUPPRESS)
    start_parser.add_argument('--system_id', metavar='SYS_ID', required=True,
                              help=_('unique ID of CL+ system'))
    start_parser.add_argument('--url', metavar='URL', required=True,
                              help=_('monitoring URL'))
    start_parser.add_argument('--client_ip', metavar='IP', default='*',
                              help=_('client IP to monitor requests from'))
    counters = start_parser.add_mutually_exclusive_group()
    counters.add_argument('--time', metavar='COUNT', type=int,
                          help=_('number of minutes to monitor'))
    counters.add_argument('--request_qty', metavar='COUNT', type=int,
                          help=_('number of requests to monitor'))

stop_parser = subparsers.add_parser('stop',
                                        help=_('Stop monitoring of given TASK'))
    stop_parser.add_argument('--system_id', metavar='SYS_ID', required=True,
                             help=_('unique ID of CL+ system'))
    stop_parser.add_argument('--tracing_task_id', metavar='TASK',
                             required=True, help=_('unique ID of tracing task'))

continue_parser = subparsers.add_parser('continue',
                                            help=_('Continue monitoring of given TASK'))
    continue_parser.add_argument('--system_id', metavar='SYS_ID',
                                 required=True,
                                 help=_('unique ID of CL+ system'))
    continue_parser.add_argument('--tracing_task_id', metavar='TASK',
                                 required=True,
                                 help=_('unique ID of tracing task'))

complete_parser = subparsers.add_parser('complete',
                                            help=_('Complete given TASK'))
    complete_parser.add_argument('--system_id', metavar='SYS_ID',
                                 required=True,
                                 help=_('unique ID of CL+ system'))
    complete_parser.add_argument('--tracing_task_id', metavar='TASK',
                                 required=True,
                                 help=_('unique ID of tracing task'))

delete_parser = subparsers.add_parser('delete',
                                          help=_('Delete given TASK'))
    delete_parser.add_argument('--system_id', metavar='SYS_ID',
                               required=True,
                               help=_('unique ID of CL+ system'))
    delete_parser.add_argument('--tracing_task_id', metavar='TASK',
                               required=True,
                               help=_('unique ID of tracing task'))

enable_parser = subparsers.add_parser('enable-continuous',
                                          help=_('Schedule a URL for continuous monitoring'))
    enable_parser.add_argument('--system_id', metavar='SYS_ID',
                               required=True,
                               help=_('unique ID of CL+ system'))
    enable_parser.add_argument('--url', metavar='URL', required=True,
                               help=_('monitoring URL'))
    enable_parser.add_argument('--email', metavar='EMAIL', required=True,
                               help=_('e-mail address for reporting'))

disable_parser = subparsers.add_parser('disable-continuous',
                                           help=_('Schedule to stop continuous monitoring for URL'))
    disable_parser.add_argument('--system_id', metavar='SYS_ID',
                                required=True,
                                help=_('unique ID of CL+ system'))
    disable_parser.add_argument('--url', metavar='URL', required=True,
                                help=_('monitoring URL'))

con_start_parser = subparsers.add_parser('start-continuous',
                                             help=_('Include URL into continuous monitoring'))
    con_start_parser.add_argument('--system_id', metavar='SYS_ID',
                                  required=True,
                                  help=_('unique ID of CL+ system'))
    con_start_parser.add_argument('--url', metavar='URL', required=True,
                                  help=_('monitoring URL'))

con_stop_parser = subparsers.add_parser('stop-continuous',
                                            help=_('Exclude URL from continuous monitoring'))
    con_stop_parser.add_argument('--system_id', metavar='SYS_ID',
                                 required=True,
                                 help=_('unique ID of CL+ system'))
    con_stop_parser.add_argument('--url', metavar='URL', required=True,
                                 help=_('monitoring URL'))

list_parser = subparsers.add_parser('continuous-tracing-list',
                                        help=_('Get list of scheduled continuous monitoring tasks'))
    list_parser.add_argument('--system_id', metavar='SYS_ID',
                             required=True,
                             help=_('unique ID of CL+ system'))

task_list_parser = subparsers.add_parser('tasks-list',
                                             help=_('Get list of tasks'))
    task_list_parser.add_argument('--system_id', metavar='SYS_ID',
                                  required=True,
                                  help=_('unique ID of CL+ system'))

req_parser = subparsers.add_parser('requests-list',
                                       help=_('Get list of requests for given task'))
    req_parser.add_argument('--system_id', metavar='SYS_ID',
                            required=True,
                            help=_('unique ID of CL+ system'))
    req_parser.add_argument('--tracing_task_id', metavar='TASK',
                            required=True,
                            help=_('unique ID of tracing task'))

req_data_parser = subparsers.add_parser('request-data',
                                            help=_('Get statistics for given request'))
    req_data_parser.add_argument('--system_id', metavar='SYS_ID',
                                 required=True,
                                 help=_('unique ID of CL+ system'))
    req_data_parser.add_argument('--tracing_task_id', metavar='TASK',
                                 required=True,
                                 help=_('unique ID of tracing task'))
    req_data_parser.add_argument('--request_id', metavar='REQUEST',
                                 required=True, type=int,
                                 help=_('unique ID of request'))

enable_user_agent_parser = subparsers.add_parser('enable-user-agent',
                                                     help=_('Enable X-Ray User Agent'))
    enable_user_agent_parser.add_argument('--system_id', metavar='SYS_ID',
                                          default=get_system_id(),
                                          help=_('unique ID of CL+ system'))

disable_user_agent_parser = subparsers.add_parser('disable-user-agent',
                                                      help=_('Disable X-Ray User Agent'))
    disable_user_agent_parser.add_argument('--system_id', metavar='SYS_ID',
                                           default=get_system_id(),
                                           help=_('unique ID of CL+ system'))

status_user_agent_parser = subparsers.add_parser('user-agent-status',
                                                     help=_('Get status of X-Ray User Agent'))
    status_user_agent_parser.add_argument('--system_id', metavar='SYS_ID',
                                          required=True,
                                          help=_('unique ID of CL+ system'))

autocomplete_parser = subparsers.add_parser('autocomplete-tasks',
                                                help=_('Completes all tasks run more than expected time'))
    autocomplete_parser.add_argument('--system_id', metavar='SYS_ID', default=get_system_id(),
                                     help=_('unique ID of CL+ system'))

advanced_metrics_parser = subparsers.add_parser('advanced-metrics',
                                                   help=_('Advanced metrics tool'))
    advanced_metrics_parser.add_argument('--system_id', metavar='SYS_ID', default=get_system_id(),
                                        help=_('unique ID of CL+ system'))
    advanced_metrics_parser_group = advanced_metrics_parser.add_mutually_exclusive_group()
    advanced_metrics_parser_group.add_argument('--enable', action='store_true', required=False,
                                              help=_('enable advanced metrics'))
    advanced_metrics_parser_group.add_argument('--disable', action='store_true', required=False,
                                              help=_('disable advanced metrics'))
    advanced_metrics_parser_group.add_argument('--status', action='store_true', required=False,
                                              help=_('get status (none|enabled|disabled)'))

enable_serverwide_mode_parser = subparsers.add_parser('enable-serverwide-mode')
    enable_serverwide_mode_parser.add_argument('--system_id', metavar='SYS_ID', default=get_system_id(),
                  help=_('unique ID of CL+ system'))
    disable_serverwide_mode_parser = subparsers.add_parser('disable-serverwide-mode')
    disable_serverwide_mode_parser.add_argument('--system_id', metavar='SYS_ID', default=get_system_id(),
                  help=_('unique ID of CL+ system'))

return parser

def cmd_parser_user_manager() -> 'argparse.ArgumentParser':
    """
    Command line parser for X Ray Manager for users using built-in argparse module
    :return: parser
    """
    parser = argparse.ArgumentParser(prog='cloudlinux-xray-user-manager.py',
                                     description='Utility to manage X-Ray Client for end-users')
    subparsers = parser.add_subparsers(title='Commands', dest='command',
                                       required=True)

start_parser = subparsers.add_parser('start',
                                         help=_('Start monitoring of URL by IP for COUNT of minutes or requests'),
                                         argument_default=argparse.SUPPRESS)
    start_parser.add_argument('--url', metavar='URL', required=True,
                              help=_('monitoring URL'))
    start_parser.add_argument('--client_ip', metavar='IP', default='*',
                              help=_('client IP to monitor requests from'))
    counters = start_parser.add_mutually_exclusive_group()
    counters.add_argument('--time', metavar='COUNT', type=int,
                          help=_('number of minutes to monitor'))
    counters.add_argument('--request_qty', metavar='COUNT', type=int,
                          help=_('number of requests to monitor'))

stop_parser = subparsers.add_parser('stop',
                                        help=_('Stop monitoring of given TASK'))
    stop_parser.add_argument('--tracing_task_id', metavar='TASK',
                             required=True, help=_('unique ID of tracing task'))

continue_parser = subparsers.add_parser('continue',
                                            help=_('Continue monitoring of given TASK'))
    continue_parser.add_argument('--tracing_task_id', metavar='TASK',
                                 required=True,
                                 help=_('unique ID of tracing task'))

complete_parser = subparsers.add_parser('complete',
                                            help=_('Complete given TASK'))
    complete_parser.add_argument('--tracing_task_id', metavar='TASK',
                                 required=True,
                                 help=_('unique ID of tracing task'))

delete_parser = subparsers.add_parser('delete',
                                          help=_('Delete given TASK'))
    delete_parser.add_argument('--tracing_task_id', metavar='TASK',
                               required=True,
                               help=_('unique ID of tracing task'))

subparsers.add_parser('tasks-list',
                          help=_('Get list of tasks'))

req_parser = subparsers.add_parser('requests-list',
                                       help=_('Get list of requests for given task'))
    req_parser.add_argument('--tracing_task_id', metavar='TASK',
                            required=True,
                            help=_('unique ID of tracing task'))

req_data_parser = subparsers.add_parser('request-data',
                                            help=_('Get statistics for given request'))
    req_data_parser.add_argument('--tracing_task_id', metavar='TASK',
                                 required=True,
                                 help=_('unique ID of tracing task'))
    req_data_parser.add_argument('--request_id', metavar='REQUEST',
                                 required=True, type=int,
                                 help=_('unique ID of request'))

subparsers.add_parser('user-agent-status',
                          help=_('Get status of X-Ray User Agent'))

return parser

def cmd_parser_adviser() -> 'argparse.ArgumentParser':
    """
    Command line parser for X Ray Smart Advice utility
    using built-in argparse module
    :return: parser
    """
    parser = argparse.ArgumentParser(prog='cl-smart-advice',
                                     description='Utility for X-Ray Smart Advice microservice interaction')
    parser.add_argument('--api-version', help=_('Specify version of CLI, if not passed - latest will be used by default'),
                        required=False, default=None)
    subparsers = parser.add_subparsers(title='Commands', dest='command',
                                       required=True)

list_parser = subparsers.add_parser('list',
                                        help=_('Get list of advice'))
    list_parser.add_argument('--extends', default=False, action='store_true',
                             help=_('Advice will be with extended info'))

detail_parser = subparsers.add_parser('details',
                                          help=_('Get details for a particular advice'))
    detail_parser.add_argument('--advice_id', required=True,
                               help=_('an ID of advice to get details for'))

apply_parser = subparsers.add_parser('apply',
                                         help=_('Apply a particular advice'))
    apply_parser.add_argument('--advice_id', required=True,
                              help=_('an ID of advice to apply'))
    apply_parser.add_argument('--ignore-errors', action="store_true",
                              default=False, help=_('apply advice anyway'))
    apply_parser.add_argument('--async', action="store_true", dest='async_mode',
                              default=False, help=_('run apply in a background process'))
    apply_parser.add_argument('--source', default='ACCELERATE_WP', choices=advice_action_sources,
                              help=_('Advice apply source'))
    apply_parser.add_argument('--accept_license_terms', default=False, action='store_true',
                              help=argparse.SUPPRESS)
    apply_parser.add_argument('--analytics_data', type=str, help=_('User analytics data'))

rollback_parser = subparsers.add_parser('rollback', help=_('Rollback a particular advice'))
    rollback_parser.add_argument('--advice_id', required=True, help=_('an ID of advice to rollback'))
    rollback_parser.add_argument('--async', action="store_true", dest='async_mode',
                                 default=False, help=_('run rollback in a background process'))
    rollback_parser.add_argument('--source', default='ACCELERATE_WP', choices=advice_action_sources,
                                 help=_('Advice rollback source'))
    rollback_parser.add_argument('--reason', type=str, help=_('Reason for advice rollback'))
    rollback_parser.add_argument('--analytics_data', type=str, help=_('User analytics data'))

subparsers.add_parser('counters',
                          help=_('Get advice counters total/applied'))

status_parser = subparsers.add_parser('status',
                                          help=_('Get current status and progress for a particular advice'))
    status_parser.add_argument('--advice_id', required=True,
                               help=_('an ID of advice to get status for'))

sites_parser = subparsers.add_parser('sites-status', help=_('Get websites statuses'))
    sites_parser.add_argument('--username', help=_('an username to get statuses info'))

get_options_parser = subparsers.add_parser('get-options', help=_('Get options for user'))
    get_options_parser.add_argument('--username', help=_('Username to get options'))

get_limits_parser = subparsers.add_parser('get-limits', help=_('Get limits for user'))
    get_limits_parser.add_argument('--username', help=_('Username to get limits'))

get_usage_parser = subparsers.add_parser('get-usage', help=_('Get usage for user'))
    get_usage_parser.add_argument('--username', help=_('Username to get usage'))

advice_plugin_install = subparsers.add_parser(
        'wordpress-plugin-install',
        help=_('Installs/uninstalls cl-smart-advice plugin for websites according to current advice list'))

advice_plugin_uninstall = subparsers.add_parser(
        'wordpress-plugin-uninstall',
        help=_('Uninstalls cl-smart-advice plugin for websites'))

subparsers.add_parser('update-advices-metadata', help=_('Sends actual metadata to microservice'))

get_pullzone = subparsers.add_parser('awp-cdn-get-pullzone', help=_('Get pullzone for account`s website'))
    get_pullzone.add_argument('--username', help=_('Name of target user'))
    get_pullzone.add_argument('--account_id', help=_('Unique account identifier saved in user`s home directory'))
    get_pullzone.add_argument('--domain', help=_('Account`s domain'), required=True)
    get_pullzone.add_argument('--website', help=_('Website'), required=True)

remove_pullzone = subparsers.add_parser('awp-cdn-remove-pullzone', help=_('Remove pullzone for account`s website'))
    remove_pullzone.add_argument('--username', help=_('Name of target user'))
    remove_pullzone.add_argument('--account_id', help=_('Unique account identifier saved in user`s home directory'))
    remove_pullzone.add_argument('--domain', help=_('Account`s domain'), required=True)
    remove_pullzone.add_argument('--website', help=_('Website'), required=True)

purge_cdn_cache = subparsers.add_parser('awp-cdn-purge', help=_('Purge cache for account`s website'))
    purge_cdn_cache.add_argument('--username', help=_('Name of target user'))
    purge_cdn_cache.add_argument('--account_id', help=_('Unique account identifier saved in user`s home directory'))
    purge_cdn_cache.add_argument('--domain', help=_('Account`s domain'), required=True)
    purge_cdn_cache.add_argument('--website', help=_('Website'), required=True)

purge_cdn_cache = subparsers.add_parser('awp-sync', help=_('[for internal usage] Sync account information on microservice'))
    purge_cdn_cache.add_argument('--username', help=_('Name of target user'))
    purge_cdn_cache.add_argument('--account_id',
                                 help=_('Unique account identifier saved in user`s home directory. '
                                        'Multiple could be passed split by ","'))

get_cdn_usage = subparsers.add_parser('get-cdn-usage', help=_('Get cdn usage for account'))
    get_cdn_usage.add_argument('--account_id', help=_('Unique account identifier saved in user`s home directory'))
    get_cdn_usage.add_argument('--username', help=_('Name of target user'))

subscription_parser = subparsers.add_parser('subscription',
                                                help=_('Subscription configuration'))
    subscription_parser.add_argument('--advice_id', required=True,
                                     help=_('an ID of advice to manage subscription for'))
    subscription_parser.add_argument('--listen', required=True, action='store_true')

agreement_parser = subparsers.add_parser('agreement',
                                             help=_('License agreement commands'))
    agreement_parser.add_argument('--text', required=True,
                                  help=_('name of the feature to get license text for'))

report_analytics = subparsers.add_parser('report-analytics',
                                             help=_('Sends analytics to Smart Advice'))
    report_analytics.add_argument('--feature',
                                  help=_('Feature to be tracked by analytics'),
                                  required=True)
    report_analytics.add_argument('--event',
                                  help=_('Event to be tracked by analytics'),
                                  required=True)
    report_analytics.add_argument('--source',
                                  help=_('Where event was triggered'),
                                  required=True)
    report_analytics.add_argument('--username',
                                  help=_('Users affected by event, pass multiple via "," '),
                                  required=True)
    report_analytics.add_argument('--advice_id',
                                     help=_('ID of advice to manage events')),
    report_analytics.add_argument('--user_hash',
                                     help=_('Hash of user for event'))
    report_analytics.add_argument('--journey_id',
                                     help=_('ID event journey'))
    report_analytics.add_argument('--variant_id',
                                     help=_('ID event variant_id'))

wp_plugin_data = subparsers.add_parser('wp-plugin-data',
                                           help=_('Get latest WordPress plugin version data'))
    wp_plugin_data.add_argument('--plugin_name', required=True, help=_('Name of the plugin'))

wp_get_plugin = subparsers.add_parser('wp-plugin-copy',
                                           help=_('Copy the latest WordPress plugin archive to the site if it\'s not already installed'))
    wp_get_plugin.add_argument('--plugin_name', required=True, help=_('Name of the plugin'))
    wp_get_plugin.add_argument('--plugin_version', required=True, help=_('Current plugin version'))
    wp_get_plugin.add_argument('--tmp_dir', required=True, help=_('Temp directory path'))

return parser

def parse_cmd_arguments(
        parser: 'argparse.ArgumentParser') -> 'argparse.Namespace':
    """
    Parse arguments with given parser
    :param parser: an instance of ArgumentParser
    :return: arguments Namespace
    """
    try:
        args = parser.parse_args()
        return args
    except SystemExit as e:
        logging.error('Invalid utility invocation', extra={'err': str(e)})
        raise