周伟鹏 / cnlive_shenhe · Files

package com.cnlive.shenhe.controller;

import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.cnlive.shenhe.bean.ErrorEnum;
import com.cnlive.shenhe.bean.ResponseBean;
import com.cnlive.shenhe.bean.SessionUser;
import com.cnlive.shenhe.dto.CommentsDTO;
import com.cnlive.shenhe.entity.ShyComments;
import com.cnlive.shenhe.entity.ShyContent;
import com.cnlive.shenhe.entity.ShySites;
import com.cnlive.shenhe.entity.ShyUsers;
import com.cnlive.shenhe.service.CommentsService;
import com.cnlive.shenhe.service.ContentService;
import com.cnlive.shenhe.service.SitesService;
import com.cnlive.shenhe.service.UsersService;
import com.cnlive.shenhe.utils.AuditPass;
import com.cnlive.shenhe.utils.CommonConst;
import com.cnlive.shenhe.utils.CommonUtils;
import com.github.pagehelper.PageInfo;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.*;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.*;
import java.util.List;


@Controller
@RequestMapping("/comments")
public class CommentsController extends BaseController {
    private static Logger logger = LoggerFactory.getLogger(CommentsController.class);
    @Autowired
    AuditPass pass;
    @Autowired
    CommentsService commentsService;

    @Autowired
    UsersService usersService;

    @Autowired
    SitesService sitesService;
    @Autowired
    ContentService contentService;

    /**
     * 评论列表页
     *
     * @param commentsDTO
     * @return
     */
    @GetMapping("/page")
    public String getPage(Model model, CommentsDTO commentsDTO, HttpSession session) {
        logger.info("列表请求参数:{}", JSON.toJSONString(commentsDTO));
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        if (CommonUtils.isNull(sessionUser)) {
            return "redirect:/users/login";
        }
        Integer spid = sessionUser.getSpid();
        commentsDTO.setSpid(spid);
        String userId = sessionUser.getUserId();
        commentsDTO.setUserId(userId);
        if (spid == 0) {
            commentsDTO.setSpid(null);
        }

        if (CommonUtils.isNotEmpty(commentsDTO.getStart_date())) {
            try {
                commentsDTO.setStart_date(String.valueOf(CommonUtils.parseDate(commentsDTO.getStart_date(), "yyyy-MM-dd")));
            } catch (Exception e) {
            }
        }
        if (CommonUtils.isNotEmpty(commentsDTO.getEnd_date())) {
            try {
                commentsDTO.setEnd_date(String.valueOf(CommonUtils.parseDate(commentsDTO.getEnd_date(), "yyyy-MM-dd")));
            } catch (Exception e) {
            }
        }
        PageInfo<ShyComments> commentPage = commentsService.getPage(commentsDTO);
        List<ShyComments> list = commentPage.getList();
        if (!list.isEmpty()) {
            for (ShyComments comment : list) {
                if (!comment.getState().equals(CommonConst.AUDIT_INTT)) {
                    String updater = CommonUtils.isEmpty(comment.getUpdater()) || "null".equals(comment.getUpdater()) ? "白名单" : comment.getUpdater();
                    String auditorId = comment.getAuditor_id();
                    if (CommonUtils.isNotEmpty(auditorId)) {
                        ShyUsers user = new ShyUsers();
                        user.setUser_id(auditorId);
                        user = usersService.select(user);
                        if (CommonUtils.isNotNull(user)) {
                            String email = CommonUtils.isEmpty(user.getEmail()) || "null".equals(user.getEmail()) ? "" : user.getEmail();
                            updater = CommonUtils.isEmpty(user.getUsername()) ? email : user.getUsername();
                        }
                    }
                    comment.setUpdater(updater);
                }
                if (null != comment.getSpid()) {
                    //显示spid简称
                    ShySites shySites = sitesService.findspidDescByspid(comment.getSpid());
                    if (CommonUtils.isNotNull(shySites)) {
                        comment.setSpid_desc(shySites.getName());
                    }
                } else {
                    comment.setSpid_desc("sp无法显示");
                }

                //获取朋友圈的评论内容
                if (comment.getProgram_id().split("_").length == 1
                        && comment.getProgram_id().split(":").length == 3) {
                    ShyContent shyContent = new ShyContent();
                    shyContent.setSp_id(comment.getSpid());
                    shyContent.setContent_id(comment.getProgram_id());
                    shyContent = contentService.getShyContent(shyContent);
                    if (CommonUtils.isNotNull(shyContent) && shyContent.getId() != null) {
                        String title = CommonUtils.isEmpty(shyContent.getTitle()) ? "朋友圈发布无标题内容" : shyContent.getTitle();
                        comment.setComment_title(title);
                        comment.setComment_original_url(shyContent.getContent_url());
                    }
                }
                //评论的类型
                if (CommonConst.CATEGORY_TOPIC.equals(comment.getType())) {
                    comment.setType("话题");
                } else {
                    comment.setType("其他");
                }
            }
        }
        /* 获取参数值给前台 用来判断是否需要把查看显示出来 state recive*/
        String state1 = commentsDTO.getState() == null ? "" : commentsDTO.getState().toString();
        model.addAttribute("state1", state1);
        model.addAttribute("pageSize", commentsDTO.getPageSize());
        model.addAttribute("commentPage", commentPage);
        model.addAttribute("Type", commentsDTO.getType());
        model.addAttribute("Activity_id", commentsDTO.getActivity_id());
        model.addAttribute("Comment_user_id", commentsDTO.getComment_user_id());
        model.addAttribute("Content", commentsDTO.getContent());
        model.addAttribute("Start_date", commentsDTO.getStart_date());
        model.addAttribute("End_date", commentsDTO.getEnd_date());
        model.addAttribute("CommentStatus", commentsDTO.getState());
        return "page/comment.html";
    }

