/*

 *   SONEWS News Server

 *   see AUTHORS for the list of contributors

 *

 *   This program 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.

 *

 *   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 General Public License for more details.

 *

 *   You should have received a copy of the GNU General Public License

 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

package org.sonews.util;

import java.io.BufferedReader;

import java.io.InputStreamReader;

import java.sql.Connection;

import java.sql.DriverManager;

import java.sql.Statement;

import java.util.HashMap;

import java.util.Map;

import org.sonews.daemon.BootstrapConfig;

import org.sonews.util.io.Resource;

/**

 * Database setup utility class.

 * @author Christian Lins

 * @since sonews/0.5.0

 */

public final class DatabaseSetup 

{

  private static final Map<String, String> templateMap 

    = new HashMap<String, String>();

  private static final Map<String, StringTemplate> urlMap

    = new HashMap<String, StringTemplate>();

  private static final Map<String, String> driverMap

    = new HashMap<String, String>();

  static

  {

    templateMap.put("1", "helpers/database_mysql5_tmpl.sql");

    templateMap.put("2", "helpers/database_postgresql8_tmpl.sql");

    urlMap.put("1", new StringTemplate("jdbc:mysql://%HOSTNAME/%DB"));

    urlMap.put("2", new StringTemplate("jdbc:postgresql://%HOSTNAME/%DB"));

    driverMap.put("1", "com.mysql.jdbc.Driver");

    driverMap.put("2", "org.postgresql.Driver");

  }

  public static void main(String[] args)

    throws Exception

  {

    System.out.println("sonews Database setup helper");

    System.out.println("This program will create a initial database table structure");

    System.out.println("for the sonews Newsserver.");

    System.out.println("You need to create a database and a db user manually before!");

    System.out.println("Select DBMS type:");

    System.out.println("[1] MySQL 5.x or higher");

    System.out.println("[2] PostgreSQL 8.x or higher");

    System.out.print("Your choice: ");

    BufferedReader in = new BufferedReader(new InputStreamReader(System.in));

    String dbmsType = in.readLine();

    String tmplName = templateMap.get(dbmsType);

    if(tmplName == null)

    {

      System.err.println("Invalid choice. Try again you fool!");

      main(args);

      return;

    }

    // Load JDBC Driver class

    Class.forName(driverMap.get(dbmsType));

    String tmpl = Resource.getAsString(tmplName, true);

    System.out.print("Database server hostname (e.g. localhost): ");

    String dbHostname = in.readLine();

    System.out.print("Database name: ");

    String dbName = in.readLine();

    System.out.print("Give name of DB user that can create tables: ");

    String dbUser = in.readLine();

    System.out.print("Password: ");

    String dbPassword = in.readLine();

    String url = urlMap.get(dbmsType)

      .set("HOSTNAME", dbHostname)

      .set("DB", dbName).toString();

    Connection conn = 

      DriverManager.getConnection(url, dbUser, dbPassword);

    conn.setAutoCommit(false);

    String[] tmplChunks = tmpl.split(";");

    for(String chunk : tmplChunks)

    {

      if(chunk.trim().equals(""))

        continue;

      Statement stmt = conn.createStatement();

      stmt.execute(chunk);

    }

    conn.commit();

    conn.setAutoCommit(true);

    BootstrapConfig config = BootstrapConfig.getInstance();

    config.set(BootstrapConfig.STORAGE_DATABASE, url);

    config.set(BootstrapConfig.STORAGE_DBMSDRIVER, driverMap.get(dbmsType));

    config.set(BootstrapConfig.STORAGE_PASSWORD, dbPassword);

    config.set(BootstrapConfig.STORAGE_USER, dbUser);

    config.save();

    System.out.println("Ok");

  }

}