"""对已配置的本地提供商执行显式、有界的连接冒烟测试。

默认仅生成计划；--execute 会发送一次测试请求，但不会读取凭据。
输出会将尚未验证的协议场景保留为待处理状态。
"""
import argparse
from datetime import datetime, timezone
import json
from pathlib import Path
from urllib.error import HTTPError, URLError
from urllib.parse import urlparse
from urllib.request import Request, urlopen

SCENARIOS = ['model_discovery', 'tool_roundtrip', 'stream_reasoning_and_content',
             'stream_cancel', 'cache_hit_and_miss', 'context_limit', 'context_compression']


def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--base-url', default='http://127.0.0.1:8000')
    parser.add_argument('--provider', required=True)
    parser.add_argument('--model', required=True)
    parser.add_argument('--output', required=True, type=Path)
    parser.add_argument('--execute', action='store_true', help='Perform one provider connection test; may incur provider charges')
    args = parser.parse_args()
    target = urlparse(args.base_url)
    if target.scheme != 'http' or target.hostname not in ('127.0.0.1', 'localhost', '::1') or target.username or target.password or target.query or target.fragment:
        parser.error('Use a local HTTP AI Core address without credentials or query parameters')
    result = {'date': datetime.now(timezone.utc).isoformat(), 'provider': args.provider, 'model': args.model,
              'max_test_requests': 1, 'connection': 'pending',
              'scenarios': {name: 'pending' for name in SCENARIOS}, 'overall': 'not_accepted'}
    if args.execute:
        body = json.dumps({'provider_id': args.provider, 'model': args.model}).encode()
        request = Request(args.base_url.rstrip('/') + '/api/providers/test', data=body, headers={'Content-Type': 'application/json'}, method='POST')
        try:
            with urlopen(request, timeout=60) as response:
                payload = json.load(response)
            result['connection'] = 'passed' if payload.get('success') is True else 'failed'
            result['latency_ms'] = payload.get('latency_ms')
        except HTTPError as error:
            result['connection'] = 'failed'
            result['http_status'] = error.code  # 不保存远程错误正文或响应头。
        except (URLError, TimeoutError, ValueError):
            result['connection'] = 'unavailable'
    args.output.write_text(json.dumps(result, ensure_ascii=False, indent=2), encoding='utf-8')


if __name__ == '__main__':
    main()