Util: add functions for serialising YAML maps.

src/main/java/moe/nekojimi/chords/Chords.java

@@ -52,11 +52,13 @@ public final class Chords extends ListenerAdapter
 
     private final File dataDirectory;
     private final File playlistsDirectory;
+    private final File soundboardFile;
 
     private MusicHandler musicHandler;
     private final Downloader downloader;
     private final Searcher searcher;
     private final QueueManager queueManager;
+    private final Soundboard soundboard;
 //    private Recommender recommender;
     private JDA jda;
 
@@ -141,6 +143,7 @@ public final class Chords extends ListenerAdapter
         // init dirs
         dataDirectory = new File(System.getProperty("user.dir"));
         playlistsDirectory = initDirectory(dataDirectory, "playlists");
+        soundboardFile = new File("soundboard.yml");
 
         // init downloader
         downloader = new Downloader();
@@ -153,6 +156,10 @@ public final class Chords extends ListenerAdapter
         queueManager = new QueueManager();
         queueManager.addSource(downloader);
 
+        // init soundboard
+        soundboard = new Soundboard(this);
+        soundboard.loadYAML(soundboardFile);
+
         // init commands
         addCommand(new JoinCommand(this));
         addCommand(new LeaveCommand(this));
@@ -162,6 +169,7 @@ public final class Chords extends ListenerAdapter
         addCommand(new RemoveCommand(this));
         addCommand(new RestartCommand(this));
         addCommand(new SkipCommand(this));
+        addCommand(new SoundboardCommand(this));
         helpCommand = new HelpCommand(this);
         addCommand(helpCommand);
 
@@ -201,7 +209,7 @@ public final class Chords extends ListenerAdapter
         super.onMessageReceived(event); //To change body of generated methods, choose Tools | Templates.
         Message message = event.getMessage();
         User author = message.getAuthor();
-        String content = message.getContentRaw();
+        String content = message.getContentRaw().trim();
         Guild guild = event.getGuild();
 
         // Ignore message if it's not in a music channel
@@ -220,7 +228,7 @@ public final class Chords extends ListenerAdapter
         {
             try
             {
-                URL parseURL = new URL(content.trim());
+                URL parseURL = new URL(content);
                 invocation = new Invocation(event, List.of(parseURL.toExternalForm()));
                 playCommand.call(invocation);
                 return;
@@ -268,6 +276,10 @@ public final class Chords extends ListenerAdapter
                         }
                     }
                 }
+            } else if (soundboard != null)
+            {
+                // try looking it up with the soundboard
+                soundboard.play(cmd);
             }
         } catch (Exception ex)
         {
@@ -445,6 +457,22 @@ public final class Chords extends ListenerAdapter
         return playlists;
     }
 
+    public Soundboard getSoundboard()
+    {
+        return soundboard;
+    }
+
+    public void saveSoundboard()
+    {
+        try
+        {
+            soundboard.saveYAML(soundboardFile);
+        } catch (IOException ex)
+        {
+
+        }
+    }
+
 //    public Recommender getRecommender()
 //    {
 //        return recommender;

src/main/java/moe/nekojimi/chords/Soundboard.java

@@ -0,0 +1,92 @@
+/*
+ * Copyright (C) 2024 jimj316
+ *
+ * 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 moe.nekojimi.chords;
+
+import com.amihaiemil.eoyaml.Yaml;
+import com.amihaiemil.eoyaml.YamlInput;
+import com.amihaiemil.eoyaml.YamlMapping;
+import com.amihaiemil.eoyaml.YamlPrinter;
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+import java.util.*;
+
+/**
+ *
+ * @author jimj316
+ */
+public class Soundboard
+{
+
+    private final Chords bot;
+    private final Map<String, Track> sounds = new HashMap<>();
+
+    public Soundboard(Chords bot)
+    {
+        this.bot = bot;
+    }
+
+    public void loadYAML(File file) throws IOException
+    {
+        if (!file.exists())
+            return;
+        YamlInput input = Yaml.createYamlInput(file);
+        YamlMapping mapping = input.readYamlMapping();
+        sounds.putAll(Util.yamlMappingToMap(mapping, (t) -> Track.fromYaml(t.asMapping())));
+    }
+
+    public void saveYAML(File file) throws IOException
+    {
+        YamlPrinter printer = Yaml.createYamlPrinter(new FileWriter(file));
+        printer.print(Util.mapToMapping(sounds, (t) -> t.toYaml()));
+    }
+
+    public void add(String emoji, Track t)
+    {
+        t.setKept(true);
+        sounds.put(emoji, t);
+    }
+
+    public boolean remove(String emoji)
+    {
+        if (!sounds.containsKey(emoji))
+            return false;
+        Track t = sounds.get(emoji);
+        t.setKept(false);
+        t.delete();
+        sounds.remove(emoji);
+        return true;
+    }
+
+    public boolean play(String emoji)
+    {
+        if (!sounds.containsKey(emoji))
+            return false;
+        if (bot.getMusicHandler() == null)
+            return false;
+        final Track track = sounds.get(emoji);
+        track.setKept(true);
+        bot.getMusicHandler().play(track);
+        return true;
+    }
+
+    public List<String> getEmoji()
+    {
+        return new ArrayList<>(sounds.keySet());
+    }
+
+}

