EmploymentSite/src/main/java/com/hz/employmentsite/controller/baseSettings/SiteUserController.java

package com.hz.employmentsite.controller.baseSettings;

import com.github.pagehelper.PageInfo;
import com.hz.employmentsite.filter.exception.BaseErrorEnum;
import com.hz.employmentsite.filter.exception.BaseResponse;
import com.hz.employmentsite.filter.exception.RespGenerstor;
import com.hz.employmentsite.model.SelectProps;
import com.hz.employmentsite.services.service.AccountService;
import com.hz.employmentsite.services.service.baseSettings.SiteUserService;
import com.hz.employmentsite.util.DateUtils;
import com.hz.employmentsite.util.ExcelHelper;
import com.hz.employmentsite.vo.baseSettings.SiteUserVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.UUID;

@RestController
@RequestMapping("/api/userInfo/")
public class SiteUserController {

    @Autowired
    private AccountService accountService;

    @Autowired
    private SiteUserService userInfoService;

    @Autowired
    private ExcelHelper excelHelper;

    @Autowired
    private DateUtils dateUtils;

    @ResponseBody
    @GetMapping("/getList")
    public BaseResponse<PageInfo<SiteUserVo>> getList(@RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize,
                                                      @RequestParam(required = false) List<String> siteUserIDList,
                                                      @RequestParam(required = false) String siteUserName,
                                                      @RequestParam(required = false) String siteID,
                                                      @RequestParam(required = false) String roleName,
                                                      @RequestParam(required = false) String regionCode,
                                                      @RequestParam(required = false) String userNo) {

        PageInfo<SiteUserVo> result = userInfoService.getList(pageIndex, pageSize, siteUserIDList, siteUserName, siteID, roleName, regionCode, userNo);

        return RespGenerstor.success(result);
    }

    @PostMapping("getListBySiteID")
    public BaseResponse<Integer> getListBySiteID(@RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize,@RequestParam(required = false) String siteID) {
        return RespGenerstor.success(userInfoService.getListBySiteID(pageIndex,pageSize, siteID));
    }


    @GetMapping("get")
    public BaseResponse<SiteUserVo> get(@RequestParam(required = false) String siteUserID) {
        var data =  userInfoService.get(siteUserID);
        if (data == null) {
            data = new SiteUserVo();
            data.siteUserID = UUID.randomUUID().toString();
        }
        return RespGenerstor.success(data);
    }

    @ResponseBody
    @GetMapping("/getDataList")
    public BaseResponse<ArrayList<SelectProps>> getDataList() {
        return RespGenerstor.success(userInfoService.getDataList());
    }

    @PostMapping("save")
    public BaseResponse<Integer> save(@RequestBody SiteUserVo data) {
        return RespGenerstor.success(userInfoService.save(data,accountService.getLoginUserID()));
    }

    @GetMapping("export")
    public BaseResponse export(HttpServletResponse response, @RequestParam("pageIndex") int pageIndex, @RequestParam("pageSize") int pageSize,
                               @RequestParam(required = false) List<String> siteUserIDList,
                               @RequestParam(required = false) String siteUserName,
                               @RequestParam(required = false) String siteID) throws Exception{
        PageInfo<SiteUserVo> result = userInfoService.getList(pageIndex, pageSize, siteUserIDList, siteUserName, siteID, null, null, null);
        ExcelHelper.ExcelData data = excelHelper.new ExcelData();
        data.setTitles(Arrays.asList(new String[]{"序号", "工号", "人员名称", "角色", "性别", "联系电话", "年龄", "身份证号", "所属驿站"}));
        int i = 0;
        List<List<Object>> rowsData= new ArrayList();
        for (SiteUserVo item : result.getList()) {
            List<Object> row = new ArrayList();
            ++i;
            row.add(i);
            row.add(item.userNo);
            row.add(item.siteUserName);
            row.add(item.roleName);
            row.add(item.genderName);
            row.add(item.mobile);
            row.add(dateUtils.getAgeForIdCard(item.getIdCard()));
            row.add(item.idCard);
            row.add(item.siteName);
            rowsData.add(row);
        }
        data.setRows(rowsData);
        excelHelper.exportExcel(response, data);
        return null;
    }

    @ResponseBody
    @PostMapping("/delete")
    public BaseResponse<Integer> delete(@RequestBody List<String> idList) {
        return RespGenerstor.success(userInfoService.delete(idList));
    }

    @PostMapping("/importSiteUser")
    public BaseResponse<Object> importSiteUser(@RequestBody List<SiteUserVo> dataList) {
        List<SiteUserVo> result = userInfoService.importSiteUser(dataList, accountService.getLoginUserID());

        if (result != null && result.size() > 0) {
            return RespGenerstor.fail(BaseErrorEnum.IMPORT_DATA_ERROR, result);
        } else {
            return RespGenerstor.success(true);
        }
    }

    @GetMapping("/getListForAddUser")
    public BaseResponse<List<SiteUserVo>> getListForAddUser() {

        List<SiteUserVo> result = userInfoService.getListForAddUser();

        return RespGenerstor.success(result);
    }


}