diff -r f907866f0e4b -r 6fceb66e1ad7 org/sonews/util/DatabaseSetup.java --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/org/sonews/util/DatabaseSetup.java Fri Jun 26 16:48:50 2009 +0200 @@ -0,0 +1,130 @@ +/* + * 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 . + */ + +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 templateMap + = new HashMap(); + private static final Map urlMap + = new HashMap(); + private static final Map driverMap + = new HashMap(); + + 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"); + } + +}