import json

from testapiclient.utils import command
from testapiclient.utils import urlparse


def scenarios_url():
    return urlparse.resource_join('scenarios')


def scenario_url(parsed_args):
    return urlparse.path_join(scenarios_url(), parsed_args.name)


class ScenarioGet(command.Lister):

    def get_parser(self, prog_name):
        parser = super(ScenarioGet, self).get_parser(prog_name)
        parser.add_argument('-name',
                            help='Search scenarios using name')
        parser.add_argument('-installer',
                            help='Search scenarios using installer')
        parser.add_argument('---version',
                            help='Search scenarios using version')
        parser.add_argument('-project',
                            help='Search scenarios using project')
        return parser

    def take_action(self, parsed_args):
        columns = (
            'name',
            '_id',
            'creator',
            'creation_date'
        )
        data = self.app.client_manager.get(
            urlparse.query_by(scenarios_url(),
                              ['name', 'installer', 'version', 'project'],
                              parsed_args))
        return self.format_output(columns, data.get('scenarios', []))


class ScenarioGetOne(command.ShowOne):

    def get_parser(self, prog_name):
        parser = super(ScenarioGetOne, self).get_parser(prog_name)
        parser.add_argument('name',
                            help='Search scenario by name')
        return parser

    def take_action(self, parsed_args):
        return self.format_output(
            self.app.client_manager.get(scenario_url(parsed_args)))


class ScenarioCreate(command.ShowOne):

    def get_parser(self, prog_name):
        parser = super(ScenarioCreate, self).get_parser(prog_name)
        parser.add_argument('scenario',
                            type=json.loads,
                            help='Scenario create request format :\n'
                                 '\'{ "installers": [], "name": ""}\',\n'
                                 'Intaller create request format :\n'
                                 '\'{"installer": "","versions": []}\',\n'
                                 'Version create request format :\n'
                                 '\'{"owner": "","version": "",'
                                 '"projects": []}\',\n'
                                 'Project create request format :\n'
                                 '\'{"project": "","customs": [],'
                                 '"scores": [],'
                                 '"trust_indicators": []}\',\n'
                                 'Custom create request format :\n'
                                 '\'["asf","saf"]\',\n'
                                 'Score create request format :\n'
                                 '\'{"date": "", "score": ""}\',\n'
                                 'Trust Indicators create request format :\n'
                                 '\'{"date": "", "status": ""}\'')
        return parser

    def take_action(self, parsed_args):
        return self.format_output(
            self.app.client_manager.post(
                scenarios_url(), parsed_args.scenario))


class ScenarioDelete(command.Command):

    def get_parser(self, prog_name):
        parser = super(ScenarioDelete, self).get_parser(prog_name)
        parser.add_argument('name',
                            type=str,
                            help='Delete scenario by name')
        return parser

    def take_action(self, parsed_args):
        return self.app.client_manager.delete(scenario_url(parsed_args))


class ScenarioPut(command.ShowOne):

    def get_parser(self, prog_name):
        parser = super(ScenarioPut, self).get_parser(prog_name)
        parser.add_argument('name',
                            type=str,
                            help='Update scenario by name')
        parser.add_argument('scenario',
                            type=json.loads,
                            help='Scenario create request format :\n'
                                 '\'{ "installers": [], "name": ""}\',\n'
                                 'Intaller create request format :\n'
                                 '\'{"installer": "","versions": []}\',\n'
                                 'Version create request format :\n'
                                 '\'{"owner": "","version": "",'
                                 '"projects": []}\',\n'
                                 'Project create request format :\n'
                                 '\'{"project": "","customs": [],'
                                 '"scores": [],'
                                 '"trust_indicators": []}\',\n'
                                 'Custom create request format :\n'
                                 '\'["asf","saf"]\',\n'
                                 'Score create request format :\n'
                                 '\'{"date": "", "score": ""}\',\n'
                                 'Trust Indicators create request format :\n'
                                 '\'{"date": "", "status": ""}\'')
        return parser

    def take_action(self, parsed_args):
        return self.format_output(
            self.app.client_manager.put(
                scenario_url(parsed_args), parsed_args.scenario))