"""Custom validators"""
import re
from datetime import datetime

import formencode
import pylons

from kai.model import Human, Snippet


class ExistingSnippetTitle(formencode.FancyValidator):
    def _to_python(self, value, state):
        if Snippet.exists(value):
            route_data = pylons.request.environ['pylons.routes_dict']
            slug = value.replace(" ", "_")
            slug = slug.lower()
            slug = re.sub('[^A-Za-z0-9_]+', '', slug)
            
            if route_data['action'] == 'update' and slug == route_data['id']:
                return value
            raise formencode.Invalid('Title already exists', value, state)
        return value


class ExistingEmail(formencode.FancyValidator):
    def _to_python(self, value, state):
        if not isinstance(value, basestring):
            raise formencode.Invalid('Invalid e-mail address type')
        users = list(Human.by_email(pylons.tmpl_context.db)[value])
        if not users:
            raise formencode.Invalid('No such e-mail address was found',
                                     value, state)
        user = users[0]
        
        # Check to see if the user has recently asked for an email token
        if user.password_token_issue:
            diff = datetime.utcnow() - user.password_token_issue
            if diff.days < 1 and diff.seconds < 3600:
                raise formencode.Invalid(
                    "You've already requested a password recently.  Please " 
                    "wait and try later.", value, state)
        return value


class UniqueDisplayname(formencode.FancyValidator):
    def _to_python(self, value, state):
        if list(Human.by_displayname(pylons.tmpl_context.db)[value]):
            raise formencode.Invalid('Display name already exists',
                                     value, state)
        else:
            return value


class UniqueEmail(formencode.FancyValidator):
    def _to_python(self, value, state):
        if list(Human.by_email(pylons.tmpl_context.db)[value]):
            raise formencode.Invalid('Email address already exists', value,
                                     state)
        else:
            return value


class ValidPassword(formencode.FancyValidator):
    def _to_python(self, value, state):
        if len(value) < 6:
            raise formencode.Invalid('Password is too short, must be at least'
                                     ' 6 characters', value, state)
        return value


class ValidLogin(formencode.FancyValidator):
    field_names = None
    validate_partial_form = False
    email = None
    answer = None
    password = None

    messages = {
        'badlogin': "Invalid email and/or password",
    }
    
    def is_empty(self, value):
        return False
    
    def validate_python(self, field_dict, state):
        errors = {}
        email = field_dict[self.email]
        password = field_dict[self.password]
        
        if isinstance(email, basestring):
            users = list(Human.by_email(pylons.tmpl_context.db)[email])
        else:
            users = None
        
        if users:
            user = users[0]
        else:
            user = None
        
        if not user:
            errors[self.email] = self.message('badlogin', state)
        if not errors:
            valid_password = user.verify_password(password)
            if valid_password:
                field_dict['user'] = user
            else:
                errors[self.email] = self.message('badlogin', state)
        if errors:
            error_list = errors.items()
            error_list.sort()
            error_message = '<br>\n'.join(
                ['%s: %s' % (name, value) for name, value in error_list])
            raise formencode.Invalid(error_message,
                                     field_dict, state,
                                     error_dict=errors)