/*
 *    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.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.chinaztt.mes.basic.entity.Division;
import com.chinaztt.mes.basic.entity.Staff;
import com.chinaztt.mes.basic.excel.DivisionData;
import com.chinaztt.mes.basic.excel.DivisionUploadListener;
import com.chinaztt.mes.basic.service.DivisionService;
import com.chinaztt.mes.basic.service.StaffService;
import com.chinaztt.mes.common.wrapper.QueryWrapperUtil;
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;


/**
 * 部门
 *
 * @author luh
 * @date 2020-08-14 15:20:19
 */
@RestController
@AllArgsConstructor
@RequestMapping("/division")
@Api(value = "division", tags = "部门管理")
public class DivisionController {

    private final DivisionService divisionService;
    private final StaffService staffService;

    /**
     * 分页查询
     *
     * @param page     分页对象
     * @param division 部门
     * @return
     */
    @ApiOperation(value = "分页查询", notes = "分页查询")
    @GetMapping("/page")
    @PreAuthorize("@pms.hasPermission('basic_division_view')")
    public R getDivisionPage(Page page, Division division) {
        return R.ok(divisionService.getDivisionPage(page, QueryWrapperUtil.gen(division)));
    }


    /**
     * 通过id查询部门
     *
     * @param id id
     * @return R
     */
    @ApiOperation(value = "通过id查询", notes = "通过id查询")
    @GetMapping("/{id}")
    @PreAuthorize("@pms.hasPermission('basic_division_view')")
    public R getById(@PathVariable("id") Long id) {
        return R.ok(divisionService.getById(id));
    }

    /**
     * 新增部门
     *
     * @param division 部门
     * @return R
     */
    @ApiOperation(value = "新增部门", notes = "新增部门")
    @SysLog("新增部门")
    @PostMapping
    @PreAuthorize("@pms.hasPermission('basic_division_add')")
    public R save(@RequestBody Division division) {
        return R.ok(divisionService.save(division));
    }
	/**
	 * excel上传
	 *
	 * @return
	 */
	@PostMapping("/upload")
	public R upload(@RequestParam("file") MultipartFile file) {
		try {
			EasyExcel.read(file.getInputStream(), DivisionData.class, new DivisionUploadListener(divisionService)).sheet().doRead();
		} catch (IOException e) {
			e.printStackTrace();
		}
		return R.ok();
	}
    /**
     * 修改部门
     *
     * @param division 部门
	 *
     * @return R
     */
    @ApiOperation(value = "修改部门", notes = "修改部门")
    @SysLog("修改部门")
    @PutMapping
    @PreAuthorize("@pms.hasPermission('basic_division_edit')")
    public R updateById(@RequestBody Division division) {
        return R.ok(divisionService.updateById(division));
    }

    /**
     * 通过id删除部门
     *
     * @param id id
     * @return R
     */
    @ApiOperation(value = "通过id删除部门", notes = "通过id删除部门")
    @SysLog("通过id删除部门")
    @DeleteMapping("/{id}")
    @PreAuthorize("@pms.hasPermission('basic_division_del')")
    public R removeById(@PathVariable Long id) {
    	return R.ok(divisionService.removeById(id));
    }


    /**
     * Description: 加载人员列表
     *
     * @param:
     * @return: List<Staff>
     * @author: luh
     * @date: 2020/8/19 14:35
     */
    @GetMapping("/staff")
    public List<Staff> loadCompany() {
        return staffService.list();
    }

    /**
     * Description: 持久化部门实体（自定义）
     *
     * @param: division
     * @return: R
     * @author: luh
     * @date: 2020/8/17 16:14
     */
    @ApiOperation(value = "自定义更新部门", notes = "自定义更新部门")
    @SysLog("自定义更新部门")
    @PostMapping("/adddivision")
    @PreAuthorize("@pms.hasPermission('basic_division_add')")
    public R addDivision(@RequestBody Division division) {
    	if (division.getId().equals(0L)) {
    		//判断编号是否重复
			Division divisionByFind=divisionService.getOne(Wrappers.<Division>lambdaQuery().eq(Division::getDivisionNo, division.getDivisionNo()));
			if(divisionByFind!=null){
				throw new RuntimeException("部门编号重复：" + divisionByFind.getDivisionName() + "-" + divisionByFind.getDivisionNo());
			}else{
				return R.ok(divisionService.save(division));
			}
    	} else {
    		return R.ok(divisionService.updateById(division));
    	}


    }

}