CsdnCommentServiceImpl.java

package com.kwan.springbootkwan.service.impl;

import cn.hutool.core.collection.CollectionUtil;
import cn.hutool.http.HttpResponse;
import cn.hutool.http.HttpUtil;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.kwan.springbootkwan.entity.CsdnUserInfo;
import com.kwan.springbootkwan.entity.resp.BusinessInfoResponse;
import com.kwan.springbootkwan.entity.resp.CommentListResponse;
import com.kwan.springbootkwan.entity.resp.CommentResponse;
import com.kwan.springbootkwan.enums.CommentStatus;
import com.kwan.springbootkwan.service.CsdnArticleService;
import com.kwan.springbootkwan.service.CsdnCommentService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.Objects;

@Slf4j
@Service
public class CsdnCommentServiceImpl implements CsdnCommentService {

    @Value("${csdn.cookie}")
    private String csdnCookie;
    @Value("#{'${csdn.self_comment}'.split(';')}")
    private String[] selfComment;
    @Value("#{'${csdn.self_reply}'.split(';')}")
    private String[] selfReply;
    @Value("${csdn.self_user_name}")
    private String selfUserName;
    @Value("${csdn.url.is_comment_list_url}")
    private String commentListUrl;
    @Value("${csdn.url.comment_url}")
    private String commentUrl;

    @Autowired
    private CsdnArticleService csdnArticleService;

