fix(handler): check http status code

DreamOneX · Aug 1, 2022 · be8bdf8 · be8bdf8
1 parent db428ff
commit be8bdf8
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/main/java/com/github/dreamonex/mirai/httpcat/handlers/MessageHandler.java b/src/main/java/com/github/dreamonex/mirai/httpcat/handlers/MessageHandler.java
@@ -54,6 +54,13 @@ public void run() {
                         InputStream stream;
                         try (Response response = client.newCall(request).execute()) {
                             stream = response.body().byteStream();
+                            if (response.code() != 404) {
+                                target.sendMessage("404 Not Found");
+                                return;
+                            } else if (response.code() != 200) {
+                                target.sendMessage("奇怪的错误，检查你的url");
+                                return;
+                            }
                             ExternalResource res = ExternalResource
                                                    .create(stream);
                             Image image = target.uploadImage(res);