from django.core.cache import cache

from piston.utils import rc
from piston.handler import BaseHandler

from colab.deprecated import solrutils
from super_archives.models import PageHit


class CountHandler(BaseHandler):
    allowed_methods = ('POST')

    def create(self, request):
        """Add one page view for the given url"""
        
        # If missing the path_info argument we can't do anything
        path_info = request.POST.get('path_info')
        if not path_info:
            return rc.BAD_REQUEST

        # Here we cache the user's IP to ensure that the same
        #   IP won't hit the same page again for while
        ip_addr = request.META.get('REMOTE_ADDR')
        page_hits_cache = cache.get('page_hits', {})
        duplicate = page_hits_cache.get(path_info, {}).get(ip_addr)
        
        if duplicate:
            return rc.DUPLICATE_ENTRY
        else:
            page_hits_cache.update({path_info: {ip_addr: True }})
            cache.set('page_hits', page_hits_cache)
        
        # Everything ok, so just increment the page count
        page_hit = PageHit.objects.get_or_create(url_path=path_info)[0]
        page_hit.hit_count += 1
        page_hit.save()
        
        return rc.CREATED

class SearchHandler(BaseHandler):
    allowed_methods = ('GET', )

    def read(self, request):
        query = request.GET.get('q')
        page = request.GET.get('p', 1)
        results_per_page = request.GET.get('n', 50)
        order = request.GET.get('o')

        if not query:
            return 'Query cannot be empty.'
        else:
            query = query.encode('utf-8')

        try: 
            n = int(results_per_page)
        except ValueError:
            n = 10
    
        if 1 > n > 500:
            n = 1
        
        try:
            page = int(page)
        except ValueError:
            page = 1
        
        if page < 1:
            page = 1

        return solrutils.select(query, results_per_page, page, order)