package com.yami.trading.service.impl;

import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yami.trading.bean.model.HighLevelAuthRecord;
import com.yami.trading.bean.model.IpMenu;
import com.yami.trading.common.constants.DdosRedisKeys;
import com.yami.trading.common.util.BlacklistIpSerivceTimeWindow;
import com.yami.trading.common.util.DateUtils;
import com.yami.trading.common.util.LocklistIpSerivceTimeWindow;
import com.yami.trading.dao.IpMenuMapper;
import com.yami.trading.dao.user.HighLevelAuthRecordMapper;
import com.yami.trading.service.IpMenuService;
import lombok.Data;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.util.Date;
import java.util.List;

@Service
public class IpMenuServiceImpl   extends ServiceImpl<IpMenuMapper, IpMenu> implements IpMenuService, InitializingBean {

    @Autowired
    private LocklistIpSerivceTimeWindow locklistIpSerivceTimeWindow;

    @Autowired
    private BlacklistIpSerivceTimeWindow blacklistIpSerivceTimeWindow;

    @Autowired
    private RedisTemplate  redisTemplate;

    public void init() {


        List<IpMenu> list =  list(Wrappers.<IpMenu>query().lambda().eq(IpMenu::getDeleteStatus,0));;
        for (IpMenu ipMenu : list) {
            if (DateUtils.addHour(ipMenu.getLastOperaTime(), 24).after(new Date())) { // 黑名单 时间+1天>现在
                switch (ipMenu.getType()) {
                    case IpMenu.IP_BLACK:
                        blacklistIpSerivceTimeWindow.putBlackIp(ipMenu.getIp(), ipMenu.getIp());
                        break;
                    case IpMenu.IP_LOCK:
                        locklistIpSerivceTimeWindow.putLockIp(ipMenu.getIp(), ipMenu.getIp());
                        break;
                }
            } else {
                checkTimeWindows(ipMenu);
                ipMenu.setDeleteStatus(-1);
                redisTemplate.opsForValue().set(DdosRedisKeys.IP_MENU_IP + ipMenu.getIp(), ipMenu);
            }
        }
    }

    @Override
    public void saveIpMenu(IpMenu entity) {

        if(null==entity) return;
        checkTimeWindows(entity);
        save(entity);
        redisTemplate.opsForValue().set(DdosRedisKeys.IP_MENU_IP + entity.getIp(), entity);
    }

    @Override
    public void updateIpMenu(IpMenu entity) {
        if(null==entity) return;
        checkTimeWindows(entity);
        updateById(entity);
        redisTemplate.opsForValue().set(DdosRedisKeys.IP_MENU_IP + entity.getIp(), entity);
    }

    @Override
    public void deleteIpMenu(IpMenu entity) {
        if(null==entity) return;
        removeById(entity);
        redisTemplate.delete(DdosRedisKeys.IP_MENU_IP +entity.getIp());
    }

    @Override
    public IpMenu cacheByIp(String ip) {
        return (IpMenu) redisTemplate.opsForValue().get(DdosRedisKeys.IP_MENU_IP + ip);
    }

    /**
     * 新增ip到白名单
     * @param ip
     */
    public void saveIpMenuWhite(String ip) {
        IpMenu ipMenu = this.cacheByIp(ip);
        if (null == ipMenu) {
            ipMenu = new IpMenu();
            ipMenu.setCreateTime(new Date());
            ipMenu.setDeleteStatus(0);
            ipMenu.setLastOperaTime(new Date());
            ipMenu.setType(IpMenu.IP_WHITE);
            ipMenu.setIp(ip);
            saveIpMenu(ipMenu);
        } else if (ipMenu.getDeleteStatus() == -1 || !IpMenu.IP_WHITE.equals(ipMenu.getType())) {// 名单被删除或者不是白名单
            ipMenu.setDeleteStatus(0);
            ipMenu.setLastOperaTime(new Date());
            ipMenu.setType(IpMenu.IP_WHITE);
            updateIpMenu(ipMenu);
        }
    }

    public void checkTimeWindows(IpMenu entity) {
        if (entity.getDeleteStatus() == -1) {
            blacklistIpSerivceTimeWindow.delBlackIp(entity.getIp());
            locklistIpSerivceTimeWindow.delLockIp(entity.getIp());
            return;
        }

        switch (entity.getType()) {
            case IpMenu.IP_WHITE:
                if (blacklistIpSerivceTimeWindow.getBlackIp(entity.getIp()) != null) {// 白名单直接删除黑名单缓存
                    blacklistIpSerivceTimeWindow.delBlackIp(entity.getIp());
                }
                if (locklistIpSerivceTimeWindow.getLockIp(entity.getIp()) != null) {// 白名单直接删除锁定名单缓存
                    locklistIpSerivceTimeWindow.delLockIp(entity.getIp());
                }
                break;
            case IpMenu.IP_BLACK:
                if (locklistIpSerivceTimeWindow.getLockIp(entity.getIp()) != null) {// 删除锁定名单缓存
                    locklistIpSerivceTimeWindow.delLockIp(entity.getIp());
                }
                blacklistIpSerivceTimeWindow.putBlackIp(entity.getIp(), entity.getIp());
                break;
            case IpMenu.IP_LOCK:
                if (blacklistIpSerivceTimeWindow.getBlackIp(entity.getIp()) != null) {// 删除黑名单缓存
                    blacklistIpSerivceTimeWindow.delBlackIp(entity.getIp());
                }
                locklistIpSerivceTimeWindow.putLockIp(entity.getIp(), entity.getIp());
                break;
        }
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        init();
    }
}