# -*- coding: utf-8 -*-
#
# Copyright 2013 - Mirantis, Inc.
#
#    Licensed under the Apache License, Version 2.0 (the "License");
#    you may not use this file except in compliance with the License.
#    You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
#    Unless required by applicable law or agreed to in writing, software
#    distributed under the License is distributed on an "AS IS" BASIS,
#    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#    See the License for the specific language governing permissions and
#    limitations under the License.

from mistral.engine.actions import actions
from mistral.engine.actions import action_types
from mistral.engine.actions import action_helper as a_h
import mistral.exceptions as exc
from mistral.workbook import services
from mistral.workbook import tasks


def create_action(db_task):
    action_type = a_h.get_action_type(db_task)
    task = tasks.TaskSpec(db_task['task_spec'])
    service = services.ServiceSpec(db_task['service_spec'])

    if not action_types.is_valid(action_type):
        raise exc.InvalidActionException("Action type is not supported: %s" %
                                         action_type)

    action = _get_mapping()[action_type](db_task, task, service)
    action.result_helper = _find_action_result_helper(db_task, action)
    return action


def _get_mapping():
    return {
        action_types.ECHO: get_echo_action,
        action_types.REST_API: get_rest_action,
        action_types.MISTRAL_REST_API: get_mistral_rest_action,
        action_types.OSLO_RPC: get_amqp_action,
        action_types.SEND_EMAIL: get_send_email_action
    }


def _find_action_result_helper(task, action):
    try:
        return task['service_spec']['actions'][action.name].get('output', {})
    except (KeyError, AttributeError):
        return {}


def get_echo_action(db_task, task, service):
    action_type = service.type
    action_name = task.get_action_name()

    output = service.actions.get(action_name).output

    return actions.EchoAction(action_type, action_name, output=output)


def get_rest_action(db_task, task, service):
    action_type = service.type
    action_name = task.get_action_name()
    action = service.actions.get(action_name)
    task_params = task.parameters
    url = service.parameters['baseUrl'] +\
        action.parameters['url']

    headers = {}
    headers.update(task.parameters.get('headers', {}))
    headers.update(action.parameters.get('headers', {}))

    method = action.parameters.get('method', "GET")

    # input_yaql = task.get('input')
    # TODO(nmakhotkin) extract input from context with the YAQL expression
    task_input = {}  # expressions.evaluate(input_expr, ctx)
    task_data = {}

    if method.upper() == "GET":
        task_params.update(task_input)
    elif method.upper() in ["POST", "PUT"]:
        task_data.update(task_input)

    return actions.RestAction(action_type, action_name, url,
                              params=task_params, method=method,
                              headers=headers, data=task_data)


def get_mistral_rest_action(db_task, task, service):
    mistral_headers = {
        'Mistral-Workbook-Name': db_task['workbook_name'],
        'Mistral-Execution-Id': db_task['execution_id'],
        'Mistral-Task-Id': db_task['id'],
    }

    action = get_rest_action(db_task, task, service)
    action.headers.update(mistral_headers)

    return action


def get_amqp_action(db_task, task, service):
    action_type = service.type
    action_name = task.get_action_name()
    action = service.actions.get(action_name)
    action_params = action.parameters
    task_params = task.parameters
    service_parameters = service.parameters

    host = service_parameters['host']
    port = service_parameters.get('port')
    userid = service_parameters['userid']
    password = service_parameters['password']
    virtual_host = action_params['virtual_host']
    message = task_params['message']
    routing_key = task_params.get('routing_key', None)
    exchange = action_params.get('exchange')
    queue_name = action_params['queue_name']

    return actions.OsloRPCAction(action_type, action_name, host, userid,
                                 password, virtual_host, message, routing_key,
                                 port, exchange, queue_name)


def get_send_email_action(db_task, task, service):
    #TODO(dzimine): Refactor action_type and action_name settings
    #               for all actions
    action_type = service.type
    action_name = task.get_action_name()
    task_params = task.parameters
    service_params = service.parameters

    return actions.SendEmailAction(action_type, action_name,
                                   task_params, service_params)