Merge pull request #8 from bwaggone/feature/add-audio-on-join

Feature/add audio on join

1 files changed, 73 insertions, 2 deletions

diff --git a/src/main/java/soundchan/LocalAudioManager.java b/src/main/java/soundchan/LocalAudioManager.java
index 9a4e299..e645fe7 100644
--- a/src/main/java/soundchan/LocalAudioManager.java
+++ b/src/main/java/soundchan/LocalAudioManager.java
@@ -3,6 +3,9 @@ package soundchan;
 import net.dv8tion.jda.core.entities.MessageChannel;
 
 import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
 import java.util.*;
 
 public class LocalAudioManager {
@@ -12,6 +15,7 @@ public class LocalAudioManager {
      */
 
     public Map<String, String> filenameDict;
+    public Map<String, String> usernameDict;
     private String filepath;
 
     public LocalAudioManager(String filepath_in){
@@ -20,13 +24,41 @@ public class LocalAudioManager {
         PopulateFiles();
     }
 
+    /**
+     * Constructor for when there is a file listing users and sounds to play for them
+     * @param filepath_in Path to folder where sounds are located
+     * @param userSoundFile Path to file with users and sounds
+     */
+    public LocalAudioManager(String filepath_in, String userSoundFile) {
+        filepath = filepath_in;
+        filenameDict = new HashMap<>();
+        usernameDict = new HashMap<>();
+        PopulateFiles();
+        MapUserAudio(userSoundFile);
+    }
+
+    /**
+     * Gives filepath to sound to play either from a command or when given a username
+     * @param command Command or username to play soundbite
+     * @return Path to sound, or "" if no sound for given command
+     */
     public String GetFilePath(String command){
+        String path;
         try{
-            return filepath + "/" + filenameDict.get(command);
+            path = filepath + "/" + filenameDict.get(command);
         }catch(Exception ex){
             System.out.println("File " + command + " not found!");
+            path = "";
+        }
+        if(path.contentEquals("") || path.contentEquals(filepath + "/null")) {
+            try {
+                path = filepath + "/" + usernameDict.get(command);
+            } catch (Exception ex) {
+                System.out.println("File " + command + " not found!");
+                path = "";
+            }
         }
-        return "";
+        return path;
     }
 
     public void ListSounds(MessageChannel channel){
@@ -52,5 +84,44 @@ public class LocalAudioManager {
         }
     }
 
+    /**
+     * Reads in users and their respective sounds from file, then builds a map of users to the filenames. This assumes
+     * filenames for the sounds are valid, but doesn't check for them.
+     * @param userSoundFile The file (with path if required) with listing of users and the sounds to play when they join
+     */
+    private void MapUserAudio(String userSoundFile) {
+        Properties userSoundProp = LoadProperties(userSoundFile);
+        Set<String> users = userSoundProp.stringPropertyNames();
+        for(String user : users) {
+            String soundFile = userSoundProp.getProperty(user);
+            usernameDict.put(user, soundFile);
+        }
+
+    }
+
+    /**
+     * Builds a property object from a file.
+     * @param filename File to be read
+     * @return Property object with information from file
+     */
+    private static Properties LoadProperties(String filename){
+        Properties properties = new Properties();
+        InputStream input = null;
+        try{
+            input = new FileInputStream(filename);
+            properties.load(input);
+
+        }catch (IOException ex){
+            ex.printStackTrace();
+        } finally {
+            try {
+                input.close();
+            } catch (IOException ex) {
+                ex.printStackTrace();
+            }
+        }
+        return properties;
+    }
+
 
 }