src/main/java/moe/nekojimi/chords/Track.java

@@ -9,14 +9,13 @@ import com.amihaiemil.eoyaml.Yaml;
 import com.amihaiemil.eoyaml.YamlMapping;
 import com.amihaiemil.eoyaml.YamlSequence;
 import com.amihaiemil.eoyaml.YamlSequenceBuilder;
-import java.io.File;
+import java.io.*;
-import java.io.FileInputStream;
-import java.io.FileNotFoundException;
-import java.io.InputStream;
 import java.net.MalformedURLException;
 import java.net.URL;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.logging.Level;
+import java.util.logging.Logger;
 
 /**
  *
@@ -63,31 +62,39 @@ public class Track implements Comparable<Track>
                 .add("url", url.toExternalForm())
                 .add("location", location.getAbsolutePath())
                 .add("num", Integer.toString(number))
-                .add("formats", build.build())
+                //                .add("formats", build.build())
+                .add("artist", artist)
                 //                .add("requestedBy", requestedBy)
                 //                .add("requestedIn", requestedIn)
                 .add("kept", Boolean.toString(kept))
                 .build();
     }
 
-    public static Track fromYaml(YamlMapping map) throws MalformedURLException
+    public static Track fromYaml(YamlMapping map)
     {
-        Track track = new Track(new URL(map.string("url")), null);
+        try
-        track.setArtist(map.string("artist"));
+        {
-        track.setTitle(map.string("title"));
+            Track track = new Track(new URL(map.string("url")), null);
-        track.setLocation(new File(map.string("location")));
+            track.setArtist(map.string("artist"));
-        track.setNumber(map.integer("num"));
+            track.setTitle(map.string("title"));
-        track.setKept(Boolean.parseBoolean(map.string("kept")));
+            track.setLocation(new File(map.string("location")));
+            track.setNumber(map.integer("num"));
+            track.setKept(Boolean.parseBoolean(map.string("kept")));
 //        track.setRequestedBy(map.string("requestedBy"));
 //        track.setRequestedIn(map.string("requestedIn"));
 
-        List<Format> formats = new ArrayList<>();
+//            List<Format> formats = new ArrayList<>();
-        YamlSequence formatSeq = map.yamlSequence("formats");
+//            YamlSequence formatSeq = map.yamlSequence("formats");
-        for (int i = 0; i < formats.size(); i++)
+//            for (int i = 0; i < formats.size(); i++)
-            formats.add(Format.fromYaml(formatSeq.yamlMapping(i)));
+//                formats.add(Format.fromYaml(formatSeq.yamlMapping(i)));
-        track.setFormats(formats);
+//            track.setFormats(formats);
 
-        return track;
+            return track;
+        } catch (MalformedURLException ex)
+        {
+            Logger.getLogger(Track.class.getName()).log(Level.SEVERE, null, ex);
+            return null;
+        }
     }
 
     public boolean isDownloaded()
@@ -152,6 +159,21 @@ public class Track implements Comparable<Track>
         this.inputStream = inputStream;
     }
 
+    void clearInputStream()
+    {
+        if (inputStream != null)
+        {
+            try
+            {
+                inputStream.close();
+            } catch (IOException ex)
+            {
+                Logger.getLogger(Track.class.getName()).log(Level.SEVERE, null, ex);
+            }
+            inputStream = null;
+        }
+    }
+
     void delete()
     {
         if (location != null)

src/main/java/moe/nekojimi/chords/TrackPlayer.java

@@ -31,13 +31,15 @@ public class TrackPlayer implements Closeable
     private static final int RETRY_DELAY = 100;
 
     private final CircularByteBuffer audioBuffer = new CircularByteBuffer(3840 * 1024);
+    private final Track track;
 
     private final AudioInputStream input;
 
     private boolean arrayErr = false; // supresses ArrayIndexOutOfBoundsException after the first time, to prevent spam
 
-    public TrackPlayer(Track track) throws UnsupportedAudioFileException, IOException
+    public TrackPlayer(Track t) throws UnsupportedAudioFileException, IOException
     {
+        track = t;
         AudioInputStream in = null;
         AudioFormat decodedFormat = null;
         int retry = 0;
@@ -72,11 +74,11 @@ public class TrackPlayer implements Closeable
         fillBuffer(false);
     }
 
-    public TrackPlayer(AudioInputStream input) throws IOException
+//    public TrackPlayer(AudioInputStream input) throws IOException
-    {
+//    {
-        this.input = input;
+//        this.input = input;
-        fillBuffer(false);
+//        fillBuffer(false);
-    }
+//    }
 
     public boolean has(int byteCount)
     {
@@ -168,6 +170,10 @@ public class TrackPlayer implements Closeable
         input.close(); //q
     }
 
+    public Track getTrack()
+    {
+        return track;
+    }
 
 
     public static class OutOfInputException extends RuntimeException

src/main/java/moe/nekojimi/chords/TrackRequest.java

@@ -43,6 +43,7 @@ public class TrackRequest implements Comparable<TrackRequest>
     private String requestedIn;
 
     private double priority = 1.0;
+    private boolean keepTracks = false;
 
     public YamlMapping toYAML()
     {
@@ -78,6 +79,16 @@ public class TrackRequest implements Comparable<TrackRequest>
         requestedIn = invocation.getRequestMessage().getChannel().getId();
     }
 
+    public boolean isKeepTracks()
+    {
+        return keepTracks;
+    }
+
+    public void setKeepTracks(boolean keepTracks)
+    {
+        this.keepTracks = keepTracks;
+    }
+
 //    public Message getRequestMessage()
 //    {
 //        return requestMessage;

src/main/java/moe/nekojimi/chords/Util.java

@@ -1,6 +1,14 @@
 package moe.nekojimi.chords;
 
+import com.amihaiemil.eoyaml.Yaml;
+import com.amihaiemil.eoyaml.YamlMapping;
+import com.amihaiemil.eoyaml.YamlMappingBuilder;
+import com.amihaiemil.eoyaml.YamlNode;
 import java.nio.ByteBuffer;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.function.Function;
 import java.util.regex.Matcher;
 import java.util.regex.Pattern;
 
@@ -31,6 +39,26 @@ public class Util
     private static final Pattern SIZE_PATTERN = Pattern.compile("\\b([0-9]+\\.?[0-9]*)([kkMmGg])i?[bB]\\b");
     private static final Pattern SAMPLE_RATE_PATTERN = Pattern.compile("\\b([0-9]+)k(b(ps?))?\\b");
 
+    public static <V> Map<String, V> yamlMappingToMap(YamlMapping mapping, Function<YamlNode, V> fromYamlFunction)
+    {
+        Map<String, V> ret = new HashMap<>();
+        for (YamlNode key : mapping.keys())
+        {
+            ret.put(key.asScalar().value(), fromYamlFunction.apply(mapping.value(key)));
+        }
+        return ret;
+    }
+
+    public static <K, V> YamlMapping mapToMapping(Map<K, V> map, Function<V, YamlNode> toYamlFunction)
+    {
+        YamlMappingBuilder builder = Yaml.createYamlMappingBuilder();
+        for (Entry<K, V> e : map.entrySet())
+        {
+            builder = builder.add(e.getKey().toString(), toYamlFunction.apply(e.getValue()));
+        }
+        return builder.build();
+    }
+
     public static String printSamples(ByteBuffer buf)
     {
         StringBuilder sb = new StringBuilder();

src/main/java/moe/nekojimi/chords/commands/SoundboardCommand.java

@@ -0,0 +1,98 @@
+/*
+ * Copyright (C) 2024 jimj316
+ *
+ * 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 moe.nekojimi.chords.commands;
+
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.util.List;
+import moe.nekojimi.chords.Chords;
+import moe.nekojimi.chords.TrackRequest;
+
+/**
+ *
+ * @author jimj316
+ */
+public class SoundboardCommand extends Command
+{
+
+    public SoundboardCommand(Chords bot)
+    {
+        super(bot, "board");
+    }
+
+    @Override
+    public void call(Invocation invocation)
+    {
+        String action = invocation.getArgs().isEmpty() ? "" : invocation.getArgs().get(0);
+        String emoji = invocation.getArgs().size() < 2 ? "" : invocation.getArgs().get(1);
+        if (action.equals("add"))
+        {
+            String urlString = invocation.getArgs().get(2);
+            if (emoji.isBlank())
+            {
+                invocation.respond("You must specify a trigger!");
+                return;
+            }
+            try
+            {
+                URL url;
+                if (invocation.getArgs().size() < 3)
+                {
+                    invocation.respond("No URL given!");
+                    return;
+                }
+                url = new URL(urlString);
+                TrackRequest request = new TrackRequest();
+                request.setUrl(url);
+                request.setInvocation(invocation);
+                request.setKeepTracks(true);
+
+                bot.getDownloader().requestSpecific(List.of(request), (t) ->
+                {
+                    bot.getSoundboard().add(emoji, t);
+                    bot.saveSoundboard();
+                });
+            } catch (MalformedURLException ex)
+            {
+                invocation.respond("That's not a valid URL! " + ex.getMessage());
+                return;
+            }
+        } else if (action.equals("remove"))
+        {
+            if (emoji.isBlank())
+            {
+                invocation.respond("You must specify a trigger!");
+                return;
+            }
+            bot.getSoundboard().remove(emoji);
+            bot.saveSoundboard();
+        } else if (action.equals("list"))
+        {
+            List<String> list = bot.getSoundboard().getEmoji();
+            String response = "__Soundboard commands available:__\n";
+            for (String item : list)
+            {
+                response += item + " ";
+            }
+            invocation.respond(response);
+        } else
+        {
+            invocation.respond("Specify \"add\" or \"remove\" after the command!");
+        }
+    }
+
+}