#!/bin/bash
# =============================================================================
# 验证 priority 修复 - 端到端测试 (v5 - 一次性脚本)
# =============================================================================

set -e

# 1. 在 backend 容器内生成 token 并直接调用 API
docker exec wecom_it_backend python3 - <<'PYEOF'
import redis
import json
import secrets
import urllib.request
import urllib.parse

REDIS_PWD = "WKzl7jgKTkeWxFWGIBXfzokm59Gvfr76"
r = redis.Redis(host="redis", port=6379, password=REDIS_PWD, db=0)

# 写 token
token = secrets.token_urlsafe(32)
user_info = {
    "employee_id": "sxn",
    "username": "admin",
    "name": "宋献",
    "role": "admin",
    "department": "IT支持组",
    "login_source": "test",
    "login_method": "test_script",
    "last_active": "2026-07-28T02:00:00",
}
r.setex(f"user:token:{token}", 8 * 3600, json.dumps(user_info, ensure_ascii=False))
print(f"TOKEN: {token}", flush=True)

# 直接通过容器内 localhost 调 API
def call_api(url, params):
    full_url = f"http://localhost:8000{url}?{urllib.parse.urlencode(params)}"
    req = urllib.request.Request(full_url, headers={"Authorization": f"Bearer {token}"})
    try:
        with urllib.request.urlopen(req, timeout=10) as resp:
            data = json.loads(resp.read().decode("utf-8"))
            return data
    except urllib.error.HTTPError as e:
        return {"error": str(e), "code": e.code}
    except Exception as e:
        return {"error": str(e)}

# 测试 1: rule_type=routing_target
print("\n===== Test 1: rule_type=routing_target =====", flush=True)
r1 = call_api("/api/admin/quick-rules", {"rule_type": "routing_target", "page": 1, "size": 20})
print(f"code: {r1.get('code')}", flush=True)
print(f"message: {r1.get('message')}", flush=True)
if r1.get("data"):
    items = r1["data"].get("items", [])
    print(f"total: {r1['data'].get('total')}", flush=True)
    print(f"items count: {len(items)}", flush=True)
    if items:
        print(f"first item: {json.dumps(items[0], ensure_ascii=False, indent=2)}", flush=True)

# 测试 2: rule_type=routing_target&category=行政
print("\n===== Test 2: rule_type=routing_target&category=行政 =====", flush=True)
r2 = call_api("/api/admin/quick-rules", {"rule_type": "routing_target", "category": "行政", "page": 1, "size": 20})
print(f"code: {r2.get('code')}", flush=True)
print(f"message: {r2.get('message')}", flush=True)
if r2.get("data"):
    items = r2["data"].get("items", [])
    print(f"total: {r2['data'].get('total')}", flush=True)
    print(f"items count: {len(items)}", flush=True)
    if items:
        print(f"first item: {json.dumps(items[0], ensure_ascii=False, indent=2)}", flush=True)

# 测试 3: rule_type=routing_target&category=人力资源
print("\n===== Test 3: rule_type=routing_target&category=人力资源 =====", flush=True)
r3 = call_api("/api/admin/quick-rules", {"rule_type": "routing_target", "category": "人力资源", "page": 1, "size": 20})
print(f"code: {r3.get('code')}", flush=True)
print(f"message: {r3.get('message')}", flush=True)
if r3.get("data"):
    items = r3["data"].get("items", [])
    print(f"total: {r3['data'].get('total')}", flush=True)
    print(f"items count: {len(items)}", flush=True)

# 测试 4: rule_type=routing_target&category=IT服务
print("\n===== Test 4: rule_type=routing_target&category=IT服务 =====", flush=True)
r4 = call_api("/api/admin/quick-rules", {"rule_type": "routing_target", "category": "IT服务", "page": 1, "size": 20})
print(f"code: {r4.get('code')}", flush=True)
print(f"message: {r4.get('message')}", flush=True)
if r4.get("data"):
    items = r4["data"].get("items", [])
    print(f"total: {r4['data'].get('total')}", flush=True)
    if items:
        print(f"item: {json.dumps(items[0], ensure_ascii=False, indent=2)}", flush=True)

# 清理
r.delete(f"user:token:{token}")
print("\n===== Cleanup OK =====", flush=True)
PYEOF

echo ""
echo "===== 验证完成 ====="