"""Read-only platform pair status; return only fixed statuses and GPU codes."""
import json
import os
import urllib.error
import urllib.parse
import urllib.request
from .config_params import GPU_ENUMS
from .task_add_client import resolve_token

SAFE_STATUSES = {'ok', 'transport_error', 'http_error', 'malformed_json',
                 'platform_error', 'malformed_data', 'missing/invalid_token'}


def precheck_model(model, token=None, opener=urllib.request.urlopen):
    try:
        token = resolve_token() if token is None else token
        if not isinstance(token, str) or not token.strip() or '<' in token or '>' in token:
            return 'missing/invalid_token', {}
        request = urllib.request.Request(
            os.environ.get('MODELHUB_API_BASE', 'https://modelhub.org.cn').rstrip('/')
            + '/api/computility/models/search-by-model-id?modelId='
            + urllib.parse.quote(model, safe=''),
            headers={'Xc-Token': token, 'Accept': 'application/json'}, method='GET')
    except Exception:
        return 'missing/invalid_token', {}
    try:
        with opener(request, timeout=30) as response:
            if response.status != 200:
                return 'http_error', {}
            raw = response.read()
    except urllib.error.HTTPError:
        return 'http_error', {}
    except Exception:
        return 'transport_error', {}
    try:
        body = json.loads(raw)
    except (ValueError, TypeError, UnicodeError):
        return 'malformed_json', {}
    if not isinstance(body, dict) or 'code' not in body:
        return 'malformed_data', {}
    if type(body['code']) not in (str, int) or str(body['code']) != '0':
        return 'platform_error', {}
    data = body.get('data')
    if not isinstance(data, dict) or not isinstance(data.get('verifyResult'), dict):
        return 'malformed_data', {}
    result = {}
    for gpu, row in data['verifyResult'].items():
        if gpu not in GPU_ENUMS.values():
            continue
        if not isinstance(row, dict) or not isinstance(row.get('result'), str):
            return 'malformed_data', {}
        code = {'已验证': 60027, '验证中': 60028}.get(row['result'])
        if code is not None:
            result[gpu] = code
    return 'ok', result