import asyncio
import telnetlib3
import re
import time
from influxdb import InfluxDBClient

# ================= 路由器配置 =================
HOST = "192.168.1.1"
USER = "root"
PASSWORD = "你的密码"  # 替换为真实密码
INTERVAL = 10          # 10秒采集一次
# ==============================================

print("正在连接 InfluxDB 数据库...")
db_client = InfluxDBClient(host='127.0.0.1', port=8086)
db_client.create_database('router_metrics')
db_client.switch_database('router_metrics')

def get_net_bytes(data, iface):
    for line in data.split('\n'):
        if iface + ':' in line:
            parts = line.split(':')[1].split()
            if len(parts) >= 9:
                return int(parts[0]), int(parts[8]) # RX, TX
    return 0, 0

def get_port_speed(data, port):
    m = re.search(rf'{port} rx_bps:(\d+) tx_bps:(\d+)', data)
    return (int(m.group(1)), int(m.group(2))) if m else (0, 0)

async def run_command(reader, writer, cmd):
    writer.write(cmd + '\n')
    data = ""
    while True:
        chunk = await reader.read(1024)
        data += chunk
        if '#' in chunk:
            break
    return data

async def fetch_and_push():
    current_time = time.strftime('%H:%M:%S')
    try:
        # 1. 登录路由器
        reader, writer = await telnetlib3.open_connection(HOST, 23)
        while True:
            out = await reader.read(1024)
            if not out: break
            out_lower = out.lower()
            if 'login:' in out_lower: writer.write(USER + '\n')
            elif 'password:' in out_lower:
                writer.write(PASSWORD + '\n')
                break
            elif '#' in out_lower: break

        while True:
            if '#' in await reader.read(1024): break
            
        # 2. 连续发送命令抓取数据
        temp_data   = await run_command(reader, writer, 'cat /proc/tempsensor')
        cpu_data    = await run_command(reader, writer, 'cat /proc/cpuusage')
        mem_data    = await run_command(reader, writer, 'cat /proc/meminfo')
        uptime_data = await run_command(reader, writer, 'cat /proc/uptime')
        loadavg_data = await run_command(reader, writer, "cat /proc/loadavg")
        port_data   = await run_command(reader, writer, 'cat /proc/laninfo/portspeed')
        net_data    = await run_command(reader, writer, 'cat /proc/net/dev')
        
        writer.close()

        # 3. 数据解析
        temp_match = re.search(r'temper value:\s*(\d+)', temp_data)
        temperature = float(temp_match.group(1)) if temp_match else 0.0

        cpu0_match = re.search(r'\[0\]Usage:\s*([\d\.]+)%', cpu_data)
        cpu1_match = re.search(r'\[1\]Usage:\s*([\d\.]+)%', cpu_data)
        cpu0 = float(cpu0_match.group(1)) if cpu0_match else 0.0
        cpu1 = float(cpu1_match.group(1)) if cpu1_match else 0.0

        # 系统负载 Load Average
        loadavg_match = re.search(r'\n([\d\.]+)\s+([\d\.]+)\s+([\d\.]+)', loadavg_data)
        if loadavg_match:
            load_1m, load_5m, load_15m = float(loadavg_match.group(1)), float(loadavg_match.group(2)), float(loadavg_match.group(3))
        else:
            load_1m = load_5m = load_15m = 0.0

        # 内存占用率
        mem_total = int(re.search(r'MemTotal:\s+(\d+)', mem_data).group(1)) if re.search(r'MemTotal:\s+(\d+)', mem_data) else 1
        mem_avail = int(re.search(r'MemAvailable:\s+(\d+)', mem_data).group(1)) if re.search(r'MemAvailable:\s+(\d+)', mem_data) else 0
        mem_usage_percent = round((mem_total - mem_avail) / mem_total * 100, 2)

        # 运行时间
        uptime_match = re.search(r'\n([\d\.]+)\s+[\d\.]+', uptime_data)
        uptime_sec = float(uptime_match.group(1)) if uptime_match else 0.0

        # 全局流量 (ppp0 为拨号 WAN 口)
        ppp0_rx, ppp0_tx = get_net_bytes(net_data, 'ppp0')
        
        # 4. 推送 InfluxDB
        json_body = [{
            "measurement": "zte_AX3000Pro+",
            "fields": {
                "temperature": temperature,
                "cpu0": cpu0, "cpu1": cpu1,
                "load_1m": load_1m, "load_5m": load_5m, "load_15m": load_15m,
                "mem_usage_percent": mem_usage_percent,
                "uptime_sec": uptime_sec,
                "ppp0_rx_bytes": ppp0_rx, "ppp0_tx_bytes": ppp0_tx
            }
        }]
        db_client.write_points(json_body)
        print(f"[{current_time}] 满血推送 -> 温度:{temperature} | CPU:{cpu0}% | 内存:{mem_usage_percent}%")

    except Exception as e:
        print(f"[{current_time}] 抓取失败: {e}")

async def main():
    while True:
        await fetch_and_push()
        await asyncio.sleep(INTERVAL)

if __name__ == "__main__":
    asyncio.run(main())