youxiw2000 / trash · Files

package com.trash.business.controller;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.util.Base64Utils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import com.trash.common.annotation.Log;
import com.trash.common.config.trashConfig;
import com.trash.common.core.controller.BaseController;
import com.trash.common.core.domain.AjaxResult;
import com.trash.common.enums.BusinessType;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.trash.business.domain.SupervisionThreestep;
import com.trash.business.domain.dayWork;
import com.trash.business.service.ISupervisionThreestepService;
import com.trash.common.utils.RemoteServerUtils;
import com.trash.common.utils.SecurityUtils;
import com.trash.common.utils.StringUtils;
import com.trash.common.utils.file.FileUtils;
import com.trash.common.utils.poi.ExcelUtil;
import com.trash.common.core.page.PageDomain;
import com.trash.common.core.page.TableDataInfo;
import com.trash.common.core.page.TableSupport;
import com.trash.common.core.redis.RedisCache;

/**
 * 三查机制Controller
 * 
 * @author trash
 * @date 2023-04-24
 */
@RestController
@RequestMapping("/business/threestep")
public class SupervisionThreestepController extends BaseController
{
    @Autowired
    private ISupervisionThreestepService supervisionThreestepService;
    
    @Autowired
    RedisCache redisCache;

    /**
     * 查询三查机制列表
     */
    @GetMapping("/list")
    public TableDataInfo list(SupervisionThreestep supervisionThreestep)
    {
        startPage();

	        
        List<SupervisionThreestep> list = supervisionThreestepService.selectSupervisionThreestepList(supervisionThreestep);
        return getDataTable(list);
    }
    
    @PostMapping("/list")
    public AjaxResult postList(@RequestBody SupervisionThreestep supervisionThreestep)
    {

        SupervisionThreestep list = supervisionThreestepService.getSupervisionThreestepInfo(supervisionThreestep);
        return AjaxResult.success(list);
    }
    
    @PostMapping("/getByObjects")
    public AjaxResult getLists(@RequestBody SupervisionThreestep supervisionThreestep)
    {	
    	AjaxResult ajaxResult = AjaxResult.success();
    	try {
        	ajaxResult.put("result", supervisionThreestepService.selectSupervisionThreestepList(supervisionThreestep));
		} catch (Exception e) {
			e.printStackTrace();
		}
    	
        return ajaxResult;
    	
    }
    
    String[] outside = {"长沙县", "望城区", "浏阳市", "宁乡市", "株洲市", "湘潭市", "汩罗市", "湘阴县"};
    
