package com.ltkj.web.controller.system;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ltkj.common.utils.poi.ExcelUtil;
import com.ltkj.hosp.domain.TjAdvice;
import com.ltkj.hosp.domain.TjProject;
import com.ltkj.hosp.service.ITjAdviceService;
import com.ltkj.hosp.service.ITjProjectService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
import com.ltkj.common.annotation.Log;
import com.ltkj.common.core.domain.AjaxResult;
import com.ltkj.common.enums.BusinessType;

/**
 * adviceController
 *
 * @author ltkj
 * @date 2022-11-24
 */
@RestController
@RequestMapping("/advice/advice")
@Api(tags = "体检建议管理接口")
public class TjAdviceController{
    @Resource
    private ITjAdviceService tjAdviceService;
    @Resource
    private ITjProjectService tjProjectService;

    /**
     * 查询advice列表
     */
    @PreAuthorize("@ss.hasPermi('advice:advice:list')")
    @GetMapping("/list")
    @ApiOperation(value = "体检建议列表查询")
    public AjaxResult list(@RequestParam(required = false) @ApiParam(value = "项目名") String proName,
                              @ApiParam(value = "页码数(默认1)") @RequestParam(defaultValue = "1") Integer page,
                              @ApiParam(value = "显示条数(默认10)") @RequestParam(defaultValue = "10") Integer pageSize) {
        Page<TjAdvice>page1=new Page<>(page,pageSize);
        Map<String,Object> map=new HashMap<>();
        List<TjAdvice> list =null;
        long total =0;
        if(null !=proName){
            LambdaQueryWrapper<TjProject> wq=new LambdaQueryWrapper<>();
            list=new ArrayList<>();
            wq.like(TjProject::getProName,proName);
            List<TjProject> projectList = tjProjectService.list(wq);
            if(null !=projectList && projectList.size()>0){
                for (TjProject project : projectList) {
                    LambdaQueryWrapper<TjAdvice>wqq=new LambdaQueryWrapper<>();
                    wqq.eq(TjAdvice::getProId,project.getProId());
                    IPage<TjAdvice> adviceList = tjAdviceService.page(page1,wqq);
                    if(null !=adviceList.getRecords() && adviceList.getRecords().size()>0){
                        for (TjAdvice record : adviceList.getRecords()) {
                            record.setProName(project.getProName());
                        }
                        list.addAll(adviceList.getRecords());
                    }
                     total = adviceList.getTotal();
                }
            }
            map.put("list",list);
            map.put("total",total);
            return AjaxResult.success(map);
        }
        Page<TjAdvice> page2 = tjAdviceService.page(page1);
        list =page2.getRecords();
        if (null !=list && list.size() > 0) {
            for (TjAdvice advice : list) {
                TjProject tjProject = tjProjectService.getById(advice.getProId());
                if (null != tjProject) {
                    advice.setProName(tjProject.getProName());
                }
            }
        }
        map.put("list",list);
        map.put("total",page2.getTotal());
        return AjaxResult.success(map);
    }

    /**
     * 导出advice列表
     */
    @PreAuthorize("@ss.hasPermi('advice:advice:export')")
    @Log(title = "advice", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(HttpServletResponse response, TjAdvice tjAdvice) {
        List<TjAdvice> list = tjAdviceService.selectTjAdviceList(tjAdvice);
        ExcelUtil<TjAdvice> util = new ExcelUtil<TjAdvice>(TjAdvice.class);
        util.exportExcel(response, list, "advice数据");
    }

    /**
     * 获取advice详细信息
     */
    @PreAuthorize("@ss.hasPermi('advice:advice:query')")
    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id) {
        return AjaxResult.success(tjAdviceService.selectTjAdviceById(id));
    }

    /**
     * 新增advice
     */
    @PreAuthorize("@ss.hasPermi('advice:advice:add')")
    @Log(title = "advice", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody TjAdvice tjAdvice) {
        return AjaxResult.success(tjAdviceService.insertTjAdvice(tjAdvice));
    }

    /**
     * 修改advice
     */
    @PreAuthorize("@ss.hasPermi('advice:advice:edit')")
    @Log(title = "advice", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody TjAdvice tjAdvice) {
        return AjaxResult.success(tjAdviceService.updateById(tjAdvice));
    }

    /**
     * 删除advice
     */
    @PreAuthorize("@ss.hasPermi('advice:advice:remove')")
    @Log(title = "advice", businessType = BusinessType.DELETE)
    @DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids) {
        return AjaxResult.success(tjAdviceService.deleteTjAdviceByIds(ids));
    }
}