import xml_code_engine
import rabbitmq


class Reporter(object):
    def __init__(self, rmqclient, task_id, environment_id):
        self._rmqclient = rmqclient
        self._task_id = task_id
        self._environment_id = environment_id
        rmqclient.declare('task-reports')

    def _report_func(self, id, entity, text, **kwargs):
        body = {
            'id': id,
            'entity': entity,
            'text': text,
            'environment_id': self._environment_id
        }

        msg = rabbitmq.Message()
        msg.body = body
        msg.id = self._task_id

        self._rmqclient.send(
            message=msg,
            key='task-reports')


def _report_func(context, id, entity, text, **kwargs):
    reporter = context['/reporter']
    return reporter._report_func(id, entity, text, **kwargs)

xml_code_engine.XmlCodeEngine.register_function(_report_func, "report")