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");
+ }
+
+}