    /**
     * 查询三查机制列表
     */
    @GetMapping("/dayWorkList")
    public TableDataInfo dayWorkList(SupervisionThreestep supervisionThreestep)
    {
//        startPage();
    	PageDomain pageDomain = TableSupport.buildPageRequest();
        Integer pageNum = pageDomain.getPageNum();
        Integer pageSize = pageDomain.getPageSize();
        
        if("430100".equals(supervisionThreestep.getPlace())){
        	supervisionThreestep.setPlace(null);
        }
        if("长沙市".equals(supervisionThreestep.getSubReason())){
        	supervisionThreestep.setSubReason(null);
        }
    	
    	JSONArray areas = redisCache.getCacheObject("areas");
    	
    	if(areas == null){
            areas = RemoteServerUtils.getAreas();
	        redisCache.setCacheObject("areas", areas);
    	}

        List<Map<String,String>> list = supervisionThreestepService.selectDayWorkList(supervisionThreestep);
        
        int count = 0;
        int count1 = 0;
        String place = "长沙市";
        String place1 = "长沙市";
        String string = "今日";
        
        if(supervisionThreestep.getSubReason()!=null){
        	
        	place1 = supervisionThreestep.getSubReason();
        }
        
        if(supervisionThreestep.getHis().equals("1")){
        	string = "";
        }
        
        
        List<String> names = new ArrayList<>();
        List<String> enames = new ArrayList<>();
        
        for(Map<String, String> m:list){
        	
        	if(names.indexOf(m.get("name")) == -1){

            	names.add(m.get("name"));
        	}
        	
        	if(enames.indexOf(m.get("ename")) == -1){

            	enames.add(m.get("ename"));
        	}
        	
        	
        	for(Object object:areas){
        		JSONObject area = (JSONObject)object;
        		
        		if(supervisionThreestep.getPlace() != null){
        			if(supervisionThreestep.getPlace().equals(area.get("code").toString())){
        				place = area.get("name").toString();
        			}
        		}
        		
        		if( area.get("code").toString().equals( m.get("place"))){
        			m.put("place", area.get("name").toString());
        		}
        		if( area.get("code").toString().equals( m.get("eplace"))){
        			m.put("eplace", area.get("name").toString());
        		}
        	}
        	
        	for(String out:outside){
        		if(out.equals(m.get("eplace"))){
        			if(m.get("status").equals("开工")){
        				m.put("estatus", "开工");
        				m.put("eselfCheckTime", m.get("selfCheckTime"));
        			}
        			
        		}
        	}
        	
        	if(m.get("status").equals("开工")){
        		count += 1;
        	}
        	if(m.get("estatus").equals("开工")){
        		count1 += 1;
        	}
        }
        
        TableDataInfo info = getDataTable(list);
        
        if(list.size() > (pageNum)*pageSize){
        	list = list.subList((pageNum -1)*pageSize, (pageNum)*pageSize);
        }else{
        	list = list.subList((pageNum -1)*pageSize, list.size());
        }
        
        info.setRows(list);
        
        info.setNames(names);
        info.setEnames(enames);
        
        info.setMsg(string + place+"工地开工数量" + count + "  "  + place1+"处理场所开工数量" + count1);
        
        return info;
    }
    
    
    @PostMapping("/exportDayWorkHistoryList")
    public AjaxResult exportDayWorkHistoryList(@RequestBody List<dayWork> list)
    {
    	
        ExcelUtil<dayWork> util = new ExcelUtil<dayWork>(dayWork.class);
        return util.exportExcel(list, "历史开工报表");
        
    }
    
    @PostMapping("/exportDayWorkList")
    public AjaxResult exportDayWorkList(@RequestBody List<dayWork> list)
    {
        ExcelUtil<dayWork> util = new ExcelUtil<dayWork>(dayWork.class);
        return util.exportExcel(list, "日开工报表");
    }
    
    
    @PostMapping("/todayDataList")
    public List<SupervisionThreestep> todayDataList(@RequestBody SupervisionThreestep supervisionThreestep)
    {        
        return supervisionThreestepService.selectTodayDataList(supervisionThreestep);
    }
    
    

    /**
     * 查询三查机制列表
     */
    @PostMapping("/dayWorkList")
    public AjaxResult days(@RequestBody SupervisionThreestep supervisionThreestep)
    {        
    	List<Map> list= supervisionThreestepService.selectDayWorkListByType(supervisionThreestep);
    	AjaxResult result = AjaxResult.success(list);
    	result.put("result", AjaxResult.success(list).get("data"));
    	result.put("data", null);
        return result;
    }
    
    /**
     * 导出三查机制列表
     */
    @Log(title = "三查机制", businessType = BusinessType.EXPORT)
    @GetMapping("/export")
    public AjaxResult export(SupervisionThreestep supervisionThreestep)
    {
        List<SupervisionThreestep> list = supervisionThreestepService.selectSupervisionThreestepList(supervisionThreestep);
        
        

    	for(int i = 0;i<list.size();i++){
        	list.get(i).setId((long)i+1);
        }
        
        JSONArray array = RemoteServerUtils.getAreas();
        
        for(Object object :array){
        	JSONObject jsonObject = (JSONObject)object;
        	for(SupervisionThreestep sThreestep : list){
        		if(jsonObject.getString("code").equals(sThreestep.getPlace())){
        			sThreestep.setPlace(jsonObject.getString("name"));
        		}
        	}
        }
        List<String> withOut = new ArrayList<String>();
        String str ="报工抽查";
        if(supervisionThreestep.getStatus() == null){
        	withOut.add("checkEndTime");
        	str ="报工自查";
        }

        
        ExcelUtil<SupervisionThreestep> util = new ExcelUtil<SupervisionThreestep>(SupervisionThreestep.class);
        
        return util.exportExcel(list, str,withOut);
    }

