package com.chinaztt.mes.production.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.common.wrapper.QueryWrapperUtil;
import com.chinaztt.mes.production.dto.HandymanTypeDTO;
import com.chinaztt.mes.production.entity.HandymanType;
import com.chinaztt.mes.production.entity.JoinTypeTemplate;
import com.chinaztt.mes.production.excel.HandymanTypeData;
import com.chinaztt.mes.production.excel.HandymanTypeUploadListener;
import com.chinaztt.mes.production.service.HandymanTypeService;
import com.chinaztt.mes.production.service.JoinTypeTemplateService;
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 cxf
 * @date 2020-12-01 13:13:44
 */
@RestController
@AllArgsConstructor
@RequestMapping("/handymanType")
@Api(value = "handymanType", tags = "杂工类型维护管理")
public class HandymanTypeController {

    private final HandymanTypeService handymanTypeService;
    private final JoinTypeTemplateService joinTypeTemplateService;

    /**
     * 查询杂工类型list
     *
     * @return
     */
    @ApiOperation(value = "查询杂工类型list", notes = "查询杂工类型list")
    @GetMapping("/list")
    public R getHandymanTypeList() {
        return R.ok(handymanTypeService.list());
    }

    /**
     * 分页查询
     *
     * @param page         分页对象
     * @param handymanType 杂工类型维护
     * @return
     */
    @ApiOperation(value = "分页查询", notes = "分页查询")
    @GetMapping("/page")
    @PreAuthorize("@pms.hasPermission('production_handymantype_view')")
    public R getHandymanTypePage(Page page, HandymanType handymanType) {
        return R.ok(handymanTypeService.page(page, QueryWrapperUtil.gen(handymanType)));
    }

    /**
     * 获取关联的人工类型
     *
     * @param page         分页对象
     * @param handymanType 杂工类型维护
     * @param id           杂工类型维护
     * @return
     */
    @ApiOperation(value = "获取关联的人工类型", notes = "获取关联的人工类型")
    @GetMapping("/page/join/{id}")
    public R getHandymanType(Page page, HandymanType handymanType, @PathVariable("id") Long id) {
        handymanType.setId(null);
        return R.ok(handymanTypeService.getHandymanType(page, QueryWrapperUtil.gen(handymanType), id));
    }

    /**
     * 获取未关联的人工类型
     *
     * @param page         分页对象
     * @param handymanType 杂工类型维护
     * @param id           杂工类型维护
     * @return
     */
    @ApiOperation(value = "获取未关联的人工类型", notes = "获取未关联的人工类型")
    @GetMapping("/page/{id}")
    public R getHandyman(Page page, HandymanType handymanType, @PathVariable("id") Long id) {
        handymanType.setId(null);
        return R.ok(handymanTypeService.getHandyman(page, QueryWrapperUtil.gen(handymanType), id));
    }

    /**
     * 通过id查询杂工类型维护
     *
     * @param id id
     * @return R
     */
    @ApiOperation(value = "通过id查询", notes = "通过id查询")
    @GetMapping("/{id}")
    @PreAuthorize("@pms.hasPermission('production_handymantype_view')")
    public R getById(@PathVariable("id") Long id) {
        return R.ok(handymanTypeService.getHandymanTypeById(id));
    }

    /**
     * 新增杂工类型维护
     *
     * @param handymanType 杂工类型维护
     * @return R
     */
    @ApiOperation(value = "新增杂工类型维护", notes = "新增杂工类型维护")
    @SysLog("新增杂工类型维护")
    @PostMapping
    @PreAuthorize("@pms.hasPermission('production_handymantype_add')")
    public R save(@RequestBody HandymanType handymanType) {
        return R.ok(handymanTypeService.save(handymanType));
    }

    /**
     * 修改杂工类型维护
     *
     * @param handymanType 杂工类型维护
     * @return R
     */
    @ApiOperation(value = "修改杂工类型维护", notes = "修改杂工类型维护")
    @SysLog("修改杂工类型维护")
    @PutMapping
    @PreAuthorize("@pms.hasPermission('production_handymantype_edit')")
    public R updateById(@RequestBody HandymanType handymanType) {
        return R.ok(handymanTypeService.updateById(handymanType));
    }

    /**
     * 通过id删除杂工类型维护
     *
     * @param id id
     * @return R
     */
    @ApiOperation(value = "通过id删除杂工类型维护", notes = "通过id删除杂工类型维护")
    @SysLog("通过id删除杂工类型维护")
    @DeleteMapping("/{id}")
    @PreAuthorize("@pms.hasPermission('production_handymantype_del')")
    public R removeById(@PathVariable Long id) {
        return R.ok(handymanTypeService.removeById(id));
    }

    /**
     * 通过id删除人工类型分组关系表
     *
     * @param id id
     * @return R
     */
    @ApiOperation(value = "通过id删除人工类型分组关系表", notes = "通过id删除人工类型分组关系表")
    @SysLog("通过id删除人工类型分组关系表")
    @DeleteMapping("/{id}/{id1}")
    public R delHandyTypeRelation(@PathVariable Long id, @PathVariable Long id1) {
        return R.ok(handymanTypeService.delHandyTypeRelation(id, id1));
    }

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

    /**
     * 根据模板id过滤分页查询
     *
     * @param page            分页对象
     * @param handymanTypeDTO 杂工类型维护
     * @return
     */
    @ApiOperation(value = "根据模板id过滤分页查询", notes = "根据模板id过滤分页查询")
    @GetMapping("/fetchListFilterTemplateId")
    public R fetchListFilterTemplateId(Page page, HandymanTypeDTO handymanTypeDTO) {
        Long handymanTemplateId = handymanTypeDTO.getHandymanTemplateId();
        handymanTypeDTO.setHandymanTemplateId(null);
        return R.ok(handymanTypeService.fetchListFilterTemplateId(page, QueryWrapperUtil.gen(handymanTypeDTO), handymanTemplateId));
    }

    /**
     * 根据模板id分页查询
     *
     * @param page            分页对象
     * @param handymanTypeDTO 杂工类型维护
     * @return
     */
    @ApiOperation(value = "根据模板id分页查询", notes = "根据模板id分页查询")
    @GetMapping("/fetchListByTemplateId")
    public R fetchListByTemplateId(Page page, HandymanTypeDTO handymanTypeDTO) {
        return R.ok(handymanTypeService.fetchListByTemplateId(page, QueryWrapperUtil.gen(handymanTypeDTO)));
    }

    /**
     * 模板关联杂工类型
     *
     * @param joinTypeTemplateList
     * @return
     */
    @ApiOperation(value = "模板关联杂工类型", notes = "模板关联杂工类型")
    @PostMapping("/relateTypeTemplate")
    public R relateTypeTemplate(@RequestBody List<JoinTypeTemplate> joinTypeTemplateList) {
        return R.ok(joinTypeTemplateService.saveBatch(joinTypeTemplateList));
    }

    /**
     * 模板删除关联的杂工
     *
     * @param joinTypeTemplate
     * @return
     */
    @ApiOperation(value = "模板删除关联的杂工", notes = "模板删除关联的杂工")
    @PostMapping("/delTypeTemplate")
    public R delTypeTemplate(@RequestBody JoinTypeTemplate joinTypeTemplate) {
        return R.ok(joinTypeTemplateService.remove(Wrappers.<JoinTypeTemplate>lambdaQuery()
                .eq(JoinTypeTemplate::getHandymanTemplateId, joinTypeTemplate.getHandymanTemplateId())
                .eq(JoinTypeTemplate::getHandymanTypeId, joinTypeTemplate.getHandymanTypeId())));
    }
}