package com.doumee.service.business.impl;

import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.doumee.core.annotation.excel.ExcelImporter;
import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
import com.doumee.core.model.LoginUserInfo;
import com.doumee.core.model.PageData;
import com.doumee.core.model.PageWrap;
import com.doumee.core.utils.Constants;
import com.doumee.core.utils.DateUtil;
import com.doumee.core.utils.Utils;
import com.doumee.dao.business.QuestionnaireMapper;
import com.doumee.dao.business.QuestionnaireProblemMapper;
import com.doumee.dao.business.model.Questionnaire;
import com.doumee.dao.business.model.QuestionnaireProblem;
import com.doumee.service.business.QuestionnaireService;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import java.io.IOException;
import org.apache.commons.lang3.StringUtils;
import org.apache.poi.openxml4j.exceptions.InvalidFormatException;
import org.apache.shiro.SecurityUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

import java.util.List;
import org.springframework.web.multipart.MultipartFile;

/**
 * 问卷调查基本信息表Service实现
 * @author 江蹄蹄
 * @date 2023/02/15 08:55
 */
@Service
public class QuestionnaireServiceImpl implements QuestionnaireService {

    @Autowired
    private QuestionnaireMapper questionnaireMapper;
    @Autowired
    private QuestionnaireProblemMapper questionnaireProblemMapper;

    @Override
    public Integer create(Questionnaire questionnaire) {
        questionnaireMapper.insert(questionnaire);
        return questionnaire.getId();
    }

    @Override
    public void deleteById(Integer id) {
        questionnaireMapper.deleteById(id);
    }

    @Override
    public void delete(Questionnaire questionnaire) {
        UpdateWrapper<Questionnaire> deleteWrapper = new UpdateWrapper<>(questionnaire);
        questionnaireMapper.delete(deleteWrapper);
    }

    @Override
    public void deleteByIdInBatch(List<Integer> ids) {
        if (CollectionUtils.isEmpty(ids)) {
            return;
        }
        questionnaireMapper.deleteBatchIds(ids);
    }

    @Override
    public void updateById(Questionnaire questionnaire) {
        questionnaireMapper.updateById(questionnaire);
    }

    @Override
    public void updateByIdInBatch(List<Questionnaire> questionnaires) {
        if (CollectionUtils.isEmpty(questionnaires)) {
            return;
        }
        for (Questionnaire questionnaire: questionnaires) {
            this.updateById(questionnaire);
        }
    }

    @Override
    public Questionnaire findById(Integer id) {
        Questionnaire model =questionnaireMapper.selectById(id);
        if(model !=null){
            QuestionnaireProblem qp = new QuestionnaireProblem();
            qp.setIsdeleted(Constants.ZERO);
            qp.setQuestionnaireId(id);
            List<QuestionnaireProblem> problems = questionnaireProblemMapper.selectList(new QueryWrapper<>(qp).orderByAsc("sortnum"));
            model.setProblemList(problems);
        }
        return model;
    }

    @Override
    public Questionnaire findOne(Questionnaire questionnaire) {
        QueryWrapper<Questionnaire> wrapper = new QueryWrapper<>(questionnaire);
        return questionnaireMapper.selectOne(wrapper);
    }

    @Override
    public List<Questionnaire> findList(Questionnaire questionnaire) {
        QueryWrapper<Questionnaire> wrapper = new QueryWrapper<>(questionnaire);
        return questionnaireMapper.selectList(wrapper);
    }
  
