a little code optimization

1 files changed, 5 insertions, 11 deletions

diff --git a/src/main/java/files/Parser.java b/src/main/java/files/Parser.java
index 8d06078..75faa00 100644
--- a/src/main/java/files/Parser.java
+++ b/src/main/java/files/Parser.java
@@ -8,6 +8,7 @@ import org.jsoup.select.Elements;
 import java.io.IOException;
 import java.net.URL;
 import java.util.ArrayList;
+import java.util.Objects;
 
 public class Parser {
     public static Document getPage(String url) throws IOException {
@@ -99,18 +100,11 @@ public class Parser {
 
         ArrayList<String> images = new ArrayList<>();
         for (Element image : imagesTemp) {
-            String text = image.toString();
-            int start = text.indexOf("data-src=\"") + 11;
-            int end = text.indexOf(".jpg") + 4;
-            if (end == 3) {
-                end = text.indexOf(".png") + 4;
-            }
-            char[] dst = new char[end - start];
-            text.getChars(start, end, dst, 0);
-            if (!String.valueOf(dst).contains("statics")) {
-                images.add("https://online.anidub.com/" + String.valueOf(dst));
+            String text = Objects.requireNonNull(image.select("img").first()).dataset().get("src");
+            if (!text.contains("statics")) {
+                images.add("https://online.anidub.com/" + text);
             } else {
-                images.add("h" + String.valueOf(dst));
+                images.add(text);
             }
         }