from flask import Flask, request, jsonify
import requests
from rich.console import Console
from rich.panel import Panel
import json
import os
from dotenv import load_dotenv
 
app = Flask(__name__)
console = Console()
load_dotenv()  # 加載環(huán)境變量

@app.route('/api/proxy', methods=['POST'])
def proxy():
    # 解析請(qǐng)求參數(shù)
    target_url = request.json.get('url')
    method = request.json.get('method', 'GET')
    headers = request.json.get('headers', {})
    data = request.json.get('data')
    
    # 發(fā)送請(qǐng)求
    try:
        if method == 'GET':
            resp = requests.get(target_url, headers=headers, params=data)
        elif method == 'POST':
            resp = requests.post(target_url, headers=headers, json=data)
        # 其他方法類似處理...
        
        # 記錄請(qǐng)求
        save_request_history({
            'url': target_url,
            'method': method,
            'status': resp.status_code,
            'time': resp.elapsed.total_seconds()
        })
        
        return format_response(resp)
    except Exception as e:
        return jsonify({'error': str(e)}), 500

def format_response(resp):
    content_type = resp.headers.get('Content-Type', '')
    
    if 'application/json' in content_type:
        try:
            pretty_json = json.dumps(resp.json(), indent=2, ensure_ascii=False)
            return Panel(pretty_json, title=f"[bold green]Status: {resp.status_code}")
        except:
            return Panel(resp.text, title=f"[bold yellow]Raw Response")
    elif 'xml' in content_type:
        return Panel(resp.text, title=f"[bold blue]XML Response")
    else:
        return Panel(resp.text, title=f"[bold magenta]Text Response")

HISTORY_FILE = 'request_history.json'
 
def save_request_history(record):
    try:
        if os.path.exists(HISTORY_FILE):
            with open(HISTORY_FILE) as f:
                history = json.load(f)
        else:
            history = []
            
        history.insert(0, record)
        if len(history) > 100:
            history.pop()
            
        with open(HISTORY_FILE, 'w') as f:
            json.dump(history, f, indent=2)
    except Exception as e:
        console.print(f"[bold red]Error saving history: {str(e)}")

BASE_URL=https://api.example.com
TIMEOUT=10

base_url = os.getenv('BASE_URL', 'http://localhost')
timeout = int(os.getenv('TIMEOUT', 5))

{
    "user_login": {
        "url": "/auth/login",
        "method": "POST",
        "headers": {"Content-Type": "application/json"},
        "body": {"username": "admin", "password": "123456"}
    }
}

@app.route('/api/templates', methods=['GET'])
def list_templates():
    with open('templates.json') as f:
        return jsonify(json.load(f))
 
@app.route('/api/execute_template', methods=['POST'])
def execute_template():
    template_name = request.json.get('template')
    # 加載并執(zhí)行模板...

requests.adapters.HTTPAdapter(pool_connections=10, pool_maxsize=10)

from fastapi import FastAPI, Request
 
@app.post("/async-proxy")
async def async_proxy(request: Request):
    # 使用 httpx 異步客戶端

curl -X POST http://localhost:5000/api/proxy \
-H "Content-Type: application/json" \
-d '{
    "url": "https://jsonplaceholder.typicode.com/posts/1",
    "method": "GET"
}'

[bold green]Status: 200
{
  "userId": 1,
  "id": 1,
  "title": "sunt aut facere repellat provident occaecati excepturi optio reprehenderit",
  "body": "quia et suscipit\nsuscipit recusandae consequuntur expedita et cum\nreprehenderit molestiae ut ut quas totam\nnostrum rerum est autem sunt rem eveniet architecto"
}

curl -X POST http://localhost:5000/api/proxy \
-H "Content-Type: application/json" \
-d '{
    "url": "https://jsonplaceholder.typicode.com/posts",
    "method": "POST",
    "headers": {"X-Custom-Header": "test"},
    "data": {"title": "foo", "body": "bar", "userId": 1}
}'

[bold green]Status: 201
{
    "title": "foo",
    "body": "bar",
    "userId": 1,
    "id": 101
}