# [title: SimAdmin多实例管理]
# [language: python]
# [service: BOSS]
# [disable: false]
# [admin: true]
# [rule: ^sim帮助$]
# [rule: ^sim配置$]
# [rule: ^sim状态(?:\s+\S+)?$]
# [rule: ^sim短信\s+\S+\s+\S+\s+[\s\S]+$]
# [platform: qq,wx,tg,tb,fs]
# [open_source: true]
# [version: 1.0.0]
# [public: false]
# [price: 0]
# [description: 管理员专用：兼容多个 SimAdmin 实例，支持状态查询和短信发送。配置桶 otto/simadmin_instances，JSON数组或别名=URL多行。]

"""
Autman Python 插件：SimAdmin 多实例状态查询和短信发送。

配置方式（Autman 数据桶）：
- bucket: otto
- key: simadmin_instances
- value 支持两种格式：

1) JSON 数组（推荐）：
[
  {"name":"主卡", "base_url":"http://192.168.68.1:3000", "token":"可选", "timeout":10},
  {"name":"副卡", "base_url":"http://192.168.68.2:3000"}
]

2) 多行简写：
主卡=http://192.168.68.1:3000
副卡=http://192.168.68.2:3000,token=xxx,timeout=10

指令：
- sim帮助
- sim配置
- sim状态              查询全部实例
- sim状态 主卡          查询指定实例
- sim短信 主卡 10086 CXLL
- sim短信 all 10086 测试短信   （向全部实例发送，不建议滥用）
"""

import json
import re
import sys
import traceback
import urllib.error
import urllib.parse
import urllib.request

import middleware

BUCKET = "otto"
CONFIG_KEY = "simadmin_instances"
DEFAULT_TIMEOUT = 10


class SimAdminError(Exception):
    pass


def get_sender():
    return middleware.Sender(middleware.getSenderID())


def reply(sender, text):
    sender.reply(str(text))


def is_blank(value):
    return value is None or str(value).strip() == ""


def mask_secret(value):
    if not value:
        return ""
    s = str(value)
    if len(s) <= 8:
        return "***"
    return s[:3] + "***" + s[-3:]


def normalize_base_url(url):
    if is_blank(url):
        raise SimAdminError("base_url 不能为空")
    url = str(url).strip().rstrip("/")
    if not re.match(r"^https?://", url, re.I):
        raise SimAdminError("base_url 必须以 http:// 或 https:// 开头: " + url)
    return url


def parse_bool(value, default=False):
    if value is None:
        return default
    s = str(value).strip().lower()
    if s in ("1", "true", "yes", "y", "on", "启用", "是"):
        return True
    if s in ("0", "false", "no", "n", "off", "禁用", "否"):
        return False
    return default


def normalize_instance(raw, index=0):
    if not isinstance(raw, dict):
        raise SimAdminError("实例配置必须是对象")
    name = str(raw.get("name") or raw.get("alias") or raw.get("id") or ("sim" + str(index + 1))).strip()
    base_url = normalize_base_url(raw.get("base_url") or raw.get("url") or raw.get("host"))
    timeout = raw.get("timeout", raw.get("timeout_seconds", DEFAULT_TIMEOUT))
    try:
        timeout = int(timeout)
    except Exception:
        timeout = DEFAULT_TIMEOUT
    timeout = max(3, min(timeout, 60))
    token = str(raw.get("token") or raw.get("access_token") or raw.get("api_key") or "").strip()
    headers = raw.get("headers") if isinstance(raw.get("headers"), dict) else {}
    enabled = parse_bool(raw.get("enabled"), True)
    return {
        "name": name,
        "base_url": base_url,
        "timeout": timeout,
        "token": token,
        "headers": headers,
        "enabled": enabled,
    }


def parse_line_config(text):
    instances = []
    for line_no, line in enumerate(str(text).splitlines(), 1):
        line = line.strip()
        if not line or line.startswith("#"):
            continue
        if "=" not in line:
            raise SimAdminError("第 %s 行配置缺少 =" % line_no)
        name, rest = line.split("=", 1)
        name = name.strip()
        parts = [p.strip() for p in rest.split(",") if p.strip()]
        if not parts:
            raise SimAdminError("第 %s 行 URL 不能为空" % line_no)
        raw = {"name": name, "base_url": parts[0]}
        for part in parts[1:]:
            if "=" in part:
                k, v = part.split("=", 1)
                raw[k.strip()] = v.strip()
        instances.append(normalize_instance(raw, len(instances)))
    return instances


