1 files changed, 190 insertions, 0 deletions
diff --git a/src/main/java/tlIItools/NameFileReader.java b/src/main/java/tlIItools/NameFileReader.java
new file mode 100644
index 00000000..ad2a83b4
--- /dev/null
+++ b/src/main/java/tlIItools/NameFileReader.java
@@ -0,0 +1,190 @@
+package tlIItools;
+
+import java.io.FileReader;
+import java.io.IOException;
+import java.io.PrintStream;
+import java.io.Reader;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+import java.util.Scanner;
+
+/**
+ * Reads in a list of file names to process.
+ *
+ * @author Ben Culkin
+ */
+public class NameFileReader {
+	/**
+	 * Are we attempting to guess group names?
+	 */
+	public boolean guessGroups;
+
+	/**
+	 * Regex to use for guessing group names.
+	 */
+	public String groupRx;
+
+	/**
+	 * The default group to put files into.
+	 */
+	public String defGroup;
+
+	/**
+	 * The map of file groups.
+	 */
+	public Map<String, List<String>> fNames;
+
+	/*
+	 * The current group.
+	 */
+	private String curGroup;
+	/*
+	 * The list of files for the current group.
+	 */
+	private	List<String> curList;
+
+	/**
+	 * Counts the files read in.
+	 */
+	public int fCount;
+
+	public PrintStream normOut = System.out;
+	public PrintStream errOut  = System.err;
+
+	/**
+	 * Create a new name reader using the default settings.
+	 *
+	 * Guessing groups is disabled by default.
+	 */
+	public NameFileReader() {
+		this(false);
+	}
+
+	/**
+	 * Create a new name reader using the default settings.
+	 *
+	 * @param guessGroups
+	 * 	Controls whether or not to try to guess file groups from file
+	 * 	names.
+	 */
+	public NameFileReader(boolean guessGroups) {
+		this(new HashMap<>(), "default", guessGroups);
+	}
+
+	/**
+	 * Create a new name reader using the specified settings.
+	 *
+	 * @param fNames
+	 * 	The set of groups to add files to.
+	 *
+	 * @param defGroup
+	 * 	The name of the default file group.
+	 *
+	 * @param guessGroups
+	 * 	Whether or not to attempt to guess group names.
+	 */
+	public NameFileReader(Map<String, List<String>> fNames, String defGroup, boolean guessGroups) {
+		if (!fNames.containsKey(defGroup))
+			fNames.put(defGroup, new ArrayList<>());
+
+		this.fNames = fNames;
+
+		this.defGroup = defGroup;
+
+		this.guessGroups = guessGroups;
+
+		this.curGroup = defGroup;
+		this.curList  = fNames.get(curGroup);
+
+		this.fCount = 0;
+	}
+
+	/**
+	 * Read in file names from a file.
+	 *
+	 * @param from
+	 * 	The name of the file to read from.
+	 *
+	 * @return The number of files read.
+	 */
+	public void readFrom(String from) {
+		int ret;
+
+		try (FileReader fr = new FileReader(from)) {
+			readFrom(fr);
+		} catch (IOException ioex) {
+			errOut.printf("Error reading names from file %s\n", from);
+			ioex.printStackTrace(errOut);
+			errOut.println();
+		}
+	}
+
+	/**
+	 * Read in file names from an input source.
+	 *
+	 * @param r
+	 * 	The input source to read from.
+	 *
+	 * @return The number of files read.
+	 */
+	public void readFrom(Reader r) {
+		int numFiles = 0;
+
+		Scanner scn = new Scanner(r);
+
+		while (scn.hasNextLine()) {
+			String ln = scn.nextLine();
+
+			boolean skipAdd = false;
+
+			if (ln.startsWith("#")) {
+				swapGroup(ln.substring(1));
+
+				skipAdd = true;
+			} else if (guessGroups && ln.contains("/mods/")) {
+				swapGroup(ln.replaceAll(groupRx, "$1"));
+			}
+
+			if (!skipAdd) {
+				fCount += 1;
+
+				curList.add(ln);
+			}
+
+			skipAdd = false;
+		}
+	}
+
+	/**
+	 * Swap to a new file group.
+	 *
+	 * @param groupName
+	 * 	The name of the group to swap to.
+	 */
+	public void swapGroup(String groupName) {
+		curGroup = groupName;
+
+		if (!fNames.containsKey(curGroup)) {
+			curList = new ArrayList<>();
+
+			fNames.put(curGroup, curList);
+		} else {
+			curList  = fNames.get(curGroup);
+		}
+	}
+	
+	public void addFile(String fName) {
+		curList.add(fName);
+
+		fCount += 1;
+	}
+
+	public void addFile(String groupName, String fName) {
+		fNames.computeIfAbsent(groupName, (key) -> new ArrayList<>()).add(fName);
+
+		fCount += 1;
+	}
+}