Move MapleCashDropFetcher to main module

This commit is contained in:
P0nk
2021-07-10 19:56:17 +02:00
parent 0af167d174
commit 5b4815f173
4 changed files with 124 additions and 372 deletions

View File

@@ -1,77 +1,38 @@
/* package tools.mapletools;
This file is part of the HeavenMS MapleStory Server
Copyleft (L) 2016 - 2019 RonanLana
This program is free software: you can redistribute it and/or modify import provider.wz.WZFiles;
it under the terms of the GNU Affero General Public License as import tools.Pair;
published by the Free Software Foundation version 3 as published by
the Free Software Foundation. You may not use, modify or distribute
this program under any other version of the GNU Affero General Public
License.
This program is distributed in the hope that it will be useful, import java.io.*;
but WITHOUT ANY WARRANTY; without even the implied warranty of import java.nio.charset.StandardCharsets;
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package maplecashdropfetcher;
import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.sql.Connection; import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.PreparedStatement; import java.sql.PreparedStatement;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.ArrayList; import java.util.*;
import java.util.Collections;
import java.util.List;
import java.util.HashSet;
import java.util.Set;
import java.io.File;
import tools.Pair;
/** /**
*
* @author RonanLana * @author RonanLana
* <p>
This application gets info from the WZ.XML files regarding cash itemids then searches the drop data on the DB * This application gets info from the WZ.XML files regarding cash itemids then searches the drop data on the DB
after any NX (cash item) drops and reports them. * after any NX (cash item) drops and reports them.
* <p>
Estimated parse time: 2 minutes * Estimated parse time: 2 minutes
*/ */
public class MapleCashDropFetcher { public class CashDropFetcher {
static String host = "jdbc:mysql://localhost:3306/cosmic"; private static final File OUTPUT_FILE = ToolConstants.getOutputFile("cash_drop_report.txt");
static String driver = "com.mysql.jdbc.Driver"; private static final Connection con = SimpleDatabaseConnection.getConnection();
static String username = "cosmic_server"; private static final int INITIAL_STRING_LENGTH = 50;
static String password = "snailshell"; private static final int ITEM_FILE_NAME_SIZE = 13;
static String wzPath = "../../wz"; private static final Set<Integer> nxItems = new HashSet<>();
private static final Set<Integer> nxDrops = new HashSet<>();
static String directoryName = "../.."; private static PrintWriter printWriter = null;
static String newFile = "lib/CashDropReport.txt"; private static BufferedReader bufferedReader = null;
static Connection con = null; private static byte status = 0;
static PrintWriter printWriter = null; private static int currentItemid = 0;
static InputStreamReader fileReader = null;
static BufferedReader bufferedReader = null;
static int initialLength = 200;
static int initialStringLength = 50;
static int itemFileNameSize = 13;
static Set<Integer> nxItems = new HashSet<>();
static Set<Integer> nxDrops = new HashSet<>();
static byte status = 0;
static int currentItemid = 0;
private static String getName(String token) { private static String getName(String token) {
int i, j; int i, j;
@@ -82,13 +43,15 @@ public class MapleCashDropFetcher {
i = token.indexOf("\"", i) + 1; //lower bound of the string i = token.indexOf("\"", i) + 1; //lower bound of the string
j = token.indexOf("\"", i); //upper bound j = token.indexOf("\"", i); //upper bound
if(j < i) return "0"; //node value containing 'name' in it's scope, cheap fix since we don't deal with strings anyway if (j < i) {
return "0"; //node value containing 'name' in it's scope, cheap fix since we don't deal with strings anyway
}
dest = new char[initialStringLength]; dest = new char[INITIAL_STRING_LENGTH];
token.getChars(i, j, dest, 0); token.getChars(i, j, dest, 0);
d = new String(dest); d = new String(dest);
return(d.trim()); return (d.trim());
} }
private static String getValue(String token) { private static String getValue(String token) {
@@ -100,31 +63,29 @@ public class MapleCashDropFetcher {
i = token.indexOf("\"", i) + 1; //lower bound of the string i = token.indexOf("\"", i) + 1; //lower bound of the string
j = token.indexOf("\"", i); //upper bound j = token.indexOf("\"", i); //upper bound
dest = new char[initialStringLength]; dest = new char[INITIAL_STRING_LENGTH];
token.getChars(i, j, dest, 0); token.getChars(i, j, dest, 0);
d = new String(dest); d = new String(dest);
return(d.trim()); return (d.trim());
} }
private static void forwardCursor(int st) { private static void forwardCursor(int st) {
String line = null; String line = null;
try { try {
while(status >= st && (line = bufferedReader.readLine()) != null) { while (status >= st && (line = bufferedReader.readLine()) != null) {
simpleToken(line); simpleToken(line);
} }
} } catch (Exception e) {
catch(Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
private static void simpleToken(String token) { private static void simpleToken(String token) {
if(token.contains("/imgdir")) { if (token.contains("/imgdir")) {
status -= 1; status -= 1;
} } else if (token.contains("imgdir")) {
else if(token.contains("imgdir")) {
status += 1; status += 1;
} }
} }
@@ -134,34 +95,31 @@ public class MapleCashDropFetcher {
String line = null; String line = null;
try { try {
while((line = bufferedReader.readLine()) != null) { while ((line = bufferedReader.readLine()) != null) {
translateEquipToken(line); translateEquipToken(line);
} }
} } catch (Exception e) {
catch(Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
private static void translateEquipToken(String token) { private static void translateEquipToken(String token) {
if(token.contains("/imgdir")) { if (token.contains("/imgdir")) {
status -= 1; status -= 1;
} } else if (token.contains("imgdir")) {
else if(token.contains("imgdir")) { if (status == 1) {
if(status == 1) { if (!getName(token).equals("info")) {
if(!getName(token).equals("info")) {
forwardCursor(status); forwardCursor(status);
} }
} }
status += 1; status += 1;
} } else {
else { if (status == 2) {
if(status == 2) {
String d = getName(token); String d = getName(token);
if(d.equals("cash")) { if (d.equals("cash")) {
if(!getValue(token).equals("0")) { if (!getValue(token).equals("0")) {
nxItems.add(currentItemid); nxItems.add(currentItemid);
} }
@@ -175,37 +133,33 @@ public class MapleCashDropFetcher {
String line = null; String line = null;
try { try {
while((line = bufferedReader.readLine()) != null) { while ((line = bufferedReader.readLine()) != null) {
translateItemToken(line); translateItemToken(line);
} }
} } catch (Exception e) {
catch(Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
private static void translateItemToken(String token) { private static void translateItemToken(String token) {
if(token.contains("/imgdir")) { if (token.contains("/imgdir")) {
status -= 1; status -= 1;
} } else if (token.contains("imgdir")) {
else if(token.contains("imgdir")) { if (status == 1) {
if(status == 1) { currentItemid = Integer.parseInt(getName(token));
currentItemid = Integer.valueOf(getName(token)); } else if (status == 2) {
} if (!getName(token).equals("info")) {
else if(status == 2) {
if(!getName(token).equals("info")) {
forwardCursor(status); forwardCursor(status);
} }
} }
status += 1; status += 1;
} } else {
else { if (status == 3) {
if(status == 3) {
String d = getName(token); String d = getName(token);
if(d.equals("cash")) { if (d.equals("cash")) {
if(!getValue(token).equals("0")) { if (!getValue(token).equals("0")) {
nxItems.add(currentItemid); nxItems.add(currentItemid);
} }
@@ -237,8 +191,8 @@ public class MapleCashDropFetcher {
private static int getItemIdFromFilename(String name) { private static int getItemIdFromFilename(String name) {
try { try {
return Integer.valueOf(name.substring(0, name.indexOf('.'))); return Integer.parseInt(name.substring(0, name.indexOf('.')));
} catch(Exception e) { } catch (Exception e) {
return -1; return -1;
} }
} }
@@ -257,10 +211,10 @@ public class MapleCashDropFetcher {
PreparedStatement ps = con.prepareStatement("SELECT DISTINCT itemid FROM " + getDropTableName(dropdata)); PreparedStatement ps = con.prepareStatement("SELECT DISTINCT itemid FROM " + getDropTableName(dropdata));
ResultSet rs = ps.executeQuery(); ResultSet rs = ps.executeQuery();
while(rs.next()) { while (rs.next()) {
int itemid = rs.getInt("itemid"); int itemid = rs.getInt("itemid");
if(nxItems.contains(itemid)) { if (nxItems.contains(itemid)) {
nxDrops.add(itemid); nxDrops.add(itemid);
} }
} }
@@ -275,12 +229,12 @@ public class MapleCashDropFetcher {
List<Integer> sortedNxDrops = new ArrayList<>(nxDrops); List<Integer> sortedNxDrops = new ArrayList<>(nxDrops);
Collections.sort(sortedNxDrops); Collections.sort(sortedNxDrops);
for(Integer nx : sortedNxDrops) { for (Integer nx : sortedNxDrops) {
PreparedStatement ps = con.prepareStatement("SELECT " + getDropElementName(dropdata) + " FROM " + getDropTableName(dropdata) + " WHERE itemid = ?"); PreparedStatement ps = con.prepareStatement("SELECT " + getDropElementName(dropdata) + " FROM " + getDropTableName(dropdata) + " WHERE itemid = ?");
ps.setInt(1, nx); ps.setInt(1, nx);
ResultSet rs = ps.executeQuery(); ResultSet rs = ps.executeQuery();
while(rs.next()) { while (rs.next()) {
entries.add(new Pair<>(nx, rs.getInt(getDropElementName(dropdata)))); entries.add(new Pair<>(nx, rs.getInt(getDropElementName(dropdata))));
} }
@@ -294,33 +248,32 @@ public class MapleCashDropFetcher {
private static void reportNxDropResults(boolean dropdata) throws SQLException { private static void reportNxDropResults(boolean dropdata) throws SQLException {
filterNxDropsOnDB(dropdata); filterNxDropsOnDB(dropdata);
if(!nxDrops.isEmpty()) { if (!nxDrops.isEmpty()) {
List<Pair<Integer, Integer>> nxEntries = getNxDropsEntries(dropdata); List<Pair<Integer, Integer>> nxEntries = getNxDropsEntries(dropdata);
printWriter.println("NX DROPS ON " + getDropTableName(dropdata)); printWriter.println("NX DROPS ON " + getDropTableName(dropdata));
for(Pair<Integer, Integer> nx : nxEntries) { for (Pair<Integer, Integer> nx : nxEntries) {
printWriter.println(nx.left + " : " + nx.right); printWriter.println(nx.left + " : " + nx.right);
} }
printWriter.println("\n\n\n"); printWriter.println("\n\n\n");
} }
} }
private static void ReportNxDropData() { private static void reportNxDropData() {
try { try {
Class.forName(driver).newInstance();
System.out.println("Reading Character.wz ..."); System.out.println("Reading Character.wz ...");
ArrayList<File> files = new ArrayList<>(); ArrayList<File> files = new ArrayList<>();
listFiles(wzPath + "/Character.wz", files); listFiles(WZFiles.CHARACTER.getFilePath(), files);
for(File f : files) { InputStreamReader fileReader = null;
for (File f : files) {
//System.out.println("Parsing " + f.getAbsolutePath()); //System.out.println("Parsing " + f.getAbsolutePath());
int itemid = getItemIdFromFilename(f.getName()); int itemid = getItemIdFromFilename(f.getName());
if(itemid < 0) { if (itemid < 0) {
continue; continue;
} }
fileReader = new InputStreamReader(new FileInputStream(f), "UTF-8"); fileReader = new InputStreamReader(new FileInputStream(f), StandardCharsets.UTF_8);
bufferedReader = new BufferedReader(fileReader); bufferedReader = new BufferedReader(fileReader);
currentItemid = itemid; currentItemid = itemid;
@@ -332,18 +285,18 @@ public class MapleCashDropFetcher {
System.out.println("Reading Item.wz ..."); System.out.println("Reading Item.wz ...");
files = new ArrayList<>(); files = new ArrayList<>();
listFiles(wzPath + "/Item.wz", files); listFiles(WZFiles.ITEM.getFilePath(), files);
for(File f : files) { for (File f : files) {
//System.out.println("Parsing " + f.getAbsolutePath()); //System.out.println("Parsing " + f.getAbsolutePath());
fileReader = new InputStreamReader(new FileInputStream(f), "UTF-8"); fileReader = new InputStreamReader(new FileInputStream(f), StandardCharsets.UTF_8);
bufferedReader = new BufferedReader(fileReader); bufferedReader = new BufferedReader(fileReader);
if(f.getName().length() <= itemFileNameSize) { if (f.getName().length() <= ITEM_FILE_NAME_SIZE) {
inspectItemWzEntry(); inspectItemWzEntry();
} else { // pet file structure is similar to equips, maybe there are other item-types following this behaviour? } else { // pet file structure is similar to equips, maybe there are other item-types following this behaviour?
int itemid = getItemIdFromFilename(f.getName()); int itemid = getItemIdFromFilename(f.getName());
if(itemid < 0) { if (itemid < 0) {
continue; continue;
} }
@@ -357,11 +310,8 @@ public class MapleCashDropFetcher {
System.out.println("Reporting results..."); System.out.println("Reporting results...");
// filter drop data on DB
con = DriverManager.getConnection(host, username, password);
// report suspects of missing quest drop data, as well as those drop data that may have incorrect questids. // report suspects of missing quest drop data, as well as those drop data that may have incorrect questids.
printWriter = new PrintWriter(newFile, "UTF-8"); printWriter = new PrintWriter(OUTPUT_FILE, StandardCharsets.UTF_8);
printReportFileHeader(); printReportFileHeader();
reportNxDropResults(true); reportNxDropResults(true);
@@ -377,30 +327,15 @@ public class MapleCashDropFetcher {
con.close(); con.close();
printWriter.close(); printWriter.close();
System.out.println("Done!"); System.out.println("Done!");
} } catch (SQLException e) {
catch(SQLException e) {
System.out.println("Warning: Could not establish connection to database to report quest data."); System.out.println("Warning: Could not establish connection to database to report quest data.");
System.out.println(e.getMessage()); System.out.println(e.getMessage());
} } catch (Exception e) {
catch(ClassNotFoundException e) {
System.out.println("Error: could not find class");
System.out.println(e.getMessage());
}
catch(InstantiationException e) {
System.out.println("Error: instantiation failure");
System.out.println(e.getMessage());
}
catch(Exception e) {
e.printStackTrace(); e.printStackTrace();
} }
} }
public static void main(String[] args) { public static void main(String[] args) {
ReportNxDropData(); reportNxDropData();
} }
} }

View File

@@ -1,11 +0,0 @@
# Report File autogenerated from the MapleCashDropFetcher feature by Ronan Lana.
# Generated data takes into account several data info from the underlying DB and the server-side WZ.xmls.
NX DROPS ON drop_data
5240005 : 2100100
5490000 : 9410066
5490001 : 9410066

View File

@@ -1,51 +0,0 @@
package tools;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
/**
* @author Frz (Big Daddy)
* @author The Real Spookster - some modifications to this beautiful code
*/
public class DatabaseConnection {
private static String DB_URL = "jdbc:mysql://localhost:3306/cosmic";
private static String DB_USER = "cosmic_server";
private static String DB_PASS = "snailshell";
public static final int RETURN_GENERATED_KEYS = 1;
private static ThreadLocal<Connection> con = new ThreadLocalConnection();
public static Connection getConnection() {
Connection c = con.get();
try {
c.getMetaData();
} catch (SQLException e) { // connection is dead, therefore discard old object 5ever
con.remove();
c = con.get();
}
return c;
}
private static class ThreadLocalConnection extends ThreadLocal<Connection> {
@Override
protected Connection initialValue() {
try {
Class.forName("com.mysql.jdbc.Driver"); // touch the mysql driver
} catch (ClassNotFoundException e) {
System.out.println("[SEVERE] SQL Driver Not Found. Consider death by clams.");
e.printStackTrace();
return null;
}
try {
return DriverManager.getConnection(DB_URL, DB_USER, DB_PASS);
} catch (SQLException e) {
System.out.println("[SEVERE] Unable to make database connection.");
e.printStackTrace();
return null;
}
}
}
}

View File

@@ -1,121 +0,0 @@
/*
This file is part of the OdinMS Maple Story Server
Copyright (C) 2008 ~ 2010 Patrick Huy <patrick.huy@frz.cc>
Matthias Butz <matze@odinms.de>
Jan Christian Meyer <vimes@odinms.de>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License version 3
as published by the Free Software Foundation. You may not use, modify
or distribute this program under any other version of the
GNU Affero General Public License.
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 Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
package tools;
/**
* Represents a pair of values.
*
* @author Frz
* @since Revision 333
* @version 1.0
*
* @param <E> The type of the left value.
* @param <F> The type of the right value.
*/
public class Pair<E, F> {
public E left;
public F right;
/**
* Class constructor - pairs two objects together.
*
* @param left The left object.
* @param right The right object.
*/
public Pair(E left, F right) {
this.left = left;
this.right = right;
}
/**
* Gets the left value.
*
* @return The left value.
*/
public E getLeft() {
return left;
}
/**
* Gets the right value.
*
* @return The right value.
*/
public F getRight() {
return right;
}
/**
* Turns the pair into a string.
*
* @return Each value of the pair as a string joined with a colon.
*/
@Override
public String toString() {
return left.toString() + ":" + right.toString();
}
/**
* Gets the hash code of this pair.
*/
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result + ((left == null) ? 0 : left.hashCode());
result = prime * result + ((right == null) ? 0 : right.hashCode());
return result;
}
/**
* Checks to see if two pairs are equal.
*/
@SuppressWarnings("unchecked")
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (obj == null) {
return false;
}
if (getClass() != obj.getClass()) {
return false;
}
final Pair other = (Pair) obj;
if (left == null) {
if (other.left != null) {
return false;
}
} else if (!left.equals(other.left)) {
return false;
}
if (right == null) {
if (other.right != null) {
return false;
}
} else if (!right.equals(other.right)) {
return false;
}
return true;
}
}