/*
 *    Copyright (c) 2018-2025, ztt All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 * Redistributions of source code must retain the above copyright notice,
 * this list of conditions and the following disclaimer.
 * Redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution.
 * Neither the name of the pig4cloud.com developer nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 * Author: ztt
 */

package com.chinaztt.mes.basic.controller;

import com.alibaba.excel.EasyExcel;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.chinaztt.mes.basic.dto.ParamDTO;
import com.chinaztt.mes.basic.dto.ParamPlusDto;
import com.chinaztt.mes.basic.entity.Param;
import com.chinaztt.mes.basic.excel.ParamData;
import com.chinaztt.mes.basic.excel.ParamUploadListener;
import com.chinaztt.mes.basic.service.ParamService;
import com.chinaztt.mes.basic.util.DictUtils;
import com.chinaztt.mes.common.util.JsonUtil;
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
import com.chinaztt.ztt.admin.api.entity.SysDictItem;
import com.chinaztt.ztt.common.core.util.R;
import com.chinaztt.ztt.common.log.annotation.SysLog;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import lombok.AllArgsConstructor;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.util.*;
import java.util.stream.Collectors;


/**
 * 工序参数表
 *
 * @author sunxl
 * @date 2021-01-14 15:24:37
 */
@RestController
@AllArgsConstructor
@RequestMapping("/param")
@Api(value = "param", tags = "工序参数表管理")
public class ParamController {

	private final ParamService paramService;

	private DictUtils dictUtils;
	/**
	 * 分页查询
	 *
	 * @param page  分页对象
	 * @param param 工序参数表
	 * @return
	 */
	@ApiOperation(value = "分页查询", notes = "分页查询")
	@GetMapping("/page")
	@PreAuthorize("@pms.hasPermission('technology_param_view','product_workbench')")
	public R getParamPage(Page page, ParamDTO param) {
		List<SysDictItem> dict = dictUtils.getDict("technology_param");
		System.out.println(dict);
		Page page1 = paramService.page(page, QueryWrapperUtil.gen(param));
		List records = page1.getRecords();
		List<Map>all = new ArrayList<Map>();
		records.forEach(l->{
			Map pojo = JsonUtil.jsonToPojo(JsonUtil.jsonToString(l), Map.class);
			Optional.ofNullable(pojo.get("type")).ifPresent(o->{
				pojo.put("type",dict.stream().filter(d-> Objects.equals(String.valueOf(d.getValue()),pojo.get("type"))).collect(Collectors.toList()).get(0).getLabel());
			});
			QueryWrapper<Param>queryWrapper=new QueryWrapper<>();
			queryWrapper.lambda().eq(Param::getParentId,pojo.get("id"));
			List<Param> secondList = paramService.list(queryWrapper);
			List<Map>secondMapList = new ArrayList<>();
			secondList.forEach(s->{
				Map second = JsonUtil.jsonToPojo(JsonUtil.jsonToString(s), Map.class);
				Optional.ofNullable(second.get("type")).ifPresent(o->{
					second.put("type",dict.stream().filter(d-> Objects.equals(String.valueOf(d.getValue()),second.get("type"))).collect(Collectors.toList()).get(0).getLabel());
				});
				System.out.println(second);
				QueryWrapper<Param>queryWrapperSecond=new QueryWrapper<>();
				queryWrapperSecond.lambda().eq(Param::getParentId,second.get("id"));
				List<Param> third = paramService.getBaseMapper().selectList(queryWrapperSecond);
				third.forEach(t->{
					Optional.ofNullable(t.getType()).ifPresent(o->{
						t.setType(dict.stream().filter(d-> Objects.equals(String.valueOf(d.getValue()),t.getType())).collect(Collectors.toList()).get(0).getLabel());
					});
				});
				System.out.println(third);
				second.put("children",third);
				secondMapList.add(second);
			});
			pojo.put("children",secondMapList);
			all.add(pojo);
		});
		page1.setRecords(all);
		return R.ok(page1);
	}

	/**
	 * 工序里的工序参数分页查询
	 *
	 * @param page  分页对象
	 * @param param 工序参数表
	 * @param id    工序的id
	 * @return
	 */
	@ApiOperation(value = "工序里的工序参数分页查询", notes = "工序里的工序参数分页查询")
	@GetMapping("/param/page/{id}")
	@PreAuthorize("@pms.hasPermission('technology_param_view','product_workbench')")
	public R getTechnologyParamPage(Page page, Param param, @PathVariable Long id) {
		param.setId(null);
		return R.ok(paramService.getTechnologyParamPage(page, id, QueryWrapperUtil.gen(param)));
	}

	/**
	 * 通过id查询工序参数表
	 *
	 * @param id id
	 * @return R
	 */
	@ApiOperation(value = "通过id查询", notes = "通过id查询")
	@GetMapping("/{id}")
	@PreAuthorize("@pms.hasPermission('technology_param_view','product_workbench')")
	public R getById(@PathVariable("id") Long id) {
		return R.ok(paramService.getById(id));
	}

	/**
	 * 新增工序参数表
	 *
	 * @param param 工序参数表
	 * @return R
	 */
	@ApiOperation(value = "新增工序参数表", notes = "新增工序参数表")
	@SysLog("新增工序参数表")
	@PostMapping
	@PreAuthorize("@pms.hasPermission('technology_param_add')")
	public R save(@RequestBody ParamPlusDto param) {
		return R.ok(paramService.fullSave(param));
	}

	/**
	 * 修改工序参数表
	 *
	 * @param param 工序参数表
	 * @return R
	 */
	@ApiOperation(value = "修改工序参数表", notes = "修改工序参数表")
	@SysLog("修改工序参数表")
	@PutMapping
	@PreAuthorize("@pms.hasPermission('technology_param_edit')")
	public R updateById(@RequestBody Param param) {
		return R.ok(paramService.fullUpdate(param));
	}

	/**
	 * 通过id删除工序参数表
	 *
	 * @param id id
	 * @return R
	 */
	@ApiOperation(value = "通过id删除工序参数表", notes = "通过id删除工序参数表")
	@SysLog("通过id删除工序参数表")
	@DeleteMapping("/{id}")
	@PreAuthorize("@pms.hasPermission('technology_param_del')")
	public R removeById(@PathVariable Long id) {
		return R.ok(paramService.delete(id));
	}

	/**
	 * excel上传
	 *
	 * @return
	 */
	@PostMapping("/upload")
	public R upload(@RequestParam("file") MultipartFile file) {
		try {
			EasyExcel.read(file.getInputStream(), ParamData.class, new ParamUploadListener(paramService)).sheet().doRead();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return R.ok();
	}

	/**
	 * 参数项判断是否已存在
	 *
	 * @param param 工序参数表
	 * @return R
	 */
	@ApiOperation(value = "参数项判断是否已存在", notes = "参数项判断是否已存在")
	@SysLog("参数项判断是否已存在")
	@PostMapping("/checkItemRepeat")
	public R checkItemRepeat(@RequestBody Param param) {
		return R.ok(paramService.checkItemRepeat(param));
	}
}