def load_instances():
    raw = middleware.bucketGet(BUCKET, CONFIG_KEY)
    if is_blank(raw):
        raise SimAdminError(
            "未配置 SimAdmin 实例。请在数据桶 %s/%s 写入 JSON数组或多行 alias=url 配置。" % (BUCKET, CONFIG_KEY)
        )
    text = str(raw).strip()
    if text.startswith("[") or text.startswith("{"):
        try:
            data = json.loads(text)
        except Exception as e:
            raise SimAdminError("SimAdmin 配置 JSON 解析失败: " + str(e))
        if isinstance(data, dict):
            if isinstance(data.get("instances"), list):
                data = data["instances"]
            else:
                data = [data]
        if not isinstance(data, list):
            raise SimAdminError("SimAdmin 配置必须是 JSON 数组或对象")
        instances = [normalize_instance(item, i) for i, item in enumerate(data)]
    else:
        instances = parse_line_config(text)
    instances = [x for x in instances if x.get("enabled")]
    if not instances:
        raise SimAdminError("没有启用的 SimAdmin 实例")
    names = set()
    for item in instances:
        if item["name"] in names:
            raise SimAdminError("SimAdmin 实例名称重复: " + item["name"])
        names.add(item["name"])
    return instances


def request_json(instance, method, path, body=None):
    base_url = instance["base_url"]
    url = base_url + path
    data = None
    headers = {"Accept": "application/json"}
    if body is not None:
        data = json.dumps(body, ensure_ascii=False).encode("utf-8")
        headers["Content-Type"] = "application/json"
    headers.update(instance.get("headers") or {})

    # SimAdmin 当前 API 无鉴权；这里兼容未来带 token 的反代/网关。
    token = instance.get("token") or ""
    if token:
        headers.setdefault("Authorization", "Bearer " + token)
        headers.setdefault("X-API-Token", token)

    req = urllib.request.Request(url, data=data, headers=headers, method=method.upper())
    try:
        with urllib.request.urlopen(req, timeout=instance["timeout"]) as resp:
            raw = resp.read().decode("utf-8", errors="replace")
            status_code = getattr(resp, "status", resp.getcode())
    except urllib.error.HTTPError as e:
        raw = e.read().decode("utf-8", errors="replace")
        raise SimAdminError("HTTP %s: %s" % (e.code, raw[:300]))
    except Exception as e:
        raise SimAdminError(str(e))

    if not raw:
        return {"status_code": status_code}
    try:
        payload = json.loads(raw)
    except Exception:
        return {"status_code": status_code, "raw": raw}
    if isinstance(payload, dict) and payload.get("status") == "error":
        raise SimAdminError(str(payload.get("message") or payload))
    return payload


def unwrap_data(payload):
    if isinstance(payload, dict) and isinstance(payload.get("data"), dict):
        return payload.get("data")
    return payload if isinstance(payload, dict) else {}


def pick(data, *keys, **kwargs):
    default = kwargs.get("default", "-")
    if not isinstance(data, dict):
        return default
    for key in keys:
        if key in data and data[key] not in (None, ""):
            return data[key]
    return default


def fmt_bytes(value):
    try:
        n = float(value)
    except Exception:
        return str(value) if value not in (None, "") else "-"
    units = ["B", "KB", "MB", "GB", "TB"]
    idx = 0
    while n >= 1024 and idx < len(units) - 1:
        n /= 1024.0
        idx += 1
    if idx == 0:
        return "%d%s" % (n, units[idx])
    return "%.1f%s" % (n, units[idx])


def fmt_percent(value):
    if value in (None, ""):
        return "-"
    try:
        n = float(value)
        if n <= 1:
            n *= 100
        return "%.1f%%" % n
    except Exception:
        return str(value)


def format_instance_status(instance):
    name = instance["name"]
    lines = ["【%s】" % name]
    try:
        health = request_json(instance, "GET", "/api/health")
        sim = unwrap_data(request_json(instance, "GET", "/api/sim"))
        network = unwrap_data(request_json(instance, "GET", "/api/network"))
        stats = unwrap_data(request_json(instance, "GET", "/api/stats"))
        sms_stats = unwrap_data(request_json(instance, "GET", "/api/sms/stats"))

        health_status = pick(health, "status", default="ok")
        version = pick(health, "version", default="-")
        iccid = pick(sim, "iccid", "ICCID")
        imsi = pick(sim, "imsi", "IMSI")
        msisdn = pick(sim, "phone_number", "msisdn", "number", "phone")
        operator = pick(network, "operator", "operator_name", "network_operator", "provider")
        reg = pick(network, "registration_state", "reg_state", "status", "state")
        signal = pick(network, "signal_strength", "signal", "rssi", "rsrp")
        uptime = pick(stats, "uptime", "uptime_text", "system_uptime")
        cpu = pick(stats, "cpu_usage", "cpu_percent", "cpu")
        mem = pick(stats, "memory_usage", "memory_percent", "mem")
        temp = pick(stats, "temperature", "temp", "cpu_temperature")
        sent = pick(sms_stats, "sent", "sent_count", "outbox", default="-")
        inbox = pick(sms_stats, "received", "received_count", "inbox", default="-")

        lines.append("健康: %s  版本: %s" % (health_status, version))
        lines.append("SIM: %s" % (msisdn if msisdn != "-" else "号码未知"))
        lines.append("ICCID: %s" % iccid)
        if imsi != "-":
            lines.append("IMSI: %s" % imsi)
        lines.append("网络: %s / %s / 信号:%s" % (operator, reg, signal))
        lines.append("系统: CPU %s / MEM %s / 温度 %s / 运行 %s" % (fmt_percent(cpu), fmt_percent(mem), temp, uptime))
        lines.append("短信: 收 %s / 发 %s" % (inbox, sent))
    except Exception as e:
        lines.append("查询失败: " + str(e))
    return "\n".join(lines)


