/*
 *    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.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.Factory;
import com.chinaztt.mes.basic.entity.Location;
import com.chinaztt.mes.basic.entity.Staff;
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 com.chinaztt.ztt.common.security.annotation.Inner;
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 java.util.List;


/**
 * 员工
 *
 * @author luh
 * @date 2020-08-14 15:30:26
 */
@RestController
@AllArgsConstructor
@RequestMapping("/staff")
@Api(value = "staff", tags = "员工管理")
public class StaffController {

    private final StaffService staffService;
    private final DivisionService divisionService;

    /**
     * 分页查询
     *
     * @param page  分页对象
     * @param staff 员工
     * @return
     */
    @ApiOperation(value = "分页查询", notes = "分页查询")
    @GetMapping("/page")
    @PreAuthorize("@pms.hasPermission('basic_staff_view','product_workbench')")
    public R getStaffPage(Page page, Staff staff) {
        return R.ok(staffService.getStaffPage(page, QueryWrapperUtil.gen(staff)));
    }


    /**
     * 通过id查询员工
     *
     * @param id id
     * @return R
     */
    @ApiOperation(value = "通过id查询", notes = "通过id查询")
    @GetMapping("/{id}")
    @PreAuthorize("@pms.hasPermission('basic_staff_view','product_workbench')")
    public R getById(@PathVariable("id") Long id) {
        return R.ok(staffService.getById(id));
    }

    /**
     * 新增员工
     *
     * @param staff 员工
     * @return R
     */
    @ApiOperation(value = "新增员工", notes = "新增员工")
    @SysLog("新增员工")
    @PostMapping
    @PreAuthorize("@pms.hasPermission('basic_staff_add')")
    public R save(@RequestBody Staff staff) {
        return R.ok(staffService.save(staff));
    }

    /**
     * 修改员工
     *
     * @param staff 员工
     * @return R
     */
    @ApiOperation(value = "修改员工", notes = "修改员工")
    @SysLog("修改员工")
    @PutMapping
    @PreAuthorize("@pms.hasPermission('basic_staff_edit')")
    public R updateById(@RequestBody Staff staff) {
        return R.ok(staffService.updateById(staff));
    }

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

    /**
     * Description: 加载部门列表
     *
     * @param:
     * @return: List<Division>
     * @author: luh
     * @date: 2020/8/19 14:35
     */
    @GetMapping("/division")
    public List<Division> loadCompany() {
        return divisionService.list();
    }

    /**
     * Description: 持久化员工实体（自定义）
     *
     * @param: staff
     * @return: R
     * @author: luh
     * @date: 2020/8/17 16:14
     */
    @ApiOperation(value = "自定义更新员工", notes = "自定义更新员工")
    @SysLog("自定义更新员工")
    @PostMapping("/addstaff")
    @PreAuthorize("@pms.hasPermission('basic_staff_add')")
    public R addStaff(@RequestBody Staff staff) {
		if (staff.getId().equals(0L)) {
			// 判断编号重复
			Staff staffByFind = staffService.getOne(Wrappers.<Staff>lambdaQuery().eq(Staff::getStaffNo, staff.getStaffNo()));
			if (staffByFind != null) {
				throw new RuntimeException("员工编号重复：" + staffByFind.getStaffName() + "-" + staffByFind.getStaffNo());
			}else{
				return R.ok(staffService.save(staff));
			}
		} else {
			return R.ok(staffService.updateById(staff));
		}
    }

	@GetMapping("/staff")
	@Inner(value = false)
	public R loadStaff(Staff staff) {
		return R.ok(staffService.list(Wrappers.query(staff)));
	}

}