package com.qianwen.smartman.common.utils;

import com.alibaba.fastjson.JSONObject;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.lang.invoke.SerializedLambda;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import org.smartboot.license.client.License;
import org.smartboot.license.client.LicenseEntity;
import org.smartboot.license.client.LicenseException;
import com.qianwen.core.tool.utils.Func;
import com.qianwen.core.tool.utils.SpringUtil;
import com.qianwen.smartman.modules.system.dto.LicenseDetailDTO;
import com.qianwen.smartman.modules.system.entity.Tenant;
import com.qianwen.smartman.modules.system.mapper.TenantMapper;
import org.springframework.util.DigestUtils;


public class LicenseUtil {
    private static final String CPU_LINUX = "dmidecode -t processor | grep ID | head -n 1";
    private static final String MainBORD_LINUX = "dmidecode | grep 'Serial Number' | head -n 1";
    private static Map<String, String> licenseMap = new HashMap<>();
    private static License license = new License();
    private static final String OS = System.getProperty("os.name");
    private static final TenantMapper tenantMapper = SpringUtil.getBean(TenantMapper.class);
    private static final String PROFILE = SpringUtil.getContext().getEnvironment().getActiveProfiles()[0];
    private static final String DEVICE_MAC = getMac();


    public static String getMac() {
        if (DEVICE_MAC != null) {
            return DEVICE_MAC;
        }
        try {
            String str = getCpuId() + getMainBordId();
            if ("".equals(str)) {
                return null;
            }
            return DigestUtils.md5DigestAsHex(str.getBytes(StandardCharsets.UTF_8)).toUpperCase();
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }
    }

    public static String getCpuId() throws IOException {
        String result;
        if (OS.contains("Windows")) {
            String[] windows = {"wmic", "cpu", "get", "ProcessorId"};
            System.getProperty("os.name");
            Process process = Runtime.getRuntime().exec(windows);
            process.getOutputStream().close();
            Scanner sc = new Scanner(process.getInputStream(), "utf-8");
            sc.next();
            result = sc.next();
        } else {
            result = execCommandLinux(CPU_LINUX);
        }
        System.out.println("CPU ID:" + result);
        return result;
    }

    public static String getMainBordId() {
        String result = "";
        if (OS.contains("Windows")) {
            try {
                File file = File.createTempFile("realhowto", ".vbs");
                file.deleteOnExit();
                FileWriter fw = new FileWriter(file);
                fw.write("Set objWMIService = GetObject(\"winmgmts:\\\\.\\root\\cimv2\")\nSet colItems = objWMIService.ExecQuery _ \n   (\"Select * from Win32_BaseBoard\") \nFor Each objItem in colItems \n    Wscript.Echo objItem.SerialNumber \n    exit for  ' do the first cpu only! \nNext \n");
                fw.close();
                Process p = Runtime.getRuntime().exec("cscript //NoLogo " + file.getPath());
                BufferedReader input = new BufferedReader(new InputStreamReader(p.getInputStream()));
                while (true) {
                    String line = input.readLine();
                    if (line == null) {
                        break;
                    }
                    result = result + line;
                }
                input.close();
            } catch (Exception e) {
                System.out.print("获取主板信息错误");
            }
        } else {
            result = execCommandLinux(MainBORD_LINUX);
        }
        System.out.println("Main BordID:" + result);
        return result.trim();
    }

    public static String execCommandLinux(String command) {
        String result = "";
        try {
            Process p = Runtime.getRuntime().exec(new String[]{"sh", "-c", command});
            BufferedReader br = new BufferedReader(new InputStreamReader(p.getInputStream()));
            String line = br.readLine();
            if (line != null) {
                result = result + line;
            }
            br.close();
        } catch (IOException e) {
            System.out.print("命令执行错误");
        }
        return result;
    }

    public static String getEncryptCode(String tenantId) {
        if (licenseMap.get(tenantId) != null) {
            return licenseMap.get(tenantId);
        }
        Tenant tenant = (Tenant) tenantMapper.selectOne(Lambda.eq((v0) -> {
            return v0.getTenantId();
        }, tenantId));
        if (Func.isNotEmpty(tenant.getBladeLicense())) {
            licenseMap.put(Func.toStr(tenant.getTenantId()), tenant.getBladeLicense());
        }
        return tenant.getBladeLicense();
    }

    public static void setEncryptCode(String tenantId, String encrypt) throws LicenseException {
        LicenseEntity licenseEntity = license.loadLicense(encrypt);
        macValid(licenseEntity);
        licenseMap.put(tenantId, encrypt);
        Tenant tenant = new Tenant();
        tenant.setTenantId(tenantId);
        tenant.setBladeLicense(encrypt);
        tenantMapper.update(tenant, Lambda.eq((v0) -> {
            return v0.getTenantId();
        }, tenant.getTenantId()));
    }

    public static LicenseDetailDTO loadLicense(String tenantId) throws LicenseException {
        if (!"dev".equals(PROFILE)) {
            String encryptCode = getEncryptCode(tenantId);
            LicenseEntity entity = license.loadLicense(encryptCode);
            LicenseDetailDTO detailDTO = macValid(entity);
            return detailDTO;
        }
        return null;
    }

    public static LicenseDetailDTO macValid(LicenseEntity entity) {
        String json = new String(entity.getData());
        LicenseDetailDTO detailDTO = (LicenseDetailDTO) JSONObject.parseObject(json, LicenseDetailDTO.class);
        if (detailDTO.getMac() == null || !detailDTO.getMac().equals(DEVICE_MAC)) {
            throw new LicenseException(MessageUtils.message("system.license.device.err", new Object[0]));
        }
        return detailDTO;
    }

    public static Map<String, String> getLicenseMap() {
        return licenseMap;
    }
}