package com.gkhy.exam.system.service.impl;

import com.alibaba.fastjson2.JSONObject;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gkhy.exam.common.api.CommonPage;
import com.gkhy.exam.common.api.CommonResult;
import com.gkhy.exam.common.domain.model.LoginUserDetails;
import com.gkhy.exam.common.exception.ApiException;
import com.gkhy.exam.common.utils.PageUtils;
import com.gkhy.exam.common.utils.SecurityUtils;
import com.gkhy.exam.common.utils.StringUtils;
import com.gkhy.exam.system.domain.CompanyBasic;
import com.gkhy.exam.system.domain.ExQuestion;
import com.gkhy.exam.system.domain.SysCompany;
import com.gkhy.exam.system.mapper.CompanyBasicMapper;
import com.gkhy.exam.system.service.CompanyBasicService;
import com.gkhy.exam.system.service.SysCompanyService;
import org.apache.poi.ss.usermodel.*;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@Service
public class CompanyBasicServiceImpl extends ServiceImpl<CompanyBasicMapper, CompanyBasic> implements CompanyBasicService {

    @Autowired
    private CompanyBasicMapper companyBasicMapper;
    @Autowired
    private SysCompanyService sysCompanyService;



    @Override
    public CommonPage selectCompanyBasicList(Integer companyId) {

        if (!SecurityUtils.adminUser()){
            if (companyId==null){
                throw new ApiException("非管理员操作,查询条件不可为空");
            }
        }
        PageUtils.startPage();
        List<CompanyBasic> companyBasics = companyBasicMapper.selectCompanyBasicList(companyId);
        return CommonPage.restPage(companyBasics);
    }

    @Override
    public CommonResult insertCompanyBasic(CompanyBasic companyBasic) {
        LoginUserDetails loginUser = SecurityUtils.getLoginUser();
        List<CompanyBasic> companyBasics = companyBasicMapper.selectCompanyBasicList(companyBasic.getCompanyId());
        if (companyBasics.size()>0){
            throw new ApiException("当前企业已有相关数据，请删除原有数据后重试");
        }
        SysCompany sysCompany = sysCompanyService.selectCompanyById(Long.valueOf(companyBasic.getCompanyId()));
        companyBasic.setCompanyName(sysCompany.getName());
        companyBasic.setCreateBy(loginUser.getUsername());
        companyBasic.setCreateTime(LocalDateTime.now());
        int insert = companyBasicMapper.insert(companyBasic);
        if (insert>0){
            return CommonResult.success();
        }
        return CommonResult.failed();
    }

    @Override
    public CommonResult updateCompanyBasic(CompanyBasic companyBasic) {
        LoginUserDetails loginUser = SecurityUtils.getLoginUser();
        SysCompany sysCompany = sysCompanyService.selectCompanyById(Long.valueOf(companyBasic.getCompanyId()));
        companyBasic.setCompanyName(sysCompany.getName());
        companyBasic.setUpdateBy(loginUser.getUsername());
        companyBasic.setUpdateTime(LocalDateTime.now());
        int update = companyBasicMapper.updateBasicById(companyBasic);
        if (update>0){
            return CommonResult.success();
        }
        return CommonResult.failed();
    }

    @Override
    public CommonResult deletedCompanyBasic(Integer companyBasicId) {
        LoginUserDetails loginUser = SecurityUtils.getLoginUser();
        CompanyBasic companyBasic = new CompanyBasic();
        companyBasic.setId(companyBasicId);
        companyBasic.setDelFlag(2);
        companyBasic.setUpdateBy(loginUser.getUsername());
        companyBasic.setUpdateTime(LocalDateTime.now());
        int deleted = companyBasicMapper.updateById(companyBasic);
        if (deleted>0){
            return CommonResult.success();
        }
        return CommonResult.failed();
    }

    @Override
    @Transactional
    public CommonResult uploadBasic(MultipartFile file) {
        Workbook workbook = null;
        List<CompanyBasic> companyBasicslist = new ArrayList<>();
        StringBuffer stringBuffer = new StringBuffer();
        Integer t=0;
        try {
            workbook = WorkbookFactory.create(file.getInputStream());
            Sheet sheetAt = workbook.getSheetAt(0);
            LoginUserDetails loginUser = SecurityUtils.getLoginUser();
            for (int i = 1; i <= sheetAt.getLastRowNum(); i++) {
                Row row = sheetAt.getRow(i);
                String cellValueAsString = getCellValueAsString(row.getCell(1));
                if (StringUtils.isEmpty(cellValueAsString)){
                    continue;
                }
                CompanyBasic companyBasic = new CompanyBasic();
                if (row!=null ){

                    companyBasic.setBasic(getCellValueAsString(row.getCell(1)));
                    companyBasic.setIntroduce(getCellValueAsString(row.getCell(2)));
                    companyBasic.setTarget(getCellValueAsString(row.getCell(3)));
                    companyBasic.setQuality(getCellValueAsString(row.getCell(4)));
                    companyBasic.setActivity(getCellValueAsString(row.getCell(5)));
                    companyBasic.setAudit(getCellValueAsString(row.getCell(6)));
                    companyBasic.setEpiboly(getCellValueAsString(row.getCell(7)));
                    companyBasic.setResource(getCellValueAsString(row.getCell(8)));
                    companyBasic.setCreateTime(LocalDateTime.now());
                    companyBasic.setCreateBy(SecurityUtils.getUsername());
                    SysCompany sysCompany = sysCompanyService.selectCompanyByName(getCellValueAsString(row.getCell(0)));
                    companyBasic.setCompanyName(sysCompany.getName());
                    companyBasic.setCompanyId(Math.toIntExact(sysCompany.getId()));

                }
                companyBasicslist.add(companyBasic);
            }
            for (CompanyBasic companyBasic : companyBasicslist) {
                List<CompanyBasic> companyBasics = companyBasicMapper.selectCompanyBasicList(companyBasic.getCompanyId());
                if (companyBasics.size()>0){
                    stringBuffer.append(companyBasic.getCompanyName()).append(",");
                }else {
                    if (!SecurityUtils.adminUser()){
                        if (companyBasic.getCompanyId() != null && SecurityUtils.getCompanyId() == companyBasic.getCompanyId().longValue()) {
                            t+=companyBasicMapper.insert(companyBasic);
                        } else {
                            stringBuffer.append(companyBasic.getCompanyName()).append(",");
                        }
                    }else {
                        t+=companyBasicMapper.insert(companyBasic);
                    }
                }
            }

            if (t<1){
                throw new ApiException("导入公司基本信息失败");
            }
            workbook.close();
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
        if (stringBuffer.length()>0){
            return CommonResult.failed("只能导入本企业数据或该企业已有数据 ["+stringBuffer.toString()+"]");
        }
        return CommonResult.success();
    }

    /**
     * 安全获取单元格值
     */
    private String getCellValueAsString(Cell cell) {
        if (cell == null) return "";

        switch (cell.getCellType()) {
            case STRING:
                return cell.getStringCellValue().trim();
            case NUMERIC:
                if (DateUtil.isCellDateFormatted(cell)) {
                    return cell.getDateCellValue().toString();
                } else {
                    return String.valueOf((long) cell.getNumericCellValue());
                }
            case BOOLEAN:
                return String.valueOf(cell.getBooleanCellValue());
            case FORMULA:
                try {
                    return cell.getStringCellValue();
                } catch (IllegalStateException e) {
                    return String.valueOf(cell.getNumericCellValue());
                }
            default:
                return "";
        }
    }

}