wms / hhwms · Files

package com.huaheng.pc.shipment.kuaidiHeader.controller;

import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.huaheng.api.WAYBILLNO.service.KuaidiService;
import com.huaheng.common.utils.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.huaheng.common.utils.poi.ExcelUtil;
import com.huaheng.common.utils.security.ShiroUtils;
import com.huaheng.framework.web.page.PageDomain;
import com.huaheng.framework.web.page.TableDataInfo;
import com.huaheng.framework.web.page.TableSupport;
import com.huaheng.common.utils.StringUtils;
import com.huaheng.pc.config.KDCertification.domain.KDCertification;
import com.huaheng.pc.config.KDCertification.service.KDCertificationService;
import com.huaheng.pc.config.container.domain.Container;
import com.huaheng.pc.receipt.receiptDetail.domain.ReceiptDetail;
import com.huaheng.pc.shipment.kuaidiHeader.domain.KuaidiHeader;
import com.huaheng.pc.shipment.kuaidiHeader.service.KuaidiHeaderService;
import com.huaheng.pc.shipment.shipmentHeader.domain.ShipmentHeader;
import com.huaheng.pc.system.dict.domain.DictData;
import com.huaheng.pc.system.dict.service.IDictDataService;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import com.huaheng.framework.aspectj.lang.annotation.Log;
import com.huaheng.framework.aspectj.lang.constant.BusinessType;
import com.huaheng.framework.web.controller.BaseController;
import com.huaheng.framework.web.domain.AjaxResult;
import com.huaheng.common.support.Convert;
import org.springframework.web.multipart.MultipartFile;

import javax.annotation.Resource;
import java.util.*;

/**
 * 【请填写功能名称】 信息操作处理
 *
 * @author huaheng
 * @date 2022-09-07
 */
@Controller
@RequestMapping("/shipment/kuaidiHeader")
public class KuaidiHeaderController extends BaseController {
    private String prefix = "shipment/kuaidiHeader";

	@Resource
	private KuaidiHeaderService kuaidiHeaderService;
	@Resource
	private IDictDataService dictDataService;
	@Resource
	private KDCertificationService kdCertificationService;
	@Resource
	private KuaidiService kuaidiService;

	@RequiresPermissions("shipment:kuaidiHeader:view")
	@GetMapping()
	public String kuaidiHeader() {
	    return prefix + "/list";
	}

