from flask import Flask, send_file, abort
from pathlib import Path

app = Flask(__name__)

# 固定 screenshots 目录
PROJECT_ROOT = Path(__file__).parent.resolve()
SCREENSHOTS_DIR = PROJECT_ROOT / "screenshots"


@app.route('/image/<path:filename>')
def serve_image(filename):
    app = Flask(__name__)

    PROJECT_ROOT = Path(__file__).parent.resolve()
    SCREENSHOTS_DIR = PROJECT_ROOT / "screenshots"

    @app.route('/image/<path:filename>')
    def serve_image(filename):
        file_path = SCREENSHOTS_DIR / filename

        # 防止路径越界访问
        try:
            file_path.resolve().relative_to(SCREENSHOTS_DIR.resolve())
        except ValueError:
            abort(403, description=f"禁止访问目录外文件: {file_path.resolve()}")

        if not file_path.exists():
            abort(404, description=f"文件不存在: {file_path.resolve()}")

        return send_file(file_path, as_attachment=False)


if __name__ == '__main__':
    app.run(host='0.0.0.0', debug=True, port=5000)