package com.iailab.module.infra.util;

import cn.hutool.core.util.NumberUtil;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.iailab.module.infra.api.monitor.dto.MonitorDiskDTO;
import com.iailab.module.infra.api.monitor.dto.MonitorMemDTO;
import org.apache.commons.lang3.ObjectUtils;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.lang.management.ManagementFactory;
import java.lang.management.OperatingSystemMXBean;
import java.math.BigDecimal;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.nio.file.FileStore;
import java.nio.file.FileSystems;
import java.util.ArrayList;
import java.util.List;

public class ServerInfoCollector {

    private static BigDecimal UNIT = new BigDecimal(1024 * 1024);

    private static BigDecimal PERCENT_UNIT = new BigDecimal(100);

    private static String hostName;

    private static String hostIp;

    public static void initServerInfo() throws UnknownHostException {
        // 获取本地主机对象
        InetAddress localHost = InetAddress.getLocalHost();
        // 获取主机名
        hostName = localHost.getHostName();
//        // 获取IP地址
        hostIp = localHost.getHostAddress();
//        try {
//            Enumeration<NetworkInterface> interfaces = NetworkInterface.getNetworkInterfaces();
//            while (interfaces.hasMoreElements()) {
//                NetworkInterface networkInterface = interfaces.nextElement();
//                // 过滤掉loopback接口
//                if (networkInterface.isLoopback() || !networkInterface.isUp()) {
//                    continue;
//                }
//                Enumeration<InetAddress> addresses = networkInterface.getInetAddresses();
//                while (addresses.hasMoreElements()) {
//                    InetAddress addr = addresses.nextElement();
//                    // 过滤掉IPv6地址和回环地址
//                    if (!addr.isLoopbackAddress() && addr.getHostAddress().contains(".")) { // 确保是IPv4地址
//                        System.out.println("局域网IP地址: " + addr.getHostAddress());
//                        hostIp = addr.getHostAddress();
//                    }
//                }
//            }
//        } catch (SocketException e) {
//            e.printStackTrace();
//        }
    }

