# -*- coding: utf-8 -*-

#Copyright (C) 2009  Sandro Knauß <bugs@sandroknauss.de>

#This program is free software; you can redistribute it and/or modify it under the terms

#of the GNU General Public License as published by the Free Software Foundation;

#either version 3 of the License, or any later version.

#This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;

#without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

#See the GNU General Public License for more details.

#You should have received a copy of the GNU General Public License

#along with this program; if not, see <http://www.gnu.org/licenses/>.

import smtplib

import copy

from ..model.status import Status

class SMTP(Provider):

    def __init__(self, name, config):

        Provider.__init__(self,name,config,{"mail":[None]})

    def loadConfig(self):

        """Read options from config"""

        needed=["send_from","host", "port", "user", "password"]

        for n in needed:

            setattr(self,n,None)

        Provider.loadConfig(self)

        self.bTLS = False

        self.bSSL = False

        for (n, v) in self.config:

            if n in needed:

                setattr(self,n,v)

            elif n == "TLS":

                self.bTLS = bool(v)

            elif n == "SSL":

                self.bSSL = bool(v)

            else:

                raise UnknownOption(self.name, n)

        for n in needed:

           if getattr(self,n) is None:

               raise NeededOption(self.name, n) 

    def send(self,mail,recipient):   

        if not self.testmode:

            if self.bSSL:

                smtp = smtplib.SMTP_SSL(self.host,self.port)

            else:

                smtp = smtplib.SMTP(self.host,self.port)

            if self.bTLS:

                smtp.starttls()

            if not self.user == "":

                smtp.login(self.user,self.password)

        try:

            frm=self.send_from

            if mail.getFrom():

                frm = mail.getFrom()

            tmpmail=copy.deepcopy(mail)

            tmpmail.content['From'] = frm 

            tmpmail.content['To']=recipient

            if not self.testmode:

                smtp.sendmail(frm,  recipient, tmpmail.as_string())

            return Status(self, None)

        finally:

            smtp.quit()

    def getSendFunc(self, typ, route):

        Provider.getSendFunc(self, typ, route)

        return self.send

providers["smtp"]=SMTP