from django.db import models
from django.contrib.auth.models import User
from django.conf import settings
from mailng.lib import exec_pipe
import os

class Domain(models.Model):
    name = models.CharField(max_length=100)
    quota = models.IntegerField()
    enabled = models.BooleanField()

    def create_dir(self):
        code, output = exec_pipe("sudo -u %s mkdir %s/%s" \
                                     % (settings.VIRTUAL_UID, settings.STORAGE_PATH, 
                                        self.name))
        if code:
            os.system("echo '%s' >> /tmp/vmail.log" % output)
            return False
        return True

    def rename_dir(self, newname):
        code, output = exec_pipe("sudo -u %s mv %s/%s %s/%s" \
                                     % (settings.VIRTUAL_UID, 
                                        settings.STORAGE_PATH, self.name,
                                        settings.STORAGE_PATH, newname))
        if code:
            os.system("echo '%s' >> /tmp/vmail.log" % output)
            return False
        return True

    def delete_dir(self):
        code, output = exec_pipe("sudo -u %s rm -r %s/%s" \
                                     % (settings.VIRTUAL_UID, settings.STORAGE_PATH, 
                                        self.name))
        if code:
            return False
        return True

class Mailbox(models.Model):
    name = models.CharField(max_length=100)
    address = models.CharField(max_length=100)
    full_address = models.CharField(max_length=150)
    quota = models.IntegerField()
    uid = models.IntegerField()
    gid = models.IntegerField()
    path = models.CharField(max_length=200)
    domain = models.ForeignKey(Domain)
    user = models.ForeignKey(User)
    path = models.CharField(max_length=255)

    def __str__(self):
        return "%s" % (self.address)

    def create_dir(self, domain):
        path = "%s/%s/%s" % (settings.STORAGE_PATH, domain.name, self.address)
        code, output = exec_pipe("sudo -u %s mkdir -p %s/.maildir" \
                                     % (settings.VIRTUAL_UID, path))
        if code:
            return False
        self.path = "%s/.maildir/" % path
        return True

    def rename_dir(self, domain, newaddress):
        path = "%s/%s" % (settings.STORAGE_PATH, domain)
        code, output = exec_pipe("sudo -u %s mv %s/%s %s/%s" \
                                     % (settings.VIRTUAL_UID, 
                                        path, self.address, path, newaddress))
        if code:
            os.system("echo '%s' >> /tmp/vmail.log" % output)
            return False
        self.path = "%s/%s/.maildir/" % (path, newaddress)
        return True

    def delete_dir(self):
        code, output = exec_pipe("sudo -u %s rm -r %s/%s/%s" \
                                     % (settings.VIRTUAL_UID, settings.STORAGE_PATH,
                                        self.domain.name, self.name))
        if code:
            return False
        return True

class Alias(models.Model):
    address = models.CharField(max_length=100)
    full_address = models.CharField(max_length=150)
    mbox = models.ForeignKey(Mailbox)
    enabled = models.BooleanField()