You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
234 lines
7.5 KiB
234 lines
7.5 KiB
# -*- encoding: utf-8 -*-
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import sys
|
|
import io
|
|
import time
|
|
import json
|
|
|
|
import socket
|
|
import smtplib
|
|
|
|
from email.utils import formatdate
|
|
from email.header import Header
|
|
from email.mime.text import MIMEText
|
|
|
|
try:
|
|
from urllib.parse import quote
|
|
except ImportError:
|
|
from urllib import quote
|
|
|
|
import logging
|
|
logger = logging.getLogger("isso")
|
|
|
|
try:
|
|
import uwsgi
|
|
except ImportError:
|
|
uwsgi = None
|
|
|
|
from isso.compat import PY2K
|
|
from isso import local
|
|
|
|
if PY2K:
|
|
from thread import start_new_thread
|
|
else:
|
|
from _thread import start_new_thread
|
|
|
|
|
|
class SMTPConnection(object):
|
|
|
|
def __init__(self, conf):
|
|
self.conf = conf
|
|
|
|
def __enter__(self):
|
|
klass = (smtplib.SMTP_SSL if self.conf.get(
|
|
'security') == 'ssl' else smtplib.SMTP)
|
|
self.client = klass(host=self.conf.get('host'),
|
|
port=self.conf.getint('port'),
|
|
timeout=self.conf.getint('timeout'))
|
|
|
|
if self.conf.get('security') == 'starttls':
|
|
if sys.version_info >= (3, 4):
|
|
import ssl
|
|
self.client.starttls(context=ssl.create_default_context())
|
|
else:
|
|
self.client.starttls()
|
|
|
|
username = self.conf.get('username')
|
|
password = self.conf.get('password')
|
|
if username and password:
|
|
if PY2K:
|
|
username = username.encode('ascii')
|
|
password = password.encode('ascii')
|
|
|
|
self.client.login(username, password)
|
|
|
|
return self.client
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
self.client.quit()
|
|
|
|
class SMTP(object):
|
|
|
|
def __init__(self, isso):
|
|
|
|
self.isso = isso
|
|
self.conf = isso.conf.section("smtp")
|
|
self.public_endpoint = isso.conf.get("server", "public-endpoint") or local("host")
|
|
self.admin_notify = any((n in ("smtp", "SMTP")) for n in isso.conf.getlist("general", "notify"))
|
|
self.reply_notify = isso.conf.getboolean("general", "reply-notifications")
|
|
|
|
# test SMTP connectivity
|
|
try:
|
|
with SMTPConnection(self.conf):
|
|
logger.info("connected to SMTP server")
|
|
except (socket.error, smtplib.SMTPException):
|
|
logger.exception("unable to connect to SMTP server")
|
|
|
|
if uwsgi:
|
|
def spooler(args):
|
|
try:
|
|
self._sendmail(args[b"subject"].decode("utf-8"),
|
|
args["body"].decode("utf-8"))
|
|
except smtplib.SMTPConnectError:
|
|
return uwsgi.SPOOL_RETRY
|
|
else:
|
|
return uwsgi.SPOOL_OK
|
|
|
|
uwsgi.spooler = spooler
|
|
|
|
def __iter__(self):
|
|
yield "comments.new:after-save", self.notify_new
|
|
yield "comments.activate", self.notify_activated
|
|
|
|
def format(self, thread, comment, parent_comment, recipient=None, admin=False):
|
|
|
|
rv = io.StringIO()
|
|
|
|
author = comment["author"] or "Anonymous"
|
|
if comment["email"]:
|
|
author += " <%s>" % comment["email"]
|
|
|
|
rv.write(author + " wrote:\n")
|
|
rv.write("\n")
|
|
rv.write(comment["text"] + "\n")
|
|
rv.write("\n")
|
|
|
|
if admin:
|
|
if comment["website"]:
|
|
rv.write("User's URL: %s\n" % comment["website"])
|
|
|
|
rv.write("IP address: %s\n" % comment["remote_addr"])
|
|
|
|
rv.write("Link to comment: %s\n" %
|
|
(local("origin") + thread["uri"] + "#isso-%i" % comment["id"]))
|
|
rv.write("\n")
|
|
rv.write("---\n")
|
|
|
|
if admin:
|
|
uri = self.public_endpoint + "/id/%i" % comment["id"]
|
|
key = self.isso.sign(comment["id"])
|
|
|
|
rv.write("Delete comment: %s\n" % (uri + "/delete/" + key))
|
|
|
|
if comment["mode"] == 2:
|
|
rv.write("Activate comment: %s\n" % (uri + "/activate/" + key))
|
|
|
|
else:
|
|
uri = self.public_endpoint + "/id/%i" % parent_comment["id"]
|
|
key = self.isso.sign(('unsubscribe', recipient))
|
|
|
|
rv.write("Unsubscribe from this conversation: %s\n" % (uri + "/unsubscribe/" + quote(recipient) + "/" + key))
|
|
|
|
rv.seek(0)
|
|
return rv.read()
|
|
|
|
def notify_new(self, thread, comment):
|
|
if self.admin_notify:
|
|
body = self.format(thread, comment, None, admin=True)
|
|
self.sendmail(thread["title"], body, thread, comment)
|
|
|
|
if comment["mode"] == 1:
|
|
self.notify_users(thread, comment)
|
|
|
|
def notify_activated(self, thread, comment):
|
|
self.notify_users(thread, comment)
|
|
|
|
def notify_users(self, thread, comment):
|
|
if self.reply_notify and "parent" in comment and comment["parent"] is not None:
|
|
# Notify interested authors that a new comment is posted
|
|
notified = []
|
|
parent_comment = self.isso.db.comments.get(comment["parent"])
|
|
comments_to_notify = [parent_comment] if parent_comment is not None else []
|
|
comments_to_notify += self.isso.db.comments.fetch(thread["uri"], mode=1, parent=comment["parent"])
|
|
for comment_to_notify in comments_to_notify:
|
|
email = comment_to_notify["email"]
|
|
if "email" in comment_to_notify and comment_to_notify["notification"] and email not in notified \
|
|
and comment_to_notify["id"] != comment["id"] and email != comment["email"]:
|
|
body = self.format(thread, comment, parent_comment, email, admin=False)
|
|
subject = "Re: New comment posted on %s" % thread["title"]
|
|
self.sendmail(subject, body, thread, comment, to=email)
|
|
notified.append(email)
|
|
|
|
def sendmail(self, subject, body, thread, comment, to=None):
|
|
if uwsgi:
|
|
uwsgi.spool({b"subject": subject.encode("utf-8"),
|
|
b"body": body.encode("utf-8"),
|
|
b"to": to})
|
|
else:
|
|
start_new_thread(self._retry, (subject, body, to))
|
|
|
|
def _sendmail(self, subject, body, to=None):
|
|
|
|
from_addr = self.conf.get("from")
|
|
to_addr = to or self.conf.get("to")
|
|
|
|
msg = MIMEText(body, 'plain', 'utf-8')
|
|
msg['From'] = from_addr
|
|
msg['To'] = to_addr
|
|
msg['Date'] = formatdate(localtime=True)
|
|
msg['Subject'] = Header(subject, 'utf-8')
|
|
|
|
with SMTPConnection(self.conf) as con:
|
|
con.sendmail(from_addr, to_addr, msg.as_string())
|
|
|
|
def _retry(self, subject, body, to):
|
|
for x in range(5):
|
|
try:
|
|
self._sendmail(subject, body, to)
|
|
except smtplib.SMTPConnectError:
|
|
time.sleep(60)
|
|
else:
|
|
break
|
|
|
|
|
|
class Stdout(object):
|
|
|
|
def __init__(self, conf):
|
|
pass
|
|
|
|
def __iter__(self):
|
|
|
|
yield "comments.new:new-thread", self._new_thread
|
|
yield "comments.new:finish", self._new_comment
|
|
yield "comments.edit", self._edit_comment
|
|
yield "comments.delete", self._delete_comment
|
|
yield "comments.activate", self._activate_comment
|
|
|
|
def _new_thread(self, thread):
|
|
logger.info("new thread %(id)s: %(title)s" % thread)
|
|
|
|
def _new_comment(self, thread, comment):
|
|
logger.info("comment created: %s", json.dumps(comment))
|
|
|
|
def _edit_comment(self, comment):
|
|
logger.info('comment %i edited: %s',
|
|
comment["id"], json.dumps(comment))
|
|
|
|
def _delete_comment(self, id):
|
|
logger.info('comment %i deleted', id)
|
|
|
|
def _activate_comment(self, thread, comment):
|
|
logger.info("comment %(id)s activated" % thread)
|