Add save_uri to tracks, might be usefull

DuncteBot · Nov 26, 2023 · fbfa6d3 · fbfa6d3
1 parent 7f035e2
commit fbfa6d3
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/plugin/src/main/java/com/dunctebot/lavalinkplugin/DuncteBotPluginInfoModifier.java b/plugin/src/main/java/com/dunctebot/lavalinkplugin/DuncteBotPluginInfoModifier.java
@@ -0,0 +1,31 @@
+package com.dunctebot.lavalinkplugin;
+
+import com.dunctebot.sourcemanagers.IWillUseIdentifierInstead;
+import com.sedmelluq.discord.lavaplayer.track.AudioTrack;
+import dev.arbjerg.lavalink.api.AudioPluginInfoModifier;
+import kotlinx.serialization.json.JsonElementKt;
+import kotlinx.serialization.json.JsonObject;
+import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
+import org.springframework.stereotype.Service;
+
+import java.util.Map;
+
+@Service
+public class DuncteBotPluginInfoModifier implements AudioPluginInfoModifier {
+    @Nullable
+    @Override
+    public JsonObject modifyAudioTrackPluginInfo(@NotNull AudioTrack track) {
+        final String uri;
+
+        if (track instanceof IWillUseIdentifierInstead) {
+            uri = track.getInfo().identifier;
+        } else {
+            uri = track.getInfo().uri;
+        }
+
+        return new JsonObject(Map.of(
+                "save_uri", JsonElementKt.JsonPrimitive(uri)
+        ));
+    }
+}