	/**
	 * 查询【请填写功能名称】列表
	 */
	@RequiresPermissions("shipment:kuaidiHeader:list")
	@PostMapping("/list")
	@ResponseBody
	public TableDataInfo list(KuaidiHeader kuaidiHeader,String startTime, String endTime) {
		LambdaQueryWrapper<KuaidiHeader> lambdaQueryWrapper = Wrappers.lambdaQuery();
		lambdaQueryWrapper.eq(KuaidiHeader::getDeleted,false);
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getSourceCode()),KuaidiHeader::getSourceCode,kuaidiHeader.getSourceCode());
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getTrackCode()),KuaidiHeader::getTrackCode,kuaidiHeader.getTrackCode());
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getOrderBill()),KuaidiHeader::getOrderBill,kuaidiHeader.getOrderBill());
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getCode()),KuaidiHeader::getCode,kuaidiHeader.getCode());
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getCompanyCode()),KuaidiHeader::getCompanyCode,kuaidiHeader.getCompanyCode());
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getUWarehouseCode()),KuaidiHeader::getUWarehouseCode,kuaidiHeader.getUWarehouseCode());
		lambdaQueryWrapper.eq(StringUtils.isNotEmpty(kuaidiHeader.getCreatedBy()),KuaidiHeader::getCreatedBy,kuaidiHeader.getCreatedBy());
		lambdaQueryWrapper.ge(StringUtils.isNotEmpty(startTime), KuaidiHeader::getCreated, startTime)
				.le(StringUtils.isNotEmpty(endTime), KuaidiHeader::getCreated, endTime);
		lambdaQueryWrapper.orderByDesc(KuaidiHeader::getKuaidiCom,KuaidiHeader::getCreated);
		PageDomain pageDomain = TableSupport.buildPageRequest();
		Integer pageNum = pageDomain.getPageNum();
		Integer pageSize = pageDomain.getPageSize();
		if (StringUtils.isNotNull(pageNum) && StringUtils.isNotNull(pageSize)){
			/*使用分页查询*/
			Page<KuaidiHeader> page = new Page<>(pageNum, pageSize);
			IPage<KuaidiHeader> iPage = kuaidiHeaderService.page(page, lambdaQueryWrapper);
			return getMpDataTable(iPage.getRecords(), iPage.getTotal());
		} else {
			List<KuaidiHeader> list = kuaidiHeaderService.list(lambdaQueryWrapper);
			return getDataTable(list);
		}
	}

	/**
	 * 新增【请填写功能名称】
	 */
	@GetMapping("/add")
	public String add() {
	    return prefix + "/add";
	}

	/**
	 * 新增保存【请填写功能名称】
	 */
	@RequiresPermissions("shipment:kuaidiHeader:add")
	@Log(title = "【请填写功能名称】", action = BusinessType.INSERT)
	@PostMapping("/add")
	@ResponseBody
	public AjaxResult addSave(KuaidiHeader kuaidiHeader) {
		return kuaidiHeaderService.saveHeader(kuaidiHeader);
	}

	/**
	 * 修改【请填写功能名称】
	 */
	@GetMapping("/edit/{id}")
	public String edit(@PathVariable("id") Long id, ModelMap mmap) {
		KuaidiHeader kuaidiHeader = kuaidiHeaderService.getById(id);
		mmap.put("kuaidiHeader", kuaidiHeader);
	    return prefix + "/edit";
	}

	/**
	 * 修改保存【请填写功能名称】
	 */
	@RequiresPermissions("shipment:kuaidiHeader:edit")
	@Log(title = "【请填写功能名称】", action = BusinessType.UPDATE)
	@PostMapping("/edit")
	@ResponseBody
	public AjaxResult editSave(KuaidiHeader kuaidiHeader) {
		return toAjax(kuaidiHeaderService.updateById(kuaidiHeader));
	}

	/**
	 * 删除【请填写功能名称】
	 */
	@RequiresPermissions("shipment:kuaidiHeader:remove")
	@Log(title = "【请填写功能名称】", action = BusinessType.DELETE)
	@PostMapping( "/remove")
	@ResponseBody
	public AjaxResult remove(String ids) {
		if (StringUtils.isEmpty(ids)){
			return AjaxResult.error("id不能为空");
		}
		for (Integer id : Convert.toIntArray(ids)) {
			KuaidiHeader kuaidiHeader = kuaidiHeaderService.getById(id);
			if(kuaidiHeader == null){
				return AjaxResult.error("没有此快递单");
			}
			kuaidiHeader.setDeleted(true);
			boolean tag= kuaidiHeaderService.updateById(kuaidiHeader);
			if (!tag) {
				return AjaxResult.error("删除失败");
			}
		}
		return AjaxResult.success("删除成功");
	}

	/**
	 * U8导入快递信息
	 */
	@GetMapping("/U8Save")
	public String U8Save()
	{
		return prefix + "/save";
	}

	@RequiresPermissions("shipment:kuaidiHeader:add")
	@Log(title = "出库-快递单", operating="新增快递单主单", action = BusinessType.INSERT)
	@PostMapping("/save")
	@ResponseBody
	public AjaxResult save(KuaidiHeader kuaidiHeader)
	{
		AjaxResult ajaxResult =kuaidiHeaderService.saveU8Kuaidi(kuaidiHeader);
		return ajaxResult;
	}

	@PostMapping("/getKuaidiHeader")
	@ResponseBody
	public AjaxResult<KuaidiHeader> getKuaidiHeader(int id){
		return AjaxResult.success(kuaidiHeaderService.getById(id));
	}

	@PostMapping("/importData")
	@ResponseBody
	public AjaxResult importData(MultipartFile file) throws Exception {
		ExcelUtil<KuaidiHeader> util = new ExcelUtil<>(KuaidiHeader.class);
		List<KuaidiHeader> list = util.importExcel(file.getInputStream());
		String operName = ShiroUtils.getLoginName();
		Boolean flag = true;
		flag = kuaidiHeaderService.importData(list,false);
		if (flag == true) {
			return AjaxResult.success("导入成功");
		} else {
			return AjaxResult.success("导入失败,数据格式不对");
		}
	}

	/**
	 * 导出模板
	 * @return
	 * @throws Exception
	 */
	@GetMapping("/importTemplate")
	@ResponseBody
	public AjaxResult importTemplate() {
		ExcelUtil<KuaidiHeader> util = new ExcelUtil<>(KuaidiHeader.class);
		return util.importTemplateExcel("快递导入");
	}
