import logging
import re

from .database import WithSession
from model import User

from error import UserNotFound, InterfaceException, ValidateException



def rehash(hash):
    if not re.match(r'^[a-f0-9]{15,}$', hash.lower()):
        raise ValidateException()
    return True

def validate(**kargs):
    def v(f):
        def new_f(*a,**k):
            for i in kargs:
                kargs[i](k[i])
            return f(*a,**k)
        new_f.__name__ = f.__name__
        return new_f
    return v
            

@validate(userhash=rehash)
def getuser(userhash):
    with WithSession() as session:
        user = session.query(User).filter_by(apikey=userhash).first()
        if user is None:
            raise UserNotFound()
        else:
            return user

def with_user(f):
    def new_f(*args,**kargs):
        args=list(args)
        logging.debug("Entering %s"%f.__name__)
        try:
            kargs["user"]=getuser(userhash = kargs["apikey"])
            del kargs["apikey"]
        except KeyError:
            try:
                kargs["user"]=getuser(userhash = args[1])
            except IndexError:
                raise InterfaceException() 
            del args[1]
        ret=f(*args,**kargs)
        logging.debug("Exited %s"%f.__name__)
        return ret
    new_f.__name__ = f.__name__
    return new_f