###
### User Document
###

class User(Document):
    """Bare minimum to have the concept of a User.
    """
    _private_fields = [
        'password', 'is_active',
    ]

    username_regex = re.compile('[^a-zA-Z0-9._]')
    username_min_length = 2

    username = StringField(max_length=30, required=True)
    email = EmailField(max_length=100)
    password = StringField(max_length=128)
    is_active = BooleanField(default=False)
    last_login = LongField(default=curtime)
    date_joined = LongField(default=curtime)

    def __unicode__(self):
        return u'%s' % (self.username)

    def set_password(self, raw_passwd):
        """Generates bcrypt hash and salt for storing a user's password. With
        bcrypt, the salt is kind of redundant, but this format stays friendly
        to other algorithms.
        """
        algorithm = auth.BCRYPT
        (salt, digest) = auth.gen_hexdigest(algorithm, raw_passwd)
        self.password = auth.build_passwd_line(algorithm, salt, digest)

    def check_password(self, raw_password, passwd_line):
        """Compares raw_password to password stored for user. Updates
        self.last_login on success.
        """
        algorithm, salt, hash = auth.split_passwd_line(self.password)
        if hash == auth.gen_hexdigest(algorithm, salt, raw_password):
            self.last_login = curtime()
            return True
        else:
            return False

    @classmethod
    def create_user(cls, username, password, email=str()):
        """Creates a user document with given username and password
        and saves it.

        Validation occurs only for email argument. It makes no assumptions
        about password format.
        """
        now = curtime()
        username = username.lower()

        # validate email argument
        try:
            cls.validate_class_partial(dict(email=email))
        except ValueError:
            pass
        else:
            email = email.strip()
            email = email.lower()

        user = cls(username=username, email=email, date_joined=now)
        user.set_password(password)
        return user


###
### UserProfile
###

class UserProfile(Document):
    """The basic things a user profile tends to carry. Isolated in separate
    class to keep separate from private data.
    """
    owner = ObjectIdField(required=True)
    username = StringField(max_length=30, required=True)
    name = StringField(max_length=255)
    website = URLField(max_length=255)
    bio = StringField(max_length=100)
    location_text = StringField(max_length=100)
    avatar_url = URLField(max_length=255)

    _private_fields = [
        'owner',
    ]

    def __init__(self, *args, **kwargs):
        super(UserProfile, self).__init__(*args, **kwargs)

    def __unicode__(self):
        return u'%s' % (self.name)