    /**
     * 获取三查机制详细信息
     */
    @GetMapping(value = "/{id}")
    public AjaxResult getInfo(@PathVariable("id") Long id)
    {
        return AjaxResult.success(supervisionThreestepService.selectSupervisionThreestepById(id));
    }

    @GetMapping(value = "names")
    public List<String> names(SupervisionThreestep supervisionThreestep)
    {
        return supervisionThreestepService.getNames(supervisionThreestep);
    }
    /**
     * 新增三查机制
     */
    @Log(title = "三查机制", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody SupervisionThreestep supervisionThreestep)
    {
    	
    	int i = supervisionThreestepService.insertSupervisionThreestep(supervisionThreestep);
    	
    	if(i==-1){
    		return AjaxResult.error("无法重复报工");
    	}
    	if(i==-2){
    		return AjaxResult.error("当前时间无法报工");
    	}
        return toAjax(i);
    }

    /**
     * 修改三查机制
     */
    @Log(title = "三查机制", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody SupervisionThreestep supervisionThreestep)
    {
        return toAjax(supervisionThreestepService.updateSupervisionThreestep(supervisionThreestep));
    }
    
    @Log(title = "三查机制", businessType = BusinessType.UPDATE)
    @PostMapping(value="/active")
    public AjaxResult active(@RequestBody SupervisionThreestep supervisionThreestep)
    {
    	
    	supervisionThreestep.setCheckTime(new Date());
        return toAjax(supervisionThreestepService.activeSupervisionThreestep(supervisionThreestep));
    }

    /**
     * 删除三查机制
     */
    @Log(title = "三查机制", businessType = BusinessType.DELETE)
	@DeleteMapping("/{ids}")
    public AjaxResult remove(@PathVariable Long[] ids)
    {
        return toAjax(supervisionThreestepService.deleteSupervisionThreestepByIds(ids));
    }
    
    @RequestMapping(value = "/upload")
    public String upload(@RequestParam("file") MultipartFile file) throws IOException {
        return supervisionThreestepService.upload(file);
    }
    
    
    @RequestMapping(value = "/download")
    public void download(@RequestParam("path") String path,HttpServletResponse response, HttpServletRequest request) throws IOException {  
	    try
	    {
	        String realFileName = path.split("/")[path.split("/").length-1];
	        String filePath = trashConfig.getUploadPath() + path.replace("/profile/upload", "");
	
	        response.setCharacterEncoding("utf-8");
	        response.setContentType("multipart/form-data");
	        response.setHeader("Content-Disposition","attachment;fileName=" + FileUtils.setFileDownloadHeader(request, realFileName));
	        FileUtils.writeBytes(filePath, response.getOutputStream());

	    }
	    catch (Exception e)
	    {
	        e.printStackTrace();
	    }
    }
    
    @RequestMapping(value = "/getBase64")
    public String getBase64(@RequestParam("path") String path) throws IOException {  
	    try
	    {
	        String realFileName = path.split("/")[path.split("/").length-1];
	        String filePath = trashConfig.getUploadPath() + path.replace("/profile/upload", "");
	        
	        File file = new File(filePath);
	        
	        
	        FileInputStream fileInputStream = new FileInputStream(file);
	        
	        byte[] bytes = new byte[fileInputStream.available()];
	        
	        fileInputStream.read(bytes);
	        
	        return Base64Utils.encodeToString(bytes);

	    }
	    catch (Exception e)
	    {
	        e.printStackTrace();
	    }
	    return null;
    }
}