#!/usr/local/CyberCP/bin/python2
import os,sys
sys.path.append('/usr/local/CyberCP')
import django
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "CyberCP.settings")
django.setup()
import threading as multi
from plogical.CyberCPLogFileWriter import CyberCPLogFileWriter as logging
from policyConstraint import policyConstraints
from emailPremium.models import DomainLimits, EmailLogs
from mailServer.models import Domains, EUsers
import time
from cacheManager import cacheManager

limitThreads = multi.BoundedSemaphore(10)

class HandleRequest(multi.Thread):
    cleaningPath = '/home/cyberpanel/purgeCache'
    def __init__(self, conn):
        multi.Thread.__init__(self)
        self.connection = conn

    def __del__(self):
        try:
            self.connection.close()
        except BaseException, msg:
            logging.writeToFile(str(msg) + ' [HandleRequest.__del__]')

    def run(self):
        limitThreads.acquire()
        dataComplete = ""

        try:
            while True:

                Data = self.connection.recv(64)

                # Wait for a connection
                if os.path.exists(HandleRequest.cleaningPath):
                    readFromFile = open(HandleRequest.cleaningPath, 'r')
                    command = readFromFile.read()
                    cacheManager.handlePurgeRequest(command)
                    readFromFile.close()
                    os.remove(HandleRequest.cleaningPath)
                    cacheManager.flushCache()

                if Data:
                    if len(Data) < 64:
                        dataComplete = dataComplete + Data
                        self.manageRequest(dataComplete)
                        dataComplete = ''
                    else:
                        dataComplete = dataComplete + Data
                else:
                    self.connection.close()
                    break

            limitThreads.release()
        except BaseException, msg:
            logging.writeToFile(str(msg) + ' [HandleRequest.run]')
            limitThreads.release()

    def manageRequest(self, completeData):
        try:
            completeData = completeData.split('\n')

            for items in completeData:
                tempData = items.split('=')
                if tempData[0] == 'sender':
                    emailAddress = tempData[1]
                    domainName = emailAddress.split('@')[1]
                elif tempData[0] == 'recipient':
                    destination = tempData[1]

            if domainName in cacheManager.domains:
                domainObj = cacheManager.domains[domainName]
                emailObj = domainObj.findEmailOBJ(emailAddress)
            else:
                try:
                    domain = Domains.objects.get(domain=domainName)
                    domainLTS = DomainLimits.objects.get(domain=domain)

                    newDomain = policyConstraints(domainName, domainLTS.monthlyLimit, domainLTS.monthlyUsed, domainLTS.limitStatus)
                    cacheManager.domains[domainName] = newDomain
                    domainObj = newDomain

                    emailObj = newDomain.findEmailOBJ(emailAddress)
                except:
                    raise BaseException

            #logging.writeToFile('Domain Limit Status: ' + str(domainObj.limitStatus))
            #logging.writeToFile('Email Limit Status: ' + str(domainObj.limitStatus))
            #logging.writeToFile('Email Monthly Limit: ' + str(emailObj.monthlyLimits))
            #logging.writeToFile('Email Monthly Used: ' + str(emailObj.monthlyUsed))

            if domainObj.limitStatus == 1 and emailObj.limitStatus == 1:
                if domainObj.monthlyLimits <= domainObj.monthlyUsed or emailObj.monthlyLimits <= emailObj.monthlyUsed or emailObj.hourlyLimits <= emailObj.hourlyUsed:
                    logging.writeToFile(emailAddress + ' either exceeded monthly or hourly sending limit.')
                    self.connection.sendall('action=defer_if_permit Service temporarily unavailable\n\n')
                else:
                    email = EUsers.objects.get(email=emailAddress)
                    if emailObj.logStatus == 1:
                        logEntry = EmailLogs(email=email, destination=destination, timeStamp=time.strftime("%I-%M-%S-%a-%b-%Y"))
                        logEntry.save()
                    emailObj.monthlyUsed = emailObj.monthlyUsed + 1
                    emailObj.hourlyUsed = emailObj.hourlyUsed + 1
                    domainObj.monthlyUsed = domainObj.monthlyUsed + 1
                    self.connection.sendall('action=dunno\n\n')
            else:
                email = EUsers.objects.get(email=emailAddress)
                if emailObj.logStatus == 1:
                    logEntry = EmailLogs(email=email, destination=destination,
                                         timeStamp=time.strftime("%I-%M-%S-%a-%b-%Y"))
                    logEntry.save()
                emailObj.monthlyUsed = emailObj.monthlyUsed + 1
                emailObj.hourlyUsed = emailObj.hourlyUsed + 1
                self.connection.sendall('action=dunno\n\n')


        except BaseException, msg:
            self.connection.sendall('action=dunno\n\n')
            limitThreads.release()
            logging.writeToFile(str(msg))