Move MapleIdRetriever to main module

src/main/java/tools/mapletools/IdRetriever.java

@@ -0,0 +1,192 @@
+package tools.mapletools;
+
+import java.io.*;
+import java.nio.charset.StandardCharsets;
+import java.sql.Connection;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.ArrayList;
+
+/**
+ * @author RonanLana
+ * <p>
+ * This application acts two-way: first section sets up a table on the SQL Server with all the names used within MapleStory,
+ * and the second queries all the names placed inside "fetch.txt", returning in the same line order the ids of the elements.
+ * In case of multiple entries with the same name, multiple ids will be returned in the same line split by a simple space
+ * in ascending order. An empty line means that no entry with the given name in a line has been found.
+ * <p>
+ * IMPORTANT: this will fail for fetching MAP ID (you shouldn't be using this program for these, just checking them up in the
+ * handbook is enough anyway).
+ * <p>
+ * Set whether you are first installing the handbook on the SQL Server (TRUE) or just fetching whatever is on your "fetch_ids.txt"
+ * file (FALSE) on the INSTALL_SQLTABLE property and build the project. With all done, run the Java executable.
+ * <p>
+ * Expected installing time: 30 minutes
+ */
+public class IdRetriever {
+    private static final boolean INSTALL_SQLTABLE = true;
+    private static final File INPUT_FILE = ToolConstants.getInputFile("fetch_ids.txt");
+    private static final File OUTPUT_FILE = ToolConstants.getOutputFile("fetched_ids.txt");
+    private static final Connection con = SimpleDatabaseConnection.getConnection();
+
+    private static InputStreamReader fileReader = null;
+    private static BufferedReader bufferedReader = null;
+
+    private static void listFiles(String directoryName, ArrayList<File> files) {
+        File directory = new File(directoryName);
+
+        // get all the files from a directory
+        File[] fList = directory.listFiles();
+        for (File file : fList) {
+            if (file.isFile()) {
+                files.add(file);
+            } else if (file.isDirectory()) {
+                listFiles(file.getAbsolutePath(), files);
+            }
+        }
+    }
+
+    private static void parseMapleHandbookLine(String line) throws SQLException {
+        String[] tokens = line.split(" - ", 3);
+
+        if (tokens.length > 1) {
+            PreparedStatement ps = con.prepareStatement("INSERT INTO `handbook` (`id`, `name`, `description`) VALUES (?, ?, ?)");
+            try {
+                ps.setInt(1, Integer.parseInt(tokens[0]));
+            } catch (NumberFormatException npe) {   // odd...
+                String num = tokens[0].substring(1);
+                ps.setInt(1, Integer.parseInt(num));
+            }
+            ps.setString(2, tokens[1]);
+            ps.setString(3, tokens.length > 2 ? tokens[2] : "");
+            ps.execute();
+
+            ps.close();
+        }
+    }
+
+    private static void parseMapleHandbookFile(File fileObj) throws SQLException {
+        if (shouldSkipParsingFile(fileObj.getName())) {
+            return;
+        }
+
+        String line;
+
+        try {
+            fileReader = new InputStreamReader(new FileInputStream(fileObj), StandardCharsets.UTF_8);
+            bufferedReader = new BufferedReader(fileReader);
+
+            System.out.println("Parsing file '" + fileObj.getCanonicalPath() + "'.");
+
+            while ((line = bufferedReader.readLine()) != null) {
+                try {
+                    parseMapleHandbookLine(line);
+                } catch (SQLException e) {
+                    System.err.println("Failed to parse line: " + line);
+                    throw e;
+                }
+            }
+
+            bufferedReader.close();
+            fileReader.close();
+        } catch (IOException ex) {
+            System.out.println(ex.getMessage());
+        }
+    }
+
+    // Quest.txt has different formatting: id is last token on the line, instead of the first
+    private static boolean shouldSkipParsingFile(String fileName) {
+        return "Quest.txt".equals(fileName);
+    }
+
+    private static void setupSqlTable() throws SQLException {
+        PreparedStatement ps = con.prepareStatement("DROP TABLE IF EXISTS `handbook`;");
+        ps.execute();
+        ps.close();
+
+        ps = con.prepareStatement("CREATE TABLE `handbook` ("
+                + "`key` int(10) unsigned NOT NULL AUTO_INCREMENT,"
+                + "`id` int(10) DEFAULT NULL,"
+                + "`name` varchar(200) DEFAULT NULL,"
+                + "`description` varchar(1000) DEFAULT '',"
+                + "PRIMARY KEY (`key`)"
+                + ");");
+        ps.execute();
+        ps.close();
+    }
+
+    private static void parseMapleHandbook() throws SQLException {
+        ArrayList<File> files = new ArrayList<>();
+
+        listFiles(ToolConstants.HANDBOOK_PATH, files);
+        if (files.isEmpty()) {
+            return;
+        }
+
+        setupSqlTable();
+
+        for (File f : files) {
+            parseMapleHandbookFile(f);
+        }
+    }
+
+    private static void fetchDataOnMapleHandbook() throws SQLException {
+        String line;
+
+        try {
+            fileReader = new InputStreamReader(new FileInputStream(INPUT_FILE), StandardCharsets.UTF_8);
+            bufferedReader = new BufferedReader(fileReader);
+
+            PrintWriter printWriter = new PrintWriter(OUTPUT_FILE, StandardCharsets.UTF_8);
+
+            while ((line = bufferedReader.readLine()) != null) {
+                if (line.isEmpty()) {
+                    printWriter.println("");
+                    continue;
+                }
+
+                PreparedStatement ps = con.prepareStatement("SELECT `id` FROM `handbook` WHERE `name` LIKE ? ORDER BY `id` ASC;");
+                ps.setString(1, line);
+
+                ResultSet rs = ps.executeQuery();
+
+                String str = "";
+                while (rs.next()) {
+                    int id = rs.getInt("id");
+
+                    str += Integer.toString(id);
+                    str += " ";
+                }
+
+                rs.close();
+                ps.close();
+
+                printWriter.println(str);
+            }
+
+            printWriter.close();
+            bufferedReader.close();
+            fileReader.close();
+        } catch (IOException ex) {
+            System.out.println(ex.getMessage());
+        }
+    }
+
+    public static void main(String[] args) {
+
+        try {
+            if (INSTALL_SQLTABLE) {
+                parseMapleHandbook();
+            } else {
+                fetchDataOnMapleHandbook();
+            }
+
+            con.close();
+        } catch (SQLException e) {
+            System.out.println("Error: invalid SQL syntax");
+            e.printStackTrace();
+        }
+    }
+}
+

src/main/java/tools/mapletools/ToolConstants.java

@@ -2,17 +2,17 @@ package tools.mapletools;
 
 import java.io.File;
 
-public class ToolConstants {
-    public static final File INPUT_DIRECTORY = new File("tools/input");
-    public static final File OUTPUT_DIRECTORY = new File("tools/output");
-    public static final String SCRIPTS_PATH = "scripts";
-    public static final String HANDBOOK_PATH = "handbook";
+class ToolConstants {
+    static final File INPUT_DIRECTORY = new File("tools/input");
+    static final File OUTPUT_DIRECTORY = new File("tools/output");
+    static final String SCRIPTS_PATH = "scripts";
+    static final String HANDBOOK_PATH = "handbook";
 
-    public static File getInputFile(String fileName) {
+    static File getInputFile(String fileName) {
         return new File(INPUT_DIRECTORY, fileName);
     }
 
-    public static File getOutputFile(String fileName) {
+    static File getOutputFile(String fileName) {
         return new File(OUTPUT_DIRECTORY, fileName);
     }
 }