    /**
     * 内存信息处理
     */
    public static MonitorMemDTO collectMonitorMem(String serverName) throws UnknownHostException {
        initServerInfo();
        OperatingSystemMXBean operatingSystemBean = ManagementFactory.getOperatingSystemMXBean();
        JSONObject operatingSystemJson = JSON.parseObject(JSON.toJSONString(operatingSystemBean));
        BigDecimal totalPhysicalMemory = operatingSystemJson.getBigDecimal("totalPhysicalMemorySize");
        BigDecimal freePhysicalMemory = operatingSystemJson.getBigDecimal("freePhysicalMemorySize");
        BigDecimal usedPhysicalMemory = totalPhysicalMemory.subtract(freePhysicalMemory);
        BigDecimal systemCpuLoad = operatingSystemJson.getBigDecimal("systemCpuLoad");
        BigDecimal processCpuLoad = operatingSystemJson.getBigDecimal("processCpuLoad");
        Runtime runtime = Runtime.getRuntime();
        MonitorMemDTO monitorMemDTO = new MonitorMemDTO();
        monitorMemDTO.setHostName(hostName)
                .setHostIp(hostIp)
                .setServerName(serverName)
                .setPhysicalTotal(totalPhysicalMemory.divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setPhysicalUsed(usedPhysicalMemory.divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setPhysicalFree(freePhysicalMemory.divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setPhysicalUsage(NumberUtil.div(usedPhysicalMemory, totalPhysicalMemory).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setRuntimeTotal(new BigDecimal(runtime.totalMemory()).divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setRuntimeFree(new BigDecimal(runtime.freeMemory()).divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setRuntimeMax(new BigDecimal(runtime.maxMemory()).divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setRuntimeUsed(new BigDecimal(runtime.maxMemory() - runtime.totalMemory() + runtime.freeMemory()).divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setRuntimeUsage(new BigDecimal(NumberUtil.div(runtime.totalMemory()/1f - runtime.freeMemory(), runtime.totalMemory())).setScale(2, BigDecimal.ROUND_HALF_UP))
                .setSystemCpuLoad(systemCpuLoad.multiply(PERCENT_UNIT))
                .setProcessCpuLoad(processCpuLoad.multiply(PERCENT_UNIT));
        return monitorMemDTO;
    }

    /**
     * 磁盘信息处理
     */
    public static List<MonitorDiskDTO> collectMonitorDisk() throws IOException {
        List<MonitorDiskDTO> monitorDiskSaveReqVOS = new ArrayList<>();
        String os = System.getProperty("os.name").toLowerCase();
        System.out.println(os);
        if (os.contains("win")) {
            getWindowsDiskUsage(monitorDiskSaveReqVOS);
        } else if (os.contains("nix") || os.contains("nux") || os.contains("mac")) {
            try {
                getLinuxDiskUsage(monitorDiskSaveReqVOS);
            } catch (IOException | InterruptedException e) {
                e.printStackTrace();
            }
        } else {
            System.out.println("不支持的服务器系统！");
        }
        return monitorDiskSaveReqVOS;
    }

    private static void getWindowsDiskUsage(List<MonitorDiskDTO> monitorDiskSaveReqVOS) throws IOException {
        initServerInfo();
        // 获取默认文件系统的所有存储设备
        Iterable<FileStore> fileStores = FileSystems.getDefault().getFileStores();
        for (FileStore store : fileStores) {
            MonitorDiskDTO monitorDiskSaveReqVO = new MonitorDiskDTO();
            // 获取磁盘的总空间、已用空间和可用空间
            BigDecimal totalSpace = new BigDecimal(store.getTotalSpace()).divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP); // 总空间
            BigDecimal usableSpace = new BigDecimal(store.getUsableSpace()).divide(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP); // 可用空间
            BigDecimal usedSpace = totalSpace.subtract(usableSpace).setScale(2, BigDecimal.ROUND_HALF_UP); // 计算已用空间
            monitorDiskSaveReqVO.setHostName(hostName)
                    .setHostIp(hostIp)
                    .setDisk(store.toString())
                    .setDiskName(ObjectUtils.isNotEmpty(store.name()) ? store.name(): "/")
                    .setSpaceTotal(totalSpace)
                    .setSpaceUsable(usableSpace)
                    .setSpaceUsed(usedSpace)
                    .setSpaceRatio(usedSpace.divide(totalSpace, BigDecimal.ROUND_HALF_UP).setScale(4, BigDecimal.ROUND_HALF_UP));
            monitorDiskSaveReqVOS.add(monitorDiskSaveReqVO);
        }
    }

    /**
     * 获取 Linux 系统的磁盘使用情况
     */
    private static void getLinuxDiskUsage(List<MonitorDiskDTO> monitorDiskSaveReqVOS) throws IOException, InterruptedException {
        Process process = Runtime.getRuntime().exec("df -h");
        BufferedReader reader = new BufferedReader(new InputStreamReader(process.getInputStream()));
        String line;
        boolean isHeader = true;
        while ((line = reader.readLine()) != null) {
            if (isHeader) {
                isHeader = false;
                continue;
            }
            String[] parts = line.trim().split("\\s+");
            System.out.println(parts);
            if (parts.length >= 6) {
                MonitorDiskDTO monitorDiskSaveReqVO = new MonitorDiskDTO();
                String filesystem = parts[0];
                String size = parts[1];
                String used = parts[2];
                String available = parts[3];
                String mountedOn = parts[5];
                BigDecimal totalSpace = formatSize(size);
                BigDecimal usableSpace = formatSize(available);
                BigDecimal usedSpace = formatSize(used);
                monitorDiskSaveReqVO.setHostName(hostName)
                        .setHostIp(hostIp)
                        .setDisk(filesystem + "(" + mountedOn + ")")
                        .setDiskName(mountedOn)
                        .setSpaceTotal(totalSpace)
                        .setSpaceUsable(usableSpace)
                        .setSpaceUsed(usedSpace)
                        .setSpaceRatio(usedSpace.divide(totalSpace, BigDecimal.ROUND_HALF_UP).setScale(4, BigDecimal.ROUND_HALF_UP));
                monitorDiskSaveReqVOS.add(monitorDiskSaveReqVO);
            }
        }
        int exitCode = process.waitFor();
        if (exitCode != 0) {
            System.err.println("Command execution failed with exit code: " + exitCode);
        }
        reader.close();
    }

    /**
     * 将字节大小转换为可读的格式
     * @param size 字节数
     * @return 人类可读的大小(MB)
     */
    private static BigDecimal formatSize(String size) {
        if(size.contains("T")){
            return new BigDecimal(Float.parseFloat(size.replace("T",""))).multiply(UNIT).setScale(2, BigDecimal.ROUND_HALF_UP);
        } else if (size.contains("G")){
            return new BigDecimal(Float.parseFloat(size.replace("G","")) * 1024).setScale(2, BigDecimal.ROUND_HALF_UP);
        } else if (size.contains("M")){
            return new BigDecimal(Float.parseFloat(size.replace("M",""))).setScale(2, BigDecimal.ROUND_HALF_UP);
        } else if (size.contains("K")){
            return new BigDecimal(Float.parseFloat(size.replace("K","")) / 1024).setScale(2, BigDecimal.ROUND_HALF_UP);
        } else {
            return new BigDecimal(0);
        }
    }
}