#!/usr/bin/env python
# ==============================================================================
# Copyright and license info is available in the LICENSE file included with
# the Server Deployment Package (SDP), and also available online:
# https://swarm.workshop.perforce.com/projects/perforce-software-sdp/view/main/LICENSE
# ------------------------------------------------------------------------------

import os
import sys
import getopt
import smtplib
from email.mime.text import MIMEText
import ConfigParser

SDP_INSTANCE = '1'

config = ConfigParser.RawConfigParser()
config.read('maintenance.cfg')

##########################################################################
#####                                                                #####
#####  CONFIGURATION VARIABLES: Modify in maintenance.cfg as needed. #####
#####                                                                #####

fromAddr = config.get(SDP_INSTANCE, 'administrator')
replyToAddr = ''
mailSMTPhost = config.get(SDP_INSTANCE, 'mailhost')
mailSMTPport = int(config.get(SDP_INSTANCE, 'mailport'))
mailsecure = int(config.get(SDP_INSTANCE, 'mailsecure'))
if mailsecure:
    mailUser = config.get(SDP_INSTANCE, 'mailuser')
    mailPass = config.get(SDP_INSTANCE, 'mailpass')


def usage():
    print("pymail.py -t <to-address or address file> -s <subject> -i <input file>")
    sys.exit()


def mail(inputfile, subject, toAddr):
    fp = open(inputfile, 'rb')
    msg = MIMEText(fp.read())
    fp.close()

    msg['Subject'] = subject
    msg['From'] = fromAddr
    if (len(replyToAddr)):
        msg.add_header('reply-to', replyToAddr)
    msg['To'] = toAddr

    try:
        print("Sending mail to: %s" % toAddr)
        s = smtplib.SMTP(mailSMTPhost, mailSMTPport)
        if mailsecure:
            s.starttls()
            s.login(mailUser, mailPass)
        s.sendmail(msg['From'], msg['To'], msg.as_string())
        s.quit()
    except smtplib.SMTPException:
        print("Error: unable to send email to %s." % msg['To'])


def main(argv):
    inputfile = ''
    subject = ''
    toAddr = ''
    addressfile = ''

    try:
        opts, args = getopt.getopt(argv, "ht:s:i:")
    except:
        usage()

    for opt, arg in opts:
        if opt == '-h':
            usage()
        elif opt == '-i':
            inputfile = arg
        elif opt == '-t':
            toAddr = arg
            if os.path.isfile(arg):
                addressfile = arg
        elif opt == '-s':
            subject = arg

    if (len(subject) == 0 or len(toAddr) == 0 or len(inputfile) == 0):
        usage()

    if addressfile:
        addresses = open(addressfile, "r")
        for address in addresses.readlines():
            mail(inputfile, subject, address.strip())
        addresses.close()
    else:
        mail(inputfile, subject, toAddr)


if __name__ == "__main__":
    main(sys.argv[1:])