from flask import Flask, request, jsonify, Response
import requests
from qushuiyin import clean_mask
import urllib.parse

app = Flask(__name__)


@app.route("/get_urls", methods=["POST"])
def get_urls():
    data = request.get_json()  # 从请求中获取JSON数据
    if not data:
        return jsonify({"error": "No data provided"}), 400

    # 处理数据
    url = clean_mask(data["share_url"])
    print(url)
    return jsonify(url), 200


@app.route("/proxy")
def proxy():
    # 从请求参数中获取视频链接
    video_url = request.args.get("url")
    print(video_url)

    # 检查视频链接是否提供
    if not video_url:
        return Response("No video URL provided", status=400)

    headers = {
        "User-Agent": "Mozilla/5.0 (iPhone; CPU iPhone OS 16_6 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/16.6 Mobile/15E148 Safari/604.1 Edg/122.0.0.0"
    }

    # 发送请求到视频链接
    try:
        response = requests.get(video_url, headers=headers, stream=True)
        response.raise_for_status()  # 检查请求是否成功
    except requests.RequestException as e:
        return Response(f"Error fetching video: {e}", status=500)

    # 将视频内容作为响应返回
    return Response(response.content, content_type=response.headers["Content-Type"])


if __name__ == "__main__":
    app.run(debug=True)