/**
 * 
 */
package net.sourceforge.fenixedu.tools;

import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

/**
 * @author - Shezad Anavarali (shezad@ist.utl.pt)
 * 
 */
public class DBColumnInfo {

    public static String getInfoWithType(String tableName, String columnName, boolean withType) {

        try {

            final Connection connection = getConnection("localhost", "ciapl", "root", "");
            final Statement statement = connection.createStatement();
            final ResultSet resultSet = statement.executeQuery("show create table " + tableName);

            resultSet.next();

            final String[] tableInfo = resultSet.getString(2).split("\n");
            for (String tableInfoLine : tableInfo) {
                tableInfoLine = tableInfoLine.trim();
                if (!tableInfoLine.startsWith("`")) {
                    continue;
                }

                String[] tableInfoLineSplitted = tableInfoLine.split(" ");
                if (tableInfoLineSplitted.length > 1
                        && tableInfoLineSplitted[0].substring(1).startsWith(columnName)) {

                    if (withType) {
                        tableInfoLine = tableInfoLine.substring(0, tableInfoLine.length() - 1);
                        return tableInfoLine.replaceFirst(tableInfoLineSplitted[0], "");
                    } else {
                        StringBuilder result = new StringBuilder();
                        for (int i = 2; i < tableInfoLineSplitted.length; i++) {
                            result.append(tableInfoLineSplitted[i]);
                            result.append(" ");
                        }
                        return result.toString().substring(0, result.toString().length() - 2);
                    }
                }
            }

        } catch (Exception e) {
            System.err.println("ERROR: " + tableName + " - " + columnName);
            e.printStackTrace();
        }

        return null;

    }

    private static Connection getConnection(final String host, final String db, final String user,
            final String pass) throws ClassNotFoundException, SQLException {
        Class.forName("com.mysql.jdbc.Driver");

        final String connectionString = getConnectionString(host, db);

        return DriverManager.getConnection(connectionString, user, pass);
    }

    private static String getConnectionString(final String host, final String db) {
        final StringBuilder url = new StringBuilder();
        url.append("jdbc:mysql://");
        url.append(host);
        url.append("/");
        url.append(db);
        return url.toString();
    }

}