mcMMO/src/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;

	public Database(mcMMO instance) {
		this.plugin = instance;
		this.connectionString = "jdbc:mysql://" + LoadProperties.MySQLserverName + ":" + LoadProperties.MySQLport + "/" + LoadProperties.MySQLdbName + "?user=" + LoadProperties.MySQLuserName + "&password=" + LoadProperties.MySQLdbPass;
		// 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 {
			Connection conn = DriverManager.getConnection(connectionString);
			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);
				}
			}
			conn.close();
		} 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) {
		try {
			Connection conn = DriverManager.getConnection(connectionString);
			PreparedStatement stmt = conn.prepareStatement(sql);
			stmt.executeUpdate();
			conn.close();
			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;
		}
	}

	// Get Int
	// only return first row / first field
	public Integer GetInt(String sql) {
		ResultSet rs = null;
		Integer result = 0;
		try {
			Connection conn = DriverManager.getConnection(connectionString);
			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;
				}
			}
			conn.close();
		} catch (SQLException ex) {
			System.out.println("SQLException: " + ex.getMessage());
			System.out.println("SQLState: " + ex.getSQLState());
			System.out.println("VendorError: " + ex.getErrorCode());
		}

		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>>();
		try {
			Connection conn = DriverManager.getConnection(connectionString);
			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);
				}
			}
			conn.close();
		} catch (SQLException ex) {
			System.out.println("SQLException: " + ex.getMessage());
			System.out.println("SQLState: " + ex.getSQLState());
			System.out.println("VendorError: " + ex.getErrorCode());
		}
		return Rows;
	}
}