mcMMO/src/main/java/com/gmail/nossr50/Database.java

/*
	This file is part of mcMMO.

    mcMMO 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.

    mcMMO 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 mcMMO.  If not, see <http://www.gnu.org/licenses/>.
*/
package com.gmail.nossr50;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.HashMap;
import java.util.ArrayList;
import java.sql.PreparedStatement;

import com.gmail.nossr50.config.LoadProperties;

public class Database {

	private mcMMO plugin;
	private String connectionString = "jdbc:mysql://" + LoadProperties.MySQLserverName + ":" + LoadProperties.MySQLport + "/" + LoadProperties.MySQLdbName + "?user=" + LoadProperties.MySQLuserName + "&password=" + LoadProperties.MySQLdbPass;
	private boolean isConnected = false;
	private Connection conn = null;
	
	public void connect()
	{
	    try 
	    {
	        System.out.println("[mcMMO] Attempting connection to MySQL...");
	        conn = DriverManager.getConnection(connectionString);
	        isConnected = true;
	        System.out.println("[mcMMO] Connection to MySQL established!");
	    } catch (SQLException ex) 
	    {
	        isConnected = false;
	        ex.printStackTrace();
	        System.out.println("SQLException: " + ex.getMessage());
	        System.out.println("SQLState: " + ex.getSQLState());
	        System.out.println("VendorError: " + ex.getErrorCode());
	    }
	}

	public Database(mcMMO instance) {
	    connect(); //Connect to MySQL
		this.plugin = instance;
		// Load the driver instance
		try {
			Class.forName("com.mysql.jdbc.Driver");
			DriverManager.getConnection(connectionString);
		} catch (ClassNotFoundException e) {
			plugin.getServer().getLogger().warning(e.getLocalizedMessage());
		} catch (SQLException e) {
			plugin.getServer().getLogger().warning(e.getLocalizedMessage());
			System.out.println("SQLException: " + e.getMessage());
			System.out.println("SQLState: " + e.getSQLState());
			System.out.println("VendorError: " + e.getErrorCode());
		}
	}
	
	//Create the DB structure
	public void createStructure() {
		Write("CREATE TABLE IF NOT EXISTS `" + LoadProperties.MySQLtablePrefix + "huds` (`user_id` int(10) unsigned NOT NULL,"
				+ "`hudtype` varchar(50) NOT NULL DEFAULT '',"
				+ "PRIMARY KEY (`user_id`)) ENGINE=MyISAM DEFAULT CHARSET=latin1;");
		Write("CREATE TABLE IF NOT EXISTS `" + LoadProperties.MySQLtablePrefix + "users` (`id` int(10) unsigned NOT NULL AUTO_INCREMENT,"
				+ "`user` varchar(40) NOT NULL,"
				+ "`lastlogin` int(32) unsigned NOT NULL,"
				+ "`party` varchar(100) NOT NULL DEFAULT '',"
				+ "PRIMARY KEY (`id`),"
				+ "UNIQUE KEY `user` (`user`)) ENGINE=MyISAM  DEFAULT CHARSET=latin1 AUTO_INCREMENT=1;");
		Write("CREATE TABLE IF NOT EXISTS `" + LoadProperties.MySQLtablePrefix + "cooldowns` (`user_id` int(10) unsigned NOT NULL,"
				+ "`taming` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`mining` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`woodcutting` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`repair` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`unarmed` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`herbalism` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`excavation` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`archery` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`swords` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`axes` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "`acrobatics` int(32) unsigned NOT NULL DEFAULT '0',"
				+ "PRIMARY KEY (`user_id`)) ENGINE=MyISAM DEFAULT CHARSET=latin1;");
		Write("CREATE TABLE IF NOT EXISTS `" + LoadProperties.MySQLtablePrefix + "skills` (`user_id` int(10) unsigned NOT NULL,"
				+ "`taming` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`mining` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`woodcutting` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`repair` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`unarmed` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`herbalism` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`excavation` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`archery` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`swords` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`axes` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`acrobatics` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "PRIMARY KEY (`user_id`)) ENGINE=MyISAM DEFAULT CHARSET=latin1;");
		Write("CREATE TABLE IF NOT EXISTS `" + LoadProperties.MySQLtablePrefix + "experience` (`user_id` int(10) unsigned NOT NULL,"
				+ "`taming` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`mining` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`woodcutting` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`repair` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`unarmed` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`herbalism` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`excavation` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`archery` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`swords` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`axes` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "`acrobatics` int(10) unsigned NOT NULL DEFAULT '0',"
				+ "PRIMARY KEY (`user_id`)) ENGINE=MyISAM DEFAULT CHARSET=latin1;");
		Write("CREATE TABLE IF NOT EXISTS `" + LoadProperties.MySQLtablePrefix + "spawn` (`user_id` int(10) NOT NULL,"
				+ "`x` int(64) NOT NULL DEFAULT '0',"
				+ "`y` int(64) NOT NULL DEFAULT '0',"
				+ "`z` int(64) NOT NULL DEFAULT '0',"
				+ "`world` varchar(50) NOT NULL DEFAULT '',"
				+ "PRIMARY KEY (`user_id`)) ENGINE=MyISAM DEFAULT CHARSET=latin1;");
		
		Write("DROP TABLE IF EXISTS `"+LoadProperties.MySQLtablePrefix+"skills2`");
		Write("DROP TABLE IF EXISTS `"+LoadProperties.MySQLtablePrefix+"experience2`");
		
		checkDatabaseStructure();
	}

