package com.zy.asrs.wms.controller; 
 | 
  
 | 
import com.alibaba.excel.EasyExcel; 
 | 
import com.alibaba.excel.write.style.column.LongestMatchColumnWidthStyleStrategy; 
 | 
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; 
 | 
import com.baomidou.mybatisplus.extension.plugins.pagination.Page; 
 | 
import com.zy.asrs.common.domain.CodeRes; 
 | 
import com.zy.asrs.common.domain.entity.MatExcel; 
 | 
import com.zy.asrs.common.domain.entity.MatPrint; 
 | 
import com.zy.asrs.common.utils.*; 
 | 
import com.zy.asrs.common.wms.entity.Mat; 
 | 
import com.zy.asrs.common.wms.entity.Tag; 
 | 
import com.zy.asrs.common.wms.service.MatService; 
 | 
import com.zy.asrs.common.wms.service.TagService; 
 | 
import com.zy.asrs.framework.annotations.ManagerAuth; 
 | 
import com.zy.asrs.framework.common.BaseRes; 
 | 
import com.zy.asrs.framework.common.Cools; 
 | 
import com.zy.asrs.framework.common.R; 
 | 
import com.zy.asrs.framework.domain.KeyValueVo; 
 | 
import com.zy.asrs.framework.common.DateUtils; 
 | 
import com.zy.asrs.common.web.BaseController; 
 | 
import com.zy.asrs.wms.config.AdminInterceptor; 
 | 
import org.springframework.beans.factory.annotation.Autowired; 
 | 
import org.springframework.transaction.annotation.Transactional; 
 | 
import org.springframework.web.bind.annotation.*; 
 | 
import org.springframework.web.multipart.MultipartFile; 
 | 
  
 | 
import javax.imageio.ImageIO; 
 | 
import javax.servlet.http.HttpServletResponse; 
 | 
import java.awt.image.BufferedImage; 
 | 
import java.io.IOException; 
 | 
import java.net.URLEncoder; 
 | 
import java.util.*; 
 | 
  
 | 
@RestController 
 | 
public class TagController extends BaseController { 
 | 
  
 | 
    @Autowired 
 | 
    private TagService tagService; 
 | 
    @Autowired 
 | 
    private MatService matService; 
 | 
    @Autowired 
 | 
    private TreeUtils treeUtils; 
 | 
  
 | 
  
