From 7079484d428321c9b3db09394577efda4d591a4e Mon Sep 17 00:00:00 2001
From: Luck <git@lucko.me>
Date: Mon, 19 Sep 2022 18:57:02 +0100
Subject: Provide extra metadata about sources in sampler data

---
 .../java/me/lucko/spark/velocity/VelocitySparkPlugin.java  | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

(limited to 'spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java')

diff --git a/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java b/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java
index 7d9ced8..4a89a4e 100644
--- a/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java
+++ b/spark-velocity/src/main/java/me/lucko/spark/velocity/VelocitySparkPlugin.java
@@ -34,11 +34,13 @@ import me.lucko.spark.common.SparkPlatform;
 import me.lucko.spark.common.SparkPlugin;
 import me.lucko.spark.common.monitor.ping.PlayerPingProvider;
 import me.lucko.spark.common.platform.PlatformInfo;
-import me.lucko.spark.common.util.ClassSourceLookup;
+import me.lucko.spark.common.sampler.source.ClassSourceLookup;
+import me.lucko.spark.common.sampler.source.SourceMetadata;
 
 import org.slf4j.Logger;
 
 import java.nio.file.Path;
+import java.util.Collection;
 import java.util.List;
 import java.util.logging.Level;
 import java.util.stream.Stream;
@@ -133,6 +135,16 @@ public class VelocitySparkPlugin implements SparkPlugin, SimpleCommand {
         return new VelocityClassSourceLookup(this.proxy.getPluginManager());
     }
 
+    @Override
+    public Collection<SourceMetadata> getKnownSources() {
+        return SourceMetadata.gather(
+                this.proxy.getPluginManager().getPlugins(),
+                plugin -> plugin.getDescription().getId(),
+                plugin -> plugin.getDescription().getVersion().orElse("unspecified"),
+                plugin -> String.join(", ", plugin.getDescription().getAuthors())
+        );
+    }
+
     @Override
     public PlayerPingProvider createPlayerPingProvider() {
         return new VelocityPlayerPingProvider(this.proxy);
-- 
cgit