def find_instances(instances, target):
    target = (target or "").strip()
    if target in ("", "全部", "all", "ALL", "*"):
        return instances
    matched = [x for x in instances if x["name"] == target]
    if matched:
        return matched
    matched = [x for x in instances if target.lower() in x["name"].lower()]
    if matched:
        return matched
    raise SimAdminError("未找到 SimAdmin 实例: %s。可用: %s" % (target, ", ".join(x["name"] for x in instances)))


def handle_status(sender, message):
    instances = load_instances()
    parts = message.split(None, 1)
    target = parts[1].strip() if len(parts) > 1 else ""
    selected = find_instances(instances, target)
    blocks = [format_instance_status(x) for x in selected]
    reply(sender, "\n\n".join(blocks))


def valid_phone(phone):
    # 兼容手机号、服务号、国际号码；只做安全字符校验，不强制中国手机号。
    return bool(re.match(r"^\+?[0-9][0-9\- ]{1,30}$", phone or ""))


def handle_sms(sender, message):
    m = re.match(r"^sim短信\s+(\S+)\s+(\S+)\s+([\s\S]+)$", message)
    if not m:
        raise SimAdminError("格式错误：sim短信 <实例名|all> <手机号> <内容>")
    target, phone, content = m.group(1), m.group(2), m.group(3).strip()
    if not valid_phone(phone):
        raise SimAdminError("手机号/号码格式不合法: " + phone)
    if not content:
        raise SimAdminError("短信内容不能为空")
    if len(content) > 1000:
        raise SimAdminError("短信内容过长，已拒绝发送")

    instances = load_instances()
    selected = find_instances(instances, target)
    results = []
    for instance in selected:
        try:
            payload = request_json(instance, "POST", "/api/sms/send", {
                "phone_number": phone,
                "content": content,
            })
            data = unwrap_data(payload)
            path = pick(data, "path", "message_path", "id", default="")
            suffix = "" if not path else " path=" + str(path)
            results.append("✅ %s 发送成功%s" % (instance["name"], suffix))
        except Exception as e:
            results.append("❌ %s 发送失败: %s" % (instance["name"], str(e)))
    reply(sender, "短信发送结果：\n" + "\n".join(results))


def handle_config(sender):
    instances = load_instances()
    lines = ["SimAdmin 实例配置："]
    for item in instances:
        lines.append("- %s => %s timeout=%ss token=%s" % (
            item["name"], item["base_url"], item["timeout"], mask_secret(item.get("token")) or "未设置"
        ))
    reply(sender, "\n".join(lines))


def handle_help(sender):
    reply(sender, "\n".join([
        "SimAdmin 多实例管理（仅管理员）",
        "配置桶：otto / simadmin_instances",
        "指令：",
        "- sim帮助",
        "- sim配置",
        "- sim状态              查询全部实例",
        "- sim状态 <实例名>      查询指定实例",
        "- sim短信 <实例名|all> <号码> <内容>",
        "示例：sim短信 主卡 10086 CXLL",
    ]))


def main():
    sender = get_sender()
    try:
        message = (sender.getMessage() or "").strip().strip('"')
        if not sender.isAdmin():
            reply(sender, "仅管理员可用")
            return
        if message == "sim帮助":
            handle_help(sender)
        elif message == "sim配置":
            handle_config(sender)
        elif message.startswith("sim状态"):
            handle_status(sender, message)
        elif message.startswith("sim短信"):
            handle_sms(sender, message)
        else:
            handle_help(sender)
    except SimAdminError as e:
        reply(sender, "SimAdmin插件错误：" + str(e))
    except Exception as e:
        traceback.print_exc(file=sys.stderr)
        reply(sender, "SimAdmin插件异常：" + str(e))


if __name__ == "__main__":
    main()