from django.db import models
from django.utils.encoding import python_2_unicode_compatible
from django.utils.translation import ugettext as _
from passlib.hash import sha512_crypt


@python_2_unicode_compatible
class MailDomain(models.Model):
    domain = models.CharField(max_length=128, unique=True)

    class Meta:
        verbose_name = _('Mail domain')
        verbose_name_plural = _('Mail domains')

    def __str__(self):
        return self.domain


class Mailbox(models.Model):
    username = models.CharField(max_length=128, unique=True)
    domain = models.ForeignKey(MailDomain)
    password = models.CharField(max_length=255)
    home = models.CharField(max_length=255)
    uid = models.PositiveSmallIntegerField()
    gid = models.PositiveSmallIntegerField()
    active = models.BooleanField(default=True)

    class Meta:
        verbose_name = _('Mailbox')
        verbose_name_plural = _('Mailboxes')

    def set_password(self, password):
        self.password = sha512_crypt.encrypt(password)


@python_2_unicode_compatible
class MailAddress(models.Model):
    localpart = models.CharField(max_length=128)
    domain = models.ForeignKey(MailDomain)
    active = models.BooleanField(default=True)

    class Meta:
        unique_together = ('localpart', 'domain')
        verbose_name = _('Mail address')
        verbose_name_plural = _('Mail addresses')

    def __str__(self):
        return "{0}@{1}".format(self.localpart, self.domain)


class MailAddressMailbox(models.Model):
    mailaddress = models.ForeignKey(MailAddress)
    mailbox = models.ForeignKey(Mailbox)

    class Meta:
        unique_together = ('mailaddress', 'mailbox')


class MailAddressForward(models.Model):
    mailaddress = models.ForeignKey(MailAddress)
    target = models.EmailField(max_length=254)

    class Meta:
        unique_together = ('mailaddress', 'target')