from flask import Flask, request, send_from_directory, abort
import os
import threading
import time
import uuid
import logging


MAX_UPLOAD_SIZE_MB = 10
UPLOAD_FOLDER = "uploads"
IP_ADDRESS = "42.193.147.60"
PORT = 4244

logging.basicConfig(level=logging.INFO)
app = Flask(__name__)
app.config["MAX_CONTENT_LENGTH"] = MAX_UPLOAD_SIZE_MB * 1024 * 1024
os.makedirs(UPLOAD_FOLDER, exist_ok=True)


@app.route("/portal/push", methods=["POST"])
def push_something():
    if "file" not in request.files:
        return "No file part", 400
    file = request.files["file"]
    if file:
        filename = str(uuid.uuid4())
        filepath = os.path.join(UPLOAD_FOLDER, filename)
        logging.info(f"Saving file: {filename}")
        file.save(filepath)
        threading.Thread(target=delete_file, args=(filepath,)).start()
        return f"http://{IP_ADDRESS}:{PORT}/portal/pull/{filename}"


@app.route("/portal/pull/<filename>", methods=["GET"])
def pull_something(filename):
    if filename:
        filepath = os.path.join(UPLOAD_FOLDER, filename)
        if os.path.exists(filepath):
            logging.info(f"Serving file: {filename}")
            return send_from_directory(UPLOAD_FOLDER, filename, as_attachment=True)
        else:
            abort(404)
    else:
        return "File parameter is missing", 400


def delete_file(filepath):
    time.sleep(60)
    if os.path.exists(filepath):
        os.remove(filepath)
        logging.info(f"Deleted file: {os.path.basename(filepath)}")


if __name__ == "__main__":
    for filename in os.listdir(UPLOAD_FOLDER):
        filepath = os.path.join(UPLOAD_FOLDER, filename)
        if os.path.isfile(filepath):
            os.remove(filepath)
            logging.info(f"Deleted file: {filename}")
    
    app.run(host="0.0.0.0", port=PORT)