raex_back/src/main/java/com/izouma/nineth/web/UserDetailController.java

package com.izouma.nineth.web;

import com.izouma.nineth.domain.User;
import com.izouma.nineth.domain.UserDetail;
import com.izouma.nineth.dto.MetaRestResult;
import com.izouma.nineth.dto.PageQuery;
import com.izouma.nineth.exception.BusinessException;
import com.izouma.nineth.repo.UserDetailRepo;
import com.izouma.nineth.service.ContentAuditService;
import com.izouma.nineth.service.UserDetailService;
import com.izouma.nineth.service.UserService;
import com.izouma.nineth.utils.SecurityUtils;
import com.izouma.nineth.utils.excel.ExcelUtils;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.domain.Page;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.List;
import java.util.Objects;

@RestController
@RequestMapping("/userDetail")
@AllArgsConstructor
public class UserDetailController extends BaseController {
    private UserDetailService userDetailService;
    private UserDetailRepo userDetailRepo;
    private ContentAuditService contentAuditService;
    private UserService userService;

    //@PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/save")
    public UserDetail save(@RequestBody UserDetail record) {
        if (Objects.isNull(record)) {
            throw new BusinessException("参数为空");
        }
        if (StringUtils.isNotBlank(record.getAutograph())) {
            if (!contentAuditService.auditText(record.getAutograph())) {
                throw new BusinessException("简介包含非法内容");
            }
        }
        UserDetail newUserDetail = userDetailRepo.save(record);
        if (StringUtils.isNotBlank(record.getNickname())) {
            User user = userService.metaUpdate(SecurityUtils.getAuthenticatedUser().getId(), record.getNickname());
            if (Objects.nonNull(user)) {
                newUserDetail.setNickname(user.getNickname());
            }
        }
        return newUserDetail;
    }

    @PostMapping("/meta/save")
    public MetaRestResult<UserDetail> metaSave(@RequestBody UserDetail record) {
        UserDetail userDetail;
        try {
            userDetail = save(record);
        } catch (Exception e) {
            return MetaRestResult.returnError(e.getMessage());
        }
        return MetaRestResult.returnSuccess(userDetail);
    }

    //@PreAuthorize("hasRole('ADMIN')")
    @PostMapping("/all")
    public Page<UserDetail> all(@RequestBody PageQuery pageQuery) {
        return userDetailService.all(pageQuery);
    }

    @GetMapping("/get/{id}")
    public UserDetail get(@PathVariable Long id) {
        UserDetail userDetail = userDetailRepo.findById(id).orElse(null);
        if (Objects.isNull(userDetail)) {
            userDetail = new UserDetail();
            userDetail.setUserId(id);
            userDetailRepo.save(userDetail);
        }
        userDetail.setNftCount(userDetailRepo.findNftCountById(id));
        userDetail.setBuildCount(userDetailRepo.findBuildCountById(id));
        userDetail.setPlotCount(userDetailRepo.findPlotCountById(id));
        userDetail.setFansCount(userDetailRepo.findFansCountById(id));
        userDetail.setReikiCount(userDetailRepo.findReikiCountById(id));
        userDetail.setMedalCount(userDetailRepo.findMedalCountById(id));
        userDetail.setPersonalheatCount(userDetailRepo.findPersonalHeatCountById(id));
        return userDetail;
    }

    @GetMapping("/light/{userId}")
    public boolean light(@PathVariable Long userId) {
        UserDetail userDetail = userDetailRepo.findById(userId).orElse(null);
        if (Objects.isNull(userDetail)) {
            return false;
        }
        if(StringUtils.isNotBlank(userDetail.getAutograph())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getBgpic())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getBirthday())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getBlood())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getCompany())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getConstellation())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getHome())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getMail())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getOccupation())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getPic())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getSchool())) {
            return true;
        }
        if(StringUtils.isNotBlank(userDetail.getSex())) {
            return true;
        }
        return false;
    }



    @GetMapping("/{userId}/metaQuery")
    public MetaRestResult<UserDetail> metaQuery(@PathVariable Long userId) {
        return MetaRestResult.returnSuccess(get(userId));
    }

    @PostMapping("/del/{id}")
    public void del(@PathVariable Long id) {
        userDetailRepo.softDelete(id);
    }

    @GetMapping("/excel")
    @ResponseBody
    public void excel(HttpServletResponse response, PageQuery pageQuery) throws IOException {
        List<UserDetail> data = all(pageQuery).getContent();
        ExcelUtils.export(response, data);
    }
}