    @Override
    public PageData<Questionnaire> findPage(PageWrap<Questionnaire> pageWrap) {
        IPage<Questionnaire> page = new Page<>(pageWrap.getPage(), pageWrap.getCapacity());
        QueryWrapper<Questionnaire> queryWrapper = new QueryWrapper<>();
        Utils.MP.blankToNull(pageWrap.getModel());
        if (pageWrap.getModel().getId() != null) {
            queryWrapper.lambda().eq(Questionnaire::getId, pageWrap.getModel().getId());
        }

        if (pageWrap.getModel().getCreator() != null) {
            queryWrapper.lambda().eq(Questionnaire::getCreator, pageWrap.getModel().getCreator());
        }
        if (pageWrap.getModel().getCreateDate() != null) {
            queryWrapper.lambda().ge(Questionnaire::getCreateDate, Utils.Date.getStart(pageWrap.getModel().getCreateDate()));
            queryWrapper.lambda().le(Questionnaire::getCreateDate, Utils.Date.getEnd(pageWrap.getModel().getCreateDate()));
        }
        if (pageWrap.getModel().getEditor() != null) {
            queryWrapper.lambda().eq(Questionnaire::getEditor, pageWrap.getModel().getEditor());
        }
        if (pageWrap.getModel().getEditDate() != null) {
            queryWrapper.lambda().ge(Questionnaire::getEditDate, Utils.Date.getStart(pageWrap.getModel().getEditDate()));
            queryWrapper.lambda().le(Questionnaire::getEditDate, Utils.Date.getEnd(pageWrap.getModel().getEditDate()));
        }
        if (pageWrap.getModel().getIsdeleted() != null) {
            queryWrapper.lambda().eq(Questionnaire::getIsdeleted, pageWrap.getModel().getIsdeleted());
        }
        if (pageWrap.getModel().getTitle() != null) {
            queryWrapper.lambda().eq(Questionnaire::getTitle, pageWrap.getModel().getTitle());
        }
        if (pageWrap.getModel().getRemark() != null) {
            queryWrapper.lambda().eq(Questionnaire::getRemark, pageWrap.getModel().getRemark());
        }
        if (pageWrap.getModel().getStatus() != null) {
            queryWrapper.lambda().eq(Questionnaire::getStatus, pageWrap.getModel().getStatus());
        }
        for(PageWrap.SortData sortData: pageWrap.getSorts()) {
            if (sortData.getDirection().equalsIgnoreCase(PageWrap.DESC)) {
                queryWrapper.orderByDesc(sortData.getProperty());
            } else {
                queryWrapper.orderByAsc(sortData.getProperty());
            }
        }
        return PageData.from(questionnaireMapper.selectPage(page, queryWrapper));
    }

    @Override
    public long count(Questionnaire questionnaire) {
        QueryWrapper<Questionnaire> wrapper = new QueryWrapper<>(questionnaire);
        return questionnaireMapper.selectCount(wrapper);
    }
    //[{"score":"1","opt":"不足20%"},{"score":"2","opt":"20%-40%"},{"score":"3","opt":"40%-60%"},{"score":"4","opt":"60%-80%"},{"score":"5","opt":"80%以上"}]
    @Transactional(rollbackFor = {Exception.class,BusinessException.class})
    @Override
    public Integer importBatch(MultipartFile file){
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
            String title = file.getOriginalFilename();
            String title1 = file.getOriginalFilename();
        ExcelImporter ie = null;
        List<QuestionnaireProblem> dataList =null;
        try {
            ie = new ExcelImporter(file,0,0);
            dataList = ie.getDataList(QuestionnaireProblem.class,null);
        }  catch (Exception e) {
            e.printStackTrace();
        }
        if(dataList == null || dataList.size() ==0){
            throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，录入数据为空！");
        }
        System.out.println(dataList);
        //批量导入数据
        Questionnaire model = new Questionnaire();
        model.setCreateDate(DateUtil.getCurrentDate());
        model.setIsdeleted(Constants.ZERO);
        model.setCreator(user.getId());
        model.setStatus(Constants.ONE);
        model.setTitle(title.substring(0,title.lastIndexOf(".")));
        questionnaireMapper.insert(model);
        int num =1;
        for(QuestionnaireProblem m: dataList){
            if(StringUtils.isBlank(m.getTitle())
                ||StringUtils.isBlank(m.getOpt())
                ||StringUtils.isBlank(m.getRemark())){
                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，第【"+(num)+"】行数据无效，请检查输入！");
            }
            String[] opts = m.getOpt().split("\n");
            String[] ss = m.getRemark().split("\n");
            if(opts.length > ss.length){
                throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起，第【"+(num)+"】行数据无效，请检查输入！");
            }
            QuestionnaireProblem t = new QuestionnaireProblem();
            t.setIsdeleted(Constants.ZERO);
            t.setQuestionnaireId(model.getId());
            t.setCreator(model.getCreator());
            t.setCreateDate(model.getCreateDate());
            t.setTitle(m.getTitle());
            t.setSortnum(num);
            t.setType(Constants.ZERO+"");
            if(StringUtils.equals(m.getType(),"多选")){
                t.setType(Constants.ONE+"");
            }

            JSONArray array = new JSONArray();
            for(int i=0;i<opts.length;i++){
                JSONObject j = new JSONObject();
                j.put("opt",opts[i]);
                j.put("score",ss[i]);
                array.add(j);
            }
            t.setOpt(array.toJSONString());
            questionnaireProblemMapper.insert(t);
            num++;
        }
        return  model.getId();

    }
}