From 991aad650217276548ebbe8854a238244084ecf5 Mon Sep 17 00:00:00 2001
From: Liam <liam@fascinated.cc>
Date: Mon, 29 Apr 2024 07:05:50 +0100
Subject: [PATCH] use new servers not total for the log

---
 .../piaservers/model/PiaServerToken.java      |  4 ++--
 .../fascinated/piaservers/pia/PiaManager.java | 19 +++++++++++--------
 2 files changed, 13 insertions(+), 10 deletions(-)

diff --git a/src/main/java/cc/fascinated/piaservers/model/PiaServerToken.java b/src/main/java/cc/fascinated/piaservers/model/PiaServerToken.java
index 8eb6ded..198c4e8 100644
--- a/src/main/java/cc/fascinated/piaservers/model/PiaServerToken.java
+++ b/src/main/java/cc/fascinated/piaservers/model/PiaServerToken.java
@@ -6,9 +6,9 @@ import lombok.Getter;
 @AllArgsConstructor @Getter
 public class PiaServerToken {
     /**
-     * The hostname for this server.
+     * The ip for this server.
      */
-    private final String hostname;
+    private final String ip;
 
     /**
      * The region this server is in.
diff --git a/src/main/java/cc/fascinated/piaservers/pia/PiaManager.java b/src/main/java/cc/fascinated/piaservers/pia/PiaManager.java
index e61ca40..ffb1c7f 100644
--- a/src/main/java/cc/fascinated/piaservers/pia/PiaManager.java
+++ b/src/main/java/cc/fascinated/piaservers/pia/PiaManager.java
@@ -12,7 +12,6 @@ import org.xbill.DNS.Record;
 import org.xbill.DNS.*;
 
 import java.io.File;
-import java.net.InetAddress;
 import java.net.URI;
 import java.net.http.HttpClient;
 import java.net.http.HttpRequest;
@@ -44,14 +43,14 @@ public class PiaManager {
 
         GitUtils.cloneRepo(); // Clone the repository
 
-        // Update the servers every 5 minutes
+        // Update the servers every 2 minutes
         new Timer().scheduleAtFixedRate(new TimerTask() {
             @Override
             public void run() {
                 updateServers(serversFile); // Update the servers
                 README_PATH = ReadMeManager.updateReadme(); // Update the README.md
             }
-        }, 0, TimeUnit.MINUTES.toMillis(5));
+        }, 0, TimeUnit.MINUTES.toMillis(2));
 
         // Commit the files every hour
         new Timer().scheduleAtFixedRate(new TimerTask() {
@@ -76,20 +75,24 @@ public class PiaManager {
                 toRemove.add(server);
             }
         }
-        SERVERS.removeAll(toRemove); // Remove the servers
+        toRemove.forEach(SERVERS::remove); // Remove the servers
         System.out.printf("Removed %s old servers\n", toRemove.size());
 
+        int newServers = 0;
+
         // Add the new servers to the list
         for (PiaServerToken serverToken : piaDomain) {
-            InetAddress address = InetAddress.getByName(serverToken.getHostname());
-
             // Add the server to the list
-            SERVERS.add(new PiaServer(address.getHostAddress(), serverToken.getRegion(), new Date()));
+            PiaServer server = new PiaServer(serverToken.getIp(), serverToken.getRegion(), new Date());
+            SERVERS.add(server);
+            if (!SERVERS.contains(server)) {
+                newServers++;
+            }
         }
 
         // Save the servers to the file
         Files.writeString(serversFile.toPath(), Main.GSON.toJson(SERVERS));
-        System.out.printf("Wrote %s servers to the file\n", SERVERS.size());
+        System.out.printf("Wrote %s new servers to the file\n", newServers);
     }
 
     @SneakyThrows