63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
from time import sleep
|
|
import requests
|
|
from aiosmtpd.controller import Controller
|
|
from configargparse import ArgParser
|
|
from xdg import xdg_config_home
|
|
|
|
|
|
def parse_args():
|
|
parser = ArgParser(default_config_files=['/etc/libertyforward/*.conf',
|
|
str(xdg_config_home() / 'LibertyTechForce' / 'libertyforward.conf')])
|
|
parser.add('-c', '--config', is_config_file=True, help='config file path', env_var='LF_CONFIG_FILE')
|
|
parser.add('--mailer', help='name of the mailer backend to use', env_var='LF_MAILER')
|
|
parser.add('--api-token', help='token to use to authenticate against API', env_var='LF_API_TOKEN')
|
|
return parser.parse_args()
|
|
|
|
|
|
class MailerHandlerMixin:
|
|
"""Base class for the mail handlers"""
|
|
_api_token = None
|
|
|
|
@property
|
|
def api_token(self):
|
|
return self._api_token
|
|
|
|
@api_token.setter
|
|
def api_token(self, value):
|
|
self._api_token = value
|
|
|
|
|
|
class KingMailerHandler(object):
|
|
|
|
async def handle_DATA(self, server, session, envelope):
|
|
print('Incoming mesage from: {}'.format(envelope.mail_from), flush=True)
|
|
payload = {
|
|
'mail_from': envelope.mail_from,
|
|
'rcpt_to': envelope.rcpt_tos,
|
|
'data': envelope.content
|
|
}
|
|
headers = {'X-Server-API-Key': self.api_token}
|
|
response = requests.post('https://api.kingmailer.co/api/v1/send/raw', data=payload, headers=headers)
|
|
print(response)
|
|
if response.ok:
|
|
print('Message accepted', flush=True)
|
|
return '250 Message accepted for delivery'
|
|
else:
|
|
print('Could not process', flush=True)
|
|
return '500 Could not process your message'
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print('Starting LibertyForward', flush=True)
|
|
args = parse_args()
|
|
handler = KingMailerHandler()
|
|
handler.api_token = args.api_token
|
|
controller = Controller(handler, hostname='127.0.0.1', port=25)
|
|
print('Starting controller thread', flush=True)
|
|
controller.start()
|
|
while True:
|
|
try:
|
|
sleep(1)
|
|
except:
|
|
break
|