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("/") 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/", 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/") 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/") 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")