 | 
    @RequestMapping(value = "/tag/list/pda/auth") 
 | 
    @ManagerAuth 
 | 
    public R pdaList(@RequestParam(defaultValue = "1")Integer curr, 
 | 
                     @RequestParam(defaultValue = "10")Integer limit, 
 | 
                     @RequestParam(required = false)Long parentId){ 
 | 
        LambdaQueryWrapper<Tag> tagLambdaQueryWrapper = new LambdaQueryWrapper<>(); 
 | 
        if(!Cools.isEmpty(parentId)) { 
 | 
            tagLambdaQueryWrapper.eq(Tag::getParentId, parentId == null ? getOriginTag().getId() : parentId); 
 | 
        } 
 | 
        List<Tag> tags = tagService.list(tagLambdaQueryWrapper); 
 | 
        return R.ok().add(tags); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/tag/{id}/auth") 
 | 
    @ManagerAuth 
 | 
    public R get(@PathVariable("id") String id) { 
 | 
        return R.ok(tagService.getById(String.valueOf(id))); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/tag/page/auth") 
 | 
    @ManagerAuth 
 | 
    public R page(@RequestParam(defaultValue = "1")Integer curr, 
 | 
                  @RequestParam(defaultValue = "10")Integer limit, 
 | 
                  @RequestParam(required = false)String orderByField, 
 | 
                  @RequestParam(required = false)String orderByType, 
 | 
                  @RequestParam Map<String, Object> param) { 
 | 
        Long hostId = getHostId(); 
 | 
        LambdaQueryWrapper<Tag> wrapper = new LambdaQueryWrapper<>(); 
 | 
        wrapper.eq(Tag::getHostId, hostId); 
 | 
        excludeTrash(param); 
 | 
        return R.parse("0-操作成功").add(tagService.list(wrapper)); 
 | 
    } 
 | 
  
 | 
  
 | 
    @RequestMapping(value = "/tag/add/auth") 
 | 
    @ManagerAuth 
 | 
    public R add(Tag tag) { 
 | 
        tag.setHostId(getHostId()); 
 | 
        tag.setStatus(1); 
 | 
        tagService.save(tag); 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/tag/update/auth") 
 | 
    @ManagerAuth 
 | 
    public R update(Tag tag){ 
 | 
        if (Cools.isEmpty(tag) || null== tag.getId()){ 
 | 
            return R.error(); 
 | 
        } 
 | 
        tagService.updateById(tag); 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/tag/delete/auth") 
 | 
    @ManagerAuth 
 | 
    public R delete(@RequestParam(value="ids[]") Long[] ids){ 
 | 
         for (Long id : ids){ 
 | 
            tagService.removeById(id); 
 | 
        } 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    @RequestMapping(value = "/tagQuery/auth") 
 | 
    @ManagerAuth 
 | 
    public R query(String condition) { 
 | 
        LambdaQueryWrapper<Tag> wrapper = new LambdaQueryWrapper<>(); 
 | 
        wrapper.like(Tag::getId, condition); 
 | 
        Page<Tag> page = tagService.page(new Page<>(0, 10), wrapper); 
 | 
        List<Map<String, Object>> result = new ArrayList<>(); 
 | 
        for (Tag tag : page.getRecords()){ 
 | 
            Map<String, Object> map = new HashMap<>(); 
 | 
            map.put("id", tag.getId()); 
 | 
            map.put("value", tag.getId()); 
 | 
            result.add(map); 
 | 
        } 
 | 
        return R.ok(result); 
 | 
    } 
 | 
  
 | 
    @RequestMapping("/tag/all/get/kv") 
 | 
    @ManagerAuth 
 | 
    public R getDataKV(@RequestParam(required = false) String condition) { 
 | 
        List<KeyValueVo> vos = new ArrayList<>(); 
 | 
        LambdaQueryWrapper<Tag> wrapper = new LambdaQueryWrapper<>(); 
 | 
        if (!Cools.isEmpty(condition)) { 
 | 
            wrapper.like(Tag::getId, condition); 
 | 
        } 
 | 
        tagService.page(new Page<>(1, 30), wrapper).getRecords().forEach(item -> vos.add(new KeyValueVo(String.valueOf(item.getId()), item.getId()))); 
 | 
        return R.ok().add(vos); 
 | 
    } 
 | 
  
 | 
    @PostMapping(value = "/tag/tree/auth") 
 | 
    @ManagerAuth 
 | 
    public R tree(@RequestParam(required = false, defaultValue = "") String condition) throws IOException, ClassNotFoundException { 
 | 
        ArrayList<Map> tree = treeUtils.getTree(getOriginTag().getId(), getHostId()); 
 | 
        // 深拷贝 
 | 
        List<Map> result = ListUtils.deepCopy(tree); 
 | 
        if (!Cools.isEmpty(condition)) { 
 | 
            treeUtils.remove(condition, result); 
 | 
            treeUtils.remove(condition, result); 
 | 
        } 
 | 
        return R.ok(result); 
 | 
    } 
 | 
  
 | 
    /*************************************** 打印相关 ***********************************************/ 
 | 
  
 | 
    @RequestMapping(value = "/mac/code/auth") 
 | 
//    @ManagerAuth(memo = "物料编码条形码获取(type:1(条形码);2(二维码)") 
 | 
    public R matCodeBarcode(@RequestParam(defaultValue = "1") Integer type 
 | 
            , @RequestParam String param 
 | 
            , HttpServletResponse response) throws Exception { 
 | 
        AdminInterceptor.cors(response); 
 | 
        if (Cools.isEmpty(param)){ 
 | 
            return R.parse(BaseRes.EMPTY); 
 | 
        } 
 | 
        BufferedImage img; 
 | 
        if (type == 1) { 
 | 
            img = BarcodeUtils.encode(param); 
 | 
        } else { 
 | 
            img = QrCode.createImg(param); 
 | 
        } 
 | 
        if (!ImageIO.write(img, "jpg", response.getOutputStream())) { 
 | 
            throw new IOException("Could not write an image of format jpg"); 
 | 
        } 
 | 
        response.getOutputStream().flush(); 
 | 
        response.getOutputStream().close(); 
 | 
        return R.ok(); 
 | 
    } 
 | 
  
 | 
    // 打印 
 | 
    @RequestMapping(value = "/mat/print/auth") 
 | 
    @ManagerAuth(memo = "商品编码打印") 
 | 
    public R matCodePrint(@RequestParam(value = "param[]") String[] param) { 
 | 
        if(Cools.isEmpty(param)) { 
 | 
            return R.parse(CodeRes.EMPTY); 
 | 
        } 
 | 
        List<MatPrint> res = new ArrayList<>(); 
 | 
        for (String matnr : param){ 
 | 
            Mat mat = matService.getOne(new LambdaQueryWrapper<Mat>().eq(Mat::getMatnr, matnr)); 
 | 
            // 打印数据注入 
 | 
            MatPrint print = new MatPrint(); 
 | 
            print.setMatnr(mat.getMatnr()); 
 | 
            print.setBarcode(mat.getBarcode()); 
 | 
            print.setMaktx(mat.getMaktx()); 
 | 
            print.setSpecs(mat.getSpecs()); 
 | 
            print.setUnit(mat.getUnit()); 
 | 
            print.setMemo(mat.getMemo()); 
 | 
  
 | 
            print.setMaterial(mat.getMaterial()); 
 | 
            print.setDecrees(mat.getDecrees()); 
 | 
            print.setImgNum(mat.getImgNum()); 
 | 
            print.setWeight(String.valueOf(mat.getWeight())); 
 | 
            print.setModel(mat.getModel()); 
 | 
            print.setLength(String.valueOf(mat.getLength())); 
 | 
            print.setWidth(mat.getWidth()); 
 | 
            print.setHeight(mat.getHeight()); 
 | 
            res.add(print); 
 | 
        } 
 | 
        return R.ok().add(res); 
 | 
    } 
 | 
  
 | 
  
 | 
    /*************************************** 数据相关 ***********************************************/ 
 | 
  
 | 
    /** 
 | 
     * excel导入模板下载 
 | 
     */ 
 | 
    @RequestMapping(value = "/mat/excel/import/mould") 
 | 
    public void matExcelImportMould(HttpServletResponse response) throws IOException { 
 | 
        List<MatExcel> excels = new ArrayList<>(); 
 | 
        response.setContentType("application/vnd.ms-excel"); 
 | 
        response.setCharacterEncoding("utf-8"); 
 | 
        String fileName = URLEncoder.encode("商品档案Excel导入模板", "UTF-8"); 
 | 
        response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx"); 
 | 
        EasyExcel.write(response.getOutputStream(), MatExcel.class) 
 | 
                .registerWriteHandler(new LongestMatchColumnWidthStyleStrategy()) 
 | 
                .sheet("sheet1") 
 | 
                .doWrite(excels); 
 | 
    } 
 | 
  
 | 
    // excel导入 
 | 
    @PostMapping(value = "/mat/excel/import/auth") 
 | 
    @ManagerAuth(memo = "商品档案数据导入") 
 | 
    @Transactional 
 | 
    public R matExcelImport(MultipartFile file) throws IOException { 
 | 
        MatExcelListener listener = new MatExcelListener(getUserId(), getHostId()); 
 | 
        EasyExcel.read(file.getInputStream(), MatExcel.class, listener).sheet().doRead(); 
 | 
        return R.ok("成功同步"+listener.getTotal()+"条商品数据"); 
 | 
    } 
 | 
  
 | 
} 
 |