/*
 *    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.equipment.controller;

import com.alibaba.excel.EasyExcel;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
import com.chinaztt.mes.equipment.dto.EquipmentDTO;
import com.chinaztt.mes.equipment.entity.Equipment;
import com.chinaztt.mes.equipment.excel.EquipmentData;
import com.chinaztt.mes.equipment.excel.EquipmentUploadListener;
import com.chinaztt.mes.equipment.service.EquipmentService;
import com.chinaztt.mes.equipment.vo.FactoryVO;
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.List;
import java.util.Map;
import java.util.stream.Collectors;


/**
 * 工厂信息表
 *
 * @author fenglang
 * @date 2020-08-18 14:19:03
 */
@RestController
@AllArgsConstructor
@RequestMapping("/equipment")
@Api(value = "equipment", tags = "设备信息表管理")
public class EquipmentController {

	private final EquipmentService equipmentService;


	/**
	 * 分页查询
	 *
	 * @param page         分页对象
	 * @param equipmentDTO 工厂信息表
	 * @return
	 */
	@ApiOperation(value = "分页查询", notes = "分页查询")

	@GetMapping("/page")
	@PreAuthorize("@pms.hasPermission('equipment_equipment_view')")
	public R getEquipmentPage(Page page, EquipmentDTO equipmentDTO) {
		return R.ok(equipmentService.getEquipmentPage(page, QueryWrapperUtil.gen(equipmentDTO)));
	}

	/**
	 * 通过工作站查询设备
	 *
	 * @param id 分页对象
	 * @return
	 */
	@ApiOperation(value = "通过工作站查询设备", notes = "通过工作站查询设备")

	@GetMapping("/page/{id}")
	public R getEquipmentPageByWorkstation(@PathVariable("id") Long id) {
		return R.ok(equipmentService.getEquipmentPageByWorkstation(id));
	}

	/**
	 * 通过id查询工厂信息表
	 *
	 * @param id id
	 * @return R
	 */
	@ApiOperation(value = "通过id查询", notes = "通过id查询")
	@GetMapping("/{id}")
	@PreAuthorize("@pms.hasPermission('equipment_equipment_view')")
	public R getById(@PathVariable("id") Long id) {
		return R.ok(equipmentService.getById(id));
	}

	/**
	 * 新增工厂信息表
	 *
	 * @param equipment 工厂信息表
	 * @return R
	 */
	@ApiOperation(value = "新增设备信息表", notes = "新增设备信息表")
	@SysLog("新增设备信息表")
	@PostMapping
	@PreAuthorize("@pms.hasPermission('equipment_equipment_add')")
	public R save(@RequestBody Equipment equipment) {
		return R.ok(equipmentService.save(equipment));
	}

	/**
	 * excel上传
	 *
	 * @return
	 */
	@PostMapping("/upload")
	public R upload(@RequestParam("file") MultipartFile file) {
		try {
			List<FactoryVO> factoryVOS = equipmentService.getFactorys();
			Map<String, FactoryVO> map = factoryVOS.stream().collect(Collectors.toMap(FactoryVO::getFactoryName, factoryVO -> factoryVO, (k1, k2) -> k1));
			EasyExcel.read(file.getInputStream(), EquipmentData.class, new EquipmentUploadListener(equipmentService, map)).sheet().doRead();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return R.ok();
	}

	/**
	 * 修改工厂信息表
	 *
	 * @param equipment 工厂信息表
	 * @return R
	 */
	@ApiOperation(value = "修改设备信息表", notes = "修改设备信息表")
	@SysLog("修改设备信息表")
	@PutMapping
	@PreAuthorize("@pms.hasPermission('equipment_equipment_edit')")
	public R updateById(@RequestBody Equipment equipment) {
		return R.ok(equipmentService.updateById(equipment));
	}

	/**
	 * 通过id删除工厂信息表
	 *
	 * @param id id
	 * @return R
	 */
	@ApiOperation(value = "通过id删除设备信息表", notes = "通过id删除设备信息表")
	@SysLog("通过id删除工厂信息表")
	@DeleteMapping("/{id}")
	@PreAuthorize("@pms.hasPermission('equipment_equipment_del')")
	public void removeById(@PathVariable Long id) {
		equipmentService.deleteEquipment(id);
	}
}