package com.qianwen.smartman.modules.cps.controller;

import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
import org.apache.commons.codec.Charsets;
import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.disk.DiskFileItem;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import java.util.Arrays;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.multipart.commons.CommonsMultipartFile;

import com.alibaba.excel.EasyExcel;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.github.xiaoymin.knife4j.annotations.ApiOperationSupport;
import com.qianwen.core.boot.ctrl.BladeController;
import com.qianwen.core.excel.util.ExcelUtil;
import com.qianwen.core.mp.support.Condition;
import com.qianwen.core.mp.support.Query;
import com.qianwen.core.oss.model.BladeFile;
import com.qianwen.core.scanner.modular.annotation.GetResource;
import com.qianwen.core.scanner.modular.annotation.PostResource;
import com.qianwen.core.scanner.modular.annotation.PutResource;
import com.qianwen.core.scanner.modular.stereotype.ApiResource;
import com.qianwen.core.secure.annotation.PreAuth;
import com.qianwen.core.tenant.annotation.NonDS;
import com.qianwen.core.tool.api.R;
import com.qianwen.core.tool.utils.DateUtil;
import com.qianwen.smartman.modules.cps.dto.EmployeeDTO;
import com.qianwen.smartman.modules.cps.entity.WorkstationDatapoints;
import com.qianwen.smartman.modules.cps.service.WorkstationDatapointsService;
import com.qianwen.smartman.modules.cps.vo.EmployeeSelectVO;
import com.qianwen.smartman.modules.cps.vo.WorkstationDatapointsSubmitVO;
import com.qianwen.smartman.modules.cps.vo.WorkstationDatapointsVO;
import com.qianwen.smartman.modules.cps.wrapper.EmployeeExcelWrapper;
import com.qianwen.smartman.modules.resource.builder.oss.OssBuilder;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;

@Api(value = "工位管理", tags = {"工位管理"})
@RestController
@ApiResource({"smis/workstation"})
@NonDS
@Validated
public class WorkstationDpController extends BladeController {
    @Autowired
    private WorkstationDatapointsService workstationDatapointsService;
    @Autowired
    private OssBuilder ossBuilder;
   
    //@PreAuth
    @GetResource({"/listDatapointsByWorkstationId"})
    @ApiOperation("根据工位id获取数据点")
    public R<WorkstationDatapointsVO> listDatapointsByWorkstationId(@RequestParam Long workstationId) {
        return R.data(workstationDatapointsService.getDatapoints(workstationId));
    }
    
    @PutResource({"/saveDatapoints"})
    @ApiOperation("保存工位数据点配置")
    public R<Boolean> saveDatapoints(@RequestBody WorkstationDatapointsSubmitVO datapoints) {
    	//return R.data(true);
    	
    	try {
    		workstationDatapointsService.submit(datapoints);
    	}catch(Exception e) {
    		//R.fail(null).
    		return R.fail(e.getMessage());
    	}
        return R.data(true);
    }
    
    @ApiOperationSupport(order = 7)
    @GetResource({"/export-dp"})
    @ApiOperation("导出工位数据点")
    @PreAuth
    public R<BladeFile> exportDatapoints(long workstationId) {
        //List<EmployeeExcel> list = EmployeeExcelWrapper.build().pageVO(pages).getRecords();
        String fileName = String.format("%s-%s.xlsx", "点位配置", DateUtil.time());
        
        WorkstationDatapointsVO dpVO = workstationDatapointsService.getDatapoints(workstationId);
        
        
        List<List<String>> head = new ArrayList<>();
        head.add(Arrays.asList("名称","别名"));
        
        List<List<Object>> data = new ArrayList<>();
        
        data.add(Arrays.asList("DeviceStatus","alias"));
        
        MultipartFile multipartFile = exportExcel("sheet1",fileName,head(dpVO.getDpHead()),data(dpVO.getDpHead(),dpVO.getDpConfig()));
        BladeFile bladeFile = this.ossBuilder.tempTemplate().putFile(multipartFile.getOriginalFilename(), multipartFile);
        return R.data(bladeFile);
        //return null;
    }
    
    List<List<Object>> data(String dpHead,String dpConfig){
    	List<List<Object>> result = new ArrayList<>();
    	JSONArray headsJson = JSONArray.parseArray(dpHead);
    	
    	JSONArray dataJson = JSONArray.parseArray(dpConfig);
    	JSONObject rowJson;
    	JSONObject headOne ;
    	String prop;
    	List<Object> rowData;
    	for(int i=0;i<dataJson.size();i++) {
    		rowJson = dataJson.getJSONObject(i);
    		rowData = new ArrayList<>();
    		
    		for(int j=0;j<headsJson.size();j++) {
    			headOne = headsJson.getJSONObject(j);
    			prop = headOne.getString("prop");
    			if(rowJson.containsKey(prop)) {
    				rowData.add(rowJson.getString(prop));
    			}else {
    				rowData.add("");
    			}
    		}
    		
    		result.add(rowData);
    	}
    	
    	return result;
    }
    
    List<List<String>> head(String dpHeadFullJsonStr){
    	List<List<String>> head = new ArrayList<>();
    	JSONArray headsJson = JSONArray.parseArray(dpHeadFullJsonStr);
    	
    	JSONObject headOne ;
    	boolean isRequired;
    	String title;
    	for(int i=0;i<headsJson.size();i++) {
    		List<String> head0 = new ArrayList<>();
    		
    		headOne = headsJson.getJSONObject(i);
    		if(headOne.containsKey("isRequired")) {
    			isRequired = headOne.getBooleanValue("isRequired");
    		}else {
    			isRequired = false;
    		}
    		title = (isRequired?"*":"") + headOne.getString("label");
	       	head0.add(title);
	       	 
	       	head.add(head0);
    	}
    	return head;
    }
    
    /**
     * 动态表头导出
     *
     * @param response HttpServletResponse
     * @param fileName 文件名
     * @param Myhead   Excel表头信息
     * @param data     Excel数据
     */
    public MultipartFile exportExcel(String sheetName,String filename, List<List<String>> myhead, List<List<Object>> data) {
        try {
        	
          DiskFileItemFactory factory = new DiskFileItemFactory();
  	      factory.setRepository(new File(System.getProperty("java.io.tmpdir")));
  	    //factory.setRepository(new File("d:/"));
  	      FileItem fileItem = factory.createItem("excel", "application/vnd.ms-excel", true, filename);
  	      
            EasyExcel.write(fileItem.getOutputStream())
                    .head(myhead)
                    .sheet(sheetName)
                    .doWrite(data);
            
            return new CommonsMultipartFile(fileItem);
        } catch (IOException e) {
            e.printStackTrace();
        }
        return null;
    }
}