package com.izouma.nineth.web; import com.izouma.nineth.domain.Rice; import com.izouma.nineth.domain.RiceUserWaterDropRecord; import com.izouma.nineth.domain.User; import com.izouma.nineth.dto.PageQuery; import com.izouma.nineth.dto.R; import com.izouma.nineth.dto.RiceDTO; import com.izouma.nineth.enums.AuthorityName; import com.izouma.nineth.enums.RiceOperationType; import com.izouma.nineth.enums.RiceWaterType; import com.izouma.nineth.exception.BusinessException; import com.izouma.nineth.repo.*; import com.izouma.nineth.service.*; import com.izouma.nineth.utils.ObjUtils; import com.izouma.nineth.utils.SecurityUtils; import com.izouma.nineth.utils.excel.ExcelUtils; import lombok.AllArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.apache.commons.lang.time.DateUtils; import org.springframework.data.domain.Page; import org.springframework.security.access.prepost.PreAuthorize; import org.springframework.web.bind.annotation.*; import javax.servlet.http.HttpServletResponse; import java.io.IOException; import java.time.LocalDateTime; import java.util.*; @RestController @RequestMapping("/rice") @AllArgsConstructor @Slf4j public class RiceController extends BaseController { private RiceService riceService; private RiceRepo riceRepo; private UserDetailService userDetailService; private UserDetailRepo userDetailRepo; private UserService userService; private UserRepo userRepo; private RiceInviteRepo RiceInviteRepo; private RiceInviteService RiceInviteService; private SysConfigService sysConfigService; private SysConfigRepo sysConfigRepo; private static final int MAX_NICKNAME_LENGTH = 14; private static final String UPDATE_SUCCESS_MSG = "修改成功"; @PreAuthorize("hasRole('ADMIN')") @PostMapping("/save") public Rice save(@RequestBody Rice record) { if (record.getId() != null) { Rice orig = riceRepo.findById(record.getId()).orElseThrow(new BusinessException("无记录")); ObjUtils.merge(orig, record); return riceRepo.save(orig); } return riceRepo.save(record); } @PreAuthorize("hasRole('ADMIN')") @PostMapping("/all") public Page all(@RequestBody PageQuery pageQuery) { if (!SecurityUtils.hasRole(AuthorityName.ROLE_ADMIN) & !SecurityUtils.hasRole(AuthorityName.ROLE_ORDERINFO)) { pageQuery.getQuery().put("userId", SecurityUtils.getAuthenticatedUser().getId()); } return riceService.all(pageQuery); } @GetMapping("/get/{id}") public Rice get(@PathVariable Long id) { return riceRepo.findById(id).orElseThrow(new BusinessException("无记录")); } /* @PostMapping("/del/{id}") public void del(@PathVariable Long id) { riceRepo.softDelete(id); }*/ @PreAuthorize("hasRole('ADMIN')") @GetMapping("/excel") @ResponseBody public void excel(HttpServletResponse response, PageQuery pageQuery) throws IOException { if (!SecurityUtils.hasRole(AuthorityName.ROLE_ADMIN) & !SecurityUtils.hasRole(AuthorityName.ROLE_ORDERINFO)) { List data = all(pageQuery).getContent(); ExcelUtils.export(response, data); } } //点击水稻游戏后对riceuser进行初始化赋值 @GetMapping("/current") public R getCurrentUser() throws BusinessException { return R.success(riceService.getCurrentRiceUser(SecurityUtils.getAuthenticatedUser())); } //修改用户昵称 @PostMapping("/updateNickName") public R updateNickName(@RequestParam("userId") Long userId, @RequestParam("nickname") String nickname) { return riceService.updateNickName(userId, nickname); } //等级显示 @GetMapping("/showLevel") public R> showLevel() { return riceService.showLevel(); } //获取用户积分 @GetMapping("/selfScore") public R getCurrentScore() { return riceService.getCurrentScore(); } //一个获取积分排行榜的接口 @GetMapping("/scoreRanking") public R> getTop100AndSelf() { return R.success(riceService.getTop100(SecurityUtils.getAuthenticatedUser().getId())); } //只获取自己的排名 @GetMapping("/riceUserRank") public R getRiceUserRank() { return riceService.getRiceUserRank(SecurityUtils.getAuthenticatedUser().getId()); } //浇水 @PostMapping("/waterDrop") public R waterDrop(@RequestParam("riceWaterType") RiceWaterType riceWaterType, @RequestParam Long riceId) { return riceService.waterDrop(riceWaterType, riceId); } // //获取今日已浇水次数和还需浇水次数升级的接口 // @GetMapping("/watering/count") // public R getWateringCount() { // Long todayWateringCount = riceService.getTodayWateringCount(SecurityUtils.getAuthenticatedUser().getId()); // Rice rice = riceRepo.findByUserId(SecurityUtils.getAuthenticatedUser().getId()).orElseThrow(new BusinessException("没找到记录")); // Long waterDropNeededForLevelUp = riceService.getWaterDropNeededForLevelUp(rice); // return R.success(Map.of("todayWateringCount", todayWateringCount, "waterDropNeededForLevelUp", waterDropNeededForLevelUp)); // } //签到 @GetMapping("/signin") public R signIn() { return riceService.signIn(SecurityUtils.getAuthenticatedUser().getId()); } // 任务初始化.显示各个任务能否点击 @GetMapping("/taskInitialization") public R taskInitialization() { return riceService.taskInitialization(); } //积分兑换水滴 @GetMapping("/exchangeScoreForWaterDrop") public R exchangeScoreForWaterDrop() { return riceService.exchangeScoreForWaterDrop(SecurityUtils.getAuthenticatedUser()); } //活动积分兑换水滴 @GetMapping("/exchangeActivityScoreForWaterDrop") public R exchangeActivityScoreForWaterDrop() { return riceService.exchangeActivityScoreForWaterDrop(SecurityUtils.getAuthenticatedUser()); } //测试添加用户 @GetMapping("/newRiceUser") public R newRiceUser() { User authenticatedUser = SecurityUtils.getAuthenticatedUser(); Long id = authenticatedUser.getId(); Optional byId = userRepo.findByIdAndDelFalse(id); String nickname = null; String avatar = null; if (byId.isPresent()) { User user = byId.get(); nickname = user.getNickname(); avatar = user.getAvatar(); } else { throw new BusinessException("用户不存在"); } UUID uuid = UUID.randomUUID(); Rice rice = new Rice(); rice.setUserId(generateUniqueId()); rice.setAvatar(avatar); rice.setNickname(nickname); rice.setLevel(0L); rice.setSignCount(0L); rice.setSelfScore(0L); rice.setSelfActivityScore(0L); rice.setEmpiricalValue(0L); riceRepo.save(rice); return R.success("添加成功"); } public static Long generateUniqueId() { UUID uuid = UUID.randomUUID(); long lsb = uuid.getLeastSignificantBits(); long msb = uuid.getMostSignificantBits(); return new Long((lsb & Long.MAX_VALUE) | (msb & Long.MAX_VALUE)); } }