from personalisation.models import AbstractBaseRule, Segment
import time

class SegmentMiddleware(object):
    """Middleware for testing and putting a user in a segment"""

    def __init__(self, get_response=None):
        self.get_response = get_response

    def __call__(self, request):
        if request.path.startswith('/admin/') or request.path.startswith('/django-admin/'):
            return self.get_response(request)

        if 'segments' not in request.session:
            request.session['segments'] = []
            request.session['visit_count'] = 1

        segments = Segment.objects.all().filter(status='enabled')

        chosen_segments = []

        for segment in segments:
            rules = AbstractBaseRule.objects.filter(segment=segment)
            result = self.test_rules(rules, request)

            print(result)

            if result:
                self.add_segment_to_user(segment, request)

        response = self.get_response(request)
        return response

    def test_rules(self, rules, request):
        """Test wether the user matches a segment's rules'"""
        if len(rules) > 0:
            for rule in rules:
                result = rule.test_user(request)

                if result is False:
                    return False

            return True
        return False

    def add_segment_to_user(self, segment, request):
        """Save the segment in the user session"""
        def check_if_segmented(segment):
            """Check if the user has been segmented"""
            for seg in request.session['segments']:
                if seg['encoded_name'] == segment.encoded_name():
                    return True
            return False

        if not check_if_segmented(segment):
            # Clear segments session on browser close.
            segdict = {
                "encoded_name": segment.encoded_name(),
                "id": segment.pk,
                "timestamp": int(time.time()),
            }
            request.session['segments'].append(segdict)