from flask import Flask, render_template, request, jsonify
import requests
from bs4 import BeautifulSoup
import re

app = Flask(__name__)

# 定义支持的网盘列表
SUPPORTED_PAN = [
    "百度网盘",
    "阿里云盘",
    "蓝奏云",
    "天翼云盘",
    "迅雷云盘",
    "夸克网盘",
    "115网盘",
    "Google Drive",
    "OneDrive",
    "Mega"
]

# 模拟搜索函数 - 实际应用中需要替换为真实的API或爬虫逻辑
def search_pan(query, pan_type=None):
    results = []
    
    # 这里应该是实际的搜索逻辑
    # 示例中只是返回模拟数据
    for i in range(1, 6):
        results.append({
            "title": f"{query}的资源{i}",
            "url": f"https://example.com/resource/{i}",
            "pan_type": pan_type if pan_type else "百度网盘",
            "size": f"{i}GB",
            "date": "2023-01-01",
            "valid": True
        })
    
    return results

@app.route('/')
def index():
    return render_template('index.html', pan_list=SUPPORTED_PAN)

@app.route('/search', methods=['GET', 'POST'])
def search():
    if request.method == 'POST':
        query = request.form.get('query')
        pan_type = request.form.get('pan_type')
    else:
        query = request.args.get('query')
        pan_type = request.args.get('pan_type')
    
    if not query:
        return jsonify({"error": "请输入搜索关键词"}), 400
    
    results = search_pan(query, pan_type)
    return jsonify({"results": results})

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