    @Override
    public Boolean isComment(BusinessInfoResponse.ArticleData.Article article, CsdnUserInfo csdnUserInfo) {
        final Integer commentStatus = csdnUserInfo.getCommentStatus();
        if (CommentStatus.HAVE_ALREADY_COMMENT.getCode().equals(commentStatus)
                || CommentStatus.COMMENT_IS_FULL.getCode().equals(commentStatus)
                || CommentStatus.RESTRICTED_COMMENTS.getCode().equals(commentStatus)) {
            return true;
        }
        final String urlInfo = article.getUrl();
        String articleId = urlInfo.substring(urlInfo.lastIndexOf("/") + 1);
        String url = commentListUrl + articleId;
        HttpResponse response = HttpUtil.createPost(url)
                .header("Cookie", csdnCookie)
                .form("page", 1)
                .form("size", 50)
                .form("fold", "unfold")
                .execute();
        final String body = response.body();
        ObjectMapper objectMapper = new ObjectMapper();
        CommentListResponse articleInfo;
        try {
            articleInfo = objectMapper.readValue(body, CommentListResponse.class);
            final CommentListResponse.DataInfo data = articleInfo.getData();
            final List<CommentListResponse.Comment> list = data.getList();
            if (CollectionUtil.isNotEmpty(list)) {
                for (CommentListResponse.Comment comment : list) {
                    final CommentListResponse.Info info = comment.getInfo();
                    final String userName = info.getUserName();
                    if (StringUtils.equals(userName, selfUserName)) {
                        log.info("文章{}已经评论过", articleId);
                        csdnUserInfo.setCommentStatus(CommentStatus.HAVE_ALREADY_COMMENT.getCode());
                        return true;
                    }
                }
            }
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return false;
    }

    @Override
    public Boolean comment(String articleId, CsdnUserInfo csdnUserInfo) {
        final Integer commentStatus = csdnUserInfo.getCommentStatus();
        if (CommentStatus.HAVE_ALREADY_COMMENT.getCode().equals(commentStatus)
                || CommentStatus.COMMENT_IS_FULL.getCode().equals(commentStatus)
                || CommentStatus.RESTRICTED_COMMENTS.getCode().equals(commentStatus)) {
            return true;
        }
        //评论
        int start = -1;
        int end = selfComment.length;
        int temp_count = (int) (Math.floor(Math.random() * (start - end + 1)) + end);
        CommentResponse comment = this.dealComment(articleId, selfComment[temp_count], null);
        final int code = comment.code;
        final String message = comment.getMessage();
        if (code == 200) {
            log.info("文章{}评论成功", articleId);
            csdnUserInfo.setCommentStatus(CommentStatus.COMMENT_SUCCESSFUL.getCode());
        } else if (code == 400 && StringUtils.equals(message, "您已达到当日发送上限，请明天尝试！")) {
            log.info(message);
            csdnUserInfo.setCommentStatus(CommentStatus.COMMENT_IS_FULL.getCode());
        } else if (code == 400 && message.contains("因存在恶意评论嫌疑，您的账号已被禁言")) {
            log.info("因存在恶意评论嫌疑，您的账号已被禁言");
            csdnUserInfo.setCommentStatus(CommentStatus.RESTRICTED_COMMENTS.getCode());
        } else if (code == 400 && StringUtils.equals(message, "您评论太快了，请休息一下！")) {
            log.info("您评论太快了，请休息一下！");
            csdnUserInfo.setCommentStatus(CommentStatus.COMMENT_TOO_FAST.getCode());
            try {
                Thread.sleep(20000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        } else {
            log.info("其他错误");
            csdnUserInfo.setCommentStatus(CommentStatus.OTHER_ERRORS.getCode());
        }
        return true;
    }

    @Override
    public void commentSelf() {
        List<BusinessInfoResponse.ArticleData.Article> list = csdnArticleService.getArticles(selfUserName);
        if (list == null) {
            return;
        }
        for (BusinessInfoResponse.ArticleData.Article article : list) {
            final String type = article.getType();
            if (!StringUtils.equals("blog", type)) {
                continue;
            }
            final String urlInfo = article.getUrl();
            String articleId = urlInfo.substring(urlInfo.lastIndexOf("/") + 1);
            String url = commentListUrl + articleId;
            HttpResponse response = HttpUtil.createPost(url)
                    .header("Cookie", csdnCookie)
                    .form("page", 1)
                    .form("size", 100)
                    .execute();
            final String body = response.body();
            ObjectMapper objectMapper = new ObjectMapper();
            CommentListResponse articleInfo;
            try {
                articleInfo = objectMapper.readValue(body, CommentListResponse.class);
                final CommentListResponse.DataInfo data = articleInfo.getData();
                final List<CommentListResponse.Comment> otherCommentList = data.getList();
                if (CollectionUtil.isNotEmpty(otherCommentList)) {
                    for (CommentListResponse.Comment oneComment : otherCommentList) {
                        final CommentListResponse.Info info = oneComment.getInfo();
                        final String userName = info.getUserName();
                        final Integer commentId = info.getCommentId();
                        if (!StringUtils.equals(userName, selfUserName)) {
                            final List<CommentListResponse.SubComment> sub = oneComment.getSub();
                            boolean flag = false;
                            if (CollectionUtil.isNotEmpty(sub)) {
                                for (CommentListResponse.SubComment subComment : sub) {
                                    //如果没有自己的评论,需要评论
                                    final String subUserName = subComment.getUserName();
                                    if (StringUtils.equals(subUserName, selfUserName)) {
                                        flag = true;
                                    }
                                }
                            }
                            if (CollectionUtil.isEmpty(sub) || !flag) {
                                //需要评论
                                int start = -1;
                                int end = selfReply.length;
                                int temp_count = (int) (Math.floor(Math.random() * (start - end + 1)) + end);
                                CommentResponse reply = this.dealComment(articleId, selfReply[temp_count], commentId);
                                log.info(reply.toString());
                            }
                        }
                    }
                }
            } catch (JsonProcessingException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 评论文章
     *
     * @param articleId
     * @return
     */
    private CommentResponse dealComment(String articleId, String commentInfo, Integer commentId) {
        HttpResponse response;
        if (Objects.nonNull(commentId)) {
            response = HttpUtil.createPost(commentUrl)
                    .header("Cookie", csdnCookie)
                    .form("articleId", articleId)
                    .form("content", commentInfo)
                    .form("commentId", commentId)
                    .execute();
        } else {
            response = HttpUtil.createPost(commentUrl)
                    .header("Cookie", csdnCookie)
                    .form("articleId", articleId)
                    .form("content", commentInfo)
                    .execute();
        }

        final String body = response.body();
        log.info(body);
        ObjectMapper objectMapper = new ObjectMapper();
        CommentResponse commentResponse = null;
        try {
            commentResponse = objectMapper.readValue(body, CommentResponse.class);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return commentResponse;
    }
}