package com.flow.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.flow.mapper.CompanyMapper;
import com.flow.mapper.LoginFailRecordMapper;
import com.flow.mapper.ManagerMapper;
import com.flow.pojo.Company;
import com.flow.pojo.LoginFailRecord;
import com.flow.pojo.Manager;
import com.flow.service.LoginFailRecordService;
import com.flow.util.DESUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

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

@Service
public class LoginFailRecordServiceImpl extends ServiceImpl<LoginFailRecordMapper, LoginFailRecord> implements LoginFailRecordService {
    @Autowired
    CompanyMapper companyMapper;

    @Autowired
    ManagerMapper managerMapper;
    private static final int MAX_FAIL_COUNT = 5; // 最大失败次数
    
    @Override
    public void recordLoginFail(String account) throws Exception {
        String phone = null;
        QueryWrapper queryWrapper1 = new QueryWrapper();

            queryWrapper1.eq("companyabbName", account);

//        queryWrapper.eq("loginPhone", compan.getLoginphone());


//        Company company1 = new Company();
//        QueryWrapper queryWrapper = new QueryWrapper();
        queryWrapper1.eq("none", 0);
        List<Company> company = this.companyMapper.selectList(queryWrapper1);
        if (company.size()>0){
            phone= DESUtil.decrypt(company.get(0).getLoginphone(), DESUtil.key);
        }else{
            QueryWrapper queryWrapper2 = new QueryWrapper();

                queryWrapper2.eq("name",account);
            Manager manager = managerMapper.selectOne(queryWrapper2);
            phone=manager.getPhone();


        }

        QueryWrapper<LoginFailRecord> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("phone", phone);
        
        LoginFailRecord record = this.getOne(queryWrapper);
        
        if (record == null) {
            // 创建新记录
            record = new LoginFailRecord(account);
            record.setFailCount(1);
            record.setPhone(phone);
            this.save(record);
        } else {
            // 更新现有记录
            record.setFailCount(record.getFailCount() + 1);
            record.setUpdateTime(new Date());
            
            // 如果达到最大失败次数，锁定账号
            if (record.getFailCount() >= MAX_FAIL_COUNT) {
                record.setIsLocked(true);
                record.setLockTime(new Date());
            }
            
            this.updateById(record);
        }
    }
    
    @Override
    public boolean isAccountLocked(String account) {
        QueryWrapper<LoginFailRecord> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("account", account);
        
        LoginFailRecord record = this.getOne(queryWrapper);
        
        if (record == null) {
            return false;
        }
        
        return record.getIsLocked() != null && record.getIsLocked();
    }
    
    @Override
    public void unlockAccount(String account) {
        QueryWrapper<LoginFailRecord> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("phone", account);
        
        LoginFailRecord record = this.getOne(queryWrapper);
        
        if (record != null) {
            record.setFailCount(0);
            record.setIsLocked(false);
            record.setLockTime(null);
            record.setUpdateTime(new Date());
            this.updateById(record);
        }
    }
    
    @Override
    public int getFailCount(String account) {
        QueryWrapper<LoginFailRecord> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("account", account);
        
        LoginFailRecord record = this.getOne(queryWrapper);
        
        if (record == null) {
            return 0;
        }
        
        return record.getFailCount();
    }
}