from afew.filters.BaseFilter import Filter
from afew.FilterRegistry import register_filter
import re

@register_filter
class RTFilter(Filter):
    message = "Create tags for an RT ticket system"
    rt_header = 'X-RT-Ticket'

    def __init__(self, database, **kwargs):
        super(RTFilter, self).__init__(database, **kwargs)
        self.queue_pattern = re.compile(r'^<([^@]+)@.+$')

    def handle_message(self, message):
        if not self._tag_blacklist.intersection(message.get_tags()):
            if message.get_header(self.rt_header):
                self.add_tags(message, 'rt')
                reply_to = message.get_header('Reply-To')
                self.add_tags(message, 'rt/' + re.match(self.queue_pattern, reply_to).group(1))