    /**
     * 评论详情
     *
     * @param id
     * @return
     */
    @GetMapping("/details/{id}")
    public String getDetailsComments(@PathVariable Integer id, Model model) {
        ShyComments comment = commentsService.getDetailsComment(id);
        model.addAttribute("comment", comment);
        return "page/commentDetail.html";
    }


    /**
     * 审核通过
     *
     * @param ids
     * @return
     */
    @PostMapping("/pass")
    @ResponseBody
    public ResponseBean pass(String ids, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        String userId = sessionUser.getUserId();
        boolean success = commentsService.updateState(ids, CommonConst.AUDIT_SUCCESS, userId, "", "审核通过");
        if (success) {
            return new ResponseBean();
        } else {
            return new ResponseBean(ErrorEnum.ERROR);
        }
    }

    /**
     * 审核拒绝
     *
     * @param ids
     * @return
     */
    @PostMapping("/refuse")
    @ResponseBody
    public ResponseBean refuse(String ids, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        String userId = sessionUser.getUserId();
        boolean success = commentsService.updateState(ids, CommonConst.AUDIT_REFUSE, userId, "", "审核拒绝");
        if (success) {
            return new ResponseBean();
        } else {
            return new ResponseBean(ErrorEnum.ERROR);
        }

    }


    /**
     * 认领
     *
     * @param ids     根据id认领
     * @param num     认领条数
     * @param session
     * @return
     */
    @PostMapping("/receive")
    @ResponseBody
    public ResponseBean receive(String ids, Integer num, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        String userId = sessionUser.getUserId();
        Integer spid = sessionUser.getSpid();
        Integer n = commentsService.receive(ids, num, userId, spid);
        return new ResponseBean(ErrorEnum.SUCCESS.getErrorCode(), "认领到" + n + "条内容");
    }


    @PostMapping("/retReceive")
    @ResponseBody
    public ResponseBean retReceive(String ids, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        String userId = sessionUser.getUserId();
        Integer n = commentsService.retReceive(ids, userId);
        return new ResponseBean(ErrorEnum.SUCCESS.getErrorCode(), "退领了" + n + "条内容");
    }

