raex_back/src/main/java/com/izouma/nineth/service/netease/TeamService.java

package com.izouma.nineth.service.netease;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.izouma.nineth.config.RedisKeys;
import com.izouma.nineth.domain.netease.NeteaseMessage;
import com.izouma.nineth.domain.netease.Team;
import com.izouma.nineth.dto.PageQuery;
import com.izouma.nineth.exception.BusinessException;
import com.izouma.nineth.repo.netease.NeteaseMessageRepo;
import com.izouma.nineth.repo.netease.TeamRepo;
import com.izouma.nineth.utils.DateTimeUtils;
import com.izouma.nineth.utils.JpaUtils;
import com.izouma.nineth.utils.SecurityUtils;
import lombok.AllArgsConstructor;
import org.apache.commons.lang3.StringUtils;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.PageImpl;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Service
@AllArgsConstructor
public class TeamService {

    private TeamRepo                      teamRepo;
    private NeteaseUserService            neteaseUserService;
    private RedisTemplate<String, Object> redisTemplate;
    private NeteaseMessageRepo            neteaseMessageRepo;

    public Page<Team> all(PageQuery pageQuery) {
        Long userId = SecurityUtils.getAuthenticatedUser().getId();
        Page<Team> teams = teamRepo
                .findAll(JpaUtils.toSpecification(pageQuery, Team.class), JpaUtils.toPageRequest(pageQuery));
        List<Team> content = teams.getContent();
        List<Team> newContent = new ArrayList<>();
        content.forEach(team -> {
            List<String> members = new ArrayList<>(team.getMembers());
            boolean inTeam = false;
            if (StringUtils.equals(team.getOwnerid(), userId.toString())) {
                inTeam = true;
            } else {
                for (String member : members) {
                    String accid = userId.toString();
                    if (StringUtils.equals(accid, member)) {
                        inTeam = true;
                    }
                }
            }
            team.setLastMsg(getTeamLastMessage(team.getTid()));
            team.setUnread(getUnreadCount(userId.toString(), team.getTid().toString()));
            team.setInTeam(inTeam);
            newContent.add(team);
        });
        return new PageImpl<>(newContent, teams.getPageable(), teams.getTotalElements());
    }

    public Team create(Team team) {
        Map<String, Object> params = new HashMap<>();
        params.put("tname", team.getName());
        params.put("owner", team.getOwnerid());
        params.put("members", JSONObject.toJSONString(team.getMembers()));
        params.put("joinmode", "0");
        params.put("custom", team.getCustom().toString());
        params.put("msg", team.getMsg());
        String result = neteaseUserService
                .httpPost("team/create.action", "application/x-www-form-urlencoded;charset=utf-8", params);
        JSONObject jsonObject = JSON.parseObject(result);
        Integer code = jsonObject.getInteger("code");
        if (code != 200) {
            throw new BusinessException("注册出错，请核查后重新注册");
        }
        Long id = jsonObject.getLong("tid");
        team.setTid(id);
        return teamRepo.save(team);
    }

    public Team update(Team team) {
        Map<String, Object> params = new HashMap<>();
        params.put("tid", team.getTid());
        params.put("tname", team.getName());
        params.put("owner", team.getOwnerid());
        params.put("announcement", team.getAnnouncement());
        params.put("intro", team.getIntro());
        params.put("icon", "0");
        String result = neteaseUserService
                .httpPost("team/update.action", "application/x-www-form-urlencoded;charset=utf-8", params);
        JSONObject jsonObject = JSON.parseObject(result);
        Integer code = jsonObject.getInteger("code");
        if (code != 200) {
            throw new BusinessException("注册出错，请核查后重新注册");
        }
        return teamRepo.save(team);
    }

