Replace username cache with squirrelid cache

Author: md5sha256

realty-paper/build.gradle.kts

@@ -14,6 +14,8 @@ dependencies {
     compileOnly("com.sk89q.worldguard:worldguard-bukkit:7.0.15") {
         exclude(group = "org.bukkit", module = "bukkit")
     }
+    implementation("org.enginehub:squirrelid:0.3.2")
+    implementation("org.xerial:sqlite-jdbc:3.46.1.0")
     compileOnly("net.essentialsx:EssentialsX:2.21.2") {
         exclude(group = "org.bukkit", module = "bukkit")
         exclude(group = "org.spigotmc", module = "spigot-api")
@@ -54,6 +56,9 @@ tasks {
         relocate("org.intellij.lang", "${base}.org.intellij.lang")
         relocate("net.kyori.option", "${base}.net.kyori.option")
         relocate("org.incendo.cloud", "${base}.org.incendo.cloud")
+        relocate("org.enginehub.squirrelid", "${base}.org.enginehub.squirrelid")
+        relocate("org.sqlite", "${base}.org.sqlite")
+        mergeServiceFiles()
     }
 
     processResources {

realty-paper/src/main/java/io/github/md5sha256/realty/Realty.java

@@ -69,8 +69,8 @@
 import io.github.md5sha256.realty.util.DateFormatter;
 import io.github.md5sha256.realty.util.EssentialsNotificationService;
 import io.github.md5sha256.realty.util.EssentialsSafeBlockPredicate;
-import io.github.md5sha256.realty.util.PlayerNameCache;
 import io.github.md5sha256.realty.util.SimpleDateFormatSerializer;
+import io.github.md5sha256.realty.util.SquirrelIdUsernameResolver;
 import io.github.md5sha256.realty.util.TransientNotificationService;
 import io.papermc.paper.util.Tick;
 import net.kyori.adventure.text.Component;
@@ -124,7 +124,7 @@ public final class Realty extends JavaPlugin {
     private final AtomicReference<RealtyTags> realtyTags = new AtomicReference<>();
     private final RegionProfileService regionProfileService = new RegionProfileService(getLogger());
     private final SignCache signCache = new SignCache();
-    private final PlayerNameCache nameCache = new PlayerNameCache(getServer());
+    private SquirrelIdUsernameResolver nameResolver;
     private ExecutorState executorState;
     private RealtyBackend logic;
     private ProfileApplicator profileApplicator;
@@ -209,6 +209,16 @@ public void onEnable() {
         };
         this.executorState = new ExecutorState(getServer().getScheduler()
                 .getMainThreadExecutor(this), Executors.newFixedThreadPool(4, threadFactory));
+        try {
+            this.nameResolver = new SquirrelIdUsernameResolver(
+                    new File(getDataFolder(), "profiles.sqlite"),
+                    this.executorState.dbExec());
+        } catch (IOException ex) {
+            getLogger().severe("Failed to initialize profile cache!");
+            ex.printStackTrace();
+            getServer().getPluginManager().disablePlugin(this);
+            return;
+        }
         MariaDatabase mariaDatabase = new MariaDatabase(this.databaseSettings, getLogger());
         this.database = mariaDatabase;
         try {
@@ -220,7 +230,7 @@ public void onEnable() {
             return;
         }
         this.logic = new RealtyBackendImpl(mariaDatabase,
-                this.nameCache::getUsername,
+                this.nameResolver::getUsername,
                 dateTime -> DateFormatter.format(this.settings.get(), dateTime),
                 () -> this.settings.get().offerPaymentDurationSeconds());
         var economyProvider = getServer().getServicesManager().getRegistration(Economy.class);

realty-paper/src/main/java/io/github/md5sha256/realty/util/PlayerNameCache.java

@@ -0,0 +1,58 @@
+package io.github.md5sha256.realty.util;
+
+import org.enginehub.squirrelid.Profile;
+import org.enginehub.squirrelid.cache.ProfileCache;
+import org.enginehub.squirrelid.cache.SQLiteCache;
+import org.enginehub.squirrelid.resolver.CacheForwardingService;
+import org.enginehub.squirrelid.resolver.CombinedProfileService;
+import org.enginehub.squirrelid.resolver.HttpRepositoryService;
+import org.enginehub.squirrelid.resolver.PaperPlayerService;
+import org.enginehub.squirrelid.resolver.ProfileService;
+import org.jetbrains.annotations.NotNull;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.Optional;
+import java.util.UUID;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.Executor;
+
+public class SquirrelIdUsernameResolver {
+
+    private final ProfileCache cache;
+    private final ProfileService service;
+    private final Executor asyncExecutor;
+
+    public SquirrelIdUsernameResolver(@NotNull File cacheFile,
+                                      @NotNull Executor asyncExecutor) throws IOException {
+        this.cache = new SQLiteCache(cacheFile);
+        this.service = new CacheForwardingService(
+                new CombinedProfileService(
+                        PaperPlayerService.getInstance(),
+                        HttpRepositoryService.forMinecraft()),
+                this.cache);
+        this.asyncExecutor = asyncExecutor;
+    }
+
+    @NotNull
+    public Optional<String> getUsernameIfCached(@NotNull UUID uuid) {
+        Profile profile = this.cache.getIfPresent(uuid);
+        return Optional.ofNullable(profile).map(Profile::getName);
+    }
+
+    @NotNull
+    public CompletableFuture<String> getUsername(@NotNull UUID uuid) {
+        return CompletableFuture.supplyAsync(() -> {
+            try {
+                Profile profile = this.service.findByUuid(uuid);
+                if (profile != null && profile.getName() != null) {
+                    return profile.getName();
+                }
+                return uuid.toString();
+            } catch (Exception ex) {
+                return uuid.toString();
+            }
+        }, this.asyncExecutor);
+    }
+
+}