Initial Commit

2 files changed, 110 insertions, 0 deletions

diff --git a/src/main/java/org/openslx/imagemaster/db/DbUser.java b/src/main/java/org/openslx/imagemaster/db/DbUser.java
new file mode 100644
index 0000000..f8400a9
--- /dev/null
+++ b/src/main/java/org/openslx/imagemaster/db/DbUser.java
@@ -0,0 +1,30 @@
+package org.openslx.imagemaster.db;
+
+import org.openslx.imagemaster.session.User;
+
+
+public class DbUser extends User
+{
+	public DbUser(String username, String password, String organization, String firstName, String lastName, String eMail,
+			String satelliteAddress)
+	{
+		super( username, password, organization, firstName, lastName, eMail, satelliteAddress );
+	}
+
+	/**
+	 * Query database for user with given login
+	 * @param login (user@organization)
+	 * @return instance of DbUser for matching entry from DB, or null if not found
+	 */
+	public static DbUser forLogin( final String login )
+	{
+		final String[] parts = login.split( "@" );
+		if ( parts.length != 2 )
+			return null;
+		return MySQL.findUniqueOrNull( DbUser.class,
+				"SELECT user.username, user.password, user.organization, user.firstname, user.lastname, user.email, satellite.address FROM user" +
+				" LEFT JOIN satellite USING (organization)" +
+				" WHERE user.username = ? AND user.organization = ? LIMIT 1", parts[0], parts[1] );
+	}
+
+}
diff --git a/src/main/java/org/openslx/imagemaster/db/MySQL.java b/src/main/java/org/openslx/imagemaster/db/MySQL.java
new file mode 100644
index 0000000..15bf5e2
--- /dev/null
+++ b/src/main/java/org/openslx/imagemaster/db/MySQL.java
@@ -0,0 +1,80 @@
+package org.openslx.imagemaster.db;
+
+import java.io.FileInputStream;
+import java.io.BufferedInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.util.List;
+import java.util.Properties;
+import com.mysql.jdbc.jdbc2.optional.MysqlDataSource;
+
+import org.apache.log4j.Logger;
+import org.openslx.imagemaster.util.Util;
+
+import fi.evident.dalesbred.Database;
+
+/**
+ * Class for talking to the DB via the dalesbred jdbc wrapper. Package private,
+ * so only the Db* classes can actually communicate with the database.
+ * 
+ */
+class MySQL
+{
+	private static final Logger log = Logger.getLogger( MySQL.class );
+	private static Database db = null;
+
+	static
+	{
+		// Load connection info from class (TODO: Make pretty)
+		Properties properties = new Properties();
+		try {
+			final BufferedInputStream stream = new BufferedInputStream( new FileInputStream( "config/mysql.properties" ) );
+			properties.load( stream );
+			stream.close();
+		} catch ( FileNotFoundException e ) {
+			log.fatal( "config/mysql.properties not found!" );
+			System.exit( 1 );
+		} catch ( IOException e ) {
+			log.fatal( "Error reading from config/mysql.properties: " + e.getMessage() );
+			System.exit( 1 );
+		} catch ( Exception e ) {
+			log.fatal( "Generic error loading mysql properties file." );
+			e.printStackTrace();
+			System.exit( 1 );
+		}
+		final String host = properties.getProperty( "host" );
+		final String dbname = properties.getProperty( "db" );
+		final String user = properties.getProperty( "user" );
+		final String password = properties.getProperty( "password" );
+		
+		Util.notNullFatal( host, "host not set in mysql properties" );
+		Util.notNullFatal( dbname, "db not set in mysql properties" );
+		Util.notNullFatal( user, "user not set in mysql properties" );
+		Util.notNullFatal( password, "password not set in mysql properties" );
+
+		// Setup db connection
+		try {
+			MysqlDataSource ds = new MysqlDataSource();
+			ds.setServerName( host );
+			ds.setDatabaseName( dbname );
+			ds.setUser( user );
+			ds.setPassword( password );
+			db = Database.forDataSource( ds );
+		} catch ( Exception e ) {
+			log.fatal( "Error initializing mysql data source!" );
+			e.printStackTrace();
+			System.exit( 1 );
+		}
+	}
+
+	protected static <T> List<T> findAll( final Class<T> clazz, final String sql, final Object... args )
+	{
+		return db.findAll( clazz, sql, args );
+	}
+
+	protected static <T> T findUniqueOrNull( final Class<T> clazz, final String sql, final Object... args )
+	{
+		return db.findUniqueOrNull( clazz, sql, args );
+	}
+
+}