fix: 图片解析

server/src/main/java/com/pica/cloud/online/exam/analysis/server/service/AntiSpamService.java

@@ -9,4 +9,7 @@ import com.pica.cloud.foundation.entity.PicaResponse;
 public interface AntiSpamService {
     PicaResponse processString(String content);
 
+    boolean processImageContent(String content);
+
+    boolean processImage(String url);
 }

server/src/main/java/com/pica/cloud/online/exam/analysis/server/service/impl/AntiSpamServiceImpl.java

@@ -17,9 +17,8 @@ import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 
 import java.io.UnsupportedEncodingException;
-import java.util.HashMap;
-import java.util.Map;
-import java.util.Random;
+import java.util.*;
+import java.util.regex.Pattern;
 
 /**
  * @author wuminghao
@@ -31,9 +30,16 @@ public class AntiSpamServiceImpl implements AntiSpamService {
     private PropertiesConfiguration properties;
 
     private static HttpClient httpClient = HttpClient4Utils.createHttpClient(100, 20, 2000, 2000, 2000);
-
+    private static HttpClient httpClientImage = HttpClient4Utils.createHttpClient(100, 20, 10000, 2000, 2000);
     @Override
     public PicaResponse processString(String content) {
+        /**
+         * 过滤图片
+         */
+        if (!processImageContent(content)) {
+            return ReturnUtil.getPicaResponse("400002", "图片包含违禁内容，请修改后重新输入");
+        }
+
         Map<String, String> params = new HashMap<String, String>();
 
         /**
@@ -96,9 +102,6 @@ public class AntiSpamServiceImpl implements AntiSpamService {
             if (action == 0) {
 
             } else {
-                /**
-                 * TODO: 可以进一步处理 显示信息被过滤原因
-                 */
                 for (JsonElement labelElement : labelArray) {
                     JsonObject lObject = labelElement.getAsJsonObject();
                     int label = lObject.get("label").getAsInt();
@@ -124,4 +127,130 @@ public class AntiSpamServiceImpl implements AntiSpamService {
 
         return ReturnUtil.getPicaResponse(PicaResultCode.SUCCESS, "");
     }
+
+    @Override
+    public boolean processImageContent(String content) {
+        List<String> imageUrlList = new ArrayList<>();
+
+        /**
+         * 如果不包含图片,直接返回
+         */
+        if (!content.contains("<imgsrc=")) {
+            return true;
+        }
+        int index = -1;
+        int end = -1;
+        while (-1 != (index = content.indexOf("<imgsrc=\\\""))) {
+            end = content.substring(index, content.length()).indexOf("\\\">");
+            imageUrlList.add(content.substring(index,index + end).replace("<imgsrc=\\\"", ""));
+            content = content.replaceFirst("(<imgsrc[\\s\\S]*?>)", "");
+        }
+
+        for (String url : imageUrlList) {
+            /**
+             * 检测出某张图片失败 则返回
+             */
+            if (!processImage(url)) {
+                return false;
+            }
+        }
+        return true;
+    }
+
+    @Override
+    public boolean processImage(String url) {
+        Map<String, String> params = new HashMap<String, String>();
+
+        /**
+         * 1. 设置公共参数
+         */
+        params.put("secretId", properties.getNeteaseAntispamSecretId());
+        params.put("businessId", properties.getNeteaseAntispamImageBusinessId());
+        params.put("version", "v3.2");
+        params.put("timestamp", String.valueOf(System.currentTimeMillis()));
+        params.put("nonce", String.valueOf(new Random().nextInt()));
+
+        /**
+         * 2. 设置私有参数
+         */
+        JsonArray jsonArray = new JsonArray();
+        JsonObject image = new JsonObject();
+        image.addProperty("name", url);
+        image.addProperty("type", 1);
+        image.addProperty("data", url);
+        jsonArray.add(image);
+
+        params.put("images", jsonArray.toString());
+        params.put("account", "picahealth@picahealth.com");
+
+        /**
+         * 3. 生成签名信息
+         */
+        try {
+            String signature = SignatureUtils.genSignature(properties.getNeteaseAntispamSecretKey(), params);
+            params.put("signature", signature);
+        } catch (UnsupportedEncodingException e) {
+            e.printStackTrace();
+        }
+
+        /**
+         * 4. 发送HTTP请求
+         */
+        String response = HttpClient4Utils.sendPost(httpClientImage, properties.getNeteaseAntispamImageApiUrl(), params, Consts.UTF_8);
+
+        /**
+         * 5. 解析返回值
+         */
+        JsonObject resultObject = new JsonParser().parse(response).getAsJsonObject();
+        int code = resultObject.get("code").getAsInt();
+        String msg = resultObject.get("msg").getAsString();
+        if (code == 200) {
+            JsonArray resultArray = resultObject.getAsJsonArray("result");
+            for (JsonElement jsonElement : resultArray) {
+                JsonObject jObject = jsonElement.getAsJsonObject();
+                String name = jObject.get("name").getAsString();
+                int status = jObject.get("status").getAsInt();
+                String taskId = jObject.get("taskId").getAsString();
+                JsonArray labelArray = jObject.get("labels").getAsJsonArray();
+                System.out.println(String.format("taskId=%s，status=%s，name=%s，labels：", taskId, status, name));
+
+                /**
+                 * 检测失败 0：成功  610：图片下载失败 620：图片格式错误 630：其他
+                 */
+                if (status != 0) {
+                    continue;
+                }
+
+                int maxLevel = -1;
+                // 产品需根据自身需求，自行解析处理，本示例只是简单判断分类级别
+                for (JsonElement labelElement : labelArray) {
+                    JsonObject lObject = labelElement.getAsJsonObject();
+                    int label = lObject.get("label").getAsInt();
+                    int level = lObject.get("level").getAsInt();
+                    double rate = lObject.get("rate").getAsDouble();
+                    if (label == 100) {
+                        System.out.println(String.format("label:%s, level=%s, rate=%s", label, level, rate));
+                        maxLevel = level > maxLevel ? level : maxLevel;
+                    }
+                }
+                switch (maxLevel) {
+                    case 0:
+                        System.out.println("#图片机器检测结果：最高等级为\"正常\"\n");
+                        break;
+                    case 1:
+                        System.out.println("#图片机器检测结果：最高等级为\"嫌疑\"\n");
+                        break;
+                    case 2:
+                        System.out.println("#图片机器检测结果：最高等级为\"确定\"\n");
+                        break;
+                    default:
+                        break;
+                }
+            }
+        } else {
+            System.out.println(String.format("ERROR: code=%s, msg=%s", code, msg));
+        }
+
+        return true;
+    }
 }