messaggeria/app.py

import hashlib
import random
from pathlib import Path
import json
import logging
import os
import re
import sys
from datetime import datetime
from email.mime.text import MIMEText
from subprocess import PIPE, Popen
from uuid import uuid4

from flask import (Flask, abort, make_response, render_template, request,
                   url_for)

app = Flask(__name__)
if "MESSAGGERIA_SETTING" in os.environ:
    app.config.from_envvar("MESSAGGERIA_SETTING")


UPLOAD_DIR = os.getenv("UPLOAD_DIR", "./uploads/")

logging.basicConfig(level=logging.DEBUG)
app.logger.setLevel(logging.DEBUG)

def testo_random(basepath):
    '''Per sembrare meno spammosi, alleghiamo una bella poesia.
    Che idea di merda'''
    if not basepath:
        return ''
    texts = list(Path(basepath).rglob('*.txt'))
    if not texts:
        return ''
    return random.sample(texts, k=1)[0].open(encoding='utf8').read().strip()

def sendmail(sender, to, subject, body):
    # msg = MIMEText(body)
    # msg["From"] = sender
    # msg["To"] = to
    # msg["Subject"] = subject
    # p = Popen(["/usr/sbin/sendmail", "-t", "-oi"], stdin=PIPE)
    args = ["/usr/bin/mail", "-s", subject, "--"] + to
    p = Popen(args, stdin=PIPE)
    # p.communicate(msg.as_bytes())
    p.communicate(body.encode("utf8"))
    logging.info("mando mail a %s", str(to))


def read_config():
    try:
        cfgname = os.getenv("CONFIG_FILE", "./config.json")
        with open(cfgname) as buf:
            cfg = json.load(buf)
    except:
        app.logger.exception("Error reading conf")
        cfg = {}
    cfg.setdefault("sites", {})
    return cfg


@app.route("/")
def home():
    return "casella di destinazione non specificata"


@app.route("/<site>")
def site(site):
    logging.debug("Visito %s", site)
    try:
        sitedata = read_config()["sites"][site]
    except KeyError:
        logging.debug("Sito %s non definito", site)
        sitedata = {}
    return render_template("index.htm", siteid=site, sitedata=sitedata)


@app.route("/upload/<site>", methods=["POST"])
def upload(site):
    temp_fname = "_%s.ogg" % uuid4().hex
    temp_fpath = os.path.join(UPLOAD_DIR, temp_fname)
    # prima scrivi su un file temporaneo, poi fai rename
    h = hashlib.new("sha1")
    with open(temp_fpath, "wb") as buf:
        while True:
            some_data = request.stream.read(1024)
            if not some_data:
                break
            buf.write(some_data)
            h.update(some_data)
    # rinomina con l'hash
    app.logger.info("hash = %s", h.hexdigest())
    fname = "%s.ogg" % h.hexdigest()
    os.rename(temp_fpath, os.path.join(UPLOAD_DIR, fname))
    if site in read_config()["sites"]:
        to = read_config()["sites"][site].get("email", [])
        if to:
            sender = os.getenv("MAIL_FROM", "")
            if not sender:
                logging.info("Not sending email (unconfigured FROM)")
            else:
                logging.debug("Sending email for `%s` to `%s`", site, ";".join(to))
                url = url_for("play", fname=fname, _external=True, _scheme="https")
                sendmail(
                    sender,
                    to,
                    subject="Nuovo messaggio (%s)" % site,
                    body="Alle {now:%H:%M} hai ricevuto un messaggio nella segreteria di {site}\n"
                    "Puoi ascoltarlo cliccando su\n{url}\n"
                    "Per scaricare l'audio, troverai un link dentro la pagina\n\n"
                    "Il servizio è gentilmente offerto da degenerazione.xyz\n\n"
                    "Al fine di aumentare la vostra cultura, "
                    "e di far contento spamssassain, "
                    "ecco una bella poesia:\n\n{text}".format(
                        now=datetime.now(), site=site, url=url,
                        text=testo_random(os.getenv('TEXT_DIR', None)),
                    ),
                )
    else:
        logging.warn("Sito senza configurazione!")
    return fname


@app.route("/listen/<fname>")
def play(fname):
    # prevent path traversal or any other trick
    if "/" in fname or not re.match(r"^[a-z0-9]*.(ogg|wav)", fname):
        abort(400)
    fpath = os.path.join(UPLOAD_DIR, fname)
    if not os.path.exists(fpath):
        abort(404)
    return render_template("player.html", fname=fname)


@app.route("/download/<fname>")
def dl(fname):
    # prevent path traversal or any other trick
    if "/" in fname or not re.match(r"^[a-z0-9]*.(ogg|wav)", fname):
        abort(400)
    fpath = os.path.join(UPLOAD_DIR, fname)
    if not os.path.exists(fpath):
        abort(404)
    with open(fpath, "rb") as buf:
        content = buf.read()
    r = make_response(content)
    r.headers["Content-Type"] = "audio/ogg"  # TODO: better detect
    return r


@app.route("/info/license")
def license():
    return render_template("license.html")