    public void mute(Long tid) {
        Map<String, Object> params = new HashMap<>();
        Team team = teamRepo.findById(tid).orElseThrow(new BusinessException("暂无"));
        params.put("tid", tid);
        params.put("owner", team.getOwnerid());
        params.put("muteType", 1);
        String result = neteaseUserService
                .httpPost("team/muteTlistAll.action", "application/x-www-form-urlencoded;charset=utf-8", params);
        JSONObject jsonObject = JSON.parseObject(result);
        Integer code = jsonObject.getInteger("code");
        if (code != 200) {
            throw new BusinessException("禁言操作出错");
        }
        team.setMute(true);
        teamRepo.save(team);
    }

    public void cancelMute(Long tid) {
        Map<String, Object> params = new HashMap<>();
        Team team = teamRepo.findById(tid).orElseThrow(new BusinessException("暂无"));
        params.put("tid", tid);
        params.put("owner", team.getOwnerid());
        params.put("muteType", 0);
        String result = neteaseUserService
                .httpPost("team/muteTlistAll.action", "application/x-www-form-urlencoded;charset=utf-8", params);
        JSONObject jsonObject = JSON.parseObject(result);
        Integer code = jsonObject.getInteger("code");
        if (code != 200) {
            throw new BusinessException("禁言操作出错");
        }
        team.setMute(false);
        teamRepo.save(team);
    }

    public Team invite(String id, String tid) {
        List<String> ids = new ArrayList<>();
        ids.add(id);
        Team team = teamRepo.findById(Long.valueOf(tid)).orElseThrow(new BusinessException("未找到群聊"));
        Map<String, Object> params = new HashMap<>();
        params.put("tid", tid);
        params.put("owner", team.getOwnerid());
        params.put("members", JSONObject.toJSONString(ids));
        params.put("magree", "0");
        params.put("custom", team.getCustom().toString());
        params.put("msg", "欢迎加入大厅测试群~~");
        String result = neteaseUserService
                .httpPost("team/add.action", "application/x-www-form-urlencoded;charset=utf-8", params);
        JSONObject jsonObject = JSON.parseObject(result);
        Integer code = jsonObject.getInteger("code");
        if (code != 200) {
            throw new BusinessException("注册出错，请核查后重新注册");
        }
        List<String> members = new ArrayList<>(team.getMembers());
        members.add(id);
        team.setMembers(members);
        return teamRepo.save(team);
    }

    public void leave(String id, String tid) {
        Team team = teamRepo.findById(Long.valueOf(tid)).orElseThrow(new BusinessException("未找到群聊"));
        Map<String, Object> params = new HashMap<>();
        params.put("tid", tid);
        params.put("accid", id);
        String result = neteaseUserService
                .httpPost("team/leave.action", "application/x-www-form-urlencoded;charset=utf-8", params);
        JSONObject jsonObject = JSON.parseObject(result);
        Integer code = jsonObject.getInteger("code");
        if (code != 200) {
            throw new BusinessException("退群出错");
        }
        List<String> members = new ArrayList<>(team.getMembers());
        members.removeIf(member -> StringUtils.equals(member, id));
        team.setMembers(members);
        teamRepo.save(team);
    }

    public void checkout(String accid, String tid) {
        Long ts = DateTimeUtils.toTimestamp(LocalDateTime.now());
        redisTemplate.opsForValue().set(RedisKeys.USER_CHECKOUT_TIME + accid + ":" + tid, ts);
    }

    public Long getUnreadCount(String accid, String tid) {
        Long ts = (Long) redisTemplate.opsForValue().get(RedisKeys.USER_CHECKOUT_TIME + accid + ":" + tid);
        Long unreadCount = 0L;
        if (ts != null) {
            LocalDateTime localDateTime = DateTimeUtils.toLocalDateTime(ts);
            unreadCount = neteaseMessageRepo.countAllByToIdAndOpeAndCreatedAtAfter(tid, 1, localDateTime);
        } else {
            unreadCount = neteaseMessageRepo.countAllByToIdAndOpe(tid, 1);
        }
        return unreadCount;
    }

    public NeteaseMessage getTeamLastMessage(Long tid) {
        return neteaseMessageRepo.findFirstByToIdAndOpeOrderByCreatedAtDesc(tid.toString(), 1);
    }
}