package com.zhgd.xmgl.config;

import com.zhgd.xmgl.modules.basicdata.entity.ComputerAuth;
import com.zhgd.xmgl.modules.basicdata.mapper.ComputerAuthMapper;
import com.zhgd.xmgl.util.ComputerInfo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.CommandLineRunner;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.util.List;

/**
 * @program: wisdomSite
 * @description: 校验服务器
 * @author: Mr.Peng
 * @create: 2021-04-23 16:39
 **/
@Component
@Order(value = 1)
@Slf4j
public class CheckMac implements CommandLineRunner, ApplicationContextAware {
    @Autowired
    private ComputerAuthMapper computerAuthMapper;

    private ApplicationContext context;

    @Value("${checkAuthEnable}")
    private boolean checkAuthEnable;

    @Override
    public void run(String... args) throws Exception {
        if (checkAuthEnable) {
            ConfigurableApplicationContext ctx = (ConfigurableApplicationContext) context;
            List<String> macList = ComputerInfo.getMd5MacAddress();
            ComputerAuth auth = computerAuthMapper.getOneComputerAuth();
            if (auth != null) {
                if (!macList.contains(auth.getAddress())) {
                    log.info("---------------服务器未授权，请联系管理员----------------");
                    ctx.close();
                }
            }
        }
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        context = applicationContext;
    }
}