//	@RequiresPermissions("shipment:kuaidi:edit")
	@Log(title = "出库-快递单", operating = "快递导入", action = BusinessType.OTHER)
	@PostMapping("/importKD")
	@ResponseBody
	public AjaxResult importKD(KuaidiHeader kuaidiHeader){
		AjaxResult ajaxResult =kuaidiHeaderService.sendKD(kuaidiHeader);
		return ajaxResult;
	}

	@Log(title = "出库-快递单", operating = "快递批量导入", action = BusinessType.OTHER)
	@PostMapping("/importKDS")
	@ResponseBody
	public AjaxResult importKDS(KuaidiHeader kuaidiHeader){
		AjaxResult ajaxResult =kuaidiHeaderService.sendKDS(kuaidiHeader);
		return ajaxResult;
	}

//	@RequiresPermissions("shipment:kuaidi:edit")
	@Log(title = "出库-快递单", operating = "快递打印", action = BusinessType.OTHER)
	@PostMapping("/printKD")
	@ResponseBody
	public AjaxResult printKD(Integer id){
		KuaidiHeader kuaidiHeader =new KuaidiHeader();
		kuaidiHeader =kuaidiHeaderService.getById(id);
		if(kuaidiHeader == null){
			return AjaxResult.error("要打印的单据不存在");
		}
		AjaxResult ajaxResult =kuaidiHeaderService.printKD(kuaidiHeader.getKuaidiNum());
		return ajaxResult;
	}

//	@RequiresPermissions("shipment:kuaidi:edit")
	@Log(title = "出库-快递单", operating = "快递批量打印", action = BusinessType.OTHER)
	@PostMapping("/printKDS")
	@ResponseBody
	public AjaxResult printKDS(String ids){
		AjaxResult ajaxResult =kuaidiHeaderService.printKDS(ids);
		return ajaxResult;
	}

	/**
	 * 选择快递公司
	 */
	@GetMapping("/importt/{id}")
	public String editt(@PathVariable("id") Integer id, ModelMap mmap)
	{
		DictData dictData =new DictData();
		dictData.setDictType("kuaidi");
		dictData =dictDataService.selectModel(dictData);
		KuaidiHeader kuaidiHeader =new KuaidiHeader();
		kuaidiHeader.setId(id);
		if(dictData !=null) {
			kuaidiHeader.setKuaidiCom(dictData.getDictValue());
		}
		mmap.put("kuaidiHeader", kuaidiHeader);
		return prefix + "/import";
	}

	/**
	 * 批量选择快递公司
	 */
	@GetMapping("/importKDS/{ids}")
	public String importKDS(@PathVariable("ids") String ids, ModelMap mmap)
	{
		DictData dictData =new DictData();
		dictData.setDictType("kuaidi");
		dictData =dictDataService.selectModel(dictData);
		KuaidiHeader kuaidiHeader =new KuaidiHeader();
		kuaidiHeader.setOrderBill(ids);
		if(dictData !=null) {
			kuaidiHeader.setKuaidiCom(dictData.getDictValue());
		}
		mmap.put("kuaidiHeader", kuaidiHeader);
		return prefix + "/imports";
	}
	@RequiresPermissions("shipment:kuaidiHeader:fresh")
	@Log(title = "出库-快递单", operating = "刷新token", action = BusinessType.OTHER)
	@PostMapping("/fresh")
	@ResponseBody
	public AjaxResult freshKdToken( ) throws Exception {
		KDCertification kdCertification = new KDCertification();
		kdCertification.setId("1");
		kdCertification = kdCertificationService.getOne(kdCertification);
		if(kdCertification == null){
			return AjaxResult.error("没有此token");
		}
		String json =kuaidiService.refreshToken(kdCertification.getRefreshToken());
		Map hashMap = JSON.parseObject(json, HashMap.class);
		if(!hashMap.isEmpty() && StringUtils.isNotNull(hashMap.get("access_token")) && StringUtils.isNotNull(hashMap.get("refresh_token")) ) {
			kdCertification.setAccessToken(hashMap.get("access_token").toString());
			kdCertification.setRefreshToken(hashMap.get("refresh_token").toString());
			kdCertificationService.updateById(kdCertification);
		}else {
			return AjaxResult.error(hashMap.get("message").toString());
		}
		return AjaxResult.success("成功");
	}


}