	public void checkDatabaseStructure()
	{
		String sql = "SELECT * FROM  `mcmmo_experience` ORDER BY  `"+LoadProperties.MySQLtablePrefix+"experience`.`fishing` ASC LIMIT 0 , 30";
		
		ResultSet rs = null;
		HashMap<Integer, ArrayList<String>> Rows = new HashMap<Integer, ArrayList<String>>();
		try {
			PreparedStatement stmt = conn.prepareStatement(sql);
			if (stmt.executeQuery() != null) {
				stmt.executeQuery();
				rs = stmt.getResultSet();
				while (rs.next()) {
					ArrayList<String> Col = new ArrayList<String>();
					for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) {
						Col.add(rs.getString(i));
					}
					Rows.put(rs.getRow(), Col);
				}
			}
		} catch (SQLException ex) {
			System.out.println("Updating mcMMO MySQL tables...");
			Write("ALTER TABLE `"+LoadProperties.MySQLtablePrefix + "skills` ADD `fishing` int(10) NOT NULL DEFAULT '0' ;");
			Write("ALTER TABLE `"+LoadProperties.MySQLtablePrefix + "experience` ADD `fishing` int(10) NOT NULL DEFAULT '0' ;");
		}
	}
	
	// write query
	public boolean Write(String sql) {
	    if(conn != null)
	    {
    		try {
    			PreparedStatement stmt = conn.prepareStatement(sql);
    			stmt.executeUpdate();
    			return true;
    		} catch (SQLException ex) {
    			System.out.println("SQLException: " + ex.getMessage());
    			System.out.println("SQLState: " + ex.getSQLState());
    			System.out.println("VendorError: " + ex.getErrorCode());
    			return false;
    		}
	    } else
	    {
	        isConnected = false;
	        connect(); //Attempt to reconnect
	        if(isConnected = true)
	        {
	            Write(sql); //Try the same operation again now that we are connected
	        } else {
	            System.out.println("[mcMMO] Unable to connect to MySQL! Make sure the SQL server is online!");
	        }
	    }
	    return false;
	}

	// Get Int
	// only return first row / first field
	public Integer GetInt(String sql) {
		ResultSet rs = null;
		Integer result = 0;
		if(conn != null)
		{
    		try {
    			PreparedStatement stmt = conn.prepareStatement(sql);
    			stmt = conn.prepareStatement(sql);
    			if (stmt.executeQuery() != null) {
    				stmt.executeQuery();
    				rs = stmt.getResultSet();
    				if (rs.next()) {
    					result = rs.getInt(1);
    				} else {
    					result = 0;
    				}
    			}
    		} catch (SQLException ex) {
    			System.out.println("SQLException: " + ex.getMessage());
    			System.out.println("SQLState: " + ex.getSQLState());
    			System.out.println("VendorError: " + ex.getErrorCode());
    		}
		} else {
		    isConnected = false;
            connect(); //Attempt to reconnect
            if(isConnected = true)
            {
                GetInt(sql); //Try the same operation again now that we are connected
            } else {
                System.out.println("[mcMMO] Unable to connect to MySQL! Make sure the SQL server is online!");
            }
		}
		return result;
	}

	// read query
	public HashMap<Integer, ArrayList<String>> Read(String sql) {
		ResultSet rs = null;
		HashMap<Integer, ArrayList<String>> Rows = new HashMap<Integer, ArrayList<String>>();
		if(conn != null)
		{
    		try {
    			PreparedStatement stmt = conn.prepareStatement(sql);
    			if (stmt.executeQuery() != null) {
    				stmt.executeQuery();
    				rs = stmt.getResultSet();
    				while (rs.next()) {
    					ArrayList<String> Col = new ArrayList<String>();
    					for (int i = 1; i <= rs.getMetaData().getColumnCount(); i++) {
    						Col.add(rs.getString(i));
    					}
    					Rows.put(rs.getRow(), Col);
    				}
    			}
    		} catch (SQLException ex) {
    			System.out.println("SQLException: " + ex.getMessage());
    			System.out.println("SQLState: " + ex.getSQLState());
    			System.out.println("VendorError: " + ex.getErrorCode());
    		}
		} else {
		    isConnected = false;
            connect(); //Attempt to reconnect
            if(isConnected = true)
            {
                Read(sql); //Attempt the same operation again now that we are connected
            } else {
                System.out.println("[mcMMO] Unable to connect to MySQL! Make sure the SQL server is online!");
            }
		}
		return Rows;
	}
}