    /**
     * 打回(审核拒绝，应要求不加权限、状态等限制)
     *
     * @param ids
     * @param msg
     * @param session
     * @return
     */
    @PostMapping("/repulse")
    @ResponseBody
    public ResponseBean repulse(String ids, String msg, HttpSession session) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        String userId = sessionUser.getUserId();
        ResponseBean responseBean = new ResponseBean();
        String[] commentsIds = ids.split(",");
        StringBuilder showMsg = new StringBuilder();
        for (String commentsId : commentsIds) {
            ShyComments comments = commentsService.selectByPrimaryKey(Integer.parseInt(commentsId));
            JSONObject result;
            JSONObject json = new JSONObject();
            json.put("activity_id", comments.getActivity_id());
            json.put("state", 4);
            json.put("attr_tags", "");
            json.put("msg", msg);
            try {
                result = pass.commentPass(json, comments.getCallback());
            } catch (Exception e) {
                logger.error("评论回调失败，activity_id:{}", comments.getActivity_id());
                showMsg.append(",").append(comments.getComment_user_name()).append("的评论拒绝请求失败");
                break;
            }
            if (result == null) {
                logger.error("评论回调错误，且无返回信息。activity_id:{}", comments.getActivity_id());
                showMsg.append(",").append(comments.getComment_user_name()).append("的评论拒绝请求失败");
                break;
            } else {
                Integer code = result.getInteger("errorCode");
                if (code != 0) {
                    logger.error("评论回调错误，activity_id:{}，errorCode:{},errorMessage:{}", comments.getActivity_id(), code, result.getString("errorMessage"));
                    showMsg.append(",").append(comments.getComment_user_name()).append("的评论拒绝请求失败");
                    break;
                }
                comments.setState(2);
                comments.setAuditor_id(userId);
                comments.setMsg(msg);
                int i = commentsService.updateByPrimaryKeySelective(comments);
                if (i != 1) {
                    logger.error("评论回调失败，修改结果为0条activity_id:{}", comments.getActivity_id());
                    showMsg.append(",").append(comments.getComment_user_name()).append("的评论拒绝请求失败");
                }
            }
        }
        if (CommonUtils.isNotEmpty(showMsg.toString())) {
            responseBean.setErrorCode(ErrorEnum.ERROR.getErrorCode());
            responseBean.setErrorMessage(showMsg.toString());
        }
        return responseBean;
    }


    /**
     * 根据条件导出excel
     */
    @GetMapping("excel")
    public void getExcel(CommentsDTO commentsDTO, HttpSession session, HttpServletResponse response) {
        SessionUser sessionUser = (SessionUser) session.getAttribute(SessionUser.getSessionKey());
        Integer spid = sessionUser.getSpid();
        commentsDTO.setSpid(spid);
        String userId = sessionUser.getUserId();
        commentsDTO.setUserId(userId);
        if (spid == 0) {
            commentsDTO.setSpid(null);
        }

        if (CommonUtils.isNotEmpty(commentsDTO.getStart_date())) {
            try {
                commentsDTO.setStart_date(String.valueOf(CommonUtils.parseDate(commentsDTO.getStart_date(), "yyyy-MM-dd")));
            } catch (Exception e) {
            }
        }
        if (CommonUtils.isNotEmpty(commentsDTO.getEnd_date())) {
            try {
                commentsDTO.setEnd_date(String.valueOf(CommonUtils.parseDate(commentsDTO.getEnd_date(), "yyyy-MM-dd")));
            } catch (Exception e) {
            }
        }
        PageInfo<ShyComments> commentPage = commentsService.getPageExcel(commentsDTO);
        List<ShyComments> list = commentPage.getList();
        String fileName = "评论数据";

        XSSFWorkbook wb = new XSSFWorkbook();
        ServletOutputStream out = null;
        BufferedInputStream bis = null;
        BufferedOutputStream bos = null;
        try {
            XSSFSheet sheetlist = wb.createSheet(fileName);
            //行
            XSSFRow row = sheetlist.createRow(0);
            XSSFCell cell = row.createCell(0);//列
            cell.setCellValue("id");
            cell = row.createCell(1);//列
            cell.setCellValue("sp昵称");
            cell = row.createCell(2);//列
            cell.setCellValue("评论用户id");
            cell = row.createCell(3);//列
            cell.setCellValue("评论用户昵称");
            cell = row.createCell(4);//列
            cell.setCellValue("评论内容");
            cell = row.createCell(5);//列
            cell.setCellValue("评论标题");
            cell = row.createCell(6);//列
            cell.setCellValue("评论类型");
            cell = row.createCell(7);//列
            cell.setCellValue("评论层级");
            cell = row.createCell(8);//列
            cell.setCellValue("评论来源");
            cell = row.createCell(9);//列
            cell.setCellValue("评论送审时间");
            cell = row.createCell(10);//列
            cell.setCellValue("审核状态");
            cell = row.createCell(11);//列
            cell.setCellValue("审核人");
            cell = row.createCell(12);//列
            cell.setCellValue("审核意见");

            for (int i = 0; i < list.size(); i++) {
                row = sheetlist.createRow((short) (i + 1));//行
                cell = row.createCell(0);//列
                cell.setCellValue(list.get(i).getId());
                cell = row.createCell(1);//列
                cell.setCellValue(list.get(i).getSpid() == null ? "" : sitesService.findspidDescByspid(list.get(i).getSpid()).getName());
                cell = row.createCell(2);//列
                cell.setCellValue(list.get(i).getComment_user_id());
                cell = row.createCell(3);//列
                cell.setCellValue(list.get(i).getComment_user_name());

                cell = row.createCell(4);//列
                cell.setCellValue(list.get(i).getComment_content());

                cell = row.createCell(5);//列
                cell.setCellValue(list.get(i).getComment_title());



                /*状态需要处理成用户可读*/
                String commonType = "";//评论类型，topic:话题
                String type = list.get(i).getType();
                if (type != null) {
                    if (type.equals(CommonConst.CATEGORY_TOPIC)) {
                        commonType = "话题";
                    }
                } else {
                    commonType = "其他";
                }
                cell = row.createCell(6);//列
                cell.setCellValue(commonType);
                cell = row.createCell(7);//列
                Integer level = list.get(i).getLevel();
                if (level != null) {
                    cell.setCellValue(level);
                }
                /*状态需要处理成用户可读*/
                String platform = "";//评论所属的平台
                if ("a".equals(list.get(i).getPlatform())) {
                    platform = "安卓";
                } else if ("i".equals(list.get(i).getPlatform())) {
                    platform = "ios";
                } else {
                    platform = "未知";
                }
                cell = row.createCell(8);//列
                cell.setCellValue(platform);


                cell = row.createCell(9);//列
                cell.setCellValue(list.get(i).getCreated_at() == null ? "" : CommonUtils.formatDate(list.get(i).getCreated_at(), "yyyy-MM-dd HH:mm:ss"));

                /*状态需要处理成用户可读*/
                //状态 0.未审核,1:已审核,2审核失败
                String shenheStatus = "";
                if (list.get(i).getState() == 1) {
                    shenheStatus = "已审核";
                } else if (list.get(i).getState() == 2) {
                    shenheStatus = "已拒绝";
                } else {
                    shenheStatus = "未审核";
                }
                cell = row.createCell(10);//列
                cell.setCellValue(shenheStatus);

                /* 审核员需要做处理*/
                String updater = "";//审核员名字
                if (!list.get(i).getState().equals(CommonConst.AUDIT_INTT)) {
                    updater = CommonUtils.isEmpty(list.get(i).getUpdater()) || "null".equals(list.get(i).getUpdater()) ? "自动通过" : list.get(i).getUpdater();
                    String auditorId = list.get(i).getAuditor_id();
                    if (CommonUtils.isNotEmpty(auditorId)) {
                        ShyUsers user = new ShyUsers();
                        user.setUser_id(auditorId);
                        user = usersService.select(user);
                        if (CommonUtils.isNotNull(user)) {
                            String email = CommonUtils.isEmpty(user.getEmail()) || "null".equals(user.getEmail()) ? "" : user.getEmail();
                            updater = CommonUtils.isEmpty(user.getUsername()) ? email : user.getUsername();
                        }
                    }
                }
                cell = row.createCell(11);//列
                cell.setCellValue(updater);
                cell = row.createCell(12);//列
                cell.setCellValue(list.get(i).getMsg());

            }

            ByteArrayOutputStream os = new ByteArrayOutputStream();
            wb.write(os);
            byte[] contents = os.toByteArray();
            InputStream is = new ByteArrayInputStream(contents);
            response.setContentType("application/vnd.ms-excel;charset=utf-8");
            response.setHeader("Content-Disposition", "attachment;filename=" + new String((fileName + ".xlsx").getBytes(), "iso-8859-1"));
            out = response.getOutputStream();
            bis = new BufferedInputStream(is);
            bos = new BufferedOutputStream(out);
            byte[] buff = new byte[2048];
            int bytesRead;
            while (-1 != (bytesRead = bis.read(buff, 0, buff.length))) {
                bos.write(buff, 0, bytesRead);
            }
        } catch (final Exception e) {
            e.printStackTrace();
            try {
                if (bis != null) {
                    bis.close();
                }
                if (bos != null) {
                    bos.close();
                }
            } catch (Exception e1) {
                e1.printStackTrace();
            }
        } finally {
            try {
                if (bis != null) {
                    bis.close();
                }
                if (bos != null) {
                    bos.close();
                }
            } catch (Exception e2) {
                e2.printStackTrace();
            }
        }
    }
}