增加发送消息前检测消息内容是否符合要求

core/src/main/java/com/wecloud/im/action/NormalChatAction.java

@@ -12,6 +12,8 @@ import java.util.Date;
 import java.util.HashMap;
 import java.util.HashMap;
 import java.util.List;
 import java.util.List;
 import java.util.Map;
 import java.util.Map;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 import java.util.stream.Collectors;
 import java.util.stream.Collectors;
 
 
 import org.apache.commons.collections4.CollectionUtils;
 import org.apache.commons.collections4.CollectionUtils;
@@ -140,6 +142,11 @@ public class NormalChatAction {
         if (muted(conversation, reqId, imClientSender, membersList, request.getSenderChannel())) {
         if (muted(conversation, reqId, imClientSender, membersList, request.getSenderChannel())) {
             return;
             return;
         }
         }
+        // 消息内容校验
+        if (checkMsg(conversation, reqId, data.getText(), request.getSenderChannel())) {
+            return;
+        }
+
         ImMessageOnlineSend imMessageOnlineSend = assembleImMessageOnlineSend(data, imClientSender, imApplication.getId());
         ImMessageOnlineSend imMessageOnlineSend = assembleImMessageOnlineSend(data, imClientSender, imApplication.getId());
 
 
         // 再给所有人发 todo 需要改成批量
         // 再给所有人发 todo 需要改成批量
@@ -378,9 +385,47 @@ public class NormalChatAction {
             responseModel.setCode(apiResult.getCode());
             responseModel.setCode(apiResult.getCode());
             responseModel.setMsg(apiResult.getMessage());
             responseModel.setMsg(apiResult.getMessage());
             responseModel.setReqId(reqId);
             responseModel.setReqId(reqId);
-            channelSender.sendMsgLocal((NioSocketChannel)channel, responseModel);
+            channelSender.sendMsgLocal((NioSocketChannel) channel, responseModel);
         }
         }
         return result;
         return result;
     }
     }
 
 
+    /**
+     * 是否禁止发送链接
+     * @Author luozh
+     * @Date 2022年04月27日 04:39:04
+     * @Return
+     */
+    private boolean checkMsg(ImConversationQueryVo conversation, String reqId,
+                             String msgContent, Channel channel) {
+        boolean result = false;
+        // 判断是否开启禁止发送链接
+        if (conversation.getForbidSendLink()) {
+            String regex = "((ht|f)tp(s?)\\:\\/\\/)?[0-9a-zA-Z]([-.\\w]*[0-9a-zA-Z])*." +
+                    "(html|htm|shtml|jsp|asp|php|com|cn|net|com.cn|org|vip)+(:(0-9)*)*(\\/?)([a-zA-Z0-9\\-\\.\\?\\," +
+                    "\\'\\/\\\\&%\\+\\$#_=]*)?";
+
+            Pattern pat = Pattern.compile(regex);
+            Matcher mat = pat.matcher(msgContent.trim());
+            result = mat.find();
+        }
+
+        if (result) {
+            log.info("群: {}, 已被禁止发送链接", conversation.getName());
+            // 响应发送方
+            WsResponse<HashMap<String, Long>> responseModel = new WsResponse<>();
+            ApiResult<Boolean> apiResult = ApiResult.result(ApiCode.IS_BE_FORBID_SEND_LINK);
+            responseModel.setCmd(WsResponseCmdEnum.RES.getCmdCode());
+            responseModel.setCode(apiResult.getCode());
+            responseModel.setMsg(apiResult.getMessage());
+            responseModel.setReqId(reqId);
+            channelSender.sendMsgLocal((NioSocketChannel) channel, responseModel);
+        }
+
+        return result;
+
+
+    }
+
+
 }
 }

core/src/main/java/com/wecloud/im/param/ImConversationQueryVo.java

@@ -61,4 +61,16 @@ public class ImConversationQueryVo implements Serializable {
      */
      */
     @ApiModelProperty("禁言开关 1-未禁言 2-禁言")
     @ApiModelProperty("禁言开关 1-未禁言 2-禁言")
     private Integer muted;
     private Integer muted;
+
+    @ApiModelProperty("禁止互加好友")
+    private Boolean forbidAddFriend;
+
+    @ApiModelProperty("禁止发红包")
+    private Boolean forbidSendRedPackets;
+
+    @ApiModelProperty("禁止发图片")
+    private Boolean forbidSendPic;
+
+    @ApiModelProperty("禁止发链接")
+    private Boolean forbidSendLink;
 }
 }

core/src/main/java/com/wecloud/im/vo/ConversationVo.java

@@ -76,4 +76,16 @@ public class ConversationVo implements Serializable {
     @ApiModelProperty("禁言开关 1-未禁言 2-禁言")
     @ApiModelProperty("禁言开关 1-未禁言 2-禁言")
     private Integer beAtCount;
     private Integer beAtCount;
 
 
+    @ApiModelProperty("禁止互加好友")
+    private Boolean forbidAddFriend;
+
+    @ApiModelProperty("禁止发红包")
+    private Boolean forbidSendRedPackets;
+
+    @ApiModelProperty("禁止发图片")
+    private Boolean forbidSendPic;
+
+    @ApiModelProperty("禁止发链接")
+    private Boolean forbidSendLink;
+
 }
 }

core/src/main/resources/mapper/ImConversationMapper.xml

@@ -5,7 +5,9 @@
     <!-- 通用查询结果列 -->
     <!-- 通用查询结果列 -->
     <sql id="Base_Column_List">
     <sql id="Base_Column_List">
         id
         id
-        , create_time, update_time, last_message, member_count, chat_type, fk_appid, creator, name, attributes, system_flag,muted
+        , create_time, update_time, last_message, member_count, chat_type, fk_appid, creator, name, attributes,
+        system_flag,muted, is_forbid_add_friend as forbid_add_friend, is_forbid_send_red_packets as forbid_send_red_packets,
+        is_forbid_send_pic as forbid_send_pic, is_forbid_send_link as forbid_send_link
     </sql>
     </sql>
 
 
     <select id="getImConversationById" resultType="com.wecloud.im.param.ImConversationQueryVo">
     <select id="getImConversationById" resultType="com.wecloud.im.param.ImConversationQueryVo">
@@ -30,6 +32,10 @@
         imConversation.attributes as attribute,
         imConversation.attributes as attribute,
         imConversation.system_flag,
         imConversation.system_flag,
         imConversation.muted as muted,
         imConversation.muted as muted,
+        imConversation.is_forbid_add_friend as forbid_add_friend,
+        imConversation.is_forbid_send_red_packets as forbid_send_red_packets,
+        imConversation.is_forbid_send_pic as forbid_send_pic,
+        imConversation.is_forbid_send_link as forbid_send_link,
         im_client.client_id AS creator,
         im_client.client_id AS creator,
         (SELECT COUNT(im_inbox.id)
         (SELECT COUNT(im_inbox.id)
         FROM im_inbox
         FROM im_inbox

framework/src/main/java/io/geekidea/springbootplus/framework/common/api/ApiCode.java

@@ -142,6 +142,11 @@ public enum ApiCode {
      */
      */
     IS_BE_DISBAND(6016, "api.response.code.IS_BE_DISBAND"),
     IS_BE_DISBAND(6016, "api.response.code.IS_BE_DISBAND"),
 
 
+    /**
+     * 群已禁止发链接
+     */
+    IS_BE_FORBID_SEND_LINK(6017, "api.response.code.IS_BE_FORBID_SEND_LINK"),
+
     ;
     ;
 
 
     private final int code;
     private final int code;