From 963d34cd3122283f88c71686bb3e5d6969b86eb0 Mon Sep 17 00:00:00 2001
From: 44trent3 <trenten102579@gmail.com>
Date: Wed, 16 Feb 2022 15:33:54 -0600
Subject: Add class source lookups for Sponge7 (#179)

---
 .../spark/sponge/Sponge7ClassSourceLookup.java     | 44 ++++++++++++++++++++++
 .../me/lucko/spark/sponge/Sponge7SparkPlugin.java  |  6 +++
 2 files changed, 50 insertions(+)
 create mode 100644 spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7ClassSourceLookup.java

(limited to 'spark-sponge7/src/main/java/me/lucko')

diff --git a/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7ClassSourceLookup.java b/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7ClassSourceLookup.java
new file mode 100644
index 0000000..90f3b8f
--- /dev/null
+++ b/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7ClassSourceLookup.java
@@ -0,0 +1,44 @@
+/*
+ * This file is part of spark.
+ *
+ *  Copyright (c) lucko (Luck) <luck@lucko.me>
+ *  Copyright (c) contributors
+ *
+ *  This program is free software: you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation, either version 3 of the License, or
+ *  (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+package me.lucko.spark.sponge;
+
+import me.lucko.spark.common.util.ClassSourceLookup;
+
+import org.spongepowered.api.Game;
+
+import java.nio.file.Path;
+
+public class Sponge7ClassSourceLookup extends ClassSourceLookup.ByCodeSource {
+    private final Path modsDirectory;
+
+    public Sponge7ClassSourceLookup(Game game) {
+        this.modsDirectory = game.getGameDirectory().resolve("mods").toAbsolutePath().normalize();
+    }
+
+    @Override
+    public String identifyFile(Path path) {
+        if (!path.startsWith(this.modsDirectory)) {
+            return null;
+        }
+
+        return super.identifyFileName(this.modsDirectory.relativize(path).toString());
+    }
+}
diff --git a/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7SparkPlugin.java b/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7SparkPlugin.java
index 1aed04a..dc34949 100644
--- a/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7SparkPlugin.java
+++ b/spark-sponge7/src/main/java/me/lucko/spark/sponge/Sponge7SparkPlugin.java
@@ -30,6 +30,7 @@ import me.lucko.spark.common.platform.PlatformInfo;
 import me.lucko.spark.common.sampler.ThreadDumper;
 import me.lucko.spark.common.tick.TickHook;
 
+import me.lucko.spark.common.util.ClassSourceLookup;
 import org.slf4j.Logger;
 import org.spongepowered.api.Game;
 import org.spongepowered.api.command.CommandCallable;
@@ -145,6 +146,11 @@ public class Sponge7SparkPlugin implements SparkPlugin {
         return new Sponge7TickHook(this);
     }
 
+    @Override
+    public ClassSourceLookup createClassSourceLookup() {
+        return new Sponge7ClassSourceLookup(this.game);
+    }
+
     @Override
     public PlayerPingProvider createPlayerPingProvider() {
         if (this.game.isServerAvailable()) {
-- 
cgit